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

Method _CalculateOutputShapes

lingvo/core/gpipe.py:339–375  ·  view source on GitHub ↗

Calcuate the output shape of intermediate layers. Given the FPropMeta function in each FeatureExtractionLayer, calcuates the shapes of outputs of that layer. This is used to recover the shape information in StackedRecurrent. Args: input_shapes: NestedMap or tuple of input Ten

(self, input_shapes)

Source from the content-addressed store, hash-verified

source not stored for this graph (policy: none)

Callers 1

FPropMethod · 0.95

Calls 3

_ToTupleFunction · 0.70
TransformMethod · 0.45
FPropMetaMethod · 0.45

Tested by

no test coverage detected