Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/torch/csrc/api/include/torch/nn/modules/
H A Dtransformercoder.h101 TransformerDecoderLayer decoder_layer, in TransformerDecoderImpl()
/aosp_15_r20/external/pytorch/torch/csrc/api/src/nn/options/
H A Dtransformer.cpp29 TransformerDecoderLayer decoder_layer, in TransformerDecoderOptions()
/aosp_15_r20/external/pytorch/test/cpp/api/
H A Dtransformer.cpp847 TransformerDecoderLayer decoder_layer = in transformer_decoder_test_helper() local