Home
last modified time | relevance | path

Searched defs:ps_parameter (Results 1 – 2 of 2) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/nn/optim/
Dlazyadam.py39 … beta1, beta2, eps, lr, gradient, params, m, v, ps_parameter, cache_enable, argument
93ps_parameter, cache_enable, distributed_opt, use_flag, distributed_sparse_opt, argument
133 … beta1, beta2, eps, lr, gradient, params, moment1, moment2, ps_parameter, cache_enable, argument
154 beta1, beta2, eps, lr, gradient, params, m, v, ps_parameter, cache_enable): argument
202 … beta2_power, beta1, beta2, eps, lr, gradient, params, m, v, ps_parameter, argument
235 … beta1, beta2, eps, lr, gradient, params, moment1, moment2, ps_parameter, cache_enable): argument
/third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/nn/wrap/
Dgrad_reducer.py162 def _tensors_allreduce_ps(degree, mean, allgather, allreduce, allreduce_filter, grad, ps_parameter): argument
216 …allreduce_with_sparse_ps(degree, mean, allgather, allreduce, allreduce_filter, grad, ps_parameter): argument