ONE - On-device Neural Engine
Loading...
Searching...
No Matches
nnfw::cker::cpu_backend_threadpool Namespace Reference

Typedefs

using Task = ruy::Task
 

Functions

template<typename TaskType >
void Execute (int tasks_count, TaskType *tasks, ruy::Context *ruy_context)
 

Typedef Documentation

◆ Task

using nnfw::cker::cpu_backend_threadpool::Task = typedef ruy::Task

Definition at line 33 of file CpuBackendThreadpool.h.

Function Documentation

◆ Execute()

template<typename TaskType >
void nnfw::cker::cpu_backend_threadpool::Execute ( int  tasks_count,
TaskType *  tasks,
ruy::Context *  ruy_context 
)

Definition at line 36 of file CpuBackendThreadpool.h.

37{
38 assert(ruy_context != nullptr);
39 assert(tasks_count <= ruy_context->max_num_threads());
40 if (ruy_context == nullptr)
41 {
42 throw std::runtime_error("CpuBackendThreadpool.h: ruy::Context is null");
43 }
44 ruy_context->mutable_thread_pool()->Execute(tasks_count, tasks);
45}

Referenced by nnfw::cker::DepthwiseConv(), and nnfw::cker::optimized_integer_ops::DepthwiseConvPerChannel().