!188 Fix aclInferShapeAndType.

Merge pull request !188 from 董铎/development
pull/188/MERGE
王涛 5 years ago committed by Gitee
commit 1eacb83358

@ -442,6 +442,12 @@ graphStatus aclgrphInferShapeAndType(ge::Graph &graph) {
auto compute_graph = GraphUtils::GetComputeGraph(graph);
GE_CHECK_NOTNULL(compute_graph);
auto ret = compute_graph->InferOriginFormat();
if (ret != GRAPH_SUCCESS) {
GELOGE(ret, "Acl InferOriginFormat failed.");
return ret;
}
for (auto &node: compute_graph->GetAllNodes()) {
graphStatus ret = ShapeRefiner::InferShapeAndType(node);
if (ret == GRAPH_PARAM_INVALID) {

Loading…
Cancel
Save