aboutsummaryrefslogtreecommitdiff
path: root/src/armnn/layers/ReduceLayer.cpp
diff options
context:
space:
mode:
Diffstat (limited to 'src/armnn/layers/ReduceLayer.cpp')
-rw-r--r--src/armnn/layers/ReduceLayer.cpp6
1 files changed, 4 insertions, 2 deletions
diff --git a/src/armnn/layers/ReduceLayer.cpp b/src/armnn/layers/ReduceLayer.cpp
index bebd043e9b..9fbbc39bd1 100644
--- a/src/armnn/layers/ReduceLayer.cpp
+++ b/src/armnn/layers/ReduceLayer.cpp
@@ -51,7 +51,8 @@ void ReduceLayer::ValidateTensorShapesFromInputs()
const TensorInfo& input = GetInputSlot(0).GetTensorInfo();
- if (auto inputDims = input.GetNumDimensions(); inputDims != std::clamp(inputDims, 1u, 4u))
+ auto inputDims = input.GetNumDimensions();
+ if (inputDims != std::clamp(inputDims, 1u, 4u))
{
throw armnn::LayerValidationException("ReduceLayer: Reduce supports up to 4D input.");
}
@@ -71,7 +72,8 @@ std::vector<TensorShape> ReduceLayer::InferOutputShapes(const std::vector<Tensor
const TensorShape& input = inputShapes[0];
- if (auto inputDims = input.GetNumDimensions(); inputDims != std::clamp(inputDims, 1u, 4u))
+ auto inputDims = input.GetNumDimensions();
+ if (inputDims != std::clamp(inputDims, 1u, 4u))
{
throw armnn::Exception("ReduceLayer: Reduce supports up to 4D input.");
}