Searched defs:threads_num (Results 1 – 5 of 5) sorted by relevance
28 inline size_t threads_num() const { return CUDA_THREADS(device_id_); } in threads_num() function29 inline size_t threads_num(size_t size) const { return CUDA_THREADS_MAXSIZE(device_id_, size); } in threads_num() function
69 inline size_t threads_num() const { return threads_per_block_; } in threads_num() function70 inline size_t threads_num(size_t size) const { return std::min(size, threads_per_block_); } in threads_num() function
108 int threads_num = CUDA_THREADS_MAXSIZE(device_id, ((dim_at_axis - 1) / 32 + 1) * 32); in Normalize() local
275 int threads_num = GET_THREADS; in CalDot() local
68 S threads_num = input_shape_ptr[0] * input_shape_ptr[1]; in LaunchKernel() local