| /third_party/mindspore/mindspore/nn/optim/ |
| D | ftrl.py | 30 gradient, weight, moment, ps_parameter, cache_enable): argument
47 gradient, weight, moment, ps_parameter, cache_enable): argument
|
| D | lazyadam.py | 35 beta1, beta2, eps, lr, gradient, params, m, v, ps_parameter, cache_enable): argument
82 … beta1, beta2, eps, lr, gradient, params, moment1, moment2, ps_parameter, cache_enable): argument
|
| D | momentum.py | 28 def _tensor_run_opt_ext(opt, momentum, learning_rate, gradient, weight, moment, ps_parameter, cache… argument
|
| D | adam.py | 94 … beta2_power, beta1, beta2, eps, lr, gradient, param, m, v, ps_parameter, cache_enable): argument
156 moment1, moment2, ps_parameter, cache_enable): argument
|
| /third_party/mindspore/mindspore/nn/wrap/ |
| D | grad_reducer.py | 130 def _tensors_allreduce_ps(degree, mean, allgather, allreduce, allreduce_filter, grad, ps_parameter): argument
184 …allreduce_with_sparse_ps(degree, mean, allgather, allreduce, allreduce_filter, grad, ps_parameter): argument
|
| /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
177 moment1, moment2, ps_parameter, cache_enable): 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
177 moment1, moment2, ps_parameter, cache_enable): 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
176 moment1, moment2, ps_parameter, cache_enable): argument
|