Browse Source

display model info

pull/688/head
wangwenhua1@huawei.com 5 years ago
parent
commit
8ae9370dbf
1 changed files with 3 additions and 1 deletions
  1. +3
    -1
      ge/common/helper/model_helper.cc

+ 3
- 1
ge/common/helper/model_helper.cc View File

@@ -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(),


Loading…
Cancel
Save