Searched defs:lr_ptr (Results 1 – 5 of 5) sorted by relevance
18 const float* lr_ptr, in sgd_math()165 float* lr_ptr = nullptr; in _fused_sgd_with_momentum_kernel_cuda_() local306 float* lr_ptr = nullptr; in _fused_sgd_kernel_cuda_() local
32 const float* lr_ptr = nullptr; in _fused_adam_cuda_impl_() local78 const float* lr_ptr = lr.const_data_ptr<float>(); in _fused_adam_cuda_impl_() local
33 const float* lr_ptr = nullptr; in _fused_adamw_cuda_impl_() local79 const float* lr_ptr = lr.const_data_ptr<float>(); in _fused_adamw_cuda_impl_() local
37 const float* lr_ptr = nullptr; in _fused_adam_amsgrad_cuda_impl_() local88 const float* lr_ptr = lr.const_data_ptr<float>(); in _fused_adam_amsgrad_cuda_impl_() local
38 const float* lr_ptr = nullptr; in _fused_adamw_amsgrad_cuda_impl_() local89 const float* lr_ptr = lr.const_data_ptr<float>(); in _fused_adamw_amsgrad_cuda_impl_() local