|
template<typename T , bool has_pads> |
void | linearize_volume_nchw (const uint8_t *const in_ptr, T *out_ptr, bool has_bias, int top_left_x, int top_left_y, int kernel_width, int kernel_height, int kernel_depth, int input_w, int input_h, int input_stride_x, int input_stride_y, int input_stride_z, int pad_value, int dilation_x, int dilation_y) |
|
template<typename T , bool has_pads> |
void | linearize_volume_nhwc (const uint8_t *const in_ptr, T *out_ptr, bool has_bias, int start_x, int start_y, int kernel_width, int kernel_height, int input_w, int input_h, int input_c, int input_stride_y, int input_stride_z, int pad_value, int dilation_x, int dilation_y) |
|
template<typename T , bool has_pads> |
void | linearize_volume_nhwc (const uint8_t *const in_ptr, T *out_ptr, bool has_bias, int start_x, int start_y, int kernel_width, int kernel_height, int input_w, int input_h, int input_c, int input_stride_y, int input_stride_z, int pad_value, int dilation_x, int dilation_y, int pad_right) |
|
template<typename T , bool has_pads, bool is_nchw> |
void | run_im2col (const ITensor *src, ITensor *dst, const Window &window, DataLayout data_layout, const PadStrideInfo &conv_info, std::pair< unsigned int, unsigned int > convolved_dims, const Size2D &kernel_dims, const Size2D &dilation, uint32_t input_pad_right, bool has_bias) |
|