Searched defs:momentum_buffer_list (Results 1 – 4 of 4) sorted by relevance
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/ |
| H A D | FusedSgdKernel.cu | 148 at::TensorList momentum_buffer_list, in _fused_sgd_with_momentum_kernel_cuda_()
194 at::TensorList momentum_buffer_list, in _fused_sgd_with_momentum_kernel_cuda_()
270 at::TensorList momentum_buffer_list, in _fused_sgd_kernel_cuda_()
334 at::TensorList momentum_buffer_list, in _fused_sgd_kernel_cuda_()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
| H A D | FusedSGD.cpp | 21 at::TensorList momentum_buffer_list, in _fused_sgd_kernel_cpu_()
66 at::TensorList momentum_buffer_list, in _fused_sgd_kernel_cpu_()
|
| /aosp_15_r20/external/pytorch/torch/optim/ |
| H A D | sgd.py | 80 def _init_group(self, group, params, grads, momentum_buffer_list): argument
|
| H A D | rmsprop.py | 97 momentum_buffer_list, argument
|