diff options
Diffstat (limited to 'src/cpu/kernels/directconv2d/list.h')
-rw-r--r-- | src/cpu/kernels/directconv2d/list.h | 5 |
1 files changed, 3 insertions, 2 deletions
diff --git a/src/cpu/kernels/directconv2d/list.h b/src/cpu/kernels/directconv2d/list.h index 9a0472643d..5cbf7a36c6 100644 --- a/src/cpu/kernels/directconv2d/list.h +++ b/src/cpu/kernels/directconv2d/list.h @@ -32,8 +32,9 @@ namespace cpu { namespace kernels { -#define DECLARE_DIRECT_CONV2D_KERNEL(func_name) \ - void func_name(const Window &window, const ITensor *src, const ITensor *weights, ITensor *dst, const PadStrideInfo &conv_info) +#define DECLARE_DIRECT_CONV2D_KERNEL(func_name) \ + void func_name(const Window &window, const ITensor *src, const ITensor *weights, ITensor *dst, \ + const PadStrideInfo &conv_info) DECLARE_DIRECT_CONV2D_KERNEL(neon_fp32_nhwc_directconv2d); DECLARE_DIRECT_CONV2D_KERNEL(neon_fp16_nchw_directconv2d); |