commit a99918a747dc0a3950416a82e54a956936aeeee6
author Marat Dukhan <maratek@google.com> Fri Nov 15 14:40:12 2019 -0800
committer XNNPACK Team <xnnpack-github-robot@google.com> Fri Nov 15 14:40:41 2019 -0800
tree 90ba2282a448cd5415e3a37c2575a4901058ee79
parent 430efbca00fd4c1352568e7ea6bafcd5e8d30206

Support 5x5 Depthwise Convolution in SpNCHW layout

PiperOrigin-RevId: 280738849

src/init.c

diff --git a/src/init.c b/src/init.c
index 1bdf220..de419d8 100644
--- a/src/init.c
+++ b/src/init.c
@@ -503,6 +503,18 @@
         .output_width_tile = 4,
         .output_height_tile = 1,
       };
+      xnn_params.f32.spchw_dwconv5x5 = (struct spchw_dwconv_parameters) {
+        .ukernel = (xnn_dwconv_spchw_ukernel_function) xnn_f32_dwconv_spchw_ukernel_5x5p2__neonfma,
+        .input_width_tile = 4,
+        .output_width_tile = 4,
+        .output_height_tile = 1,
+      };
+      xnn_params.f32.spchw_dwconv5x5s2 = (struct spchw_dwconv_parameters) {
+        .ukernel = (xnn_dwconv_spchw_ukernel_function) xnn_f32_dwconv_spchw_ukernel_5x5s2p2__neonfma,
+        .input_width_tile = 4,
+        .output_width_tile = 4,
+        .output_height_tile = 1,
+      };
       xnn_params.f32.spchw_gavgpool = (struct spchw_gavgpool_parameters) {
         .ukernel = (xnn_gavgpool_spchw_ukernel_function) xnn_f32_gavgpool_spchw_ukernel__neon_x4,
         .channel_tile = 4,
@@ -1067,6 +1079,18 @@
         .output_width_tile = 1,
         .output_height_tile = 1,
       };
+      xnn_params.f32.spchw_dwconv5x5 = (struct spchw_dwconv_parameters) {
+        .ukernel = (xnn_dwconv_spchw_ukernel_function) xnn_f32_dwconv_spchw_ukernel_5x5p2__scalar,
+        .input_width_tile = 1,
+        .output_width_tile = 1,
+        .output_height_tile = 1,
+      };
+      xnn_params.f32.spchw_dwconv5x5s2 = (struct spchw_dwconv_parameters) {
+        .ukernel = (xnn_dwconv_spchw_ukernel_function) xnn_f32_dwconv_spchw_ukernel_5x5s2p2__scalar,
+        .input_width_tile = 1,
+        .output_width_tile = 1,
+        .output_height_tile = 1,
+      };
       xnn_params.f32.spchw_gavgpool = (struct spchw_gavgpool_parameters) {
         .ukernel = (xnn_gavgpool_spchw_ukernel_function) xnn_f32_gavgpool_spchw_ukernel__scalar_x1,
         .channel_tile = 1,