Feature: Get default op format from ge graph

pull/621/head
l00444296 4 years ago
parent 54b6ce9eea
commit c85ad855e0

@ -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