Searched defs:l2_regularization (Results 1 – 5 of 5) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/cuda_impl/ |
| D | ftrl_impl.cu | 51 … const T *l1_regularization, const T *l2_regularization, const T *learning_rate_power, in ApplyFtrlKernel()
73 … const T *l2_regularization, const T *learning_rate_power, T *variable, T *accumulation, T *linear, in ApplyFtrl()
|
| D | sparse_apply_proximal_adagrad_impl.cu | 61 … const T *l1_regularization, const T *l2_regularization, in SparseApplyProximalAdagradUpdate()
85 … const T *l1_regularization, const T *l2_regularization, const T *gradient, in CalSparseApplyProximalAdagrad()
|
| D | sparse_ftrl_impl.cu | 54 … const float l2_regularization, const float learning_rate_power, in SparseApplyFtrlKernel()
85 … const float learning_rate, const float l1_regularization, const float l2_regularization, in CalSparseApplyFtrl()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/nn/ |
| D | ftrl_gpu_kernel.h | 58 T *l2_regularization = GetDeviceAddress<T>(inputs, 6); in Launch() local
|
| D | sparse_apply_proximal_adagrad_kernel.h | 52 T *l2_regularization = GetDeviceAddress<T>(inputs, 4); in Launch() local
|