|
|
|
@@ -129,7 +129,11 @@ Status ModelHelper::SaveModelDef(std::shared_ptr<OmFileSaveHelper> &om_file_save |
|
|
|
model_tmp->SetGraph(ge_model->GetGraph()); |
|
|
|
model_tmp->SetVersion(ge_model->GetVersion()); |
|
|
|
model_tmp->SetAttr(ge_model->MutableAttrMap()); |
|
|
|
SaveSizeToModelDef(ge_model); |
|
|
|
Status ret = SaveSizeToModelDef(ge_model); |
|
|
|
if (ret != SUCCESS) { |
|
|
|
GELOGE(ret, "SaveSizeToModelDef failed"); |
|
|
|
return ret; |
|
|
|
} |
|
|
|
|
|
|
|
(void)model_tmp->Save(model_buffer); |
|
|
|
GELOGD("MODEL_DEF size is %zu", model_buffer.GetSize()); |
|
|
|
|