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

Class GPipeEvolvedTransformerDecoderLayer

lingvo/core/layers_with_gpipe.py:289–352  ·  view source on GitHub ↗

GPipe-compatible Evolved Transformer decoder layer.

Source from the content-addressed store, hash-verified

source not stored for this graph (policy: none)

Callers 3

testEvolvedTransformerDecoderLayerConstructionMethod · 0.90
testEvolvedTransformerDecoderLayerFPropMethod · 0.90
testEvolvedTransformerDecoderLayerExtendStepMethod · 0.90

Calls

no outgoing calls

Tested by 3

testEvolvedTransformerDecoderLayerConstructionMethod · 0.72
testEvolvedTransformerDecoderLayerFPropMethod · 0.72
testEvolvedTransformerDecoderLayerExtendStepMethod · 0.72

Used in the wild real call sites across dependent graphs

searching dependent graphs…