aboutsummaryrefslogtreecommitdiff
path: root/OutputShapeUtils.hpp
diff options
context:
space:
mode:
authorAron Virginas-Tar <Aron.Virginas-Tar@arm.com>2019-07-10 13:01:41 +0100
committerAron Virginas-Tar <Aron.Virginas-Tar@arm.com>2019-07-10 13:01:41 +0100
commit366e0a66f4566cf71dff3f850556350709ee66a8 (patch)
treef85badd07900acca485b6b66c8fd4b9b59f777fd /OutputShapeUtils.hpp
parentf03fcf0dd180ba2c87648a524fcca9214e1f979b (diff)
downloadandroid-nn-driver-366e0a66f4566cf71dff3f850556350709ee66a8.tar.gz
IVGCVSW-3482 Report operations with dynamic output size as unsupported
Signed-off-by: Aron Virginas-Tar <Aron.Virginas-Tar@arm.com> Change-Id: Ifafe2a6fbfd6019b3395d51ed9967db794d2b034
Diffstat (limited to 'OutputShapeUtils.hpp')
-rw-r--r--OutputShapeUtils.hpp2
1 files changed, 2 insertions, 0 deletions
diff --git a/OutputShapeUtils.hpp b/OutputShapeUtils.hpp
index f314252f..6e2a76db 100644
--- a/OutputShapeUtils.hpp
+++ b/OutputShapeUtils.hpp
@@ -10,6 +10,8 @@
namespace armnn_driver
{
+bool IsDynamicOutput(const armnn::TensorInfo& outputInfo);
+
armnn::TensorShape InferPreluOutputShape(const armnn::TensorShape& inputShape, const armnn::TensorShape& alphaShape);
} // namespace armnn_driver