Searched refs:kPrimCall (Results 1 – 12 of 12) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/pass/ |
| D | convert_tuple_output_to_maketuple.cc | 81 !AnfAlgo::CheckPrimitiveType(input, prim::kPrimCall)) { in Process()
|
| D | convert_tuple_input_to_dynamic_input.cc | 62 if (AnfAlgo::CheckPrimitiveType(cnode_ptr, prim::kPrimCall) || in ConvertMakeTupleInputToPlantInputs()
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/common/ |
| D | node_pass.cc | 46 } else if (AnfAlgo::CheckPrimitiveType(cnode, prim::kPrimCall)) { in AddOutputAndCallerToMap()
|
| /third_party/mindspore/mindspore/ccsrc/frontend/optimizer/ |
| D | auto_monad_eliminate.cc | 43 …bool is_special_op = IsPrimitiveCNode(cnode, prim::kPrimCall) || IsValueNode<FuncGraph>(cnode->inp… in GenerateLoadGroups()
|
| /third_party/mindspore/mindspore/ccsrc/backend/session/ |
| D | session_basic.cc | 780 graph->NewValueNode(NewValueNode(std::make_shared<Primitive>(prim::kPrimCall->name())))}; in CreateCallSwitchInputs()
831 graph->NewValueNode(NewValueNode(std::make_shared<Primitive>(prim::kPrimCall->name())))}; in ProcessNodeRetFunc()
877 graph->NewValueNode(NewValueNode(std::make_shared<Primitive>(prim::kPrimCall->name())))}; in CreateCallSwitchLayerInputs()
945 graph->NewValueNode(NewValueNode(std::make_shared<Primitive>(prim::kPrimCall->name())))}; in CreateSwitchOrPartialNode()
987 …cnode_inputs = {graph->NewValueNode(NewValueNode(std::make_shared<Primitive>(prim::kPrimCall->name… in CreateValueNode()
1060 if (AnfAlgo::CheckPrimitiveType(new_cnode, prim::kPrimCall) && in CreateNewCNode()
1064 if (AnfAlgo::CheckPrimitiveType(new_cnode, prim::kPrimCall) && in CreateNewCNode()
|
| D | kernel_graph.cc | 69 if (new_output.size() == 1 && AnfAlgo::CheckPrimitiveType(new_output[0], prim::kPrimCall)) { in GetCallRealOutputs()
73 if (!AnfAlgo::CheckPrimitiveType(node, prim::kPrimCall)) { in GetCallRealOutputs()
1299 auto call_nodes = FindNodeByPrimitive({std::make_shared<Primitive>(prim::kPrimCall->name()), in UpdateChildGraphOrder()
|
| D | ascend_auto_monad.cc | 422 } else if (AnfAlgo::CheckPrimitiveType(node, prim::kPrimCall)) { in FindCallSites()
1085 const bool is_call = AnfAlgo::CheckPrimitiveType(cnode, prim::kPrimCall); in MakeLabelGotoSwitch()
|
| D | anf_runtime_algorithm.cc | 1698 …if (!(AnfAlgo::CheckPrimitiveType(cnode, prim::kPrimCall) || AnfAlgo::CheckPrimitiveType(cnode, pr… in GetCallSwitchKernelGraph()
1720 if (AnfAlgo::CheckPrimitiveType(cnode, prim::kPrimCall)) { in GetCallSwitchKernelGraph()
1744 if (!CheckPrimitiveType(call_node, prim::kPrimCall)) { in IsSwitchCall()
|
| /third_party/mindspore/mindspore/ccsrc/frontend/optimizer/irpass/ |
| D | updatestate_eliminate.cc | 100 …if (IsPrimitiveCNode(first_input_node, prim::kPrimCall) || IsPrimitiveCNode(first_input_node, prim… in CheckHasMonadInput()
|
| /third_party/mindspore/mindspore/lite/tools/anf_exporter/ |
| D | anf_exporter.cc | 508 if (opt::CheckPrimitiveType(fg_output, prim::kPrimCall)) { in GetFinalGraph()
|
| /third_party/mindspore/mindspore/ccsrc/pipeline/jit/static_analysis/ |
| D | auto_monad.cc | 838 if (IsPrimitiveEquals(prim, prim::kPrimCall)) { in DetectEffectInfo()
|
| /third_party/mindspore/mindspore/core/base/ |
| D | core_ops.h | 543 inline const PrimitivePtr kPrimCall = std::make_shared<Primitive>("call"); variable
|