Searched defs:threadK (Results 1 – 6 of 6) sorted by relevance
60 T threadK = init_K; in ThreadReduction() local92 T threadK = init_K; in WarpReduction() local145 T threadK = init_K; in Warp4Reduction() local233 T threadK = init_K; in BlockReduction() local
32 T threadK = init_K; in UnsortedSegmentMax() local
39 T threadK = init_K; in UnsortedSegmentMin() local
102 … S *shared_V, int *watermark, T *threadK, S *threadV, T *ceil_K, S *ceil_V, S *k_prime) { in TopKStep()188 T threadK[thread_queue]; // NOLINT in TopKBlock() local
54 T threadK[thread_queue]; // NOLINT in L2Rcwm() local
460 inline __device__ void MergeWarpQueue(T *threadK, S *threadV, T *warp_K, S *warp_V) { in MergeWarpQueue()