Home
last modified time | relevance | path

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

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