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

Method _OutfeedDequeue

lingvo/core/program.py:1896–1922  ·  view source on GitHub ↗

Collect outfeed dequeue from all devices. Args: decode_nm: A NestedMap containing decoded tensors. Returns: A list of tensors corresponding to stacked decoded outputs. The decoder outputs are stacked on the first dimension (usually corresponds to batch size).

(self, decode_nm)

Source from the content-addressed store, hash-verified

source not stored for this graph (policy: none)

Callers 1

DecodeFuncMethod · 0.95

Calls 1

FlattenMethod · 0.45

Tested by

no test coverage detected