diff options
Diffstat (limited to 'src/armnn/layers/QuantizeLayer.hpp')
| -rw-r--r-- | src/armnn/layers/QuantizeLayer.hpp | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/src/armnn/layers/QuantizeLayer.hpp b/src/armnn/layers/QuantizeLayer.hpp index 32cd53f810..a223f59470 100644 --- a/src/armnn/layers/QuantizeLayer.hpp +++ b/src/armnn/layers/QuantizeLayer.hpp @@ -21,7 +21,8 @@ public: Layer* Clone(Graph& graph) const override; - void ValidateTensorShapesFromInputs() override; + void ValidateTensorShapesFromInputs( + ShapeInferenceMethod shapeInferenceMethod = ShapeInferenceMethod::ValidateOnly) override; void Accept(ILayerVisitor& visitor) const override; |