Searched defs:optimizer_ctor (Results 1 – 3 of 3) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/testing/_internal/ |
| H A D | common_cuda.py | 246 def _create_scaling_models_optimizers(device="cuda", optimizer_ctor=torch.optim.SGD, optimizer_kwar… argument
264 def _create_scaling_case(device="cuda", dtype=torch.float, optimizer_ctor=torch.optim.SGD, optimize… argument
|
| /aosp_15_r20/external/pytorch/test/ |
| H A D | test_cuda.py | 2887 self, steps_warmup, steps_train, optimizer_ctor, kwargs argument
|
| H A D | test_torch.py | 5852 …def _run_scaling_case(self, device, run, unskipped, skipped, atol=1e-7, optimizer_ctor=torch.optim… argument
|