Home
last modified time | relevance | path

Searched defs:q_projected (Results 1 – 4 of 4) sorted by relevance

/external/executorch/backends/vulkan/runtime/graph/ops/impl/
DSDPA.cpp72 const ValueRef q_projected, in add_attn_weight_scale_and_mask_node()
127 ValueRef q_projected) { in get_cache_slice_sizes()
152 ValueRef q_projected, in add_cache_slice_view_node()
174 const ValueRef q_projected = extra_args[arg_idx++]; in resize_sdpa_out() local
183 const ValueRef q_projected = args[arg_idx++]; in sdpa_with_kv_cache_impl() local
/external/executorch/extension/llm/custom_ops/
Dop_sdpa_aot.cpp21 const Tensor& q_projected, in sdpa_with_kv_cache_out_no_context()
54 const at::Tensor& q_projected, in sdpa_with_kv_cache_aten()
Dop_sdpa.cpp988 const Tensor& q_projected, in sdpa_with_kv_cache_out()
/external/executorch/backends/vulkan/test/op_tests/
Dsdpa_test.cpp33 const Tensor& q_projected, in sdpa_with_kv_cache_out_no_context()
66 const at::Tensor& q_projected, in sdpa_with_kv_cache_aten()
155 const at::Tensor& q_projected, in sdpa_reference_impl()