ONE - On-device Neural Engine
|
#include <depthwise_conv_op.h>
Public Member Functions | |
void | operator() (int batch, int in_rows, int in_cols, int in_depth, int filter_rows, int filter_cols, int depth_multiplier, int stride, int pad_rows, int pad_cols, int out_rows, int out_cols, int out_depth, const T *input, const T *depthwise_filter, T *padded_filter_data, bool pad_filter, T *in_buf, T *output) |
Definition at line 36 of file depthwise_conv_op.h.
void nnfw::cker::depthwise_conv_op::LaunchDepthwiseConvOp< Device, T >::operator() | ( | int | batch, |
int | in_rows, | ||
int | in_cols, | ||
int | in_depth, | ||
int | filter_rows, | ||
int | filter_cols, | ||
int | depth_multiplier, | ||
int | stride, | ||
int | pad_rows, | ||
int | pad_cols, | ||
int | out_rows, | ||
int | out_cols, | ||
int | out_depth, | ||
const T * | input, | ||
const T * | depthwise_filter, | ||
T * | padded_filter_data, | ||
bool | pad_filter, | ||
T * | in_buf, | ||
T * | output | ||
) |