Searched refs:out_cnode (Results 1 – 5 of 5) sorted by relevance
43 auto out_cnode = func_graph->output()->cast<CNodePtr>(); in FindAssignAndOutputVal() local44 MS_EXCEPTION_IF_NULL(out_cnode); in FindAssignAndOutputVal()47 if (!IsPrimitiveCNode(out_cnode, prim::kPrimMakeTuple)) { in FindAssignAndOutputVal()58 const auto &inputs = out_cnode->inputs(); in FindAssignAndOutputVal()
74 auto out_cnode = out_node->cast<CNodePtr>(); in EliminateMakeTuple() local75 for (auto out : out_cnode->inputs()) { in EliminateMakeTuple()85 if (output_args.size() != out_cnode->inputs().size()) { in EliminateMakeTuple()
443 for (auto &out_cnode : out_nodes) { in HandleGraphMultiNode() local444 manager->Replace(out_cnode, out_cnode->input(1)); in HandleGraphMultiNode()497 auto out_cnode = cnode->input(index)->cast<CNodePtr>(); in SetSubGraphInput() local498 MS_ASSERT(out_cnode != nullptr); in SetSubGraphInput()500 auto out_prim = GetValueNode<PrimitivePtr>(out_cnode->input(0)); in SetSubGraphInput()
255 auto out_cnode = cnode->input(index)->cast<CNodePtr>(); in SetSubGraphInput() local257 auto out_prim = GetValueNode<PrimitivePtr>(out_cnode->input(0)); in SetSubGraphInput()
106 auto out_cnode = out_node->cast<CNodePtr>(); in Run() local108 new_cnode->CloneCNodeInfo(out_cnode); in Run()