HomeSort by relevance Sort by last modified time
    Searched refs:pop_mean (Results 1 - 3 of 3) sorted by null

  /external/tensorflow/tensorflow/core/kernels/
fused_batch_norm_op.h 76 typename TTypes<U>::ConstVec pop_mean(pop_mean_input.vec<U>());
82 const int depth = pop_mean.dimension(0);
99 // scale_backprop = y_backprop * ((x - pop_mean) * rsqrt(pop_var + epsilon))
115 pop_mean.reshape(one_by_depth).broadcast(rest_by_one)))
  /external/tensorflow/tensorflow/python/ops/
nn_fused_batchnorm_test.py 198 pop_mean = None
201 pop_mean = np.random.random_sample(scale_shape).astype(scale_dtype)
207 mean=pop_mean,
223 mean=pop_mean,
261 pop_mean = None
264 pop_mean = np.random.random_sample(scale_shape).astype(scale_dtype)
270 mean=pop_mean,
281 grad_internal = nn_grad._BatchNormGrad(grad_y, x, scale, pop_mean,
311 mean=pop_mean,
nn_grad.py 774 sum(grad_y * (x - pop_mean) * rsqrt(pop_variance + epsilon))
800 pop_mean = op.inputs[3]
809 pop_mean,
832 pop_mean,
843 pop_mean: A `Tensor` of 1 dimension for the population mean. Only used when
896 shape = [1, array_ops.size(pop_mean), 1, 1]
897 pop_mean = array_ops.reshape(pop_mean, shape)
904 grad_y * (x - pop_mean) * var_rsqrt, axis=reduce_axis)
930 pop_mean = op.inputs[3
    [all...]

Completed in 83 milliseconds