|
|
|
@@ -170,9 +170,11 @@ std::string IrExportBuilder::GetProtoString(const FuncGraphPtr &func_graph) { |
|
|
|
} |
|
|
|
|
|
|
|
void IrExportBuilder::BuildModelInfo() { |
|
|
|
model_.set_ir_version("0.1.0"); |
|
|
|
model_.set_producer_name("MindSpore"); |
|
|
|
model_.set_model_version("1.1.0"); |
|
|
|
constexpr auto ir_version = "0.1.0"; |
|
|
|
constexpr auto mindspore_name = "MindSpore"; |
|
|
|
model_.set_ir_version(ir_version); |
|
|
|
model_.set_producer_name(mindspore_name); |
|
|
|
model_.set_model_version(VERSION); |
|
|
|
} |
|
|
|
|
|
|
|
void IrExportBuilder::BuildModel(const FuncGraphPtr &func_graph, bool save_tensor_data) { |
|
|
|
|