Feature: Get default op format from ge graph

4 years ago · 842d3f2716
parent c85ad855e0
commit 842d3f2716
1 changed files with 1 additions and 1 deletions
--- a/ge/ir_build/ge_ir_build.cc
+++ b/ge/ir_build/ge_ir_build.cc
@ -378,7 +378,7 @@ graphStatus Impl::GetDefaultInputFormat(const Graph &graph, string &default_form
  return GRAPH_SUCCESS;
 }

-graphStatus Impl::Init(const Graph &graph, string &default_shape, bool &dynamic_shape_flag) {
+graphStatus Impl::GetDefaultInputShape(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()) {