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

Method StreamStep

lingvo/core/conformer_layer.py:390–442  ·  view source on GitHub ↗

Streams t steps. Args: theta: A NestedMap of layer params. inputs: [b, t, d]. paddings: A 0/1 valued tensor of shape [b, t]. state0: A NestedMap of tensors of the same struct as returned by zero_state(). Returns: output: the same shape as inputs, with

(self, theta, inputs, paddings, state0)

Source from the content-addressed store, hash-verified

source not stored for this graph (policy: none)

Callers 2

_NormalizeStepMethod · 0.45
StreamStepMethod · 0.45

Calls 5

_GLUMethod · 0.95
_NormalizeStepMethod · 0.95
_ApplyActivationMethod · 0.95
splitMethod · 0.80
FPropMethod · 0.45

Tested by

no test coverage detected