Searched defs:all_to_all_v (Results 1 – 6 of 6) sorted by relevance
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/optimizer/ |
| D | alltoall_fusion.cc | 117 auto all_to_all_v = graph->NewCNode(all_to_all_v_input); in CreateAllToAllvNode() local
140 …teConcatNode(const FuncGraphPtr &graph, const CNodePtr &all_to_all, const CNodePtr &all_to_all_v) { in CreateConcatNode()
199 auto all_to_all_v = CreateAllToAllvNode(graph, all_to_all, split); in Process() local
|
| D | neighbor_exchange_v2_fusion.cc | 428 auto all_to_all_v = pass.NewCNode(all_to_all_v_input, graph); in CreateAllToAllvNode() local
908 const CNodePtr &all_to_all_v, in CreateSplitGradNodes()
1009 …auto all_to_all_v = CreateAllToAllvNode(graph, neighbor_exchange_v2, split_nodes, split_num, false… in Process() local
1025 …auto all_to_all_v = CreateAllToAllvNode(graph, neighbor_exchange_v2_grad, split_nodes, split_num, … in Process() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/optimizer/ge/ |
| D | all_to_all_v_for_ge.cc | 129 std::vector<ShapeVector> GetAllToAllvOutputShapes(const CNodePtr &all_to_all_v) { in GetAllToAllvOutputShapes()
278 auto all_to_all_v = node->cast<CNodePtr>(); in Process() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/optimizer/mindir/ |
| D | all_to_all_unify_mindir.cc | 116 auto all_to_all_v = NewCNode(all_to_all_v_input, graph); in CreateAllToAllvNode() local
314 auto all_to_all_v = CreateAllToAllvNode(graph, neighbor_exchange); in Process() local
|
| D | neighbor_exchange_v2_unify_mindir.cc | 349 auto all_to_all_v = pass.NewCNode(all_to_all_v_input, graph); in CreateAllToAllvNode() local
844 const CNodePtr &all_to_all_v, in CreateSplitGradNodes()
959 …auto all_to_all_v = CreateAllToAllvNode(graph, neighbor_exchange_v2, split_nodes, split_num, false… in Process() local
982 …auto all_to_all_v = CreateAllToAllvNode(graph, neighbor_exchange_v2_grad, split_nodes, split_num, … in Process() local
|
| /third_party/mindspore/mindspore-src/source/tests/ut/cpp/device/ |
| D | hccl_adapter_test.cc | 39 auto all_to_all_v = graph->NewCNode(all_to_all_v_input); in CreateAllToAllvNode() local
|