Searched defs:init_rrefs (Results 1 – 3 of 3) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/distributed/_shard/sharded_tensor/ |
| H A 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
|
| H A 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
|
| /aosp_15_r20/external/pytorch/torch/testing/_internal/distributed/_shard/sharded_tensor/ |
| H A D | _test_st_common.py | 39 init_rrefs=True argument
56 init_rrefs=True argument
|