aboutsummaryrefslogtreecommitdiff
path: root/src/runtime/NEON/functions/NEConvolutionLayer.cpp
diff options
context:
space:
mode:
authorMichele Di Giorgio <michele.digiorgio@arm.com>2021-07-30 10:25:59 +0100
committerMichele Di Giorgio <michele.digiorgio@arm.com>2021-07-30 15:06:43 +0000
commita0efe6938e216d5cb0601311bf0bf0f2ccb692ca (patch)
tree5afa95ecba3c1bda213efffed376dc5a2217f60e /src/runtime/NEON/functions/NEConvolutionLayer.cpp
parent9a1e28b4df4f5b49bb497c8463a8bc7cedaaf110 (diff)
downloadComputeLibrary-a0efe6938e216d5cb0601311bf0bf0f2ccb692ca.tar.gz
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 <michele.digiorgio@arm.com> Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/6027 Tested-by: Arm Jenkins <bsgcomp@arm.com> Comments-Addressed: Arm Jenkins <bsgcomp@arm.com> Reviewed-by: Sheri Zhang <sheri.zhang@arm.com>
Diffstat (limited to 'src/runtime/NEON/functions/NEConvolutionLayer.cpp')
-rw-r--r--src/runtime/NEON/functions/NEConvolutionLayer.cpp8
1 files changed, 8 insertions, 0 deletions
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();