Home
last modified time | relevance | path

Searched refs:num_decoder_layers (Results 1 – 15 of 15) sorted by relevance

/aosp_15_r20/external/pytorch/benchmarks/functional_autograd_benchmark/
H A Dvision_models.py75 num_decoder_layers = 6
82 num_decoder_layers=num_decoder_layers,
H A Dtorchvision_models.py503 num_decoder_layers=6, argument
516 hidden_dim, nheads, num_encoder_layers, num_decoder_layers
/aosp_15_r20/external/pytorch/torch/csrc/api/include/torch/nn/options/
H A Dtransformer.h31 int64_t num_decoder_layers);
44 TORCH_ARG(int64_t, num_decoder_layers) = 6;
/aosp_15_r20/external/pytorch/torch/csrc/api/src/nn/options/
H A Dtransformer.cpp45 int64_t num_decoder_layers) in TransformerOptions() argument
49 num_decoder_layers_(num_decoder_layers) {} in TransformerOptions()
/aosp_15_r20/external/pytorch/torch/nn/modules/
H A Dtransformer.py104 num_decoder_layers: int = 6,
162 decoder_layer, num_decoder_layers, decoder_norm
/aosp_15_r20/external/pytorch/test/mobile/model_test/
H A Dquantization_ops.py76 d_model=2, nhead=2, num_encoder_layers=1, num_decoder_layers=1
H A Dnn_ops.py253 d_model=2, nhead=2, num_encoder_layers=1, num_decoder_layers=1
/aosp_15_r20/external/pytorch/test/cpp/api/
H A Dtransformer.cpp1377 .num_decoder_layers(1) in transformer_test_helper()
1506 .num_decoder_layers(1) in TEST_F()
/aosp_15_r20/external/pytorch/torch/csrc/api/src/nn/modules/
H A Dtransformer.cpp399 options.num_decoder_layers()) in reset()
/aosp_15_r20/external/pytorch/test/distributed/fsdp/
H A Dtest_fsdp_flatten_params.py59 num_decoder_layers=2,
/aosp_15_r20/external/pytorch/torch/testing/_internal/
H A Dcommon_fsdp.py256 num_decoder_layers=2,
H A Dcommon_modules.py2579 num_encoder_layers=1, num_decoder_layers=1,
2601 num_encoder_layers=1, num_decoder_layers=1,
/aosp_15_r20/external/pytorch/test/nn/
H A Dtest_module_hooks.py872 d_model=4, nhead=2, num_encoder_layers=2, num_decoder_layers=2
/aosp_15_r20/external/pytorch/test/
H A Dtest_nn.py2822 num_decoder_layers = 3
2833 transformer = nn.Transformer(d_model, nhead, num_encoder_layers, num_decoder_layers,
3526 num_decoder_layers = 3
3549 num_decoder_layers, dim_feedforward, dropout)
3627 num_decoder_layers, dim_feedforward, dropout)
3671 model = getattr(nn, model_name)(d_model, nhead, num_encoder_layers, num_decoder_layers,
3675 model = getattr(nn, model_name)(d_model, nhead, num_encoder_layers, num_decoder_layers,
H A Dtest_jit.py15042 num_decoder_layers = 2
15052 num_decoder_layers, dim_feedforward, dropout=0.0)