aboutsummaryrefslogtreecommitdiff
path: root/src/armnn/layers/ReshapeLayer.cpp
diff options
context:
space:
mode:
Diffstat (limited to 'src/armnn/layers/ReshapeLayer.cpp')
-rw-r--r--src/armnn/layers/ReshapeLayer.cpp4
1 files changed, 3 insertions, 1 deletions
diff --git a/src/armnn/layers/ReshapeLayer.cpp b/src/armnn/layers/ReshapeLayer.cpp
index b496dbb642..0257ca9b15 100644
--- a/src/armnn/layers/ReshapeLayer.cpp
+++ b/src/armnn/layers/ReshapeLayer.cpp
@@ -36,8 +36,10 @@ std::vector<TensorShape> ReshapeLayer::InferOutputShapes(const std::vector<Tenso
return std::vector<TensorShape>({ m_Param.m_TargetShape });
}
-void ReshapeLayer::ValidateTensorShapesFromInputs()
+void ReshapeLayer::ValidateTensorShapesFromInputs(ShapeInferenceMethod shapeInferenceMethod)
{
+ IgnoreUnused(shapeInferenceMethod);
+
VerifyLayerConnections(1, CHECK_LOCATION());
auto inferredShapes = InferOutputShapes({ });