浏览代码

Feature: Get default from ge ir graph while no user input shape

tags/v1.2.0
l00444296 5 年前
父节点
当前提交
07800ed03c
共有 1 个文件被更改,包括 2 次插入1 次删除
  1. +2
    -1
      ge/ir_build/ge_ir_build.cc

+ 2
- 1
ge/ir_build/ge_ir_build.cc 查看文件

@@ -334,7 +334,8 @@ graphStatus Impl::Init(const Graph &graph, const std::map<std::string, std::stri

string input_shape;
if (options_.find("input_shape") == options_.end()) {
GE_CHK_BOOL_RET_STATUS_NOLOG(GetDefaultInputShape(graph, input_shape), GRAPH_PARAM_INVALID);
GE_CHK_BOOL_EXEC(GetDefaultInputShape(graph, input_shape) == ge::SUCCESS, return ge::GRAPH_PARAM_INVALID,
"Get defaule data op shape from graph failed!");
} else {
input_shape = options_["input_shape"];
}


正在加载...
取消
保存