Browse Source

Feature: Get default op format from ge graph

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

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

@@ -230,7 +230,7 @@ class Impl {
graphStatus CreateInputsForIRBuild(const ge::Graph &graph, vector<ge::GeTensor> &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<std::string, std::string> &options);
graphStatus BuildModel(const Graph &graph, const std::map<std::string, std::string> &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()) {


Loading…
Cancel
Save