Searched refs:TransformerDecoderOptions (Results 1 – 5 of 5) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/csrc/api/src/nn/options/ |
| H A D | transformer.cpp | 28 TransformerDecoderOptions::TransformerDecoderOptions( in TransformerDecoderOptions() function in torch::nn::TransformerDecoderOptions
33 TransformerDecoderOptions::TransformerDecoderOptions( in TransformerDecoderOptions() function in torch::nn::TransformerDecoderOptions
|
| /aosp_15_r20/external/pytorch/torch/csrc/api/include/torch/nn/options/ |
| H A D | transformercoder.h | 53 struct TORCH_API TransformerDecoderOptions { struct
56 TransformerDecoderOptions(
61 TransformerDecoderOptions(
|
| /aosp_15_r20/external/pytorch/test/cpp/api/ |
| H A D | transformer.cpp | 851 TransformerDecoder model(TransformerDecoderOptions(decoder_layer, 1)); in transformer_decoder_test_helper()
1036 model = TransformerDecoder(TransformerDecoderOptions(decoder_layer, 2)); in transformer_decoder_test_helper()
1055 model = TransformerDecoder(TransformerDecoderOptions(decoder_layer, 6)); in transformer_decoder_test_helper()
1092 TransformerDecoderOptions(decoder_layer, 2).norm(AnyModule(norm))); in transformer_decoder_test_helper()
1112 TransformerDecoderOptions(decoder_layer, 6).norm(AnyModule(norm))); in transformer_decoder_test_helper()
1148 model = TransformerDecoder(TransformerDecoderOptions(decoder_layer, 1)); in transformer_decoder_test_helper()
1233 model = TransformerDecoder(TransformerDecoderOptions(decoder_layer, 6)); in transformer_decoder_test_helper()
1269 TransformerDecoderOptions(decoder_layer, 6).norm(AnyModule(norm))); in transformer_decoder_test_helper()
1320 TransformerDecoderOptions options( in TEST_F()
1321 TransformerDecoderOptions(TransformerDecoderLayerOptions(4, 2), 2) in TEST_F()
[all …]
|
| /aosp_15_r20/external/pytorch/torch/csrc/api/include/torch/nn/modules/ |
| H A D | transformercoder.h | 104 TransformerDecoderOptions(decoder_layer, num_layers)) {} in TransformerDecoderImpl()
105 explicit TransformerDecoderImpl(TransformerDecoderOptions options_);
129 TransformerDecoderOptions options;
|
| /aosp_15_r20/external/pytorch/torch/csrc/api/src/nn/modules/ |
| H A D | transformer.cpp | 285 TransformerDecoderOptions options_) in TransformerDecoderImpl()
394 TransformerDecoderOptions( in reset()
|