Searched refs:AnfNodeIsPrimitive (Results 1 – 3 of 3) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/frontend/parallel/graph_util/ |
| D | node_info.cc | 78 …(AnfNodeIsPrimitive(node_inputs[1], MAKE_TUPLE) || AnfNodeIsPrimitive(node_inputs[1], MAKE_LIST)))… in ExtractInputParameterByNode()
174 …(AnfNodeIsPrimitive(node_inputs[1], MAKE_TUPLE) || AnfNodeIsPrimitive(node_inputs[1], MAKE_LIST)))… in ExtractInputTypeLengthByNode()
265 bool AnfNodeIsPrimitive(const AnfNodePtr &anf_node, const std::string &prim_name) { in AnfNodeIsPrimitive() function
|
| D | node_info.h | 46 bool AnfNodeIsPrimitive(const AnfNodePtr &anf_node, const std::string &prim_name);
|
| /third_party/mindspore/mindspore/ccsrc/frontend/parallel/ |
| D | step_parallel.cc | 554 if (AnfNodeIsPrimitive(node, MAKE_TUPLE) || AnfNodeIsPrimitive(node, MAKE_LIST)) { in StepRedistribution()
1073 …(AnfNodeIsPrimitive(node->input(1), MAKE_TUPLE) || AnfNodeIsPrimitive(node->input(1), MAKE_LIST)))… in CheckInsertMirrorOps()
2977 if (!AnfNodeIsPrimitive(node, MAKE_TUPLE) && !AnfNodeIsPrimitive(node, MAKE_LIST)) { in HandleForwardMakeTupleAndMakeList()
2989 std::string op_type = AnfNodeIsPrimitive(node, MAKE_TUPLE) ? MAKE_TUPLE : MAKE_LIST; in HandleForwardMakeTupleAndMakeList()
|