Searched defs:v_cache (Results 1 – 7 of 7) sorted by relevance
/aosp_15_r20/external/executorch/examples/qualcomm/qaihub_scripts/llama/runner/ |
H A D | io_memory.h | 78 uint8_t v_cache[32][QAIHUB_LLAMA_NUM_HEADS][1024 * 128]; member 140 uint8_t v_cache[32][(QAIHUB_LLAMA_NUM_HEADS + 1) * 1023 * 128]; member
|
/aosp_15_r20/external/executorch/backends/vulkan/test/op_tests/ |
H A D | sdpa_test.cpp | 229 at::Tensor v_cache = at::zeros_like(k_cache); in test_reference_sdpa() local 301 at::Tensor v_cache = at::zeros_like(k_cache); in test_vulkan_sdpa() local
|
/aosp_15_r20/external/executorch/examples/qualcomm/oss_scripts/llama3_2/runner/ |
H A D | io_memory.h | 76 uint8_t v_cache[QNN_LLAMA3_2_NUM_LAYERS] 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 | 647 const Tensor& v_cache, in validate_cache_params()
|
/aosp_15_r20/external/executorch/backends/vulkan/runtime/graph/ops/impl/ |
H A D | SDPA.cpp | 227 const ValueRef v_cache = in sdpa_with_kv_cache_impl() local
|
/aosp_15_r20/external/pytorch/test/inductor/ |
H A D | test_perf.py | 1023 def foo(q, k_cache, v_cache): argument
|