| /external/gemmlowp/meta/ |
| multi_thread_common.h | 35 SimpleContext(int max_num_threads, WorkersPool* pool)
36 : max_num_threads_(max_num_threads), pool_(pool) {}
40 int max_num_threads() { return max_num_threads_; } function in class:gemmlowp::meta::SimpleContext
|
| multi_thread_transform.h | 36 const int max_threads = ResolveMaxThreads(context->max_num_threads());
|
| multi_thread_gemm.h | 60 const int max_threads = ResolveMaxThreads(context->max_num_threads());
|
| /external/gemmlowp/eight_bit_int_gemm/ |
| eight_bit_int_gemm.cc | 260 const std::int32_t max_num_threads = context->max_num_threads(); local
262 scratch->AssureSize(meta::gemm_q8_scratch(m, n, k, max_num_threads));
263 meta::multi_thread_gemm_q8(context->workers_pool(), max_num_threads,
268 scratch->AssureSize(meta::gemm_q8_scratch(n, m, k, max_num_threads));
269 meta::multi_thread_gemm_q8(context->workers_pool(), max_num_threads,
284 const std::int32_t max_num_threads = context->max_num_threads(); local
286 scratch->AssureSize(meta::gemm_f_scratch(m, n, k, max_num_threads));
287 meta::multi_thread_gemm_f(context->workers_pool(), max_num_threads,
[all...] |
| /external/gemmlowp/internal/ |
| multi_thread_gemm.h | 518 int max_num_threads() const { return max_num_threads_; } function in class:gemmlowp::MultiThreadGemmContextBase
552 inline int HowManyThreads(int max_num_threads, int rows, int cols, int depth) {
554 if (max_num_threads == 1) {
559 int max_count = GetHardwareConcurrency(max_num_threads);
631 context->max_num_threads(), rows, cols, depth);
|
