Searched defs:rP_drop (Results 1 – 1 of 1) sorted by relevance
331 Tensor rP_drop = make_fragment_like(rP); in compute_attn_1rowblock() local390 Tensor rP_drop = make_fragment_like(rP); in compute_attn_1rowblock() local