hub / github.com/tensorflow/lingvo / FProp

Method FProp

lingvo/core/conformer_layer.py:281–344 · view source on GitHub ↗

Builds FProp graph. Args: theta: A NestedMap of Tensors, see base class. inputs: A Tensor of shape [batch, seqlen, dim0]. paddings: A Tensor of shape [batch, seqlen]. Returns: output: A Tensor of shape [batch, seqlen, dim0]. out_paddings: A Tensor of shape [ba

(self, theta, inputs, paddings)

Source from the content-addressed store, hash-verified

source not stored for this graph (policy: none)

Callers 9

_NormalizeMethod · 0.45

_NormalizeStepMethod · 0.45

StreamStepMethod · 0.45

_SelfAttenMethod · 0.45

_LConvMethod · 0.45

_MaybeFFLayerAdapterMethod · 0.45

_MoeOrFFLayerMethod · 0.45

_FPropMethod · 0.45

StreamStepMethod · 0.45

Calls 7

_GLUMethod · 0.95

_NormalizeMethod · 0.95

_ApplyActivationMethod · 0.95

_CastToFPropDtypeMethod · 0.80

splitMethod · 0.80

copyMethod · 0.80

MeshSplitMethod · 0.45

Tested by

no test coverage detected