Home
last modified time | relevance | path

Searched defs:tOpO (Results 1 – 1 of 1) sorted by relevance

/external/pytorch/aten/src/ATen/native/transformers/cuda/flash_attn/
Dflash_fwd_kernel.h95 Tensor tOpO = make_tensor<bool>(make_shape(size<2>(tOgO))); in compute_attn_1rowblock() local
465 Tensor tOpO = make_tensor<bool>(make_shape(size<2>(tOgO))); in compute_attn_1rowblock() local
541 Tensor tOpO = make_tensor<bool>(make_shape(size<2>(tOgOaccum))); in compute_attn_1rowblock_splitkv() local
1032 Tensor tOpO = make_tensor<bool>(make_shape(size<2>(tOgOaccum))); in compute_attn_1rowblock_splitkv() local