diff options
Diffstat (limited to 'src/armnn/layers/ResizeBilinearLayer.hpp')
-rw-r--r-- | src/armnn/layers/ResizeBilinearLayer.hpp | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/src/armnn/layers/ResizeBilinearLayer.hpp b/src/armnn/layers/ResizeBilinearLayer.hpp index 2cefedb0b8..e6798ce531 100644 --- a/src/armnn/layers/ResizeBilinearLayer.hpp +++ b/src/armnn/layers/ResizeBilinearLayer.hpp @@ -18,6 +18,7 @@ public: ResizeBilinearLayer* Clone(Graph& graph) const override; void ValidateTensorShapesFromInputs() override; + std::vector<TensorShape> InferOutputShapes(const std::vector<TensorShape>& inputShapes) const override; protected: ResizeBilinearLayer(const ResizeBilinearDescriptor& param, const char* name); |