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)
|