Searched defs:nesterov (Results 1 – 11 of 11) sorted by relevance
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/ |
| H A D | FusedSgdKernel.cu | 21 const bool nesterov, in sgd_math()
153 const bool nesterov, in _fused_sgd_with_momentum_kernel_cuda_()
199 const bool nesterov, in _fused_sgd_with_momentum_kernel_cuda_()
275 const bool nesterov, in _fused_sgd_kernel_cuda_()
339 const bool nesterov, in _fused_sgd_kernel_cuda_()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cpu/ |
| H A D | FusedSGDKernel.cpp | 26 const bool nesterov, in sgd_math()
118 const bool nesterov, in sgd_math()
193 const bool nesterov, in sgd_fused_step_impl()
243 const bool nesterov, in fused_sgd_kernel()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
| H A D | FusedSGD.cpp | 26 const bool nesterov, in _fused_sgd_kernel_cpu_()
71 const bool nesterov, in _fused_sgd_kernel_cpu_()
|
| /aosp_15_r20/external/executorch/extension/training/optimizer/ |
| H A D | sgd.h | 115 bool nesterov() const { in nesterov() function
|
| H A D | sgd.cpp | 91 auto nesterov = options.nesterov(); in step() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/keras/optimizer_v2/ |
| H A D | gradient_descent.py | 103 nesterov=False, argument
|
| /aosp_15_r20/external/executorch/extension/training/pybindings/ |
| H A D | _training_lib.cpp | 129 bool nesterov = false) { in get_sgd_optimizer()
|
| /aosp_15_r20/external/pytorch/torch/csrc/jit/mobile/train/optim/ |
| H A D | sgd.cpp | 91 auto nesterov = options.nesterov(); in step() local
|
| /aosp_15_r20/external/pytorch/torch/csrc/api/src/optim/ |
| H A D | sgd.cpp | 74 auto nesterov = options.nesterov(); in step() local
|
| /aosp_15_r20/external/pytorch/torch/optim/ |
| H A D | sgd.py | 32 nesterov=False, argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/keras/ |
| H A D | optimizer_v1.py | 175 def __init__(self, lr=0.01, momentum=0., decay=0., nesterov=False, **kwargs): argument
|