| /aosp_15_r20/external/pytorch/test/ |
| H A D | test_optim.py | 116 def test_optim_infos_do_not_specify_global_cliquey_kwargs( argument
126 def test_errors(self, device, dtype, optim_info): argument
163 self, device, dtype, optim_info, contiguous, with_lrsched argument
230 self, device, dtype, optim_info, with_lrsched argument
281 self, device, dtype, optim_info argument
318 def test_tensor_lr(self, device, dtype, optim_info): argument
377 def test_rosenbrock_sparse(self, device, dtype, optim_info, with_lrsched): argument
507 def test_complex(self, device, dtype, optim_info): argument
584 def test_complex_2d(self, device, dtype, optim_info): argument
724 optim_info, argument
[all …]
|
| H A D | test_cuda.py | 4461 def test_graph_optims(self, device, dtype, optim_info): argument
4539 def test_graph_scaling_fused_optimizers(self, device, dtype, optim_info): argument
4626 def test_grad_scaling_autocast_fused_optimizers(self, device, dtype, optim_info): argument
4707 self, device, dtype, optim_info, in_place_unscale argument
4742 def test_graph_grad_scaling(self, device, dtype, optim_info, foreach, fused): argument
|
| H A D | test_torch.py | 5894 def test_grad_scaling_autocast(self, device, dtype, optim_info, foreach, fused): argument
5948 …arams_invalidated_with_grads_invalidated_between_unscale_and_step(self, device, dtype, optim_info): argument
|
| /aosp_15_r20/external/pytorch/test/inductor/ |
| H A D | test_compiled_optimizers.py | 498 def test_correctness(self, device, dtype, optim_info, use_closure): argument
|
| /aosp_15_r20/external/pytorch/torch/testing/_internal/ |
| H A D | common_optimizers.py | 1182 device, dtype, optim_info, skip=() argument
|