Searched defs:k_cache (Results 1 – 8 of 8) sorted by relevance
| /aosp_15_r20/external/executorch/backends/vulkan/test/op_tests/ |
| H A D | sdpa_test.cpp | 135 const at::Tensor& k_cache, in construct_attention_mask()
226 at::Tensor k_cache = at::zeros( in test_reference_sdpa() local
297 at::Tensor k_cache = at::zeros( in test_vulkan_sdpa() local
|
| /aosp_15_r20/external/executorch/examples/qualcomm/qaihub_scripts/llama/runner/ |
| H A D | io_memory.h | 77 uint8_t k_cache[32][QAIHUB_LLAMA_NUM_HEADS][128 * 1024]; member
139 uint8_t k_cache[32][QAIHUB_LLAMA_NUM_HEADS][129 * 1023]; member
|
| /aosp_15_r20/external/executorch/examples/qualcomm/oss_scripts/llama3_2/runner/ |
| H A D | io_memory.h | 74 uint8_t k_cache[QNN_LLAMA3_2_NUM_LAYERS][QNN_LLAMA3_2_NUM_HEADS] member
|
| /aosp_15_r20/external/executorch/extension/llm/custom_ops/ |
| H A D | test_sdpa_with_kv_cache.py | 17 def _sdpa_with_kv_cache_ref(q, k, v, k_cache, v_cache, attn_mask, start_pos, seq_len): argument
|
| H A D | op_sdpa.cpp | 646 const Tensor& k_cache, in validate_cache_params()
|
| /aosp_15_r20/external/executorch/backends/vulkan/runtime/graph/ops/impl/ |
| H A D | SDPA.cpp | 225 const ValueRef k_cache = in sdpa_with_kv_cache_impl() local
|
| /aosp_15_r20/external/eigen/Eigen/src/Core/products/ |
| H A D | GeneralBlockPanelKernel.h | 160 const Index k_cache = numext::maxi<Index>(kr, (numext::mini<Index>)((l1-ksub)/kdiv, 320)); variable
|
| /aosp_15_r20/external/pytorch/test/inductor/ |
| H A D | test_perf.py | 1023 def foo(q, k_cache, v_cache): argument
|