Home
last modified time | relevance | path

Searched defs:TransformerDecoderLayer (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/pytorch/torch/nn/modules/
H A Dtransformer.py547 decoder_layer: "TransformerDecoderLayer",
935 class TransformerDecoderLayer(Module): class
/aosp_15_r20/external/pytorch/test/cpp/api/
H A Dtransformer.cpp438 TEST_F(TransformerTest, TransformerDecoderLayer) { in TEST_F() argument
/aosp_15_r20/external/pytorch/test/quantization/eager/
H A Dtest_quantize_eager_ptq.py1073 class TransformerDecoderLayer(nn.Module): class