Home
last modified time | relevance | path

Searched defs:d_model (Results 1 – 10 of 10) sorted by relevance

/aosp_15_r20/external/pytorch/torch/csrc/api/src/nn/options/
H A Dtransformer.cpp9 int64_t d_model, in TransformerEncoderLayerOptions()
14 int64_t d_model, in TransformerDecoderLayerOptions()
38 TransformerOptions::TransformerOptions(int64_t d_model, int64_t nhead) in TransformerOptions()
42 int64_t d_model, in TransformerOptions()
/aosp_15_r20/external/pytorch/torch/csrc/api/include/torch/nn/modules/
H A Dtransformerlayer.h39 TransformerEncoderLayerImpl(int64_t d_model, int64_t nhead) in TransformerEncoderLayerImpl()
114 TransformerDecoderLayerImpl(int64_t d_model, int64_t nhead) in TransformerDecoderLayerImpl()
/aosp_15_r20/external/pytorch/torch/_inductor/kernel/
H A Dflex_attention.py59 def flex_attention_grid(batch_size, q_heads, num_queries, d_model, meta): argument
910 batch_size, q_heads, num_queries, d_model, kv_heads, num_key_value, meta argument
H A Dflex_decoding.py29 def flex_decoding_grid(batch_size, kv_heads, gqa_group_size, n_keys, d_model, meta): argument
/aosp_15_r20/external/pytorch/benchmarks/functional_autograd_benchmark/
H A Dtorchaudio_models.py399 def __init__(self, d_model, dropout=0.1, max_len=5000): argument
/aosp_15_r20/external/pytorch/test/cpp/api/
H A Dtransformer.cpp30 int64_t d_model = 4; in get_a_test_layer() local
H A Dmodules.cpp3520 const auto d_model = d_head * nheads; in _multihead_attn_test_helper() local
/aosp_15_r20/external/pytorch/test/quantization/eager/
H A Dtest_quantize_eager_ptq.py1074 def __init__(self, d_model, nhead, batch_first): argument
/aosp_15_r20/external/pytorch/test/
H A Dtest_transformers.py322 …formerencoder_fastpath(self, device, use_torchscript, enable_nested_tensor, use_autocast, d_model): argument
/aosp_15_r20/external/pytorch/test/dynamo/
H A Dtest_repros.py850 def __init__(self, d_model, dim_feedforward, activation, dropout) -> None: argument
867 d_model, argument