MCPcopy Create free account
hub / github.com/tensorflow/lingvo / ComputeAndUpdateMoments

Method ComputeAndUpdateMoments

lingvo/core/bn_layers.py:306–375  ·  view source on GitHub ↗

Computes moments and updates state. Args: theta: A `.NestedMap` object containing weights' values of this layer and its children layers. inputs: The inputs tensor. Shaped [..., dim]. paddings: The paddings tensor. Shaped [..., 1], with the same rank as the in

(self, theta, inputs, paddings=None, **kwargs)

Source from the content-addressed store, hash-verified

source not stored for this graph (policy: none)

Callers 4

FPropMethod · 0.95
_GetWeightsMethod · 0.80
_GetWeightsMethod · 0.80
FPropMethod · 0.80

Calls 4

_GetDefaultPaddingsMethod · 0.95
_GetBetaGammaMethod · 0.95
ComputeMomentsFunction · 0.85
rankMethod · 0.80

Tested by

no test coverage detected