Searched defs:init_rrefs (Results 1 – 3 of 3) sorted by relevance
| /external/pytorch/torch/distributed/_shard/sharded_tensor/ |
| D | __init__.py | 35 init_rrefs=False, argument
90 init_rrefs=False, argument
144 init_rrefs=False, argument
200 init_rrefs=False, argument
254 init_rrefs=False, argument
310 init_rrefs=False, argument
359 local_shards: List[Shard], *global_size, process_group=None, init_rrefs=False argument
|
| D | api.py | 254 init_rrefs=False, argument
287 def _prepare_init(self, process_group=None, init_rrefs=False): argument
721 init_rrefs=False, argument
782 init_rrefs=False, argument
882 init_rrefs=False, argument
|
| /external/pytorch/torch/testing/_internal/distributed/_shard/sharded_tensor/ |
| D | _test_st_common.py | 39 init_rrefs=True argument
56 init_rrefs=True argument
|