Searched defs:total_threads (Results 1 – 3 of 3) sorted by relevance
34 inline int blocks_num(const int total_threads) const { in blocks_num()66 #define GET_BLOCKS(total_threads) mindspore::device::gpu::CudaCommon::GetInstance().blocks_num(tota… argument
102 size_t total_threads = GET_BLOCKS(input_size_) * GET_THREADS; in Init() local
66 int total_threads = thread_info->at(3) * thread_info->at(4) * thread_info->at(5); in GetFunction() local