!12434 fix operator_info is null

From: @Margaret_wangrui
Reviewed-by: 
Signed-off-by:
pull/12434/MERGE
mindspore-ci-bot 4 years ago committed by Gitee
commit 2c5d19260e

@ -2250,6 +2250,10 @@ LossNodeInfo FindLossCNode(const FuncGraphPtr &func_graph) {
}
AnfNodePtr pre_node = return_node->input(1);
MS_EXCEPTION_IF_NULL(pre_node);
if (IsPrimitiveCNode(pre_node, prim::kPrimDepend)) {
pre_node = pre_node->cast<CNodePtr>()->input(1);
MS_EXCEPTION_IF_NULL(pre_node);
}
auto pre_cnode = pre_node->cast<CNodePtr>();
if (pre_cnode == nullptr || !IsValueNode<Primitive>(pre_cnode->input(0))) {

Loading…
Cancel
Save