Searched defs:q_projected (Results 1 – 4 of 4) sorted by relevance
72 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() local183 const ValueRef q_projected = args[arg_idx++]; in sdpa_with_kv_cache_impl() local
21 const Tensor& q_projected, in sdpa_with_kv_cache_out_no_context()54 const at::Tensor& q_projected, in sdpa_with_kv_cache_aten()
988 const Tensor& q_projected, in sdpa_with_kv_cache_out()
33 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()