1 files changed, 4 insertions, 0 deletions
diff --git a/OutputShapeUtils.hpp b/OutputShapeUtils.hpp
index dac4a193..58686954 100644
--- a/OutputShapeUtils.hpp
+++ b/OutputShapeUtils.hpp
@@ -12,6 +12,10 @@ namespace armnn_driver
 
 bool IsDynamicOutput(const armnn::TensorInfo& outputInfo);
 
+armnn::TensorShape InferConvolution2dOutputShape(const armnn::TensorShape& inputShape,
+                                                 const armnn::TensorShape& kernelShape,
+                                                 const armnn::Convolution2dDescriptor& descriptor);
+
 armnn::TensorShape InferMaximumOutputShape(const armnn::TensorShape& input0Shape,
                                            const armnn::TensorShape& input1Shape);