Searched refs:running_var (Results 1 – 2 of 2) sorted by relevance
| /third_party/mindspore/tests/vm_impl/ |
| D | vm_me.py | 71 def _batch_norm(x, scale, shift, running_mean=None, running_var=None, argument
80 running_var = np.zeros(c_h_w)
95 running_var = momentum * running_var + (1 - momentum) * x_var
98 x_norm = (x - running_mean) / np.sqrt(running_var + eps)
100 x_var = running_var
104 return out, x_mean, x_var, running_mean, running_var
115 out, _, _, running_mean, running_var = _batch_norm(x, scale, shift, mean, variance, \
118 return out.reshape(*input_shape), np.array(scale), np.array(shift), running_mean, running_var
130 running_var=save_inv_variance, \
|
| D | nn_ops_vm_impl.py | 106 out, x_mean, x_var, running_mean, running_var = vm.batch_norm(x, scale, b, mean, \
110 Tensor(running_mean), Tensor(running_var)
|