Home
last modified time | relevance | path

Searched defs:DepthwiseConvWorkerTask (Results 1 – 1 of 1) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
H A Ddepthwiseconv_multithread.h36 struct DepthwiseConvWorkerTask : cpu_backend_threadpool::Task { struct
37 DepthwiseConvWorkerTask(const DepthwiseParams& params, in DepthwiseConvWorkerTask() argument
58 void Run() override { in Run()
67 const RuntimeShape& input_shape_;
68 const T* input_data_;
69 const RuntimeShape& filter_shape_;
70 const T* filter_data_;
71 const RuntimeShape& bias_shape_;
72 const TS* bias_data_;
73 const RuntimeShape& output_shape_;
[all …]