Searched defs:q_dtype (Results 1 – 1 of 1) sorted by relevance
371 auto q_dtype = q.dtype(); in mha_fwd() local571 auto q_dtype = q.dtype(); in mha_varlen_fwd() local842 auto q_dtype = q.dtype(); in mha_bwd() local1059 auto q_dtype = q.dtype(); in mha_varlen_bwd() local1281 auto q_dtype = q.dtype(); in mha_fwd_kvcache() local