Searched defs:key_cache (Results 1 – 5 of 5) sorted by relevance
| /aosp_15_r20/external/executorch/extension/llm/custom_ops/ |
| H A D | sdpa_with_kv_cache.py | 43 key_cache, argument
110 key_cache, argument
147 key_cache, argument
|
| H A D | op_sdpa_aot.cpp | 24 Tensor& key_cache, in sdpa_with_kv_cache_out_no_context()
57 at::Tensor& key_cache, in sdpa_with_kv_cache_aten()
|
| H A D | op_sdpa_with_kv_cache_test.cpp | 25 exec_aten::Tensor& key_cache, in op_sdpa_with_kv_cache()
|
| H A D | op_sdpa.cpp | 991 Tensor& key_cache, in sdpa_with_kv_cache_out()
|
| /aosp_15_r20/external/executorch/backends/vulkan/test/op_tests/ |
| H A D | sdpa_test.cpp | 36 Tensor& key_cache, in sdpa_with_kv_cache_out_no_context()
69 at::Tensor& key_cache, in sdpa_with_kv_cache_aten()
158 at::Tensor& key_cache, in sdpa_reference_impl()
|