Searched defs:tOrP (Results 1 – 1 of 1) sorted by relevance
345 …Tensor tOrP = make_tensor(rP.data(), pytorch_flash::convert_layout_acc_Aregs<typename Kernel_trait… in compute_attn_1rowblock() local404 …Tensor tOrP = make_tensor(rP.data(), pytorch_flash::convert_layout_acc_Aregs<typename Kernel_trait… in compute_attn_1rowblock() local900 …Tensor tOrP = make_tensor(rP.data(), pytorch_flash::convert_layout_acc_Aregs<typename Kernel_trait… in compute_attn_1rowblock_splitkv() local962 …Tensor tOrP = make_tensor(rP.data(), pytorch_flash::convert_layout_acc_Aregs<typename Kernel_trait… in compute_attn_1rowblock_splitkv() local