Home
last modified time | relevance | path

Searched refs:_create_scaling_case (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/pytorch/torch/testing/_internal/
H A Dcommon_cuda.py264 def _create_scaling_case(device="cuda", dtype=torch.float, optimizer_ctor=torch.optim.SGD, optimize… function
/aosp_15_r20/external/pytorch/test/
H A Dtest_cuda_multigpu.py20 _create_scaling_case,
1210 ) = _create_scaling_case()
H A Dtest_cuda.py33 _create_scaling_case,
4647 ) = _create_scaling_case(
H A Dtest_torch.py61 _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)