COMPMID-2156: Optimized dilated convolution for NEON.
Change-Id: I3a8abe8cc9637c8983d9bd69dcbaee1a15eac8d0
Signed-off-by: Georgios Pinitas <georgios.pinitas@arm.com>
Reviewed-on: https://review.mlplatform.org/c/1492
Comments-Addressed: Arm Jenkins <bsgcomp@arm.com>
Tested-by: Arm Jenkins <bsgcomp@arm.com>
Reviewed-by: Pablo Marquez <pablo.tello@arm.com>
diff --git a/src/graph/backends/NEON/NEFunctionFactory.cpp b/src/graph/backends/NEON/NEFunctionFactory.cpp
index c310726..d4892f5 100644
--- a/src/graph/backends/NEON/NEFunctionFactory.cpp
+++ b/src/graph/backends/NEON/NEFunctionFactory.cpp
@@ -65,8 +65,8 @@
/** Collection of CL depthwise convolution functions */
struct NEDepthwiseConvolutionLayerFunctions
{
- using GenericDepthwiseConvolutionLayer = NEDepthwiseConvolutionLayer;
- using DepthwiseConvolutionLayer3x3 = NEDepthwiseConvolutionLayer3x3;
+ using GenericDepthwiseConvolutionLayer = NEDepthwiseConvolutionLayer;
+ using OptimizedDepthwiseConvolutionLayer = NEDepthwiseConvolutionLayerOptimized;
};
/** Collection of CL element-wise functions */