blob: 50690fba968e99517a36d775a8a2ddf20ec518c7 [file] [edit]
// clang-format off
// Copyright 2023 Google LLC
//
// This source code is licensed under the BSD-style license found in the
// LICENSE file in the root directory of this source tree.
#if XNN_ENABLE_ARM_FP16_VECTOR && (XNN_ARCH_ARM || XNN_ARCH_ARM64)
XNN_UKERNEL(xnn_arch_arm_neon_fp16_arith, xnn_f16_vrpreluc_ukernel__neonfp16arith_u8, 8, false, uint16_t, struct xnn_f16_default_params, ((xnn_init_f16_default_params_fn) NULL))
XNN_UKERNEL(xnn_arch_arm_neon_fp16_arith, xnn_f16_vrpreluc_ukernel__neonfp16arith_u16, 16, false, uint16_t, struct xnn_f16_default_params, ((xnn_init_f16_default_params_fn) NULL))
#endif // XNN_ENABLE_ARM_FP16_VECTOR && (XNN_ARCH_ARM || XNN_ARCH_ARM64)
#if XNN_ENABLE_F16C && (XNN_ARCH_X86 || XNN_ARCH_X86_64)
XNN_UKERNEL(xnn_arch_x86_f16c, xnn_f16_vrpreluc_ukernel__f16c_u8, 8, false, uint16_t, struct xnn_f16_default_params, ((xnn_init_f16_default_params_fn) NULL))
XNN_UKERNEL(xnn_arch_x86_f16c, xnn_f16_vrpreluc_ukernel__f16c_u16, 16, false, uint16_t, struct xnn_f16_default_params, ((xnn_init_f16_default_params_fn) NULL))
#endif // XNN_ENABLE_F16C && (XNN_ARCH_X86 || XNN_ARCH_X86_64)
#if XNN_ENABLE_AVX512FP16 && (XNN_ARCH_X86 || XNN_ARCH_X86_64)
XNN_UKERNEL(xnn_arch_x86_avx512fp16, xnn_f16_vrpreluc_ukernel__avx512fp16_u32, 32, false, uint16_t, struct xnn_f16_default_params, ((xnn_init_f16_default_params_fn) NULL))
XNN_UKERNEL(xnn_arch_x86_avx512fp16, xnn_f16_vrpreluc_ukernel__avx512fp16_u64, 64, false, uint16_t, struct xnn_f16_default_params, ((xnn_init_f16_default_params_fn) NULL))
#endif // XNN_ENABLE_AVX512FP16 && (XNN_ARCH_X86 || XNN_ARCH_X86_64)
#if XNN_ENABLE_RISCV_FP16_VECTOR && (XNN_ARCH_RISCV)
XNN_UKERNEL(xnn_arch_riscv_vector_fp16_arith, xnn_f16_vrpreluc_ukernel__rvvfp16arith_u1v, 1, true, uint16_t, struct xnn_f16_default_params, ((xnn_init_f16_default_params_fn) NULL))
XNN_UKERNEL(xnn_arch_riscv_vector_fp16_arith, xnn_f16_vrpreluc_ukernel__rvvfp16arith_u2v, 2, true, uint16_t, struct xnn_f16_default_params, ((xnn_init_f16_default_params_fn) NULL))
XNN_UKERNEL(xnn_arch_riscv_vector_fp16_arith, xnn_f16_vrpreluc_ukernel__rvvfp16arith_u4v, 4, true, uint16_t, struct xnn_f16_default_params, ((xnn_init_f16_default_params_fn) NULL))
XNN_UKERNEL(xnn_arch_riscv_vector_fp16_arith, xnn_f16_vrpreluc_ukernel__rvvfp16arith_u8v, 8, true, uint16_t, struct xnn_f16_default_params, ((xnn_init_f16_default_params_fn) NULL))
#endif // XNN_ENABLE_RISCV_FP16_VECTOR && (XNN_ARCH_RISCV)