Go to the source code of this file.
|
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 ¶ms, 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 ¶ms, 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) |
|