Searched defs:accu_grad (Results 1 – 7 of 7) sorted by relevance
| /third_party/mindspore/mindspore-src/source/tests/st/networks/models/bert/bert_performance/src/ |
| D | bert_for_pre_training.py | 484 def _add_grads(accu_grad, grad): argument
492 def _update_accu_grads(accu_grad, grad): argument
501 def _accumulate_accu_grads(accu_grad, grad): argument
511 def _reset_accu_grads(accu_grad): argument
|
| /third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/nn/wrap/ |
| D | grad_reducer.py | 40 def tensor_grad_scale_pipeline(scale, grad, accu_grad): argument
50 def tensor_shard_grad_scale_pipeline(scale, grad, accu_grad): argument
|
| D | loss_scale.py | 589 def tensor_grad_scale_pipeline(scale, grad, accu_grad): argument
599 def tensor_shard_grad_scale_pipeline(scale, grad, accu_grad): argument
|
| D | cell_wrapper.py | 729 def _pipeline_clear_grad(accu_grad, grad): argument
|
| /third_party/mindspore/mindspore-src/source/tests/ut/python/parallel/ |
| D | test_func_mode_pipeline.py | 83 def pipeline_clear_grad(accu_grad, grad): argument
|
| D | test_dynamic_pangu_alpha.py | 1279 def tensor_grad_scale_pipeline(scale, grad, accu_grad): argument
1289 def tensor_shard_grad_scale_pipeline(scale, grad, accu_grad): argument
|
| D | test_dynamic_pangu_alpha_without_skip_redistribution.py | 1281 def tensor_grad_scale_pipeline(scale, grad, accu_grad): argument
1291 def tensor_shard_grad_scale_pipeline(scale, grad, accu_grad): argument
|