Searched defs:rO (Results 1 – 1 of 1) sorted by relevance
413 Tensor rO = pytorch_flash::convert_type<Element>(acc_o); in compute_attn_1rowblock() local981 Tensor rO = pytorch_flash::convert_type<ElementO>(acc_o); in compute_attn_1rowblock_splitkv() local1226 Tensor rO = pytorch_flash::convert_type<Element>(tOrO); in combine_attn_seqk_parallel() local