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

Function _AddGradMom

gradient_moment.py:191–223  ·  view source on GitHub ↗

Computes gradient moment for a bias variable through an Add operation. Assumes ``Z = tf.add(Zz, b)``, where ``b`` is a bias parameter and ``Zz`` is a ``[n, ?]`` tensor (``n`` being the batch size). Broadcasting for all kinds of shapes of ``Zz`` (e.g. ``[n, d_in]`` or ``[n, h_in, w_in, c_in]

(op, b, 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