| /third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp32_grad/ |
| D | adam.cc | 35 … float beta2_power, float eps, float learning_rate, bool nesterov, int start, int end) { in DoAdam()
70 auto beta2_power = reinterpret_cast<float *>(in_tensors_.at(4)->MutableData())[0]; in Execute() local
141 auto beta2_power = reinterpret_cast<float *>(in_tensors_.at(4)->MutableData())[0]; in OptimizerStep() local
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/ |
| D | adam_cpu_kernel.cc | 37 float beta2_power = reinterpret_cast<float *>(inputs[BETA2_POWER]->addr)[kScalarIndex]; in LaunchAdam() local
72 float beta2_power = reinterpret_cast<float *>(inputs[BETA2_POWER]->addr)[kScalarIndex]; in LaunchAdamNnacl() local
|
| D | adam_delta_cpu_kernel.cc | 120 auto beta2_power = reinterpret_cast<float *>(inputs[3]->addr)[0]; in Launch() local
|
| D | sparse_apply_lazy_adam_cpu_kernel.cc | 132 auto beta2_power = reinterpret_cast<float *>(inputs[4]->addr)[0]; in LaunchKernel() local
|
| D | sparse_apply_adam_cpu_kernel.cc | 152 auto beta2_power = reinterpret_cast<float *>(inputs[4]->addr)[0]; in LaunchKernel() local
|
| /third_party/mindspore/tests/st/ops/cpu/ |
| D | test_sparse_apply_adam_op.py | 26 beta2_power = 0.999 variable
|
| /third_party/mindspore/tests/st/ops/ascend/test_aicpu_ops/ |
| D | test_fused_sparse_lazy_adam.py | 26 beta2_power = 0.999 variable
|
| D | test_fused_sparse_adam.py | 26 beta2_power = 0.999 variable
|
| /third_party/mindspore/tests/st/fl/mobile/src/ |
| D | adam.py | 109 … beta2_power, beta1, beta2, eps, lr, gradient, param, m, v, ps_parameter, cache_enable): argument
176 beta1_power, beta2_power, beta1, beta2, eps, lr, gradient, param, argument
192 def _run_off_load_opt(opt, beta1_power, beta2_power, beta1, beta2, eps, lr, gradient, param, moment… argument
|
| /third_party/mindspore/tests/st/fl/albert/src/ |
| D | adam.py | 110 … beta2_power, beta1, beta2, eps, lr, gradient, param, m, v, ps_parameter, cache_enable): argument
176 beta1_power, beta2_power, beta1, beta2, eps, lr, gradient, param, argument
192 def _run_off_load_opt(opt, beta1_power, beta2_power, beta1, beta2, eps, lr, gradient, param, moment… argument
|
| /third_party/mindspore/tests/st/fl/hybrid_lenet/src/ |
| D | adam.py | 109 … beta2_power, beta1, beta2, eps, lr, gradient, param, m, v, ps_parameter, cache_enable): argument
175 beta1_power, beta2_power, beta1, beta2, eps, lr, gradient, param, argument
191 def _run_off_load_opt(opt, beta1_power, beta2_power, beta1, beta2, eps, lr, gradient, param, moment… argument
|
| /third_party/mindspore/mindspore/nn/optim/ |
| D | lazyadam.py | 34 …th_sparse(opt, sparse_opt, push, pull, use_locking, use_nesterov, target, beta1_power, beta2_power, argument
81 …ne_number(opt, sparse_opt, push, pull, use_locking, use_nesterov, target, beta1_power, beta2_power, argument
|
| D | adam.py | 94 … beta2_power, beta1, beta2, eps, lr, gradient, param, m, v, ps_parameter, cache_enable): argument
155 beta1_power, beta2_power, beta1, beta2, eps, lr, gradient, param, argument
171 def _run_off_load_opt(opt, beta1_power, beta2_power, beta1, beta2, eps, lr, gradient, param, moment… argument
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/cuda_impl/ |
| D | adam_impl.cu | 30 …d ApplyAdamKernel(const size_t size, const T *gradient, const T *beta1_power, const T *beta2_power, in ApplyAdamKernel()
72 void ApplyAdam(const size_t size, const T *gradient, const T *beta1_power, const T *beta2_power, co… in ApplyAdam()
|
| /third_party/mindspore/tests/st/auto_monad/ |
| D | test_auto_monad_expression.py | 34 def construct(self, beta1_power, beta2_power, lr, beta1, beta2, epsilon, grad): argument
|
| D | test_effect_optimizer.py | 34 def construct(self, beta1_power, beta2_power, lr, beta1, beta2, epsilon, grad): argument
466 def construct(self, beta1_power, beta2_power, lr, beta1, beta2, epsilon, grad, indices): argument
536 def construct(self, beta1_power, beta2_power, lr, beta1, beta2, epsilon, grad, indices): argument
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/nn/ |
| D | adam_gpu_kernel.h | 58 T *beta2_power = GetDeviceAddress<T>(inputs, 4); in Launch() local
|
| /third_party/mindspore/tests/ut/python/optimizer/ |
| D | test_auto_grad.py | 292 def construct(self, beta1_power, beta2_power, lr, beta1, beta2, epsilon, grad): argument
303 def construct(self, beta1_power, beta2_power, lr, beta1, beta2, epsilon, grad): argument
333 def construct(self, beta1_power, beta2_power, lr, beta1, beta2, epsilon, grad): argument
344 def construct(self, beta1_power, beta2_power, lr, beta1, beta2, epsilon, grad): argument
|
| /third_party/mindspore/mindspore/ccsrc/ps/ |
| D | optimizer_info_builder.cc | 186 …AddressPtr beta2_power = GenInputAddrPtr<float>(kSparseAdam, "beta2_power", const_cast<float *>(va… in BuildInputs() local
|
| D | optimizer_info.cc | 306 … const AddressPtr &beta1_power, const AddressPtr &beta2_power, in SparseAdamOptimInfo()
|