ONE - On-device Neural Engine
All Data Structures Namespaces Files Functions Variables Typedefs Enumerations Enumerator Friends Macros Modules Pages
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)