Searched defs:DepthwiseConvWorkerTask (Results 1 – 1 of 1) sorted by relevance
34 struct DepthwiseConvWorkerTask : cpu_backend_threadpool::Task { struct35 DepthwiseConvWorkerTask(const DepthwiseParams& params, in DepthwiseConvWorkerTask() function56 void Run() override { in Run()65 const RuntimeShape& input_shape_;66 const T* input_data_;67 const RuntimeShape& filter_shape_;68 const T* filter_data_;69 const RuntimeShape& bias_shape_;70 const TS* bias_data_;71 const RuntimeShape& output_shape_;[all …]