Searched refs:pop_mean (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | fused_batch_norm_op.h | 76 typename TTypes<U>::ConstVec pop_mean(pop_mean_input.vec<U>()); in operator()
82 const int depth = pop_mean.dimension(0); in operator()
115 pop_mean.reshape(one_by_depth).broadcast(rest_by_one))) in operator()
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | nn_fused_batchnorm_test.py | 199 pop_mean = None
202 pop_mean = np.random.random_sample(scale_shape).astype(scale_dtype)
208 mean=pop_mean,
224 mean=pop_mean,
262 pop_mean = None
265 pop_mean = np.random.random_sample(scale_shape).astype(scale_dtype)
271 mean=pop_mean,
282 grad_internal = nn_grad._BatchNormGrad(grad_y, x, scale, pop_mean,
312 mean=pop_mean,
|
| D | nn_grad.py | 878 pop_mean = op.inputs[3]
887 pop_mean,
910 pop_mean, argument
973 shape = [1, array_ops.size(pop_mean), 1, 1]
974 pop_mean = array_ops.reshape(pop_mean, shape)
981 grad_y * (x - pop_mean) * var_rsqrt, axis=reduce_axis)
1007 pop_mean = op.inputs[3]
1017 grad_y, x, scale, pop_mean, pop_var, epsilon, data_format, is_training)
|