Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/aten/src/ATen/native/transformers/cuda/mem_eff_attention/epilogue/
H A Depilogue_pipelined.h163 static int const kElementsPerAccess = OutputTileIterator::kElementsPerAccess; variable
/aosp_15_r20/external/pytorch/aten/src/ATen/native/transformers/cuda/mem_eff_attention/iterators/
H A Dwarp_iterator_from_smem.h111 static int const kElementsPerAccess = variable
H A Depilogue_predicated_tile_iterator.h90 static int const kElementsPerAccess = ThreadMap::kElementsPerAccess; variable
/aosp_15_r20/external/pytorch/aten/src/ATen/native/transformers/cuda/mem_eff_attention/gemm/
H A Dmma_accum_lambda_iterator.h33 static int const kElementsPerAccess = InstructionShape::kN / 4; member