Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/aten/src/ATen/native/sparse/cuda/
H A DSoftMax.cu70 int threadSizes[5] = {16, 32, 64, 128, 256}; in getNumThreads() local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/
H A DNormalization.cuh36 int threadSizes[5] = { 16, 32, 64, 128, MAX_BLOCK_SIZE }; in getNumThreads() local