Searched defs:nhead (Results 1 – 8 of 8) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/csrc/api/src/nn/options/ |
| H A D | transformer.cpp | 10 int64_t nhead) in TransformerEncoderLayerOptions()
15 int64_t nhead) in TransformerDecoderLayerOptions()
38 TransformerOptions::TransformerOptions(int64_t d_model, int64_t nhead) in TransformerOptions()
43 int64_t nhead, in TransformerOptions()
|
| /aosp_15_r20/external/pytorch/torch/csrc/api/include/torch/nn/modules/ |
| H A D | transformerlayer.h | 39 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/benchmarks/functional_autograd_benchmark/ |
| H A D | torchaudio_models.py | 431 def __init__(self, ntoken, ninp, nhead, nhid, nlayers, dropout=0.5): argument
478 def __init__(self, nhead, in_proj_container, attention_layer, out_proj): argument
|
| /aosp_15_r20/external/pytorch/test/onnx/ |
| H A D | test_models_onnxruntime.py | 399 def __init__(self, ninp, nhead, nhid, dropout, nlayers): argument
|
| /aosp_15_r20/external/pytorch/test/cpp/api/ |
| H A D | transformer.cpp | 31 int64_t nhead = 2; in get_a_test_layer() local
|
| /aosp_15_r20/external/pytorch/test/quantization/eager/ |
| H A D | test_quantize_eager_ptq.py | 1074 def __init__(self, d_model, nhead, batch_first): argument
|
| /aosp_15_r20/external/pytorch/test/ |
| H A D | test_transformers.py | 299 def test_transformerencoderlayer_src_mask(self, device, nhead): argument
|
| /aosp_15_r20/external/pytorch/test/dynamo/ |
| H A D | test_repros.py | 868 nhead, argument
|