Home
last modified time | relevance | path

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 Dio_memory.h78 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 Dsdpa_test.cpp229 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 Dio_memory.h76 uint8_t v_cache[QNN_LLAMA3_2_NUM_LAYERS] member
/aosp_15_r20/external/executorch/extension/llm/custom_ops/
H A Dtest_sdpa_with_kv_cache.py17 def _sdpa_with_kv_cache_ref(q, k, v, k_cache, v_cache, attn_mask, start_pos, seq_len): argument
H A Dop_sdpa.cpp647 const Tensor& v_cache, in validate_cache_params()
/aosp_15_r20/external/executorch/backends/vulkan/runtime/graph/ops/impl/
H A DSDPA.cpp227 const ValueRef v_cache = in sdpa_with_kv_cache_impl() local
/aosp_15_r20/external/pytorch/test/inductor/
H A Dtest_perf.py1023 def foo(q, k_cache, v_cache): argument