| /third_party/mindspore/mindspore/nn/optim/ |
| D | proximal_ada_grad.py | 28 def _tensor_run_opt_with_sparse(opt, sparse_opt, l1, l2, learning_rate, gradient, weight, accum): argument
37 def _tensor_run_opt(opt, sparse_opt, l1, l2, learning_rate, gradient, weight, accum): argument
|
| D | lazyadam.py | 34 def _run_opt_with_sparse(opt, sparse_opt, push, pull, use_locking, use_nesterov, target, beta1_powe… argument
81 def _run_opt_with_one_number(opt, sparse_opt, push, pull, use_locking, use_nesterov, target, beta1_… argument
|
| D | adam.py | 93 def _run_opt_with_sparse(opt, sparse_opt, push, pull, use_locking, use_nesterov, target, beta1_powe… argument
154 def _run_opt_with_one_number(opt, sparse_opt, push, pull, use_locking, use_nesterov, target, argument
|
| /third_party/mindspore/tests/st/fl/mobile/src/ |
| D | adam.py | 108 def _run_opt_with_sparse(opt, sparse_opt, push, pull, use_locking, use_nesterov, target, beta1_powe… argument
175 def _run_opt_with_one_number(opt, sparse_opt, push, pull, use_locking, use_nesterov, target, argument
|
| /third_party/mindspore/tests/st/fl/albert/src/ |
| D | adam.py | 109 def _run_opt_with_sparse(opt, sparse_opt, push, pull, use_locking, use_nesterov, target, beta1_powe… argument
175 def _run_opt_with_one_number(opt, sparse_opt, push, pull, use_locking, use_nesterov, target, argument
|
| /third_party/mindspore/tests/st/fl/hybrid_lenet/src/ |
| D | adam.py | 108 def _run_opt_with_sparse(opt, sparse_opt, push, pull, use_locking, use_nesterov, target, beta1_powe… argument
174 def _run_opt_with_one_number(opt, sparse_opt, push, pull, use_locking, use_nesterov, target, argument
|