Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/torch/csrc/api/src/nn/options/
H A Dtransformer.cpp8 TransformerEncoderLayerOptions::TransformerEncoderLayerOptions( in TransformerEncoderLayerOptions() function in torch::nn::TransformerEncoderLayerOptions
24 const TransformerEncoderLayerOptions& encoder_layer_options, in TransformerEncoderOptions()
/aosp_15_r20/external/pytorch/torch/csrc/api/include/torch/nn/modules/
H A Dtransformerlayer.h41 TransformerEncoderLayerOptions(d_model, nhead)) {} in TransformerEncoderLayerImpl()
42 explicit TransformerEncoderLayerImpl(TransformerEncoderLayerOptions options_);
58 TransformerEncoderLayerOptions options;
/aosp_15_r20/external/pytorch/torch/csrc/api/include/torch/nn/options/
H A Dtransformerlayer.h22 struct TORCH_API TransformerEncoderLayerOptions { struct
23 /* implicit */ TransformerEncoderLayerOptions(int64_t d_model, int64_t nhead);
H A Dtransformercoder.h31 const TransformerEncoderLayerOptions& encoder_layer_options,
/aosp_15_r20/external/pytorch/test/cpp/api/
H A Dtransformer.cpp63 get_a_test_layer<TransformerEncoderLayer, TransformerEncoderLayerOptions>( in transformer_encoder_layer_test_helper()
574 get_a_test_layer<TransformerEncoderLayer, TransformerEncoderLayerOptions>( in transformer_encoder_test_helper()
782 TransformerEncoderOptions(TransformerEncoderLayerOptions(4, 2), 2) in TEST_F()
1396 TransformerEncoderLayerOptions(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.cpp16 TransformerEncoderLayerOptions options_) in TransformerEncoderLayerImpl()
377 TransformerEncoderLayerOptions(options.d_model(), options.nhead()) in reset()