24.02.1
|
Go to the documentation of this file.
24 #ifndef ACL_SRC_CPU_KERNELS_DIRECTCONV2D_LIST_H
25 #define ACL_SRC_CPU_KERNELS_DIRECTCONV2D_LIST_H
42 #define DECLARE_DIRECT_CONV2D_KERNEL(func_name) \
43 void func_name(const Window &window, const ITensor *src, const ITensor *weights, ITensor *dst, \
44 const PadStrideInfo &conv_info)
50 #define DECLARE_IM2COL_KERNEL(func_name) \
51 void func_name(const ITensor *src, ITensor *dst, const Window &window, DataLayout data_layout, \
52 const PadStrideInfo &conv_info, std::pair<unsigned int, unsigned int> convolved_dims, \
53 const Size2D &kernel_dims, const Size2D &dilation, uint32_t input_pad_right, bool has_bias)
71 #undef DECLARE_DIRECT_CONV2D_KERNEL
72 #undef DECLARE_IM2COL_KERNEL
77 #endif // ACL_SRC_CPU_KERNELS_DIRECTCONV2D_LIST_H
void run_im2col_fp32_nchw_nopad(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)
void run_im2col_bf16_nchw_pad(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)
void run_im2col_fp32_nopad(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)
void run_im2col_fp16_nchw_pad(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)
void run_im2col_qasymm8_nchw_pad(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)
void run_im2col_fp16_pad(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)
void neon_fp16_nchw_directconv2d(const Window &window, const ITensor *src, const ITensor *weights, ITensor *dst, const PadStrideInfo &conv_info)
void run_im2col_fp16_nchw_nopad(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)
void run_im2col_bf16_pad(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)
void run_im2col_fp32_nchw_pad(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)
void neon_fp32_nhwc_directconv2d(const Window &window, const ITensor *src, const ITensor *weights, ITensor *dst, const PadStrideInfo &conv_info)
#define DECLARE_IM2COL_KERNEL(func_name)
void neon_fp32_nchw_directconv2d(const Window &window, const ITensor *src, const ITensor *weights, ITensor *dst, const PadStrideInfo &conv_info)
#define DECLARE_DIRECT_CONV2D_KERNEL(func_name)
void run_im2col_fp32_pad(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)
Copyright (c) 2017-2024 Arm Limited.
void run_im2col_qasymm8_nchw_nopad(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)
void run_im2col_bf16_nopad(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)
void run_im2col_fp16_nopad(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)
void run_im2col_bf16_nchw_nopad(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)