MCPcopy Create free account
hub / github.com/alibaba/nann / _BatchNormGrad

Function _BatchNormGrad

tensorflow/tensorflow/python/ops/nn_grad.py:929–1005  ·  view source on GitHub ↗

Returns the gradients for the 3 inputs of BatchNorm. Args: grad_y: A `Tensor` of 4 dimensions for gradient for y. x: A `Tensor` of 4 dimensions for x. scale: A `Tensor` of 1 dimension for scaling. pop_mean: A `Tensor` of 1 dimension for the population mean. Only used when is

(grad_y,
                   x,
                   scale,
                   pop_mean,
                   pop_var,
                   epsilon,
                   data_format,
                   is_training=True)

Source from the content-addressed store, hash-verified

source not stored for this graph (policy: none)

Callers 1

_FusedBatchNormGradGradFunction · 0.85

Calls 5

reshapeMethod · 0.80
reduce_meanMethod · 0.80
reduce_sumMethod · 0.80
castMethod · 0.45
sizeMethod · 0.45

Tested by

no test coverage detected