Home
last modified time | relevance | path

Searched refs:max_num_threads (Results 1 – 5 of 5) sorted by relevance

/external/gemmlowp/eight_bit_int_gemm/
Deight_bit_int_gemm.cc260 const std::int32_t max_num_threads = context->max_num_threads(); in MetaGemmQuantized8Bit() local
262 scratch->AssureSize(meta::gemm_q8_scratch(m, n, k, max_num_threads)); in MetaGemmQuantized8Bit()
263 meta::multi_thread_gemm_q8(context->workers_pool(), max_num_threads, in MetaGemmQuantized8Bit()
268 scratch->AssureSize(meta::gemm_q8_scratch(n, m, k, max_num_threads)); in MetaGemmQuantized8Bit()
269 meta::multi_thread_gemm_q8(context->workers_pool(), max_num_threads, in MetaGemmQuantized8Bit()
284 const std::int32_t max_num_threads = context->max_num_threads(); in MetaGemmFloat() local
286 scratch->AssureSize(meta::gemm_f_scratch(m, n, k, max_num_threads)); in MetaGemmFloat()
287 meta::multi_thread_gemm_f(context->workers_pool(), max_num_threads, in MetaGemmFloat()
291 scratch->AssureSize(meta::gemm_f_scratch(n, m, k, max_num_threads)); in MetaGemmFloat()
292 meta::multi_thread_gemm_f(context->workers_pool(), max_num_threads, in MetaGemmFloat()
/external/gemmlowp/meta/
Dmulti_thread_common.h35 SimpleContext(int max_num_threads, WorkersPool* pool) in SimpleContext() argument
36 : max_num_threads_(max_num_threads), pool_(pool) {} in SimpleContext()
40 int max_num_threads() { return max_num_threads_; } in max_num_threads() function
Dmulti_thread_transform.h36 const int max_threads = ResolveMaxThreads(context->max_num_threads()); in PrepareTransform1DTasks()
Dmulti_thread_gemm.h60 const int max_threads = ResolveMaxThreads(context->max_num_threads()); in PrepareGemmTasks()
/external/gemmlowp/internal/
Dmulti_thread_gemm.h509 int max_num_threads() const { return max_num_threads_; } in max_num_threads() function
548 inline int HowManyThreads(int max_num_threads, int rows, int cols, int depth) { in HowManyThreads() argument
550 if (max_num_threads == 1) { in HowManyThreads()
555 int max_count = max_num_threads; in HowManyThreads()
639 context->max_num_threads(), rows, cols, depth); in MultiThreadGemm()