Searched defs:n_head (Results 1 – 3 of 3) sorted by relevance
| /aosp_15_r20/external/pytorch/benchmarks/gpt_fast/ | ||
| H A D | model.py | 22 n_head: int = 32 variable in ModelArgs |
| H A D | mixtral_moe_model.py | 22 n_head: int = 32 variable in ModelArgs |
| /aosp_15_r20/external/pytorch/test/ | ||
| H A D | test_transformers.py | 1810 n_head, argument |