diff options
Diffstat (limited to 'src/armnn/layers/BatchToSpaceNdLayer.cpp')
| -rw-r--r-- | src/armnn/layers/BatchToSpaceNdLayer.cpp | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/src/armnn/layers/BatchToSpaceNdLayer.cpp b/src/armnn/layers/BatchToSpaceNdLayer.cpp index aff818e664..e1b78b21df 100644 --- a/src/armnn/layers/BatchToSpaceNdLayer.cpp +++ b/src/armnn/layers/BatchToSpaceNdLayer.cpp @@ -10,6 +10,7 @@ #include <armnn/TypesUtils.hpp> #include <backendsCommon/CpuTensorHandle.hpp> +#include <backendsCommon/DataLayoutIndexed.hpp> #include <backendsCommon/WorkloadData.hpp> #include <backendsCommon/WorkloadFactory.hpp> @@ -50,7 +51,7 @@ void BatchToSpaceNdLayer::ValidateTensorShapesFromInputs() std::vector<TensorShape> BatchToSpaceNdLayer::InferOutputShapes(const std::vector<TensorShape>& inputShapes) const { - const DataLayoutIndexed & dataLayout = m_Param.m_DataLayout; + const DataLayoutIndexed dataLayout = m_Param.m_DataLayout; const TensorShape& inputShape = inputShapes[0]; unsigned int inBatchSize = inputShape[0]; unsigned int channelSize = inputShape[dataLayout.GetChannelsIndex()]; |