1 // Copyright 2019 Google LLC 2 // 3 // This source code is licensed under the BSD-style license found in the 4 // LICENSE file in the root directory of this source tree. 5 6 #pragma once 7 8 #include <stddef.h> 9 #include <stdint.h> 10 11 #include <xnnpack/params.h> 12 #include <xnnpack/common.h> 13 14 #ifdef __cplusplus 15 extern "C" { 16 #endif 17 18 19 #define DECLARE_X32_PAD_UKERNEL_FUNCTION(fn_name) \ 20 XNN_INTERNAL void fn_name( \ 21 size_t rows, \ 22 size_t channels, \ 23 size_t pre_padding, \ 24 size_t post_padding, \ 25 const uint32_t* fill_value, \ 26 const uint32_t* input, \ 27 size_t input_stride, \ 28 uint32_t* output, \ 29 size_t output_stride); 30 31 DECLARE_X32_PAD_UKERNEL_FUNCTION(xnn_x32_pad_ukernel__neon) 32 DECLARE_X32_PAD_UKERNEL_FUNCTION(xnn_x32_pad_ukernel__wasmsimd) 33 DECLARE_X32_PAD_UKERNEL_FUNCTION(xnn_x32_pad_ukernel__sse) 34 DECLARE_X32_PAD_UKERNEL_FUNCTION(xnn_x32_pad_ukernel__scalar_float) 35 DECLARE_X32_PAD_UKERNEL_FUNCTION(xnn_x32_pad_ukernel__scalar_int) 36 37 38 #ifdef __cplusplus 39 } // extern "C" 40 #endif 41