diff options
Diffstat (limited to 'src/armnn/layers')
-rw-r--r-- | src/armnn/layers/Convolution2dLayer.cpp | 3 |
1 files changed, 3 insertions, 0 deletions
diff --git a/src/armnn/layers/Convolution2dLayer.cpp b/src/armnn/layers/Convolution2dLayer.cpp index 71f54b88f8..07d6d7eee4 100644 --- a/src/armnn/layers/Convolution2dLayer.cpp +++ b/src/armnn/layers/Convolution2dLayer.cpp @@ -26,6 +26,9 @@ std::unique_ptr<IWorkload> Convolution2dLayer::CreateWorkload(const Graph& graph Convolution2dQueueDescriptor descriptor; descriptor.m_Weight = m_Weight.get(); + + descriptor.m_DataLayout = GetParameters().m_DataLayout; + if (m_Param.m_BiasEnabled) { BOOST_ASSERT_MSG(m_Bias != nullptr, "Convolution2dLayer: Bias data should not be null."); |