Home
last modified time | relevance | path

Searched refs:load_with_process_group (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/pytorch/torch/distributed/_shard/
H A D__init__.py1 from .api import _shard_tensor, load_with_process_group, shard_module, shard_parameter
H A Dapi.py136 def load_with_process_group(process_group): function
/aosp_15_r20/external/pytorch/test/distributed/_shard/sharded_tensor/
H A Dtest_sharded_tensor.py19 load_with_process_group,
1291 with load_with_process_group(pg):
1363 with load_with_process_group(pg):
1369 with load_with_process_group(pg):