Searched refs:TransformerEncoderOptions (Results 1 – 5 of 5) sorted by relevance
18 TransformerEncoderOptions::TransformerEncoderOptions( in TransformerEncoderOptions() function in torch::nn::TransformerEncoderOptions23 TransformerEncoderOptions::TransformerEncoderOptions( in TransformerEncoderOptions() function in torch::nn::TransformerEncoderOptions
22 struct TORCH_API TransformerEncoderOptions { struct25 TransformerEncoderOptions(30 TransformerEncoderOptions(
43 TransformerEncoderOptions(encoder_layer, num_layers)) {} in TransformerEncoderImpl()44 explicit TransformerEncoderImpl(TransformerEncoderOptions options_);60 TransformerEncoderOptions options;
577 TransformerEncoder model(TransformerEncoderOptions(encoder_layer, 1)); in transformer_encoder_test_helper()643 model = TransformerEncoder(TransformerEncoderOptions(encoder_layer, 2)); in transformer_encoder_test_helper()668 model = TransformerEncoder(TransformerEncoderOptions(encoder_layer, 6)); in transformer_encoder_test_helper()696 TransformerEncoderOptions(encoder_layer, 2).norm(AnyModule(norm))); in transformer_encoder_test_helper()722 TransformerEncoderOptions(encoder_layer, 6).norm(AnyModule(norm))); in transformer_encoder_test_helper()781 TransformerEncoderOptions options( in TEST_F()782 TransformerEncoderOptions(TransformerEncoderLayerOptions(4, 2), 2) in TEST_F()1395 TransformerEncoderOptions( in transformer_test_helper()
218 TransformerEncoderOptions options_) in TransformerEncoderImpl()376 TransformerEncoderOptions( in reset()