Home
last modified time | relevance | path

Searched +path:pytorch +path:torch +path:distributed +path:examples (Results 1 – 11 of 11) sorted by relevance

/aosp_15_r20/external/pytorch/torch/distributed/examples/
H A Dmemory_tracker_example.py
/aosp_15_r20/external/pytorch/torch/distributed/tensor/examples/
H A Dtorchrec_sharding_example.py
H A Dconvnext_example.py
H A Dcomm_mode_features_example.py
H A Dvisualize_sharding_example.py
/aosp_15_r20/external/pytorch/torch/distributed/checkpoint/examples/
H A Dasync_checkpointing_example.py
H A Dstateful_example.py
H A Dfsdp_checkpoint_example.py
/aosp_15_r20/external/pytorch/torch/testing/_internal/distributed/rpc/examples/
H A Dparameter_server_test.py
H A D__init__.py
H A Dreinforcement_learning_rpc_test.py