Searched refs:recv_tag (Results 1 – 2 of 2) sorted by relevance
| /third_party/mindspore/mindspore/boost/ |
| D | adasum.py | 225 recv_tag = self._hash(step, dest_target, weights_index)
226 recv = Receive(sr_tag=recv_tag, src_rank=dest_target, shape=shape, dtype=dtype,
236 recv_tag = self._hash(step, dest_target, weights_index)
237 recv = Receive(sr_tag=recv_tag, src_rank=dest_target, shape=shape, dtype=dtype,
|
| /third_party/mindspore/mindspore/ccsrc/frontend/parallel/pipeline_transformer/ |
| D | pipeline_transformer.cc | 625 int64_t recv_tag; in InsertReceive() local
627 recv_tag = recv_tag_map[src_rank] + 1; in InsertReceive()
630 recv_tag = 0; in InsertReceive()
633 Attr attr_tag = std::make_pair(SR_TAG, MakeValue(recv_tag)); in InsertReceive()
|