Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/torch/nn/modules/
H A Dtransformer.py333 encoder_layer: "TransformerEncoderLayer",
619 class TransformerEncoderLayer(Module): class
/aosp_15_r20/external/pytorch/test/cpp/api/
H A Dtransformer.cpp232 TEST_F(TransformerTest, TransformerEncoderLayer) { in TEST_F() argument
/aosp_15_r20/external/pytorch/test/dynamo/
H A Dtest_repros.py864 class TransformerEncoderLayer(nn.Module): class