| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
| H A D | FusedAdam.cpp | 25 at::TensorList state_steps, in _fused_adam_kernel_cpu_()
82 at::TensorList state_steps, in _fused_adam_kernel_cpu_()
101 at::TensorList state_steps, in _fused_adamw_kernel_cpu_()
158 at::TensorList state_steps, in _fused_adamw_kernel_cpu_()
|
| H A D | FusedAdagrad.cpp | 21 at::TensorList state_steps, in _fused_adagrad_kernel_cpu_()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/ |
| H A D | FusedAdamKernel.cu | 22 at::TensorList state_steps, in _fused_adam_kernel_cuda_()
81 at::TensorList state_steps, in _fused_adam_kernel_cuda_()
|
| H A D | FusedAdamWKernel.cu | 23 at::TensorList state_steps, in _fused_adamw_kernel_cuda_()
82 at::TensorList state_steps, in _fused_adamw_kernel_cuda_()
|
| H A D | fused_adam_impl.cu | 16 at::TensorList state_steps, in _fused_adam_cuda_impl_()
62 at::TensorList state_steps, in _fused_adam_cuda_impl_()
|
| H A D | fused_adamw_impl.cu | 17 at::TensorList state_steps, in _fused_adamw_cuda_impl_()
63 at::TensorList state_steps, in _fused_adamw_cuda_impl_()
|
| H A D | fused_adam_amsgrad_impl.cu | 17 at::TensorList state_steps, in _fused_adam_amsgrad_cuda_impl_()
68 at::TensorList state_steps, in _fused_adam_amsgrad_cuda_impl_()
|
| H A D | fused_adamw_amsgrad_impl.cu | 18 at::TensorList state_steps, in _fused_adamw_amsgrad_cuda_impl_()
69 at::TensorList state_steps, in _fused_adamw_amsgrad_cuda_impl_()
|
| H A D | MultiTensorApply.cuh | 300 at::TensorList state_steps, in multi_tensor_apply_for_fused_optimizer()
|
| /aosp_15_r20/external/pytorch/torch/optim/ |
| H A D | rprop.py | 78 def _init_group(self, group, params, grads, prevs, step_sizes, state_steps): argument
|
| H A D | adamax.py | 87 self, group, params_with_grad, grads, exp_avgs, exp_infs, state_steps argument
|
| H A D | asgd.py | 86 def _init_group(self, group, params_with_grad, grads, mus, axs, etas, state_steps): argument
|
| H A D | radam.py | 91 self, group, params_with_grad, grads, exp_avgs, exp_avg_sqs, state_steps argument
|
| H A D | rmsprop.py | 99 state_steps, argument
|
| H A D | _adafactor.py | 77 state_steps, argument
|
| H A D | nadam.py | 112 state_steps, argument
|
| H A D | adagrad.py | 123 def _init_group(self, group, params_with_grad, grads, state_sums, state_steps): argument
|
| H A D | adamw.py | 118 state_steps, argument
|
| H A D | adam.py | 122 state_steps, argument
|
| /aosp_15_r20/external/pytorch/test/ |
| H A D | test_proxy_tensor.py | 867 def fused_adam(params, grads, exp_avgs, exp_avg_sqs, max_exp_avg_sqs, state_steps): argument
|
| /aosp_15_r20/external/pytorch/torch/ |
| H A D | _meta_registrations.py | 3160 state_steps, argument
3186 state_steps, argument
|