Searched defs:batch_variance (Results 1 – 1 of 1) sorted by relevance
| /third_party/mindspore/mindspore/ops/operations/ |
| D | _grad_ops.py | 701 def infer_shape(self, grads, x, diff_scale, diff_offset, scale, batch_mean, batch_variance): argument
704 def infer_dtype(self, grads, x, diff_scale, diff_offset, scale, batch_mean, batch_variance): argument
716 def infer_shape(self, grads, x, batch_mean, batch_variance): argument
719 def infer_dtype(self, grads, x, batch_mean, batch_variance): argument
|