send_rank_ids (reference) in projects: third_party

Project(s)

Full Search
Definition
Symbol
File Path
History
Type

Searched refs:send_rank_ids (Results 1 – 6 of 6) sorted by relevance

/third_party/mindspore/tests/ut/cpp/device/
D	hccl_adapter_test.cc	29 … const std::vector<int64_t> &send_rank_ids, const std::vector<int64_t> &recv_rank_ids) { in CreateAllToAllvNode() argument 35 …AnfAlgo::SetNodeAttr(kAttrSendRankIds, MakeValue<std::vector<int64_t>>(send_rank_ids), all_to_all_… in CreateAllToAllvNode() 96 std::vector<int64_t> send_rank_ids = {1}; in TEST_F() local 98 …auto alltoall = CreateAllToAllvNode(graph, CreateInputs(graph, {{1}}, {type_}), send_rank_ids, rec… in TEST_F() 119 std::vector<int64_t> send_rank_ids = {}; in TEST_F() local 121 …auto alltoall = CreateAllToAllvNode(graph, CreateInputs(graph, {}, {}), send_rank_ids, recv_rank_i… in TEST_F() 142 std::vector<int64_t> send_rank_ids = {1, 2, 3}; in TEST_F() local 144 …ateAllToAllvNode(graph, CreateInputs(graph, {{1}, {1}, {1}}, {type_, type_, type_}), send_rank_ids, in TEST_F() 166 std::vector<int64_t> send_rank_ids = {1, 3}; in TEST_F() local 169 …CreateAllToAllvNode(graph, CreateInputs(graph, {{1}, {1}}, {type_, type_}), send_rank_ids, recv_ra… in TEST_F() [all …]
/third_party/mindspore/tests/ut/python/parallel/
D	test_neighborexchange.py	52 self.alltoallv = NeighborExchange(send_rank_ids=[0, 1], recv_rank_ids=[1, 2], 80 …self.alltoallv = NeighborExchange(send_rank_ids=[0], recv_rank_ids=[1, 2], recv_shapes=([32, 32], … 105 … self.alltoallv = NeighborExchange(send_rank_ids=[], recv_rank_ids=[1], recv_shapes=([1],), 127 self.alltoallv = NeighborExchange(send_rank_ids=[0], recv_rank_ids=[], recv_shapes=(), 149 self.alltoallv = NeighborExchange(send_rank_ids=[], recv_rank_ids=[], recv_shapes=(), 173 …self.alltoallv = NeighborExchange(send_rank_ids=[0], recv_rank_ids=[1, 2], recv_shapes=([32, 32],), 201 self.alltoallv = NeighborExchange(send_rank_ids=[0, 1], recv_rank_ids=[1, 2], 230 self.alltoallv = NeighborExchange(send_rank_ids=[0, 1], recv_rank_ids=[1, 2], 259 …self.alltoallv = NeighborExchange(send_rank_ids=[0], recv_rank_ids=[1, 2], recv_shapes=([32, 32], … 285 …self.alltoallv = NeighborExchange(send_rank_ids=(0), recv_rank_ids=[1, 2], recv_shapes=([32, 32], … [all …]
/third_party/mindspore/mindspore/core/ops/
D	neighborexchange.cc	`101 auto send_rank_ids = GetValue<std::vector<int64_t>>(primitive->GetAttr(kSendRankIds)); in Check() local 102 if (send_rank_ids.empty()) { in Check()`
/third_party/mindspore/tests/ut/cpp/python_input/gtest_input/pre_activate/
D	all_to_all_unify_mindir_test.py	`37 …neighbor = NeighborExchange(send_rank_ids=[0], recv_rank_ids=[1], recv_shapes=([2, 2],), send_shap…`
/third_party/mindspore/mindspore/ops/operations/
D	comm_ops.py	`643 def __init__(self, send_rank_ids, recv_rank_ids, recv_shapes, send_shapes, recv_type, argument 646 self.send_rank_ids = send_rank_ids`
/third_party/mindspore/mindspore/ops/_grad/
D	grad_comm_ops.py	`364 send_rank_ids = self.recv_rank_ids 365 recv_rank_ids = self.send_rank_ids 369 …neighborexchange_grad = NeighborExchange(send_rank_ids, recv_rank_ids, recv_shapes, send_shapes, r…`