diff options
| -rw-r--r-- | src/armnn/layers/MeanLayer.cpp | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/src/armnn/layers/MeanLayer.cpp b/src/armnn/layers/MeanLayer.cpp index 62a3923750..8e6d9e00af 100644 --- a/src/armnn/layers/MeanLayer.cpp +++ b/src/armnn/layers/MeanLayer.cpp @@ -77,7 +77,8 @@ std::vector<TensorShape> MeanLayer::InferOutputShapes(const std::vector<TensorSh 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."); } |