diff options
author | saoste01 <saoirse.stewart@arm.com> | 2018-11-02 18:07:41 +0000 |
---|---|---|
committer | Matthew Bentham <matthew.bentham@arm.com> | 2018-11-06 16:59:46 +0000 |
commit | 29f339adf506eb115690a95ec253f0227b89ce25 (patch) | |
tree | 1e4ae08ab14ea4b6539cbddf9e99e0138037db3f /src/backends/cl/workloads/ClDepthwiseConvolutionWorkload.cpp | |
parent | 72f02b890912a3323a1d477648bac013e403cfb3 (diff) | |
download | armnn-branches/nhwc-preview.tar.gz |
MLCE-65: Update layout issue on Depthwisebranches/nhwc-preview
Change-Id: I2ab6659d4e0e879f70be39b02a263067c1ac1467
Diffstat (limited to 'src/backends/cl/workloads/ClDepthwiseConvolutionWorkload.cpp')
-rw-r--r-- | src/backends/cl/workloads/ClDepthwiseConvolutionWorkload.cpp | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/src/backends/cl/workloads/ClDepthwiseConvolutionWorkload.cpp b/src/backends/cl/workloads/ClDepthwiseConvolutionWorkload.cpp index 83aa03d727..3b8bc020cb 100644 --- a/src/backends/cl/workloads/ClDepthwiseConvolutionWorkload.cpp +++ b/src/backends/cl/workloads/ClDepthwiseConvolutionWorkload.cpp @@ -88,8 +88,8 @@ ClDepthwiseConvolutionWorkload::ClDepthwiseConvolutionWorkload( const unsigned int depthMultiplier = weightInfo.GetShape()[0]; - const unsigned int widthIndex = (m_Data.m_Parameters.m_DataLayout == DataLayout::NHWC) ? 3 : 2; - const unsigned int heightIndex = (m_Data.m_Parameters.m_DataLayout == DataLayout::NHWC) ? 2 : 1; + const unsigned int widthIndex = (m_Data.m_Parameters.m_DataLayout == DataLayout::NCHW) ? 3 : 2; + const unsigned int heightIndex = (m_Data.m_Parameters.m_DataLayout == DataLayout::NCHW) ? 2 : 1; //Check for optimisation opportunities. bool use3x3Optimisation = (weightInfo.GetShape()[widthIndex] == 3) && (weightInfo.GetShape()[heightIndex] == 3); |