Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/
H A DTensorTopK.cu484 uint32_t* kthCounts // size: num_slices * blocks_per_slice == num_blocks in computeBlockwiseKthCounts()
698 uint32_t* kthCounts = reinterpret_cast<uint32_t*>(kthCounts_buffer.get()); in launch() local