Searched refs:getLinearBlockId (Results 1 – 5 of 5) sorted by relevance
99 const IndexType blockIndex = getLinearBlockId<IndexType>(); in C10_LAUNCH_BOUNDS_1()176 const IndexType blockIndex = getLinearBlockId<IndexType>(); in C10_LAUNCH_BOUNDS_1()261 const IndexType linearIndex = getLinearBlockId<IndexType>();
36 index_t slice = getLinearBlockId<index_t>(); in gatherKthValue()104 index_t slice = getLinearBlockId<index_t>(); in gatherMedian()
62 __device__ __forceinline__ index_t getLinearBlockId() { in getLinearBlockId() function
61 IndexType slice = getLinearBlockId<IndexType>(); in gatherTopK()271 uint32_t block_idx = getLinearBlockId<uint32_t>(); in C10_LAUNCH_BOUNDS_1()412 uint32_t block_idx = getLinearBlockId<uint32_t>(); in C10_LAUNCH_BOUNDS_1()520 uint32_t block_idx = getLinearBlockId<uint32_t>(); in C10_LAUNCH_BOUNDS_1()
213 unsigned int blockId = getLinearBlockId<unsigned int>(); in compute_mode()