diff options
| -rw-r--r-- | src/armnn/layers/MeanLayer.cpp | 2 | ||||
| -rw-r--r-- | src/armnn/layers/ReduceLayer.cpp | 4 |
2 files changed, 3 insertions, 3 deletions
diff --git a/src/armnn/layers/MeanLayer.cpp b/src/armnn/layers/MeanLayer.cpp index 8e6d9e00af..6a041c2e53 100644 --- a/src/armnn/layers/MeanLayer.cpp +++ b/src/armnn/layers/MeanLayer.cpp @@ -78,7 +78,7 @@ std::vector<TensorShape> MeanLayer::InferOutputShapes(const std::vector<TensorSh const TensorShape& input = inputShapes[0]; auto inputDims = input.GetNumDimensions(); - if (inputDims != std::clamp(inputDims, 1u, 4u)) + if (inputDims < 1 || inputDims > 4) { throw armnn::Exception("ReduceLayer: Reduce supports up to 4D input."); } diff --git a/src/armnn/layers/ReduceLayer.cpp b/src/armnn/layers/ReduceLayer.cpp index 9fbbc39bd1..d514aa6a6d 100644 --- a/src/armnn/layers/ReduceLayer.cpp +++ b/src/armnn/layers/ReduceLayer.cpp @@ -52,7 +52,7 @@ void ReduceLayer::ValidateTensorShapesFromInputs() const TensorInfo& input = GetInputSlot(0).GetTensorInfo(); auto inputDims = input.GetNumDimensions(); - if (inputDims != std::clamp(inputDims, 1u, 4u)) + if (inputDims < 1 || inputDims > 4) { throw armnn::LayerValidationException("ReduceLayer: Reduce supports up to 4D input."); } @@ -73,7 +73,7 @@ std::vector<TensorShape> ReduceLayer::InferOutputShapes(const std::vector<Tensor const TensorShape& input = inputShapes[0]; auto inputDims = input.GetNumDimensions(); - if (inputDims != std::clamp(inputDims, 1u, 4u)) + if (inputDims < 1 || inputDims > 4) { throw armnn::Exception("ReduceLayer: Reduce supports up to 4D input."); } |