Home
last modified time | relevance | path

Searched defs:task_parameters (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/pytorch/test/distributed/
H A Dtest_c10d_ucc.py426 def task_parameters(self): member in DistributedDataParallelTest._test_global_local_unused_params_grad.GlobalLocalUnusedParamModule
515 def task_parameters(self): member in DistributedDataParallelTest.test_find_unused_parameters_when_unused_parameters_empty.FindUnusedParamModule
H A Dtest_c10d_gloo.py1567 def task_parameters(self): member in DistributedDataParallelTest._test_global_local_unused_params_grad.GlobalLocalUnusedParamModule
1648 def task_parameters(self): member in DistributedDataParallelTest.test_find_unused_parameters_when_unused_parameters_empty.FindUnusedParamModule
/aosp_15_r20/external/pytorch/torch/testing/_internal/distributed/
H A Ddistributed_test.py7327 def task_parameters(self): member in DistributedTest._DistTestBase.test_ddp_uneven_inputs.UnusedParamModule