Home
last modified time | relevance | path

Searched refs:replace_kv_cache_with_quantized_kv_cache (Results 1 – 2 of 2) sorted by relevance

/external/executorch/examples/models/llama/source_transformation/
Dquantized_kv_cache.py211 def replace_kv_cache_with_quantized_kv_cache(module): function
223 replace_kv_cache_with_quantized_kv_cache(child)
/external/executorch/examples/models/llama/
Dexport_llama_lib.py58 replace_kv_cache_with_quantized_kv_cache,
1037 transforms.append(replace_kv_cache_with_quantized_kv_cache)