From c85ad855e08b95e30866011e0a0228cd495e768e Mon Sep 17 00:00:00 2001 From: l00444296 Date: Tue, 15 Dec 2020 10:19:42 +0800 Subject: [PATCH] Feature: Get default op format from ge graph --- ge/ir_build/ge_ir_build.cc | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/ge/ir_build/ge_ir_build.cc b/ge/ir_build/ge_ir_build.cc index 34e612a2..1b00b334 100644 --- a/ge/ir_build/ge_ir_build.cc +++ b/ge/ir_build/ge_ir_build.cc @@ -230,7 +230,7 @@ class Impl { graphStatus CreateInputsForIRBuild(const ge::Graph &graph, vector &inputs); graphStatus GetDefaultInputShape(const Graph &graph, string &default_shape, bool &dynamic_shape_flag); graphStatus GetDefaultInputFormat(const Graph &graph, string &default_format); - const Graph &graph, string &default_shape, string &input_fo graphStatus UpdateDataOpAttr(const Graph &graph); + graphStatus UpdateDataOpAttr(const Graph &graph); graphStatus Init(const Graph &graph, const std::map &options); graphStatus BuildModel(const Graph &graph, const std::map &options, ModelBufferData &ge_models); @@ -378,7 +378,7 @@ graphStatus Impl::GetDefaultInputFormat(const Graph &graph, string &default_form return GRAPH_SUCCESS; } -graphStatus Impl::(const Graph &graph, string &default_shape, bool &dynamic_shape_flag) { +graphStatus Impl::Init(const Graph &graph, string &default_shape, bool &dynamic_shape_flag) { auto compute_graph = ge::GraphUtils::GetComputeGraph(graph); GE_CHECK_NOTNULL(compute_graph); for (ge::NodePtr &input_node : compute_graph->GetDirectNode()) {