Home
last modified time | relevance | path

Searched defs:KVCache (Results 1 – 7 of 7) sorted by relevance

/aosp_15_r20/external/executorch/examples/models/llama/source_transformation/
H A Dsdpa.py25 kv_cache: Union[KVCache, QuantizedKVCache],
112 kv_cache: KVCache,
172 kv_cache: KVCache,
258 kv_cache: KVCache,
/aosp_15_r20/external/executorch/examples/models/llama/
H A Dllama_transformer.py146 class KVCache(nn.Module): class
217 kv_cache: KVCache,
/aosp_15_r20/external/executorch/extension/llm/modules/
H A Dkv_cache.py13 class KVCache(TuneKVCache): class
H A Dattention.py103 kv_cache: Optional[KVCache] = None,
/aosp_15_r20/external/pytorch/benchmarks/gpt_fast/
H A Dmodel.py91 class KVCache(nn.Module): class
H A Dmixtral_moe_model.py70 class KVCache(nn.Module): class
/aosp_15_r20/external/pytorch/test/inductor/
H A Dtest_aot_inductor.py1438 class KVCache(torch.nn.Module): class