MCPcopy Create free account
hub / github.com/benoitsteiner/tensorflow-opencl / _BatchNormGrad

Function _BatchNormGrad

tensorflow/python/ops/nn_grad.py:785–852  ·  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_train

(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 3

castMethod · 0.80
sizeMethod · 0.45
stop_gradientMethod · 0.45

Tested by

no test coverage detected