| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/ |
| H A D | FusedSgdKernel.cu | 16 const double weight_decay, in sgd_math()
149 const double weight_decay, in _fused_sgd_with_momentum_kernel_cuda_()
195 const double weight_decay, in _fused_sgd_with_momentum_kernel_cuda_()
271 const double weight_decay, in _fused_sgd_kernel_cuda_()
335 const double weight_decay, in _fused_sgd_kernel_cuda_()
|
| H A D | FusedAdamKernel.cu | 26 const double weight_decay, in _fused_adam_kernel_cuda_()
85 const double weight_decay, in _fused_adam_kernel_cuda_()
|
| H A D | FusedAdamWKernel.cu | 27 const double weight_decay, in _fused_adamw_kernel_cuda_()
86 const double weight_decay, in _fused_adamw_kernel_cuda_()
|
| H A D | fused_adam_impl.cu | 20 const double weight_decay, in _fused_adam_cuda_impl_()
66 const double weight_decay, in _fused_adam_cuda_impl_()
|
| H A D | fused_adamw_impl.cu | 21 const double weight_decay, in _fused_adamw_cuda_impl_()
67 const double weight_decay, in _fused_adamw_cuda_impl_()
|
| H A D | fused_adam_amsgrad_impl.cu | 21 const double weight_decay, in _fused_adam_amsgrad_cuda_impl_()
72 const double weight_decay, in _fused_adam_amsgrad_cuda_impl_()
|
| H A D | fused_adamw_amsgrad_impl.cu | 22 const double weight_decay, in _fused_adamw_amsgrad_cuda_impl_()
73 const double weight_decay, in _fused_adamw_amsgrad_cuda_impl_()
|
| H A D | fused_adam_utils.cuh | 32 const double& weight_decay, in adam_math()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
| H A D | FusedAdam.cpp | 29 const double weight_decay, in _fused_adam_kernel_cpu_()
86 const double weight_decay, in _fused_adam_kernel_cpu_()
105 const double weight_decay, in _fused_adamw_kernel_cpu_()
162 const double weight_decay, in _fused_adamw_kernel_cpu_()
|
| H A D | FusedSGD.cpp | 22 const double weight_decay, in _fused_sgd_kernel_cpu_()
67 const double weight_decay, in _fused_sgd_kernel_cpu_()
|
| H A D | FusedAdagrad.cpp | 24 const double weight_decay, in _fused_adagrad_kernel_cpu_()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cpu/ |
| H A D | FusedAdagradKernel.cpp | 24 const double weight_decay, in adagrad_math()
93 const double weight_decay, in adagrad_math()
147 const double weight_decay, in adagrad_fused_step_impl()
194 const double weight_decay, in fused_adagrad_kernel()
|
| H A D | FusedSGDKernel.cpp | 22 const double weight_decay, in sgd_math()
114 const double weight_decay, in sgd_math()
189 const double weight_decay, in sgd_fused_step_impl()
239 const double weight_decay, in fused_sgd_kernel()
|
| H A D | FusedAdamKernel.cpp | 31 double weight_decay, in adam_math()
174 double weight_decay, in adam_math()
276 const double weight_decay, in adam_fused_step_impl()
347 const double weight_decay, in fused_adam_kernel()
|
| /aosp_15_r20/external/executorch/extension/training/optimizer/ |
| H A D | sgd.h | 111 double weight_decay() const { in weight_decay() function
|
| H A D | sgd.cpp | 88 auto weight_decay = options.weight_decay(); in step() local
|
| /aosp_15_r20/external/executorch/extension/training/pybindings/ |
| H A D | _training_lib.cpp | 128 double weight_decay = 0, in get_sgd_optimizer()
|
| /aosp_15_r20/external/pytorch/torch/csrc/jit/mobile/train/optim/ |
| H A D | sgd.cpp | 88 auto weight_decay = options.weight_decay(); in step() local
|
| /aosp_15_r20/external/pytorch/torch/csrc/api/src/optim/ |
| H A D | sgd.cpp | 71 auto weight_decay = options.weight_decay(); in step() local
|
| /aosp_15_r20/external/pytorch/torch/ |
| H A D | _meta_registrations.py | 3165 weight_decay, argument
3191 weight_decay, argument
|