Searched refs:_create_scaling_case (Results 1 – 4 of 4) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/testing/_internal/ |
| H A D | common_cuda.py | 264 def _create_scaling_case(device="cuda", dtype=torch.float, optimizer_ctor=torch.optim.SGD, optimize… function
|
| /aosp_15_r20/external/pytorch/test/ |
| H A D | test_cuda_multigpu.py | 20 _create_scaling_case,
1210 ) = _create_scaling_case()
|
| H A D | test_cuda.py | 33 _create_scaling_case,
4647 ) = _create_scaling_case(
|
| H A D | test_torch.py | 61 _create_scaling_case, _create_scaling_models_optimizers)
5857 …) = _create_scaling_case(device=device, optimizer_ctor=optimizer_ctor, optimizer_kwargs=optimizer_…
5954 model, _, optimizer, _, data, loss_fn, _ = _create_scaling_case(
6112 _create_scaling_case(device.type)
|