| /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 void RMSPropCPUKernel<T>::LaunchRMSPropUnuseCenter(T *variable, T *mean_square, T *moment, T *gradi… in LaunchRMSPropUnuseCenter()
51 void RMSPropCPUKernel<T>::LaunchRMSPropUseCenter(T *variable, T *mean_square, T *moment, T *gradien… in LaunchRMSPropUseCenter()
103 float *mean_square = reinterpret_cast<float *>(inputs[1]->addr); in Launch() local
115 T *mean_square = reinterpret_cast<float *>(inputs[2]->addr); in Launch() local
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/nn/ |
| D | rmsprop_gpu_kernel.h | 45 T *mean_square = GetDeviceAddress<T>(inputs, 1); in Launch() local
55 T *mean_square = GetDeviceAddress<T>(inputs, 2); in Launch() local
|
| /third_party/boost/libs/numeric/ublas/test/common/ |
| D | testhelper.hpp | 119 typename distance<AE>::type mean_square(const boost::numeric::ublas::matrix_expression<AE> &me) { in mean_square() function
131 typename distance<AE>::type mean_square(const boost::numeric::ublas::vector_expression<AE> &ve) { in mean_square() function
|
| /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/auto_monad/ |
| D | test_effect_optimizer.py | 225 def construct(self, mean_grad, mean_square, moment, grad, learning_rate): argument
436 def construct(self, mean_square, moment, learning_rate, grad): argument
|
| /third_party/mindspore/mindspore/ops/operations/ |
| D | nn_ops.py | 3147 def infer_value(self, var, mean_square, moment, learning_rate, grad, decay, momentum, epsilon): argument
|