Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/torch/csrc/api/include/torch/nn/modules/
H A Dtransformercoder.h40 TransformerEncoderLayer encoder_layer, in TransformerEncoderImpl()
/aosp_15_r20/external/pytorch/torch/csrc/api/src/nn/options/
H A Dtransformer.cpp19 TransformerEncoderLayer encoder_layer, in TransformerEncoderOptions()
/aosp_15_r20/external/pytorch/test/cpp/api/
H A Dtransformer.cpp573 TransformerEncoderLayer encoder_layer = in transformer_encoder_test_helper() local