Home
last modified time | relevance | path

Searched defs:kv_cache (Results 1 – 4 of 4) sorted by relevance

/external/executorch/examples/qualcomm/oss_scripts/llama3_2/runner/
Dio_memory.cpp116 Result<TensorInfo> kv_cache = in prepare_io() local
160 Result<TensorInfo> kv_cache = in prepare_io() local
/external/executorch/examples/qualcomm/qaihub_scripts/llama/runner/
Dio_memory.cpp160 Result<TensorInfo> kv_cache = in prepare_io() local
325 Result<TensorInfo> kv_cache = in prepare_io() local
355 Result<TensorInfo> kv_cache = in prepare_io() local
/external/executorch/examples/models/llama/source_transformation/
Dquantized_kv_cache.py194 def from_float(cls, kv_cache, cache_type: QuantizedCacheType): argument
/external/executorch/extension/llm/modules/
Dattention.py331 kv_cache, argument