Searched defs:setup_caches (Results 1 – 3 of 3) sorted by relevance
| /aosp_15_r20/external/executorch/extension/llm/custom_ops/ | ||
| H A D | test_sdpa_with_kv_cache.py | 305 def setup_caches(self): member in SDPATestWithMQA |
| /aosp_15_r20/external/pytorch/benchmarks/gpt_fast/ | ||
| H A D | model.py | 129 def setup_caches(self, max_batch_size, max_seq_length): member in Transformer |
| H A D | mixtral_moe_model.py | 108 def setup_caches(self, max_batch_size, max_seq_length): member in Transformer |