diff options
Diffstat (limited to 'src/armnn/layers/AdditionLayer.hpp')
| -rw-r--r-- | src/armnn/layers/AdditionLayer.hpp | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/src/armnn/layers/AdditionLayer.hpp b/src/armnn/layers/AdditionLayer.hpp index c48c027763..37f0b5c259 100644 --- a/src/armnn/layers/AdditionLayer.hpp +++ b/src/armnn/layers/AdditionLayer.hpp @@ -19,6 +19,8 @@ public: void ValidateTensorShapesFromInputs() override; + std::vector<TensorShape> InferOutputShapes(const std::vector<TensorShape>& inputShapes) const override; + protected: AdditionLayer(const char* name); ~AdditionLayer() = default; |