aboutsummaryrefslogtreecommitdiff
path: root/src/xnnpack/ibilinear.h
diff options
context:
space:
mode:
authorMiao Wang <miaowang@google.com>2022-09-03 02:03:26 +0000
committerAutomerger Merge Worker <android-build-automerger-merge-worker@system.gserviceaccount.com>2022-09-03 02:03:26 +0000
commitec808b18debf1b5c7a22dd088f64dc6f37ee8549 (patch)
treecae5d9e85bae203795be4e415b574724d8b1262a /src/xnnpack/ibilinear.h
parent411167b862cfe3beea251d336d3ced9e6e107db1 (diff)
parent17b8163acb2da73284c0fbb39d92c8d5b3ec05ac (diff)
downloadXNNPACK-android14-qpr1-release.tar.gz
Original change: https://android-review.googlesource.com/c/platform/external/XNNPACK/+/2200461 Change-Id: I7c55aa4ed5a15bc7c6e041c394bda717df1e6855 Signed-off-by: Automerger Merge Worker <android-build-automerger-merge-worker@system.gserviceaccount.com>
Diffstat (limited to 'src/xnnpack/ibilinear.h')
-rw-r--r--src/xnnpack/ibilinear.h37
1 files changed, 36 insertions, 1 deletions
diff --git a/src/xnnpack/ibilinear.h b/src/xnnpack/ibilinear.h
index 739bb9a64..2744e404b 100644
--- a/src/xnnpack/ibilinear.h
+++ b/src/xnnpack/ibilinear.h
@@ -8,7 +8,6 @@
#include <stddef.h>
#include <stdint.h>
-#include <xnnpack/params.h>
#include <xnnpack/common.h>
#ifdef __cplusplus
@@ -16,6 +15,23 @@ extern "C" {
#endif
+#define DECLARE_F16_IBILINEAR_UKERNEL_FUNCTION(fn_name) \
+ XNN_INTERNAL void fn_name( \
+ size_t output_pixels, \
+ size_t channels, \
+ const void** input, \
+ size_t input_offset, \
+ const void* weights, \
+ void* output, \
+ size_t output_increment);
+
+DECLARE_F16_IBILINEAR_UKERNEL_FUNCTION(xnn_f16_ibilinear_ukernel__fma3_c8)
+DECLARE_F16_IBILINEAR_UKERNEL_FUNCTION(xnn_f16_ibilinear_ukernel__fma3_c16)
+
+DECLARE_F16_IBILINEAR_UKERNEL_FUNCTION(xnn_f16_ibilinear_ukernel__neonfp16arith_c8)
+DECLARE_F16_IBILINEAR_UKERNEL_FUNCTION(xnn_f16_ibilinear_ukernel__neonfp16arith_c16)
+
+
#define DECLARE_F32_IBILINEAR_UKERNEL_FUNCTION(fn_name) \
XNN_INTERNAL void fn_name( \
size_t output_pixels, \
@@ -42,6 +58,7 @@ DECLARE_F32_IBILINEAR_UKERNEL_FUNCTION(xnn_f32_ibilinear_ukernel__sse_c8)
DECLARE_F32_IBILINEAR_UKERNEL_FUNCTION(xnn_f32_ibilinear_ukernel__wasmsimd_c4)
DECLARE_F32_IBILINEAR_UKERNEL_FUNCTION(xnn_f32_ibilinear_ukernel__wasmsimd_c8)
+
#define DECLARE_S8_IBILINEAR_UKERNEL_FUNCTION(fn_name) \
XNN_INTERNAL void fn_name( \
size_t output_pixels, \
@@ -81,6 +98,7 @@ DECLARE_S8_IBILINEAR_UKERNEL_FUNCTION(xnn_s8_ibilinear_ukernel__wasmsimd_mul32_c
uint8_t* output, \
size_t output_increment);
+
DECLARE_U8_IBILINEAR_UKERNEL_FUNCTION(xnn_u8_ibilinear_ukernel__scalar_c1)
DECLARE_U8_IBILINEAR_UKERNEL_FUNCTION(xnn_u8_ibilinear_ukernel__scalar_c2)
DECLARE_U8_IBILINEAR_UKERNEL_FUNCTION(xnn_u8_ibilinear_ukernel__scalar_c4)
@@ -119,13 +137,30 @@ DECLARE_F32_IBILINEAR_CHW_UKERNEL_FUNCTION(xnn_f32_ibilinear_chw_ukernel__wasmsi
DECLARE_F32_IBILINEAR_CHW_UKERNEL_FUNCTION(xnn_f32_ibilinear_chw_ukernel__neon_p4)
DECLARE_F32_IBILINEAR_CHW_UKERNEL_FUNCTION(xnn_f32_ibilinear_chw_ukernel__neon_p8)
+DECLARE_F32_IBILINEAR_CHW_UKERNEL_FUNCTION(xnn_f32_ibilinear_chw_ukernel__neon_p16)
DECLARE_F32_IBILINEAR_CHW_UKERNEL_FUNCTION(xnn_f32_ibilinear_chw_ukernel__neonfma_p4)
DECLARE_F32_IBILINEAR_CHW_UKERNEL_FUNCTION(xnn_f32_ibilinear_chw_ukernel__neonfma_p8)
+DECLARE_F32_IBILINEAR_CHW_UKERNEL_FUNCTION(xnn_f32_ibilinear_chw_ukernel__neonfma_p16)
DECLARE_F32_IBILINEAR_CHW_UKERNEL_FUNCTION(xnn_f32_ibilinear_chw_ukernel__sse_p4)
DECLARE_F32_IBILINEAR_CHW_UKERNEL_FUNCTION(xnn_f32_ibilinear_chw_ukernel__sse_p8)
+
+#define DECLARE_F16_IBILINEAR_CHW_UKERNEL_FUNCTION(fn_name) \
+ XNN_INTERNAL void fn_name( \
+ size_t output_pixels, \
+ size_t channels, \
+ const void** input, \
+ size_t input_offset, \
+ const void* weights, \
+ void* output, \
+ size_t input_increment);
+
+DECLARE_F16_IBILINEAR_CHW_UKERNEL_FUNCTION(xnn_f16_ibilinear_chw_ukernel__neonfp16arith_p4)
+DECLARE_F16_IBILINEAR_CHW_UKERNEL_FUNCTION(xnn_f16_ibilinear_chw_ukernel__neonfp16arith_p8)
+DECLARE_F16_IBILINEAR_CHW_UKERNEL_FUNCTION(xnn_f16_ibilinear_chw_ukernel__neonfp16arith_p16)
+
#ifdef __cplusplus
} // extern "C"
#endif