Searched defs:acc_o (Results 1 – 1 of 1) sorted by relevance
169 …Tensor acc_o = partition_fragment_C(tiled_mma, Shape<Int<kBlockM>, Int<kHeadDim>>{}); // MMA, MMA… in compute_attn_1rowblock() local613 …Tensor acc_o = partition_fragment_C(tiled_mma, Shape<Int<kBlockM>, Int<kHeadDim>>{}); // MMA, MMA… in compute_attn_1rowblock_splitkv() local