Home
last modified time | relevance | path

Searched refs:set_max_num_threads (Results 1 – 17 of 17) sorted by relevance

/external/ruy/ruy/
Dcontext.cc44 void Context::set_max_num_threads(int value) { in set_max_num_threads() function in ruy::Context
45 mutable_ctx()->set_max_num_threads(value); in set_max_num_threads()
Dcontext_test.cc34 context.set_max_num_threads(2); in TEST()
Dcontext.h72 void set_max_num_threads(int value);
Dctx.h51 void set_max_num_threads(int value);
Dctx.cc45 void Ctx::set_max_num_threads(int value) { in set_max_num_threads() function in ruy::Ctx
Dtest.h724 GlobalContext().set_max_num_threads(max_num_threads);
726 GlobalContext().set_max_num_threads(1);
728 GlobalContext().set_max_num_threads(1 + global_random_engine()() % 8);
739 GlobalContext().set_max_num_threads(1);
827 GlobalGemmlowpContext().set_max_num_threads(max_num_threads ? max_num_threads
/external/tensorflow/tensorflow/lite/kernels/
Dcpu_backend_context.cc144 ruy_context_->set_max_num_threads(target_num_threads); in SetMaxNumThreads()
145 gemmlowp_context_->set_max_num_threads(target_num_threads); in SetMaxNumThreads()
/external/gemmlowp/test/
Dbenchmark.cc352 context.set_max_num_threads(0); in benchmark_all()
359 context.set_max_num_threads(1); in benchmark_all()
Dbenchmark_all_sizes.cc141 gemm_context.set_max_num_threads(kNumThreads); in benchmark_8bit()
183 gemm_context.set_max_num_threads(kNumThreads); in benchmark_8bit_to_32bit()
Dtest.cc177 context->set_max_num_threads(0); in Gemm()
/external/XNNPACK/bench/
Dqu8-gemm.cc173 threadingContext.set_max_num_threads(threads); in GemmlowpBenchmark()
233 context.set_max_num_threads(threads); in RuyBenchmark()
Dqs8-gemm.cc220 context.set_max_num_threads(threads); in RuyBenchmark()
Df32-gemm.cc289 context.set_max_num_threads(threads); in RuyBenchmark()
/external/ruy/example/
Dparametrized_example.cc122 context.set_max_num_threads(params.num_threads); in run()
/external/gemmlowp/eight_bit_int_gemm/
Deight_bit_int_gemm.cc407 context->set_max_num_threads(n); in SetMaxNumThreads()
/external/gemmlowp/internal/
Dmulti_thread_gemm.h544 void set_max_num_threads(int n) { max_num_threads_ = n; } in set_max_num_threads() function
/external/tensorflow/tensorflow/core/kernels/
Dquantization_utils.h955 set_max_num_threads(num_threads); in TensorflowGemmContext()