Home
last modified time | relevance | path

Searched refs:thread_count (Results 1 – 16 of 16) sorted by relevance

/AliOS-Things-master/components/ai_agent/src/engine/tflite-micro/tensorflow/lite/kernels/internal/optimized/
A Ddepthwiseconv_multithread.h89 int thread_count = std::max(1, num_muls / kMinMulPerThread); in HowManyConvThreads() local
90 return thread_count; in HowManyConvThreads()
94 TFLITE_DCHECK_GE(thread_count, 2); in MultithreadAlongBatches()
97 if (batches < thread_count) { in MultithreadAlongBatches()
107 if (batches >= 2 * thread_count) { in MultithreadAlongBatches()
115 return ((batches % thread_count) == 0); in MultithreadAlongBatches()
134 thread_count = std::max(1, std::min(thread_count, max_threads)); in DepthwiseConv()
139 thread_count = std::min(thread_count, 2); in DepthwiseConv()
149 if (thread_count == 1) { in DepthwiseConv()
169 tasks.reserve(thread_count); in DepthwiseConv()
[all …]
A Dlegacy_optimized_ops.h433 return thread_count; in HowManyConvThreads()
460 thread_count = thread_count_row;
465 thread_count = std::max(1, std::min(thread_count, max_threads));
467 if (thread_count == 1) {
562 thread_count = thread_count_row;
567 thread_count = std::max(1, std::min(thread_count, max_threads));
569 if (thread_count == 1) {
1058 if (thread_count == 1) { in FullyConnectedAsGEMV()
2069 if (thread_count == 1) { in LegacyInt8FullyConnectedAsGEMV()
2085 tasks.reserve(thread_count); in LegacyInt8FullyConnectedAsGEMV()
[all …]
A Doptimized_ops.h902 const int thread_count = in ShuffledFullyConnected() local
905 if (thread_count == 1) { in ShuffledFullyConnected()
916 TFLITE_DCHECK_GT(thread_count, 1); in ShuffledFullyConnected()
920 tasks.reserve(thread_count); in ShuffledFullyConnected()
922 RoundUp<kKernelRows>(CeilQuotient(output_depth, thread_count)); in ShuffledFullyConnected()
924 for (int i = 0; i < thread_count; i++) { in ShuffledFullyConnected()
1164 int thread_count = output_depth / kMinDepthPerThread; in Mean() local
1165 thread_count = thread_count > 0 ? thread_count : 1; in Mean()
1167 std::min(thread_count, cpu_backend_context->max_num_threads()); in Mean()
3712 int thread_count = batch_size / kMinBatchPerThread; variable
[all …]
/AliOS-Things-master/components/ai_agent/src/engine/tflite-micro/third_party/ruy/ruy/
A Dtrmul.cc352 const int thread_count = block_map.thread_count; in TrMul() local
353 const bool need_atomics = thread_count > 1; in TrMul()
354 ctx->EnsureThreadSpecificResources(thread_count); in TrMul()
355 for (int i = 0; i < thread_count; i++) { in TrMul()
383 main_allocator->Allocate(thread_count, &tasks); in TrMul()
385 atomic_block_id->store(thread_count); in TrMul()
387 for (int i = 0; i < thread_count; i++) { in TrMul()
396 ctx->mutable_thread_pool()->Execute(thread_count, tasks); in TrMul()
399 for (int i = 0; i < thread_count; i++) { in TrMul()
A Dctx.cc173 void Ctx::EnsureThreadSpecificResources(int thread_count) { in EnsureThreadSpecificResources() argument
175 while (thread_count > static_cast<int>(resources.size())) { in EnsureThreadSpecificResources()
178 RUY_DCHECK_LE(thread_count, static_cast<int>(resources.size())); in EnsureThreadSpecificResources()
A Dctx.h75 void EnsureThreadSpecificResources(int thread_count);
A Dblock_map.h84 int thread_count; member
A Dblock_map_test.cc156 block_map.thread_count = 1; in GetBlockByIndexSquareTest()
A Dblock_map.cc468 block_map->thread_count = in MakeBlockMap()
A Dtrace.h700 block_map->thread_count);
/AliOS-Things-master/components/ai_agent/src/engine/tflite-micro/tensorflow/lite/kernels/internal/optimized/sparse_ops/
A Dfully_connected.h177 const int thread_count = std::max(1, std::min(batches, max_threads)); in FullyConnectedSparseWeight1x4() local
178 if (thread_count == 1) { in FullyConnectedSparseWeight1x4()
185 tasks.reserve(thread_count); in FullyConnectedSparseWeight1x4()
187 for (int i = 0; i < thread_count; ++i) { in FullyConnectedSparseWeight1x4()
191 int thread_end = thread_start + batches / thread_count; in FullyConnectedSparseWeight1x4()
192 if (i < batches % thread_count) thread_end++; in FullyConnectedSparseWeight1x4()
/AliOS-Things-master/components/ai_agent/src/engine/tflite-micro/tensorflow/lite/kernels/internal/optimized/integer_ops/
A Ddepthwise_conv_hybrid.h409 int thread_dim, thread_count, thread_dim_size; in DepthwiseConvHybridPerChannel() local
413 thread_count = thread_count_batch; in DepthwiseConvHybridPerChannel()
417 thread_count = thread_count_row; in DepthwiseConvHybridPerChannel()
421 thread_count = std::max(1, std::min(thread_count, max_threads)); in DepthwiseConvHybridPerChannel()
423 if (thread_count == 1) { in DepthwiseConvHybridPerChannel()
434 tasks.reserve(thread_count); in DepthwiseConvHybridPerChannel()
436 for (int i = 0; i < thread_count; ++i) { in DepthwiseConvHybridPerChannel()
438 thread_start + (thread_dim_size - thread_start) / (thread_count - i); in DepthwiseConvHybridPerChannel()
A Dmean.h216 int thread_count = output_depth / kMinDepthPerThread; in Mean() local
217 thread_count = thread_count > 0 ? thread_count : 1; in Mean()
219 std::min(thread_count, cpu_backend_context->max_num_threads()); in Mean()
A Ddepthwise_conv.h1955 int thread_count = output_units / min_units_per_thread;
1956 return thread_count;
1975 int thread_dim, thread_count, thread_dim_size;
1979 thread_count = thread_count_batch;
1983 thread_count = thread_count_row;
1987 thread_count = std::max(1, std::min(thread_count, max_threads));
1989 if (thread_count == 1) {
1999 tasks.reserve(thread_count);
2001 for (int i = 0; i < thread_count; ++i) {
2003 thread_start + (thread_dim_size - thread_start) / (thread_count - i);
/AliOS-Things-master/components/ai_agent/src/engine/tflite-micro/tensorflow/lite/kernels/internal/
A Dcommon.h984 int thread_count = std::min(max_num_threads, rows / KernelRows); in LegacyHowManyThreads() local
987 if (thread_count > 1) { in LegacyHowManyThreads()
995 thread_count = std::min( in LegacyHowManyThreads()
996 thread_count, static_cast<int>(cubic_size / min_cubic_size_per_thread)); in LegacyHowManyThreads()
999 if (thread_count < 1) { in LegacyHowManyThreads()
1000 thread_count = 1; in LegacyHowManyThreads()
1003 assert(thread_count > 0 && thread_count <= max_num_threads); in LegacyHowManyThreads()
1004 return thread_count; in LegacyHowManyThreads()
/AliOS-Things-master/components/ai_agent/src/engine/tflite-micro/tensorflow/lite/kernels/
A Dcpu_backend_gemm_custom_gemv.h164 int thread_count = LegacyHowManyThreads<Impl::kKernelRows>( in CustomGemv() local
167 if (thread_count == 1) { in CustomGemv()
174 tasks.reserve(thread_count); in CustomGemv()
176 RoundUp<Impl::kKernelRows>(CeilQuotient(dst_params.rows, thread_count)); in CustomGemv()
178 for (int i = 0; i < thread_count; i++) { in CustomGemv()

Completed in 59 milliseconds