Searched defs:rankset (Results 1 – 1 of 1) sorted by relevance
| /external/pytorch/torch/distributed/ |
| D | _functional_collectives.py | 819 def _all_gather_into_tensor_coalesced_meta(self, tag, rankset, group_size): argument
842 def _all_gather_into_tensor_meta(shard, tag, rankset, group_size): argument
848 def _reduce_scatter_tensor_meta(input, reduce_op, tag, rankset, group_size): argument
870 def _reduce_scatter_tensor_coalesced_meta(inputs, reduceOp, tag, rankset, group_size): argument
|