|
|
|
@@ -78,6 +78,8 @@ Status ModelHelper::SaveModelPartition(std::shared_ptr<OmFileSaveHelper> &om_fil |
|
|
|
|
|
|
|
Status ModelHelper::SaveSizeToModelDef(const GeModelPtr &ge_model, ge::Buffer &model_buffer) { |
|
|
|
vector<int64_t> om_info; |
|
|
|
ModelPtr model_tmp = ge::MakeShared<ge::Model>(ge_model->GetName(), ge_model->GetPlatformVersion()); |
|
|
|
(void)model_tmp->Save(model_buffer); |
|
|
|
GELOGD("SaveSizeToModelDef modeldef_size is %zu", model_buffer.GetSize()); |
|
|
|
om_info.push_back(model_buffer.GetSize()); |
|
|
|
|
|
|
|
@@ -118,10 +120,10 @@ Status ModelHelper::SaveModelDef(std::shared_ptr<OmFileSaveHelper> &om_file_save |
|
|
|
} |
|
|
|
model_tmp->SetGraph(ge_model->GetGraph()); |
|
|
|
model_tmp->SetVersion(ge_model->GetVersion()); |
|
|
|
SaveSizeToModelDef(ge_model, model_buffer); |
|
|
|
model_tmp->SetAttr(ge_model->MutableAttrMap()); |
|
|
|
|
|
|
|
(void)model_tmp->Save(model_buffer); |
|
|
|
SaveSizeToModelDef(ge_model, model_buffer); |
|
|
|
GELOGD("MODEL_DEF size is %zu", model_buffer.GetSize()); |
|
|
|
if (model_buffer.GetSize() > 0) { |
|
|
|
if (SaveModelPartition(om_file_save_helper, ModelPartitionType::MODEL_DEF, model_buffer.GetData(), |
|
|
|
|