Home
last modified time | relevance | path

Searched refs:TransformerDecoderLayerOptions (Results 1 – 6 of 6) sorted by relevance

/aosp_15_r20/external/pytorch/torch/csrc/api/src/nn/options/
H A Dtransformer.cpp13 TransformerDecoderLayerOptions::TransformerDecoderLayerOptions( in TransformerDecoderLayerOptions() function in torch::nn::TransformerDecoderLayerOptions
34 const TransformerDecoderLayerOptions& decoder_layer_options, in TransformerDecoderOptions()
/aosp_15_r20/external/pytorch/torch/csrc/api/include/torch/nn/modules/
H A Dtransformerlayer.h116 TransformerDecoderLayerOptions(d_model, nhead)) {} in TransformerDecoderLayerImpl()
117 explicit TransformerDecoderLayerImpl(TransformerDecoderLayerOptions options_);
141 TransformerDecoderLayerOptions options;
/aosp_15_r20/external/pytorch/torch/csrc/api/include/torch/nn/options/
H A Dtransformerlayer.h51 struct TORCH_API TransformerDecoderLayerOptions { struct
52 TransformerDecoderLayerOptions(int64_t d_model, int64_t nhead);
H A Dtransformercoder.h62 const TransformerDecoderLayerOptions& decoder_layer_options,
/aosp_15_r20/external/pytorch/test/cpp/api/
H A Dtransformer.cpp254 get_a_test_layer<TransformerDecoderLayer, TransformerDecoderLayerOptions>( in transformer_decoder_layer_test_helper()
460 get_a_test_layer<TransformerDecoderLayer, TransformerDecoderLayerOptions>( in transformer_decoder_layer_test_helper_gelu()
848 get_a_test_layer<TransformerDecoderLayer, TransformerDecoderLayerOptions>( in transformer_decoder_test_helper()
1321 TransformerDecoderOptions(TransformerDecoderLayerOptions(4, 2), 2) in TEST_F()
1403 TransformerDecoderLayerOptions(4, 2).dim_feedforward(16).dropout(0.0), in transformer_test_helper()
/aosp_15_r20/external/pytorch/torch/csrc/api/src/nn/modules/
H A Dtransformer.cpp94 TransformerDecoderLayerOptions options_) in TransformerDecoderLayerImpl()
395 TransformerDecoderLayerOptions(options.d_model(), options.nhead()) in reset()