aboutsummaryrefslogtreecommitdiff
path: root/src/armnn/layers/MergerLayer.hpp
diff options
context:
space:
mode:
Diffstat (limited to 'src/armnn/layers/MergerLayer.hpp')
-rw-r--r--src/armnn/layers/MergerLayer.hpp1
1 files changed, 1 insertions, 0 deletions
diff --git a/src/armnn/layers/MergerLayer.hpp b/src/armnn/layers/MergerLayer.hpp
index ad94cb5f3a..b6261027d4 100644
--- a/src/armnn/layers/MergerLayer.hpp
+++ b/src/armnn/layers/MergerLayer.hpp
@@ -19,6 +19,7 @@ public:
MergerLayer* Clone(Graph& graph) const override;
void ValidateTensorShapesFromInputs() override;
+ std::vector<TensorShape> InferOutputShapes(const std::vector<TensorShape>& inputShapes) const override;
protected:
MergerLayer(const OriginsDescriptor& param, const char* name);