Home
last modified time | relevance | path

Searched defs:with_comms (Results 1 – 5 of 5) sorted by relevance

/aosp_15_r20/external/pytorch/torch/testing/_internal/distributed/_shard/sharded_tensor/
H A D__init__.py83 def with_comms(func=None, init_rpc=True, backend="nccl"): function
/aosp_15_r20/external/pytorch/test/distributed/
H A Dtest_c10d_logger.py35 def with_comms(func=None): function
H A Dtest_c10d_object_collectives.py30 def with_comms(func=None): function
H A Dtest_functional_api.py441 def with_comms(func=None): function
/aosp_15_r20/external/pytorch/torch/testing/_internal/distributed/_tensor/
H A Dcommon_dtensor.py359 def with_comms(func: TestFunc) -> TestFunc: function