diff options
Diffstat (limited to 'src/armnn/layers/PreCompiledLayer.hpp')
| -rw-r--r-- | src/armnn/layers/PreCompiledLayer.hpp | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/src/armnn/layers/PreCompiledLayer.hpp b/src/armnn/layers/PreCompiledLayer.hpp index 1a87f61e5b..a4b1c78f12 100644 --- a/src/armnn/layers/PreCompiledLayer.hpp +++ b/src/armnn/layers/PreCompiledLayer.hpp @@ -29,7 +29,8 @@ public: PreCompiledLayer* Clone(Graph &graph) const override; - void ValidateTensorShapesFromInputs() override; + void ValidateTensorShapesFromInputs( + ShapeInferenceMethod shapeInferenceMethod = ShapeInferenceMethod::ValidateOnly) override; void SetPreCompiledObject(PreCompiledObjectPtr preCompiledObject); |