OpenGrok
Home
Sort by relevance
Sort by last modified time
Full Search
Definition
Symbol
File Path
History
|
|
Help
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