| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/cuda_impl/cuda_ops/ |
| D | rmsprop_impl.cu | 72 … const T *decay, const T *momentum, const T *epsilon, T *variable, T *mean_square, in RmsPropKernel()
87 const Complex<double> *learning_rate, const Complex<double> *decay, in RmsPropKernel()
103 const Complex<float> *decay, const Complex<float> *momentum, in RmsPropKernel()
117 …sProp(const size_t batch_size, const size_t input_elements, const T *learning_rate, const T *decay, in RmsProp()
128 … const T *decay, const T *momentum, const T *epsilon, T *variable, T *mean_gradients, in RmsPropCenterKernel()
146 … const float *decay, const float *momentum, const float *epsilon, float *variable, in RmsPropCenterKernel()
163 … const Complex<double> *learning_rate, const Complex<double> *decay, in RmsPropCenterKernel()
184 … const Complex<float> *learning_rate, const Complex<float> *decay, in RmsPropCenterKernel()
205 … const half *decay, const half *momentum, const half *epsilon, half *variable, in RmsPropCenterKernel()
224 …enter(const size_t batch_size, const size_t input_elements, const T *learning_rate, const T *decay, in RmsPropCenter()
|
| D | adam_impl.cu | 52 … const float *beta1, const float *beta2, const float *epsilon, const float *decay, in AdamWeightDecayKernel()
66 … const float *beta1, const float *beta2, const float *epsilon, const float *decay, in AdamWeightDecayKernel()
80 … const float *beta1, const float *beta2, const float *epsilon, const float *decay, in AdamWeightDecayKernel()
107 … const float *beta2, const float *epsilon, const float *decay, S *variable, T *m, T *v, in AdamWeightDecayOp()
|
| D | lamb_impl.cu | 24 … const float *epsilon, const float *decay, const int32_t *global_step, in ApplyLambEralyKernel()
44 const float *beta2, const float *epsilon, const float *decay, in ApplyLambEralyKernel()
85 … const float *epsilon, const float *decay, const int32_t *global_step, const T *gradient, in ApplyLambEraly()
|
| /third_party/mindspore/mindspore-src/source/tests/st/ops/cpu/ |
| D | test_rmsprop.py | 30 def __init__(self, lr, decay, momentum, epsilon, var, g, mg, rms, mom): argument
49 def __init__(self, lr, decay, momentum, epsilon, var, g, mg, rms, mom): argument
67 learning_rate, decay, momentum, epsilon): argument
75 learning_rate, decay, momentum, epsilon): argument
|
| /third_party/mindspore/mindspore-src/source/tests/st/ops/gpu/ |
| D | test_rmsprop.py | 30 def __init__(self, lr, decay, momentum, epsilon, var, g, mg, rms, mom): argument
49 def __init__(self, lr, decay, momentum, epsilon, var, g, mg, rms, mom): argument
67 learning_rate, decay, momentum, epsilon): argument
75 learning_rate, decay, momentum, epsilon): argument
|
| /third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/nn/optim/ |
| D | rmsprop.py | 29 def _rmsprop_opt_(opt, decay, epsilon, momentum, learning_rate, weight, ms, mom, grad): argument
42 def _centered_rmsprop_opt_(opt, decay, epsilon, momentum, learning_rate, weight, mg, ms, mom, grad): argument
212 def __init__(self, params, learning_rate=0.1, decay=0.9, momentum=0.0, epsilon=1e-10, argument
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/ |
| D | rmsprop_cpu_kernel.cc | 48 … float *learning_rate, float *decay, float *momentum, in LaunchRMSPropUnuseCenter()
80 … T *mean_gradients, T *momentum, T *learning_rate, T *decay, in LaunchRMSPropUseCenter()
112 … std::complex<float> *learning_rate, std::complex<float> *decay, in LaunchRMSPropUseCenter()
145 … std::complex<double> *learning_rate, std::complex<double> *decay, in LaunchRMSPropUseCenter()
269 float *decay = GetDeviceAddress<float>(inputs, kNumberFive); in LaunchKernel() local
284 T *decay = static_cast<T *>(inputs[kNumberSix]->device_ptr()); in LaunchKernel() local
|
| D | adam_weight_decay_cpu_kernel.cc | 42 T decay = static_cast<T>(reinterpret_cast<float *>(inputs[kIndex7]->device_ptr())[kScalarIndex]); in LaunchAdamWeightDecay() local
73 auto decay = reinterpret_cast<float *>(inputs[kIndex7]->device_ptr())[kScalarIndex]; in LaunchAdamWeightDecayNnacl() local
|
| D | fused_cast_adam_weight_decay_cpu_kernel.cc | 51 auto decay = reinterpret_cast<float *>(inputs[kDecayIndex]->device_ptr())[kScalarIndex]; in LaunchFusedCastAdamFp32() local
106 auto decay = reinterpret_cast<float *>(inputs[kDecayIndex]->device_ptr())[kScalarIndex]; in LaunchFusedCastAdamFp16() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/include/nnacl/sse/ |
| D | adam_fp32_sse.h | 31 …ex, float *var, float *m, float *v, float lr, float beta1, float beta2, float epsilon, float decay, in AdamWeightDecayFp32SSE()
64 …_t *gradient16, float *m, float *v, float lr, float beta1, float beta2, float epsilon, float decay, in FusedCastAdamFp32Fp16SSE()
99 …at *gradient32, float *m, float *v, float lr, float beta1, float beta2, float epsilon, float decay, in FusedCastAdamFp32Fp32SSE()
134 …_t *gradient16, float *m, float *v, float lr, float beta1, float beta2, float epsilon, float decay, in FusedCastAdamFp16Fp16SSE()
168 …at *gradient32, float *m, float *v, float lr, float beta1, float beta2, float epsilon, float decay, in FusedCastAdamFp16Fp32SSE()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/include/nnacl/avx/ |
| D | adam_fp32_avx.h | 31 …ex, float *var, float *m, float *v, float lr, float beta1, float beta2, float epsilon, float decay, in AdamWeightDecayFp32AVX()
64 …_t *gradient16, float *m, float *v, float lr, float beta1, float beta2, float epsilon, float decay, in FusedCastAdamFp32Fp16AVX()
99 …at *gradient32, float *m, float *v, float lr, float beta1, float beta2, float epsilon, float decay, in FusedCastAdamFp32Fp32AVX()
134 …_t *gradient16, float *m, float *v, float lr, float beta1, float beta2, float epsilon, float decay, in FusedCastAdamFp16Fp16AVX()
168 …at *gradient32, float *m, float *v, float lr, float beta1, float beta2, float epsilon, float decay, in FusedCastAdamFp16Fp32AVX()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/include/nnacl/neon/ |
| D | adam_fp32_neon.h | 30 …ex, float *var, float *m, float *v, float lr, float beta1, float beta2, float epsilon, float decay, in AdamWeightDecayFp32NEON()
63 …_t *gradient16, float *m, float *v, float lr, float beta1, float beta2, float epsilon, float decay, in FusedCastAdamFp32Fp16NEON()
98 …at *gradient32, float *m, float *v, float lr, float beta1, float beta2, float epsilon, float decay, in FusedCastAdamFp32Fp32NEON()
133 …_t *gradient16, float *m, float *v, float lr, float beta1, float beta2, float epsilon, float decay, in FusedCastAdamFp16Fp16NEON()
167 …at *gradient32, float *m, float *v, float lr, float beta1, float beta2, float epsilon, float decay, in FusedCastAdamFp16Fp32NEON()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/include/nnacl/avx512/ |
| D | adam_fp32_avx512.h | 31 …ex, float *var, float *m, float *v, float lr, float beta1, float beta2, float epsilon, float decay, in AdamWeightDecayFp32AVX512()
64 …_t *gradient16, float *m, float *v, float lr, float beta1, float beta2, float epsilon, float decay, in FusedCastAdamFp32Fp16AVX512()
99 …at *gradient32, float *m, float *v, float lr, float beta1, float beta2, float epsilon, float decay, in FusedCastAdamFp32Fp32AVX512()
134 …_t *gradient16, float *m, float *v, float lr, float beta1, float beta2, float epsilon, float decay, in FusedCastAdamFp16Fp16AVX512()
168 …at *gradient32, float *m, float *v, float lr, float beta1, float beta2, float epsilon, float decay, in FusedCastAdamFp16Fp32AVX512()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/fp32/ |
| D | adam_fp32.c | 159 …p32(float *var, float *m, float *v, float lr, float beta1, float beta2, float epsilon, float decay, in AdamWeightDecayFp32()
176 … float beta2, float epsilon, float decay, float global_norm_reciprocal, size_t start, in FusedCastAdamFp32Fp16()
186 … float beta2, float epsilon, float decay, float global_norm_reciprocal, size_t start, in FusedCastAdamFp32Fp32()
196 … float beta2, float epsilon, float decay, float global_norm_reciprocal, size_t start, in FusedCastAdamFp16Fp16()
205 … float beta2, float epsilon, float decay, float global_norm_reciprocal, size_t start, in FusedCastAdamFp16Fp32()
|
| D | rmsprop_fp32.c | 33 … float learning_rate, float decay, float epsilon, size_t start, size_t end) { in RMSPropUnuseCenterFp32()
82 … float momentum, float learning_rate, float decay, float epsilon, size_t start, size_t end) { in RMSPropUseCenterFp32()
|
| /third_party/skia/third_party/externals/oboe/apps/OboeTester/app/src/main/cpp/analyzer/ |
| D | PeakDetector.h | 58 void setDecay(double decay) { in setDecay()
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/kernel/cpu/fp32_grad/ |
| D | adam_weight_decay.cc | 50 auto decay = reinterpret_cast<float *>(in_tensors_.at(kDecayIdx)->MutableData())[0]; in DoExecute() local
117 auto decay = reinterpret_cast<float *>(in_tensors_.at(kDecayIdx)->MutableData())[0]; in OptimizerStep() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/minddata/dataset/audio/kernels/ |
| D | phaser_op.cc | 23 PhaserOp::PhaserOp(int32_t sample_rate, float gain_in, float gain_out, float delay_ms, float decay,… in PhaserOp()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/minddata/dataset/audio/ir/kernels/ |
| D | phaser_ir.cc | 25 …n::PhaserOperation(int32_t sample_rate, float gain_in, float gain_out, float delay_ms, float decay, in PhaserOperation()
|
| /third_party/mindspore/mindspore-src/source/tests/st/networks/models/resnet50/ |
| D | train_resnet50_thor.py | 32 def get_thor_lr(global_step, lr_init, decay, total_epochs, steps_per_epoch, decay_epochs=100): argument
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/nn/ |
| D | rmsprop_gpu_kernel.h | 138 T *decay = GetDeviceAddress<T>(inputs, kNumberFive); in Launch() local
152 T *decay = GetDeviceAddress<T>(inputs, 6); in Launch() local
|
| D | adam_weight_decay_gpu_kernel.cc | 97 float *decay = GetDeviceAddress<float>(inputs, kIndex7); in LaunchKernel() local
|
| /third_party/mindspore/mindspore-src/source/tests/ut/python/dataset/ |
| D | test_phaser.py | 80 …def test_invalid_input(test_name, sample_rate, gain_in, gain_out, delay_ms, decay, mod_speed, sinu… argument
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/pyboost/customize/ |
| D | adamw.cc | 29 …const FP32ImmPtr &beta1, const FP32ImmPtr &beta2, const FP32ImmPtr &decay, const FP32ImmPtr &epsil… in AdamWAscendCustomize()
|
| /third_party/ffmpeg/libavfilter/ |
| D | af_asubboost.c | 33 double decay; member
103 const double feedback = s->feedback, decay = s->decay; in filter_channels() local
|