Searched refs:send_shapes (Results 1 – 6 of 6) sorted by relevance
| /third_party/mindspore/tests/ut/python/parallel/ |
| D | test_neighborexchange.py | 54 … send_shapes=([32, 32], [32, 16]), recv_type=ms.float32)
81 send_shapes=([32, 32],), recv_type=ms.float32)
106 send_shapes=(), recv_type=ms.float32)
128 send_shapes=([32, 16],), recv_type=ms.float32)
150 send_shapes=(), recv_type=ms.float32)
174 send_shapes=([32, 32],), recv_type=ms.float32)
203 send_shapes=([32, 32],), recv_type=ms.float32)
232 … send_shapes=([32, 32], [32, 32]), recv_type=ms.float32)
260 send_shapes=([16, 16],), recv_type=ms.float32)
286 send_shapes=([32, 16],), recv_type=ms.float32)
[all …]
|
| /third_party/mindspore/mindspore/core/ops/ |
| D | neighborexchange.cc | 116 auto send_shapes = GetValue<ValuePtrList>(primitive->GetAttr(kSendShapes)); in Check() local
117 if (abstract_element.size() != send_shapes.size()) { in Check()
119 << kSendShapes << " size " << send_shapes.size(); in Check()
123 MS_EXCEPTION_IF_NULL(send_shapes[i]); in Check()
124 auto send_shape_value = send_shapes[i]->cast<ValueSequeuePtr>(); in Check()
|
| /third_party/mindspore/tests/ut/cpp/python_input/gtest_input/pre_activate/ |
| D | all_to_all_unify_mindir_test.py | 37 …eighborExchange(send_rank_ids=[0], recv_rank_ids=[1], recv_shapes=([2, 2],), send_shapes=([2, 2],),
|
| /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
649 self.send_shapes = send_shapes
|
| /third_party/mindspore/mindspore/ops/_grad/ |
| D | grad_comm_ops.py | 366 recv_shapes = self.send_shapes
367 send_shapes = self.recv_shapes
369 …neighborexchange_grad = NeighborExchange(send_rank_ids, recv_rank_ids, recv_shapes, send_shapes, r…
|
| /third_party/mindspore/mindspore/ccsrc/frontend/parallel/ops_info/ |
| D | conv2d_info.cc | 658 Attr send_shapes = {SEND_SHAPES, MakeTupleListValue(send_shapes_)}; in CreateNeighborExchangeAttrs() local
661 OperatorAttrs attrs = {send_ranks, recv_ranks, recv_shapes, send_shapes, recv_type}; in CreateNeighborExchangeAttrs()
|