Searched defs:ddp_model (Results 1 – 3 of 3) sorted by relevance
| /aosp_15_r20/external/pytorch/benchmarks/distributed/rpc/parameter_server/trainer/ |
| H A D | iteration_steps.py | 2 self, ddp_model, criterion, optimizer, hook_state, epoch, index, batch argument
|
| /aosp_15_r20/external/pytorch/test/distributed/ |
| H A D | test_c10d_ucc.py | 639 def _run_and_verify_sparse_gradients(self, vanilla_model, ddp_model): argument
|
| H A D | test_c10d_gloo.py | 1814 def _run_and_verify_sparse_gradients(self, vanilla_model, ddp_model): argument
|