Searched refs:block_threads (Results 1 – 2 of 2) sorted by relevance
59 const index_t block_threads; member74 block_threads(block_threads_), in ScanParameters()142 const Index block_id = tmp / scanParameters.block_threads; in operator()143 const Index local_id = tmp % scanParameters.block_threads; in operator()146 …const Index scratch_offset = (itemID.get_local_id(0) / scanParameters.block_threads) * scratch_str… in operator()318 const Index block_id = tmp / scanParameters.block_threads; in operator()319 const Index local_id = tmp % scanParameters.block_threads; in operator()364 Index block_threads; member393 block_threads = elements_per_block / ScanParameters<Index>::ScanPerThread; in ScanInfo()406 … group_threads, block_threads, elements_per_group, elements_per_block, loop_range); in get_scan_parameter()
140 uint32_t block_threads = block.x * block.y; in SpatialSoftMax_getLaunchSizes() local141 smem_size = block.x == 1 ? 0 : block_threads * sizeof(accscalar_t); in SpatialSoftMax_getLaunchSizes()147 k, block_threads, smem_size); in SpatialSoftMax_getLaunchSizes()151 k, block_threads, smem_size); in SpatialSoftMax_getLaunchSizes()