diff options
Diffstat (limited to 'src/armnn/layers/LstmLayer.cpp')
| -rw-r--r-- | src/armnn/layers/LstmLayer.cpp | 4 |
1 files changed, 3 insertions, 1 deletions
diff --git a/src/armnn/layers/LstmLayer.cpp b/src/armnn/layers/LstmLayer.cpp index 653b18a1c9..af708e4e06 100644 --- a/src/armnn/layers/LstmLayer.cpp +++ b/src/armnn/layers/LstmLayer.cpp @@ -163,8 +163,10 @@ std::vector<TensorShape> LstmLayer::InferOutputShapes(const std::vector<TensorSh return outShapes; } -void LstmLayer::ValidateTensorShapesFromInputs() +void LstmLayer::ValidateTensorShapesFromInputs(ShapeInferenceMethod shapeInferenceMethod) { + IgnoreUnused(shapeInferenceMethod); + VerifyLayerConnections(3, CHECK_LOCATION()); auto inferredShapes = InferOutputShapes( { |