Searched defs:TransformerDecoderLayer (Results 1 – 3 of 3) sorted by relevance
547 decoder_layer: "TransformerDecoderLayer",935 class TransformerDecoderLayer(Module): class
438 TEST_F(TransformerTest, TransformerDecoderLayer) { in TEST_F() argument
1073 class TransformerDecoderLayer(nn.Module): class