Searched defs:exp_avg_sqs (Results 1 – 13 of 13) sorted by relevance
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
| H A D | FusedAdam.cpp | 23 at::TensorList exp_avg_sqs, in _fused_adam_kernel_cpu_()
80 at::TensorList exp_avg_sqs, in _fused_adam_kernel_cpu_()
99 at::TensorList exp_avg_sqs, in _fused_adamw_kernel_cpu_()
156 at::TensorList exp_avg_sqs, in _fused_adamw_kernel_cpu_()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/ |
| H A D | FusedAdamKernel.cu | 20 at::TensorList exp_avg_sqs, in _fused_adam_kernel_cuda_()
79 at::TensorList exp_avg_sqs, in _fused_adam_kernel_cuda_()
|
| H A D | FusedAdamWKernel.cu | 21 at::TensorList exp_avg_sqs, in _fused_adamw_kernel_cuda_()
80 at::TensorList exp_avg_sqs, in _fused_adamw_kernel_cuda_()
|
| H A D | fused_adam_impl.cu | 15 at::TensorList exp_avg_sqs, in _fused_adam_cuda_impl_()
61 at::TensorList exp_avg_sqs, in _fused_adam_cuda_impl_()
|
| H A D | fused_adamw_impl.cu | 16 at::TensorList exp_avg_sqs, in _fused_adamw_cuda_impl_()
62 at::TensorList exp_avg_sqs, in _fused_adamw_cuda_impl_()
|
| H A D | fused_adam_amsgrad_impl.cu | 15 at::TensorList exp_avg_sqs, in _fused_adam_amsgrad_cuda_impl_()
66 at::TensorList exp_avg_sqs, in _fused_adam_amsgrad_cuda_impl_()
|
| H A D | fused_adamw_amsgrad_impl.cu | 16 at::TensorList exp_avg_sqs, in _fused_adamw_amsgrad_cuda_impl_()
67 at::TensorList exp_avg_sqs, in _fused_adamw_amsgrad_cuda_impl_()
|
| /aosp_15_r20/external/pytorch/torch/optim/ |
| H A D | radam.py | 91 self, group, params_with_grad, grads, exp_avgs, exp_avg_sqs, state_steps argument
|
| H A D | nadam.py | 110 exp_avg_sqs, argument
|
| H A D | adamw.py | 116 exp_avg_sqs, argument
|
| H A D | adam.py | 120 exp_avg_sqs, 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 | 3158 exp_avg_sqs, argument
3184 exp_avg_sqs, argument
|