Searched defs:is_first_step (Results 1 – 3 of 3) sorted by relevance
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/ |
| H A D | FusedSgdKernel.cu | 23 const bool is_first_step, in sgd_math()
155 const bool is_first_step, in _fused_sgd_with_momentum_kernel_cuda_()
201 const bool is_first_step, in _fused_sgd_with_momentum_kernel_cuda_()
277 const bool is_first_step, in _fused_sgd_kernel_cuda_()
341 const bool is_first_step, in _fused_sgd_kernel_cuda_()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cpu/ |
| H A D | FusedSGDKernel.cpp | 28 const bool is_first_step, in sgd_math()
120 const bool is_first_step, in sgd_math()
195 const bool is_first_step, in sgd_fused_step_impl()
245 const bool is_first_step, in fused_sgd_kernel()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
| H A D | FusedSGD.cpp | 28 const bool is_first_step, in _fused_sgd_kernel_cpu_()
73 const bool is_first_step, in _fused_sgd_kernel_cpu_()
|