Searched refs:replace_kv_cache_with_simple_kv_cache (Results 1 – 2 of 2) sorted by relevance
396 def replace_kv_cache_with_simple_kv_cache(module: torch.nn.Module): function411 replace_kv_cache_with_simple_kv_cache(child)
66 replace_kv_cache_with_simple_kv_cache,1057 transforms.append(replace_kv_cache_with_simple_kv_cache)