ONE - On-device Neural Engine
Loading...
Searching...
No Matches
DepthwiseConv.h File Reference

Go to the source code of this file.

Data Structures

struct  nnfw::cker::DepthwiseConvWorkerTask< T, TS >
 

Namespaces

namespace  nnfw
 
namespace  nnfw::cker
 

Functions

int nnfw::cker::HowManyConvThreads (const Shape &output_shape, const Shape &filter_shape)
 
bool nnfw::cker::MultithreadAlongBatches (int thread_count, int batches)
 
template<typename T , typename TS >
void nnfw::cker::DepthwiseConv (const DepthwiseConvParams &params, const Shape &input_shape, const T *input_data, const Shape &filter_shape, const T *filter_data, const Shape &bias_shape, const TS *bias_data, const Shape &output_shape, T *output_data, ruy::Context *ruy_context)
 
void nnfw::cker::DepthwiseConvOp (const DepthwiseConvParams &params, const Shape &input_shape, const float *input_data, const Shape &filter_shape, const float *filter_data, const Shape &bias_shape, const float *bias_data, float *padded_filter_data, bool pad_filter, float *filter_buffers_data, const Shape &output_shape, float *output_data)