diff options
Diffstat (limited to 'src/backends/backendsCommon/test/Pooling2dTestImpl.hpp')
-rw-r--r-- | src/backends/backendsCommon/test/Pooling2dTestImpl.hpp | 4 |
1 files changed, 3 insertions, 1 deletions
diff --git a/src/backends/backendsCommon/test/Pooling2dTestImpl.hpp b/src/backends/backendsCommon/test/Pooling2dTestImpl.hpp index 0f33ac01a5..f5e3386fed 100644 --- a/src/backends/backendsCommon/test/Pooling2dTestImpl.hpp +++ b/src/backends/backendsCommon/test/Pooling2dTestImpl.hpp @@ -20,6 +20,8 @@ #include <test/TensorHelpers.hpp> +#include <DataLayoutIndexed.hpp> + #include <boost/numeric/conversion/cast.hpp> #include <algorithm> @@ -36,7 +38,7 @@ LayerTestResult<T, 4> SimplePooling2dTestImpl( const boost::multi_array<T, 4>& outputExpected) { const armnn::DataLayout dataLayout = descriptor.m_DataLayout; - const armnn::DataLayoutIndexed dimensionIndices = dataLayout; + const armnnUtils::DataLayoutIndexed dimensionIndices = dataLayout; auto heightIndex = dimensionIndices.GetHeightIndex(); auto widthIndex = dimensionIndices.GetWidthIndex(); auto channelsIndex = dimensionIndices.GetChannelsIndex(); |