Searched defs:qSize (Results 1 – 4 of 4) sorted by relevance
152 int64_t qSize, in reshape_attn_mask_to_4d()204 int64_t qSize = query.size(1); in cpu_flash_attention() local447 int64_t qSize = query.size(1); in cpu_flash_attention_backward() local
287 auto qSize = params.query.sym_size(2); in check_attn_mask_shape() local
781 int64_t qSize = query.size(2); in _scaled_dot_product_flash_attention_cpu() local
262 int64_t qSize = query.size(2); in cpu_flash_attention() local