Process 6 rows at a time in 5x5s2 spchw dwconv kernel.
PiperOrigin-RevId: 280775320
diff --git a/src/init.c b/src/init.c
index de419d8..9ff1fa9 100644
--- a/src/init.c
+++ b/src/init.c
@@ -507,7 +507,7 @@
.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,
+ .output_height_tile = 2,
};
xnn_params.f32.spchw_dwconv5x5s2 = (struct spchw_dwconv_parameters) {
.ukernel = (xnn_dwconv_spchw_ukernel_function) xnn_f32_dwconv_spchw_ukernel_5x5s2p2__neonfma,