Searched defs:save_inv_variance (Results 1 – 3 of 3) sorted by relevance
| /third_party/mindspore/tests/vm_impl/ |
| D | vm_me.py | 121 def _batch_norm_grad(dout, x, scale, save_mean, save_inv_variance, \ argument
143 def batch_norm_grad(dy, x, scale, save_mean, save_inv_variance): argument
|
| D | nn_ops_vm_impl.py | 210 def vm_impl(dy, x, scale, save_mean, save_inv_variance): argument
|
| /third_party/mindspore/mindspore/ops/operations/ |
| D | _grad_ops.py | 689 def __call__(self, dy, x, scale, save_mean, save_inv_variance): argument
|