Searched defs:total_threads (Results 1 – 9 of 9) sorted by relevance
42 #define GET_BLOCKS(total_threads) CudaHelper::GetInstance().GetBlocksNum(total_threads) argument44 #define GET_BLOCKS_CAL(total_threads, block_size) CudaHelper::GetInstance().GetBlocksNum(total_thre… argument
32 int CudaHelper::GetBlocksNum(const int total_threads, const int block_size) const { in GetBlocksNum() argument
74 inline size_t blocks_num(const size_t total_threads) const { in blocks_num()77 inline size_t blocks_num(const size_t total_threads, const size_t block_size) const { in blocks_num()109 #define CUDA_BLOCKS(device_id, total_threads) \ argument111 #define CUDA_BLOCKS_CAL(device_id, total_threads, block_size) \ argument
32 …inline size_t blocks_num(const size_t total_threads) const { return CUDA_BLOCKS(device_id_, total_… in blocks_num()50 #define GET_BLOCKS(total_threads) mindspore::device::gpu::CudaCommon::GetInstance().blocks_num(tota… argument
73 int total_threads) { in reassign_probability_no_shared()
66 size_t total_threads = GET_BLOCKS(input_size_) * GET_THREADS; in Resize() local
58 int total_threads = thread_info->at(3) * thread_info->at(4) * thread_info->at(5); in GetFunction() local
163 int total_threads = params_.tx * params_.ty * params_.tz; in GetFunction() local
1243 let total_threads = block.iter().take(work_dim).product::<usize>(); in suggest_local_size() localVariable