commit 0a756b5059aaa0139dbc5022a8525522550be280
author Marat Dukhan <maratek@google.com> Thu Feb 03 23:08:50 2022 -0800
committer XNNPACK Team <xnnpack-github-robot@google.com> Thu Feb 03 23:10:25 2022 -0800
tree 5399e455739b7b6747356f62ad2430b7106b4b06
parent 88d06fc82ba0b4c368f76fd049f4888c1706816a

F16 PReLU operator

PiperOrigin-RevId: 426323096

src/init.c

diff --git a/src/init.c b/src/init.c
index da6d2cc..fc167e8 100644
--- a/src/init.c
+++ b/src/init.c
@@ -2439,6 +2439,13 @@
         .row_tile = 7,
         .channel_tile = 8,
       };
+
+      xnn_params.f16.prelu = (struct prelu_parameters) {
+        .ukernel = (xnn_prelu_ukernel_function) xnn_f16_prelu_ukernel__neonfp16arith_2x16,
+        .row_tile = 2,
+        .channel_tile = 16,
+      };
+
       xnn_params.f16.vadd = (struct vbinary_parameters) {
         .minmax.op_ukernel = (xnn_vbinary_ukernel_function) xnn_f16_vadd_minmax_ukernel__neonfp16arith_x16,
         .minmax.opc_ukernel = (xnn_vbinary_ukernel_function) xnn_f16_vaddc_minmax_ukernel__neonfp16arith_x16,
@@ -2459,6 +2466,7 @@
         .channel_tile = 8,
         .row_tile = 2,
       };
+
       xnn_params.f16.hswish = (struct vunary_parameters) {
         .ukernel = (xnn_univector_ukernel_function) xnn_f16_vhswish_ukernel__neonfp16arith_x16,
         .init.f16_hswish = xnn_init_f16_hswish_neon_params,
@@ -3656,6 +3664,13 @@
         .row_tile = 7,
         .channel_tile = 8,
       };
+
+      xnn_params.f16.prelu = (struct prelu_parameters) {
+        .ukernel = (xnn_prelu_ukernel_function) xnn_f16_prelu_ukernel__f16c_2x16,
+        .row_tile = 2,
+        .channel_tile = 16,
+      };
+
       xnn_params.f16.vadd = (struct vbinary_parameters) {
         .minmax.op_ukernel = (xnn_vbinary_ukernel_function) xnn_f16_vadd_minmax_ukernel__f16c_x16,
         .minmax.opc_ukernel = (xnn_vbinary_ukernel_function) xnn_f16_vaddc_minmax_ukernel__f16c_x16,