hub / github.com/tensorflow/lingvo / DecoderLayerStack

Method DecoderLayerStack

lingvo/core/gshard_builder.py:675–708 · view source on GitHub ↗

Clean DecoderLayerStack, similar to EncoderLayerStack.

(self,
                        name,
                        sub_layers,
                        num=1,
                        conv_kernel_size=None,
                        norm_type='ln',
                        norm_policy='pre',
                        use_repeat_layer=False,
                        spmd_pipeline_stages=1,
                        spmd_pipeline_microbatches=None,
                        start_layer_id=0,
                        has_final_layer=True)

Source from the content-addressed store, hash-verified

source not stored for this graph (policy: none)

Callers 6

__init__Method · 0.45

testRepeatLayerUnrolledStateMethod · 0.45

testMoEModelDimReshapeFPropMethod · 0.45

testLayerStackMethod · 0.45

testLayerStackSummaryMethod · 0.45

Calls 1

_LayerStackMethod · 0.95

Tested by 4

testRepeatLayerUnrolledStateMethod · 0.36

testMoEModelDimReshapeFPropMethod · 0.36

testLayerStackMethod · 0.36

testLayerStackSummaryMethod · 0.36