MCPcopy Create free account
hub / github.com/MethodsOfMachineLearning/cabs / _Conv2DGradMom

Function _Conv2DGradMom

gradient_moment.py:158–189  ·  view source on GitHub ↗

Computes gradient moment for the filter of a Conv2D operation. Assumes ``Z=tf.nn.conv2d(A, f)``, where ``f`` is a ``[h_f, w_f, c_in, c_out]`` convolution filter and ``A`` are the ``[n, h_in, w_in, c_in]`` activations of the previous layer (``n`` being the batch size). ``out_grad`` is the gr

(op, f, out_grad, batch_size, mom=2)

Source from the content-addressed store, hash-verified

source not stored for this graph (policy: none)

Callers 1

_GradMomFunction · 0.85

Calls

no outgoing calls

Tested by

no test coverage detected