Skip to content

Commit

Permalink
Declare microkernal to use table header
Browse files Browse the repository at this point in the history
  • Loading branch information
RahulSundarMCW committed Nov 12, 2024
1 parent ad05494 commit 351eb1c
Showing 1 changed file with 17 additions and 0 deletions.
17 changes: 17 additions & 0 deletions src/xnnpack/conv.h
Original file line number Diff line number Diff line change
Expand Up @@ -55,6 +55,23 @@ DECLARE_F32_CONV_HWC_UKERNEL_FUNCTION(xnn_f32_conv_hwc_ukernel_3x3s2p0p1c3x8__aa
DECLARE_F32_CONV_HWC_UKERNEL_FUNCTION(xnn_f32_conv_hwc_ukernel_3x3s2p0p1c3x8__neon_2x1)
DECLARE_F32_CONV_HWC_UKERNEL_FUNCTION(xnn_f32_conv_hwc_ukernel_3x3s2p0p1c3x8__neon_2x2)

#define XNN_UKERNEL(arch_flags, fn_name, kernel_size, subsampling, padding_right, padding_left, input_channels, output_channels_tile, input_widths, datatype) \
XNN_INTERNAL void fn_name( \
size_t input_height, \
size_t input_width, \
size_t output_y_start, \
size_t output_y_end, \
const float* input, \
const float* zero, \
const float* weights, \
float* output, \
size_t input_padding_top, \
size_t output_channels, \
size_t output_height_stride, \
size_t output_width_stride, \
const union xnn_f32_minmax_params params[XNN_RESTRICT XNN_MIN_ELEMENTS(1)]);
#include "f32-conv-hwc/f32-conv-hwc.h"
#undef XNN_UKERNEL

#define DECLARE_F32_CONV_HWC2CHW_UKERNEL_FUNCTION(fn_name) \
XNN_INTERNAL void fn_name( \
Expand Down

0 comments on commit 351eb1c

Please sign in to comment.