| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/cuda_impl/ |
| D | momentum_impl.cu | 19 __global__ void MomentumUpdateVariableKernel(const size_t size, T *variable, T *accumulation, const… in MomentumUpdateVariableKernel()
34 __global__ void MomentumUpdateVariableKernel(const size_t size, half *variable, half *accumulation, in MomentumUpdateVariableKernel()
51 …global__ void MomentumUpdateVariableKernel(const size_t size, float *variable, float *accumulation, in MomentumUpdateVariableKernel()
67 void MomentumUpdateVariable(const size_t size, T *variable, T *accumulation, const S *learning_rate… in MomentumUpdateVariable()
75 … T *accumulation, const T *learning_rate, const S *gradient, in FusedMomentumWeightDecayScaleKernel()
85 …cayScaleMomentum(const size_t element_num, T *weight_decay, T *scale, T *variable, T *accumulation, in FusedWeightDecayScaleMomentum()
95 …l__ void FusedMomentumScaleKernel(const size_t element_num, T *scale, T *variable, T *accumulation, in FusedMomentumScaleKernel()
104 void FusedScaleMomentum(const size_t element_num, T *scale, T *variable, T *accumulation, const T *… in FusedScaleMomentum()
113 …dWeightDecayMomentumKernel(const size_t element_num, T *weight_decay, T *variable, T *accumulation, in FusedWeightDecayMomentumKernel()
123 …d FusedWeightDecayMomentum(const size_t element_num, T *weight_decay, T *variable, T *accumulation, in FusedWeightDecayMomentum()
[all …]
|
| D | adagrad_impl.cu | 35 T *accumulation) { in ApplyAdagradKernel()
50 half *accumulation) { in ApplyAdagradKernel()
65 float *accumulation) { in ApplyAdagradKernel()
80 float *accumulation) { in ApplyAdagradKernel()
95 half *accumulation) { in ApplyAdagradKernel()
110 T *accumulation, in ApplyAdagrad()
|
| D | ftrl_impl.cu | 52 T *variable, T *accumulation, T *linear) { 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 | 62 … const T *gradient, const int *indices, T *variable, T *accumulation, in SparseApplyProximalAdagradUpdate()
86 … const int *indices, T *variable, T *accumulation, T *variable_out, in CalSparseApplyProximalAdagrad()
|
| D | sparse_ftrl_impl.cu | 55 T *variable, T *accumulation, T *linear) { in SparseApplyFtrlKernel()
86 … const float learning_rate_power, const bool use_locking, T *variable, T *accumulation, in CalSparseApplyFtrl()
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/gpu/ |
| D | apply_momentum_weight_fusion.cc | 43 auto accumulation = utils::cast<AnfNodePtr>((*equiv)[accumulation_]); in Process() local
|
| D | apply_momentum_scale_fusion.cc | 71 auto accumulation = utils::cast<AnfNodePtr>((*equiv)[accumulation_]); in Process() local
|
| D | apply_momentum_weight_scale_fusion.cc | 104 auto accumulation = utils::cast<AnfNodePtr>((*equiv)[accumulation_]); in Process() local
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/nn/ |
| D | fused_weightdecay_momentum_gpu_kernel.h | 43 T *accumulation = GetDeviceAddress<T>(inputs, 2); in Launch() local
|
| D | fused_scale_momentum_gpu_kernel.h | 43 T *accumulation = GetDeviceAddress<T>(inputs, 2); in Launch() local
|
| D | fused_weightdecay_scale_momentum_gpu_kernel.h | 44 T *accumulation = GetDeviceAddress<T>(inputs, 3); in Launch() local
|
| D | adagrad_gpu_kernel.h | 50 T *accumulation = GetDeviceAddress<T>(inputs, 1); in Launch() local
|
| D | momentum_gpu_kernel.h | 49 T *accumulation = GetDeviceAddress<T>(inputs, 1); in Launch() local
|
| D | ftrl_gpu_kernel.h | 53 T *accumulation = GetDeviceAddress<T>(inputs, 1); in Launch() local
|
| D | sparse_apply_proximal_adagrad_kernel.h | 49 T *accumulation = GetDeviceAddress<T>(inputs, 1); in Launch() local
|
| D | sparse_ftrl_gpu_kernel.h | 43 T *accumulation = GetDeviceAddress<T>(inputs, 1); in Launch() local
|
| /third_party/mindspore/tests/vm_impl/ |
| D | nn_ops_vm_impl.py | 336 accumulation, argument
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/ |
| D | cpu_kernel.cc | 85 size_t accumulation = 1; in GetElementNumEveryDim() local
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/mem_reuse/ |
| D | mem_swap_manager.cc | 78 size_t accumulation = 0; in InitSwapThreshold() local
|