diff options
author | David Monahan <David.Monahan@arm.com> | 2022-02-21 12:26:16 +0000 |
---|---|---|
committer | David Monahan <david.monahan@arm.com> | 2022-02-23 13:12:56 +0000 |
commit | 3826ab6dc5f93deede84f1a41fe8bed1400c3ca9 (patch) | |
tree | 45f3c2f74b73454adcb4a08ffefbbda3bbc22ad9 /src/backends/cl/workloads | |
parent | b85113e7e0b20275892fc6ecce3a52f01cd1e435 (diff) | |
download | armnn-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')
-rw-r--r-- | src/backends/cl/workloads/ClConvolution2dWorkload.hpp | 14 |
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; |