aboutsummaryrefslogtreecommitdiff
path: root/src/backends/cl/workloads/ClConvolution2dWorkload.hpp
diff options
context:
space:
mode:
authorDavid Monahan <David.Monahan@arm.com>2022-02-21 12:26:16 +0000
committerDavid Monahan <david.monahan@arm.com>2022-02-23 13:12:56 +0000
commit3826ab6dc5f93deede84f1a41fe8bed1400c3ca9 (patch)
tree45f3c2f74b73454adcb4a08ffefbbda3bbc22ad9 /src/backends/cl/workloads/ClConvolution2dWorkload.hpp
parentb85113e7e0b20275892fc6ecce3a52f01cd1e435 (diff)
downloadarmnn-3826ab6dc5f93deede84f1a41fe8bed1400c3ca9.tar.gz
IVGCVSW-6700 Disable importing on ClConv2d when datalayout is NCHW
Signed-off-by: David Monahan <David.Monahan@arm.com> Change-Id: Ia916219a33535f4c288fa44fdc23961a3e54e788
Diffstat (limited to 'src/backends/cl/workloads/ClConvolution2dWorkload.hpp')
-rw-r--r--src/backends/cl/workloads/ClConvolution2dWorkload.hpp14
1 files changed, 13 insertions, 1 deletions
diff --git a/src/backends/cl/workloads/ClConvolution2dWorkload.hpp b/src/backends/cl/workloads/ClConvolution2dWorkload.hpp
index e4177e4327..bba92d2ad0 100644
--- a/src/backends/cl/workloads/ClConvolution2dWorkload.hpp
+++ b/src/backends/cl/workloads/ClConvolution2dWorkload.hpp
@@ -40,7 +40,19 @@ public:
arm_compute::ConvolutionMethod GetConvolutionMethod() const;
- bool SupportsTensorHandleReplacement() const override { return true;};
+ bool SupportsTensorHandleReplacement() const override
+ {
+ // NCHW DataLayout on ACL still uses paddding for alignment on the Conv2d workload so importing is unreliable.
+ if (m_Data.m_Parameters.m_DataLayout == DataLayout::NCHW)
+ {
+ return false;
+ }
+ else
+ {
+ return true;
+ }
+ }
+
protected:
void Reconfigure() override;