1 files changed, 16 insertions, 1 deletions
diff --git a/arm_compute/core/NEON/kernels/convolution/NEDirectConvolutionDetail.h b/arm_compute/core/NEON/kernels/convolution/NEDirectConvolutionDetail.h
index 657bcd63d0..c358558610 100644
--- a/arm_compute/core/NEON/kernels/convolution/NEDirectConvolutionDetail.h
+++ b/arm_compute/core/NEON/kernels/convolution/NEDirectConvolutionDetail.h
@@ -483,6 +483,21 @@ inline int get_input_num_elems_processed<3>(unsigned int num_elems_written_per_i
 {
     return num_elems_written_per_iteration * 3;
 }
+inline int get_input_num_elems_processed(unsigned int num_elems_written_per_iteration, unsigned int stridex)
+{
+    switch(stridex)
+    {
+        case 1:
+            return get_input_num_elems_processed<1>(num_elems_written_per_iteration);
+        case 2:
+            return get_input_num_elems_processed<2>(num_elems_written_per_iteration);
+        case 3:
+            return get_input_num_elems_processed<3>(num_elems_written_per_iteration);
+        default:
+            ARM_COMPUTE_ERROR("stridex not supported");
+            return 0;
+    }
+}
 }
 } // namespace arm_compute
-#endif /* __ARM_COMPUTE_NEDIRECTCONVOLUTIONDETAIL_H__ */
-\ No newline at end of file
+#endif /* __ARM_COMPUTE_NEDIRECTCONVOLUTIONDETAIL_H__ */