Home
last modified time | relevance | path

Searched refs:AlltoAll (Results 1 – 7 of 7) sorted by relevance

/third_party/mindspore/tests/ut/python/parallel/
Dtest_alltoall.py28 from mindspore.ops.operations.comm_ops import AlltoAll
132 self.alltoallv = AlltoAll(split_count=8, split_dim=2, concat_dim=3)
153 self.alltoallv = AlltoAll(split_count=7, split_dim=2, concat_dim=3)
175 self.alltoallv = AlltoAll(split_count=[8], split_dim=2, concat_dim=3)
197 self.alltoallv = AlltoAll(split_count=8, split_dim=4, concat_dim=3)
219 self.alltoallv = AlltoAll(split_count=8, split_dim=(3,), concat_dim=3)
241 self.alltoallv = AlltoAll(split_count=8, split_dim=3, concat_dim=4)
263 self.alltoallv = AlltoAll(split_count=8, split_dim=3, concat_dim=([3],))
285 self.alltoallv = AlltoAll(split_count=3, split_dim=3, concat_dim=3)
307 self.alltoallv = AlltoAll(split_count=8, split_dim=3, concat_dim=3, group=3)
/third_party/mindspore/tests/ut/cpp/python_input/gtest_input/pre_activate/
Dall_to_all_unify_mindir_test.py17 from mindspore.ops.operations.comm_ops import AlltoAll, NeighborExchange
48 altoall = AlltoAll(split_count=2, split_dim=2, concat_dim=3)
/third_party/mindspore/tests/ut/python/communication/
Dtest_comm.py28 from mindspore.ops.operations.comm_ops import AllReduce, AllGather, AlltoAll, ReduceOp, ReduceScatt…
116 self.alltoall = AlltoAll(1, 0, 1)
/third_party/mindspore/mindspore/ops/_grad/
Dgrad_comm_ops.py25 …m_ops import (AllGather, _MiniStepAllGather, _HostAllGather, AllReduce, NeighborExchange, AlltoAll,
377 @bprop_getters.register(AlltoAll)
380 all_to_all_grad = AlltoAll(self.split_count, self.concat_dim, self.split_dim, self.group)
/third_party/mindspore/mindspore/ops/operations/
D__init__.py38 from .comm_ops import (AllGather, AllReduce, NeighborExchange, AlltoAll, AllSwap, ReduceScatter, Br…
Dcomm_ops.py657 class AlltoAll(PrimitiveWithInfer): class
/third_party/mindspore/
DRELEASE.md53 - [STABLE] Support AlltoAll operator.