Searched defs:grad_sum (Results 1 – 1 of 1) sorted by relevance
| /third_party/mindspore/tests/st/networks/ |
| D | test_gradient_accumulation.py | 26 def _cumulative_gard(grad_sum, grad): argument
33 def _clear_grad_sum(grad_sum, zero): argument
75 def __init__(self, network, optimizer, grad_sum, sens=1.0): argument
96 def __init__(self, optimizer, grad_sum): argument
106 def __init__(self, grad_sum, zeros): argument
|