IVGCVSW-3457 Fix VTS pad dynamic_output_shape test failures
Signed-off-by: Sadik Armagan <sadik.armagan@arm.com>
Change-Id: Icb9a308f19ac6adbcf0d1de4d8733b95bab3bf0a
diff --git a/OutputShapeUtils.hpp b/OutputShapeUtils.hpp
index 6e2a76d..d1818db 100644
--- a/OutputShapeUtils.hpp
+++ b/OutputShapeUtils.hpp
@@ -12,6 +12,9 @@
bool IsDynamicOutput(const armnn::TensorInfo& outputInfo);
+armnn::TensorShape InferPadOutputShape(const armnn::TensorShape& inputShape,
+ const std::vector<std::pair<unsigned int, unsigned int>>& padList);
+
armnn::TensorShape InferPreluOutputShape(const armnn::TensorShape& inputShape, const armnn::TensorShape& alphaShape);
} // namespace armnn_driver