| /third_party/mindspore/mindspore/nn/optim/ |
| D | lamb.py | 39 …beta1, beta2, eps, global_step, lr, weight_decay, param, m, v, gradient, decay_flag, optim_filter): argument
121 …_run_op_ascend(beta1, beta2, eps, global_step, lr, weight_decay, param, m, v, gradient, decay_flag, argument
|
| D | lars.py | 28 def _tensor_run_opt(lars, learning_rate, weight_decay, gradient, weight, decay_flag, lars_flag): argument
|
| D | adam.py | 37 def _update_run_op(beta1, beta2, eps, lr, weight_decay, param, m, v, gradient, decay_flag, optim_fi… argument
|
| /third_party/mindspore/tests/st/ops/graph_kernel/ |
| D | test_fused_adam.py | 30 def __init__(self, decay_flag=True): argument
72 def __init__(self, decay_flag=True): argument
|
| /third_party/mindspore/tests/st/ops/gpu/ |
| D | test_adam_fusion.py | 32 def __init__(self, decay_flag=True): argument
|
| /third_party/mindspore/tests/ut/python/ir/ |
| D | test_row_tensor.py | 162 m, v, gradient, decay_flag): argument
168 m, v, gradient, decay_flag): argument
|
| /third_party/mindspore/tests/st/fl/mobile/src/ |
| D | adam.py | 48 def _update_run_op(beta1, beta2, eps, lr, overflow, weight_decay, param, m, v, gradient, decay_flag… argument
|
| /third_party/mindspore/tests/st/fl/albert/src/ |
| D | adam.py | 49 def _update_run_op(beta1, beta2, eps, lr, overflow, weight_decay, param, m, v, gradient, decay_flag… argument
|
| /third_party/mindspore/tests/st/fl/hybrid_lenet/src/ |
| D | adam.py | 48 def _update_run_op(beta1, beta2, eps, lr, overflow, weight_decay, param, m, v, gradient, decay_flag… argument
|