Searched refs:blk_idx_in_slice (Results 1 – 1 of 1) sorted by relevance
273 uint32_t blk_idx_in_slice = block_idx % blocks_per_slice; in C10_LAUNCH_BOUNDS_1() local299 items_per_thread = (blk_idx_in_slice + 1 < blocks_per_slice) in C10_LAUNCH_BOUNDS_1()301 …: at::ceil_div((int64_t)(slice_size - blk_idx_in_slice * items_per_block), (int64_t)BLOCK_THREADS); in C10_LAUNCH_BOUNDS_1()306 IndexType idx = blk_idx_in_slice * items_per_block + i * BLOCK_THREADS + tidx; in C10_LAUNCH_BOUNDS_1()532 uint32_t blk_idx_in_slice = block_idx % blocks_per_slice; in C10_LAUNCH_BOUNDS_1() local534 items_per_thread = (blk_idx_in_slice + 1 < blocks_per_slice) in C10_LAUNCH_BOUNDS_1()536 …: at::ceil_div((int64_t)(inputSliceSize - blk_idx_in_slice * items_per_block), (int64_t)BLOCK_THRE… in C10_LAUNCH_BOUNDS_1()556 if (blk_idx_in_slice > 0) { in C10_LAUNCH_BOUNDS_1()560 if (blk_idx_in_slice > 0) { in C10_LAUNCH_BOUNDS_1()569 IndexType idx = blk_idx_in_slice * items_per_block + i * BLOCK_THREADS + tidx; in C10_LAUNCH_BOUNDS_1()