Searched refs:_batch_norm_grad (Results 1 – 1 of 1) 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, \ function
149 dx, dgamma, dbeta = _batch_norm_grad(dy, x, scale, save_mean, save_inv_variance)
|