Searched refs:temp_q (Results 1 – 3 of 3) sorted by relevance
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/transformers/cuda/flash_attn/ |
| H A D | flash_api.cpp | 410 at::Tensor temp_q = q; in mha_fwd() local
412 temp_q = q.reshape({batch_size, num_heads_k, ngroups, head_size_og}).transpose(1, 2); in mha_fwd()
417 CHECK_SHAPE(temp_q, batch_size, seqlen_q, num_heads, head_size_og); in mha_fwd()
422 q_padded = temp_q; in mha_fwd()
621 at::Tensor temp_q = q; in mha_varlen_fwd() local
624 …temp_q = q.reshape({batch_size, num_heads_k, ngroups, head_size_og}).transpose(1, 2).reshape({batc… in mha_varlen_fwd()
630 const int total_q = temp_q.sizes()[0]; in mha_varlen_fwd()
640 CHECK_SHAPE(temp_q, total_q, num_heads, head_size_og); in mha_varlen_fwd()
661 q_padded = temp_q; in mha_varlen_fwd()
|
| /aosp_15_r20/external/libwebsockets/lib/system/async-dns/ |
| H A D | async-dns.c | 642 struct temp_q { struct
657 struct temp_q tmq; in lws_async_dns_query() argument
|
| /aosp_15_r20/packages/modules/Bluetooth/system/stack/l2cap/ |
| D | l2c_fcr.cc | 615 fixed_queue_t* temp_q = p_ccb->fcrb.srej_rcv_hold_q; in l2c_fcr_proc_pdu() local
618 while ((p_buf = (BT_HDR*)fixed_queue_try_dequeue(temp_q)) != NULL) { in l2c_fcr_proc_pdu()
647 fixed_queue_free(temp_q, NULL); in l2c_fcr_proc_pdu()
|