Searched defs:all_to_all (Results 1 – 2 of 2) sorted by relevance
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/optimizer/mindir/ |
| D | all_to_all_unify_mindir.cc | 61 …dePtr AllToAllUnifyMindIR::CreateSplitNode(const KernelGraphPtr &graph, const CNodePtr &all_to_all, in CreateSplitNode()
162 …lToAllUnifyMindIR::CreateSplitNodeWithDim0(const KernelGraphPtr &graph, const CNodePtr &all_to_all, in CreateSplitNodeWithDim0()
169 …r AllToAllUnifyMindIR::CreateAllToAllvNode(const KernelGraphPtr &graph, const CNodePtr &all_to_all, in CreateAllToAllvNode()
218 …tr AllToAllUnifyMindIR::CreateAllToAllNode(const KernelGraphPtr &graph, const CNodePtr &all_to_all, in CreateAllToAllNode()
246 …ePtr AllToAllUnifyMindIR::CreateConcatNode(const KernelGraphPtr &graph, const CNodePtr &all_to_all, in CreateConcatNode()
277 …UnifyMindIR::CreateConcatNodeWithConcatDim(const KernelGraphPtr &graph, const CNodePtr &all_to_all, in CreateConcatNodeWithConcatDim()
285 …ToAllUnifyMindIR::CreateConcatNodeWithDim0(const KernelGraphPtr &graph, const CNodePtr &all_to_all, in CreateConcatNodeWithDim0()
332 auto all_to_all = node->cast<CNodePtr>(); in Process() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/optimizer/ |
| D | alltoall_fusion.cc | 47 CNodePtr CreateSplitNode(const FuncGraphPtr &graph, const CNodePtr &all_to_all) { in CreateSplitNode()
102 CNodePtr CreateAllToAllvNode(const FuncGraphPtr &graph, const CNodePtr &all_to_all, const CNodePtr … in CreateAllToAllvNode()
140 CNodePtr CreateConcatNode(const FuncGraphPtr &graph, const CNodePtr &all_to_all, const CNodePtr &al… in CreateConcatNode()
193 auto all_to_all = node->cast<CNodePtr>(); in Process() local
|