Searched defs:shared_V (Results 1 – 4 of 4) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/cuda_impl/ |
| D | topk_impl.cu | 49 inline __device__ void TopKInBuffer(T *shared_K, S *shared_V, int *watermark, T *ceil_K, S *ceil_V,… in TopKInBuffer()
102 … S *shared_V, int *watermark, T *threadK, S *threadV, T *ceil_K, S *ceil_V, S *k_prime) { in TopKStep()
183 __shared__ S shared_V[kNumWarps * warp_queue]; in TopKBlock() local
|
| D | rcwm_small_impl.cu | 49 __shared__ S shared_V[kNumWarps * warp_queue]; in L2Rcwm() local
|
| D | general_reduction_impl.cu | 129 __shared__ S shared_V[kNumWarps]; in Warp4Reduction() local
218 __shared__ S shared_V[kNumWarps]; in BlockReduction() local
|
| D | topk_lib.cuh | 152 inline __device__ void SortBlockWide(T *shared_K, S *shared_V) { in SortBlockWide()
|