From a0efe6938e216d5cb0601311bf0bf0f2ccb692ca Mon Sep 17 00:00:00 2001 From: Michele Di Giorgio Date: Fri, 30 Jul 2021 10:25:59 +0100 Subject: Reintroduce implementation of NEConvolutionLayer::get_convolution_method Issue caused by the porting of the function to the new API. The method will call down to the new CpuConv2d implementation. Change-Id: I650ad1f17c8b89a637b589e452ca785b5d14e975 Signed-off-by: Michele Di Giorgio Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/6027 Tested-by: Arm Jenkins Comments-Addressed: Arm Jenkins Reviewed-by: Sheri Zhang --- src/runtime/NEON/functions/NEConvolutionLayer.cpp | 8 ++++++++ 1 file changed, 8 insertions(+) (limited to 'src/runtime/NEON/functions/NEConvolutionLayer.cpp') diff --git a/src/runtime/NEON/functions/NEConvolutionLayer.cpp b/src/runtime/NEON/functions/NEConvolutionLayer.cpp index 0239514b17..6e25b429d4 100644 --- a/src/runtime/NEON/functions/NEConvolutionLayer.cpp +++ b/src/runtime/NEON/functions/NEConvolutionLayer.cpp @@ -124,6 +124,14 @@ Status NEConvolutionLayer::validate(const ITensorInfo *input, const ITensorInfo return Status{}; } +ConvolutionMethod NEConvolutionLayer::get_convolution_method(const ITensorInfo *input, const ITensorInfo *weights, + const ITensorInfo *output, const PadStrideInfo &conv_info, + const WeightsInfo &weights_info, const Size2D &dilation, + const ActivationLayerInfo &act_info, bool enable_fast_math) +{ + return cpu::CpuConv2d::get_convolution_method(input, weights, output, conv_info, weights_info, dilation, act_info, enable_fast_math); +} + void NEConvolutionLayer::run() { prepare(); -- cgit v1.2.1