| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/cuda_impl/ |
| D | rmsprop_impl.cu | 23 T* mean_square, T*moment, T* gradients, const size_t size) { in RmsPropKernel()
33 …T* variable, T* mean_square, T* moment, T* gradients, const size_t size, cudaStream_t cuda_stream)… in RmsProp()
40 … T* variable, T* mean_gradients, T* mean_square, T*moment, T* gradients, in RmsPropCenterKernel()
53 T* mean_gradients, T* mean_square, T*moment, T* gradients, const size_t size, in RmsPropCenter()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/ |
| D | rmsprop_cpu_kernel.cc | 30 …RMSPropCPUKernel<T>::LaunchRMSPropUnuseCenter(T *variable, T *mean_square, T *moment, T *gradients, in LaunchRMSPropUnuseCenter()
51 …d RMSPropCPUKernel<T>::LaunchRMSPropUseCenter(T *variable, T *mean_square, T *moment, T *gradients, in LaunchRMSPropUseCenter()
106 float *gradients = reinterpret_cast<float *>(inputs[4]->addr); in Launch() local
117 T *gradients = reinterpret_cast<float *>(inputs[4]->addr); in Launch() local
|
| /third_party/mindspore/mindspore/lite/include/ |
| D | lite_session.h | 210 std::vector<tensor::MSTensor *> gradients; in GetGradients() local
218 …virtual int ApplyGradients(const std::vector<tensor::MSTensor *> &gradients) { return mindspore::l… in ApplyGradients()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/nn/ |
| D | rmsprop_gpu_kernel.h | 48 T *gradients = GetDeviceAddress<T>(inputs, 4); in Launch() local
57 T *gradients = GetDeviceAddress<T>(inputs, 4); in Launch() local
|
| /third_party/mindspore/mindspore/nn/optim/ |
| D | thor.py | 94 def clip_gradient(enable_clip_grad, gradients): argument
515 def _get_ainv_ginv_list(self, gradients, damping_step, matrix_a_allreduce, matrix_g_allreduce): argument
585 def construct(self, gradients): argument
916 …def _get_fc_ainv_ginv(self, index, damping_step, gradients, matrix_a_allreduce, matrix_g_allreduce, argument
995 …def _get_ainv_ginv_amax_gmax_list(self, gradients, damping_step, matrix_a_allreduce, matrix_g_allr… argument
1116 def _get_second_gradients_one(self, params_len, gradients, new_grads): argument
1157 def _get_second_gradients(self, new_grads, damping_step, gradients): argument
1248 def construct(self, gradients): argument
|
| D | optimizer.py | 286 def decay_weight(self, gradients): argument
310 def gradients_centralization(self, gradients): argument
329 def scale_grad(self, gradients): argument
349 def _grad_sparse_indices_deduplicate(self, gradients): argument
|
| D | adam.py | 341 def construct(self, gradients): argument
492 def construct(self, gradients): argument
646 def construct(self, gradients): argument
|
| D | momentum.py | 162 def construct(self, gradients): argument
|
| D | sgd.py | 177 def construct(self, gradients): argument
|
| D | lars.py | 155 def construct(self, gradients): argument
|
| D | rmsprop.py | 206 def construct(self, gradients): argument
|
| D | lamb.py | 312 def construct(self, gradients): argument
|
| D | lazyadam.py | 260 def construct(self, gradients): argument
|
| /third_party/mindspore/tests/st/networks/models/resnet50/src_thor/ |
| D | thor.py | 313 …def _get_Ainv_Ginv_Amax_Gmax_list(self, gradients, damping_step, matrix_a_allreduce, matrix_g_allr… argument
421 def _get_second_gradients(self, new_grads, damping_step, gradients): argument
441 def construct(self, gradients): argument
|
| /third_party/mindspore/tests/st/ops/gpu/ |
| D | test_rmsprop.py | 66 def rmsprop_numpy(variable, gradients, mean_square, moment, argument
74 def rmspropcented_numpy(variable, gradients, mean_gradients, mean_square, moment, argument
|
| /third_party/mindspore/tests/st/ops/cpu/ |
| D | test_rmsprop.py | 66 def rmsprop_numpy(variable, gradients, mean_square, moment, argument
74 def rmspropcented_numpy(variable, gradients, mean_gradients, mean_square, moment, argument
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp32/ |
| D | rmsprop_fp32.c | 31 int RMSPropUnuseCenterFp32(float *variable, float *mean_square, float *moment, float *gradients, fl… in RMSPropUnuseCenterFp32()
80 int RMSPropUseCenterFp32(float *variable, float *mean_square, float *moment, float *gradients, floa… in RMSPropUseCenterFp32()
|
| /third_party/mindspore/tests/st/ops/ascend/test_tbe_ops/ |
| D | test_stridedslice_grad.py | 50 def me_stridedslice(input_, begin, end, stride, gradients): argument
|
| D | test_layernorm_grad.py | 48 def py_me_layernorm_grad(input_data, normalized_shape, gamma, beta, axis, gradients): argument
|
| /third_party/mindspore/mindspore/ops/_grad_experimental/ |
| D | grad_nn_ops.py | 68 def bprop(features, out, gradients): argument
|
| /third_party/mindspore/tests/st/fl/mobile/src/ |
| D | adam.py | 296 def construct(self, gradients, overflow): argument
402 def construct(self, gradients): argument
|
| /third_party/mindspore/tests/st/fl/albert/src/ |
| D | adam.py | 296 def construct(self, gradients, overflow): argument
402 def construct(self, gradients): argument
|
| /third_party/mindspore/tests/st/fl/hybrid_lenet/src/ |
| D | adam.py | 295 def construct(self, gradients, overflow): argument
401 def construct(self, gradients): argument
|
| /third_party/mindspore/tests/mindspore_test_framework/utils/ |
| D | model_util.py | 79 def construct(self, gradients): argument
|
| /third_party/mindspore/mindspore/explainer/ |
| D | _utils.py | 59 def abs_max(gradients): argument
|