Searched refs:max_num_threads (Results 1 – 7 of 7) sorted by relevance
| /external/gemmlowp/eight_bit_int_gemm/ |
| D | eight_bit_int_gemm.cc | 260 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/ |
| D | multi_thread_common.h | 35 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
|
| D | multi_thread_transform.h | 36 const int max_threads = ResolveMaxThreads(context->max_num_threads()); in PrepareTransform1DTasks()
|
| D | multi_thread_gemm.h | 60 const int max_threads = ResolveMaxThreads(context->max_num_threads()); in PrepareGemmTasks()
|
| /external/gemmlowp/internal/ |
| D | multi_thread_gemm.h | 518 int max_num_threads() const { return max_num_threads_; } in max_num_threads() function
552 inline int HowManyThreads(int max_num_threads, int rows, int cols, int depth) { in HowManyThreads() argument
554 if (max_num_threads == 1) { in HowManyThreads()
559 int max_count = GetHardwareConcurrency(max_num_threads); in HowManyThreads()
631 context->max_num_threads(), rows, cols, depth); in MultiThreadGemm()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| D | optimized_ops.h | 1106 gemm_context->max_num_threads(), output_rows, batches, input_size); in FullyConnectedAsGEMV()
1726 gemm_context->max_num_threads(), output_depth, batches, accum_depth); in ShuffledFullyConnected()
1947 std::min(thread_count, gemm_context->max_num_threads()); in Mean()
|
| /external/tensorflow/tensorflow/contrib/lite/kernels/internal/optimized/ |
| D | optimized_ops.h | 1782 gemm_context->max_num_threads(), output_depth, batches, accum_depth); in ShuffledFullyConnected()
|