Home
last modified time | relevance | path

Searched defs:exp_avgs (Results 1 – 14 of 14) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A DFusedAdam.cpp22 at::TensorList exp_avgs, in _fused_adam_kernel_cpu_()
79 at::TensorList exp_avgs, in _fused_adam_kernel_cpu_()
98 at::TensorList exp_avgs, in _fused_adamw_kernel_cpu_()
155 at::TensorList exp_avgs, in _fused_adamw_kernel_cpu_()
/aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/
H A DFusedAdamKernel.cu19 at::TensorList exp_avgs, in _fused_adam_kernel_cuda_()
78 at::TensorList exp_avgs, in _fused_adam_kernel_cuda_()
H A DFusedAdamWKernel.cu20 at::TensorList exp_avgs, in _fused_adamw_kernel_cuda_()
79 at::TensorList exp_avgs, in _fused_adamw_kernel_cuda_()
H A Dfused_adam_impl.cu14 at::TensorList exp_avgs, in _fused_adam_cuda_impl_()
60 at::TensorList exp_avgs, in _fused_adam_cuda_impl_()
H A Dfused_adamw_impl.cu15 at::TensorList exp_avgs, in _fused_adamw_cuda_impl_()
61 at::TensorList exp_avgs, in _fused_adamw_cuda_impl_()
H A Dfused_adam_amsgrad_impl.cu14 at::TensorList exp_avgs, in _fused_adam_amsgrad_cuda_impl_()
65 at::TensorList exp_avgs, in _fused_adam_amsgrad_cuda_impl_()
H A Dfused_adamw_amsgrad_impl.cu15 at::TensorList exp_avgs, in _fused_adamw_amsgrad_cuda_impl_()
66 at::TensorList exp_avgs, in _fused_adamw_amsgrad_cuda_impl_()
/aosp_15_r20/external/pytorch/torch/optim/
H A Dadamax.py87 self, group, params_with_grad, grads, exp_avgs, exp_infs, state_steps argument
H A Dradam.py91 self, group, params_with_grad, grads, exp_avgs, exp_avg_sqs, state_steps argument
H A Dnadam.py109 exp_avgs, argument
H A Dadamw.py115 exp_avgs, argument
H A Dadam.py119 exp_avgs, argument
/aosp_15_r20/external/pytorch/test/
H A Dtest_proxy_tensor.py867 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.py3157 exp_avgs, argument
3183 exp_avgs, argument