Browse Source

Feature: Get default op format from ge graph

tags/v1.2.0
l00444296 4 years ago
parent
commit
842d3f2716
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      ge/ir_build/ge_ir_build.cc

+ 1
- 1
ge/ir_build/ge_ir_build.cc View File

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


Loading…
Cancel
Save