From 8800c00770ed14eb48045cfcf033d6b67595a126 Mon Sep 17 00:00:00 2001 From: Matthew Bentham Date: Mon, 19 Nov 2018 13:19:28 +0000 Subject: IVGCVSW-2169 Remove DataLayoutIndexed from public API Change-Id: If8d8087d9d365e467d3ca9bf9c40d7219cb75cfd --- src/armnn/layers/ResizeBilinearLayer.cpp | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) (limited to 'src/armnn/layers/ResizeBilinearLayer.cpp') diff --git a/src/armnn/layers/ResizeBilinearLayer.cpp b/src/armnn/layers/ResizeBilinearLayer.cpp index 69ce69eea5..f72ccfce90 100644 --- a/src/armnn/layers/ResizeBilinearLayer.cpp +++ b/src/armnn/layers/ResizeBilinearLayer.cpp @@ -7,6 +7,7 @@ #include "LayerCloneBase.hpp" #include +#include #include #include @@ -34,10 +35,10 @@ std::vector ResizeBilinearLayer::InferOutputShapes(const std::vecto { BOOST_ASSERT(inputShapes.size() == 1); const TensorShape& inputShape = inputShapes[0]; - + const DataLayoutIndexed dimensionIndices = m_Param.m_DataLayout; unsigned int outWidth = m_Param.m_TargetWidth; unsigned int outHeight = m_Param.m_TargetHeight; - unsigned int outChannels = inputShape[m_Param.m_DataLayout.GetChannelsIndex()]; + unsigned int outChannels = inputShape[dimensionIndices.GetChannelsIndex()]; unsigned int outBatch = inputShape[0]; TensorShape tensorShape = m_Param.m_DataLayout == armnn::DataLayout::NHWC ? -- cgit v1.2.1