|
|
|
@ -851,6 +851,10 @@ KernelGraphPtr SessionBasic::ConstructKernelGraph(const AnfNodePtrList &lst, con
|
|
|
|
|
if (ExistSummaryNode(graph.get())) {
|
|
|
|
|
graph->set_summary_node_exist(true);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
// Update Graph Dynamic Shape Attr
|
|
|
|
|
UpdateGraphDynamicShapeAttr(NOT_NULL(graph));
|
|
|
|
|
|
|
|
|
|
opt::BackendCommonOptimization(graph);
|
|
|
|
|
graph->SetInputNodes();
|
|
|
|
|
graph->SetOptimizerFlag();
|
|
|
|
@ -1493,6 +1497,7 @@ void SessionBasic::UpdateGraphDynamicShapeAttr(const NotNull<KernelGraphPtr> &ro
|
|
|
|
|
MS_LOG(INFO) << "Set Input Dynamic Shape Attr to Node:" << cnode->fullname_with_scope();
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
root_graph->UpdateGraphDynamicAttr();
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
#if (ENABLE_CPU && (ENABLE_D || ENABLE_GPU))
|
|
|
|
|