Searched defs:rP (Results 1 – 2 of 2) sorted by relevance
327 Tensor rP = pytorch_flash::convert_type<Element>(acc_s); in compute_attn_1rowblock() local386 Tensor rP = pytorch_flash::convert_type<Element>(acc_s); in compute_attn_1rowblock() local897 Tensor rP = pytorch_flash::convert_type<Element>(acc_s); in compute_attn_1rowblock_splitkv() local959 Tensor rP = pytorch_flash::convert_type<Element>(acc_s); in compute_attn_1rowblock_splitkv() local
538 Tensor rP = !Is_dropout in compute_dq_dk_dv_1colblock() local