diff options
author | Aron Virginas-Tar <Aron.Virginas-Tar@arm.com> | 2019-07-10 13:01:41 +0100 |
---|---|---|
committer | Aron Virginas-Tar <Aron.Virginas-Tar@arm.com> | 2019-07-10 13:01:41 +0100 |
commit | 366e0a66f4566cf71dff3f850556350709ee66a8 (patch) | |
tree | f85badd07900acca485b6b66c8fd4b9b59f777fd /OutputShapeUtils.cpp | |
parent | f03fcf0dd180ba2c87648a524fcca9214e1f979b (diff) | |
download | android-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.cpp')
-rw-r--r-- | OutputShapeUtils.cpp | 5 |
1 files changed, 5 insertions, 0 deletions
diff --git a/OutputShapeUtils.cpp b/OutputShapeUtils.cpp index de27630e..a0c624c8 100644 --- a/OutputShapeUtils.cpp +++ b/OutputShapeUtils.cpp @@ -12,6 +12,11 @@ namespace armnn_driver using namespace armnn; +bool IsDynamicOutput(const TensorInfo& outputInfo) +{ + return outputInfo.GetNumElements() == 0u; +} + TensorShape InferPreluOutputShape(const TensorShape& inputShape, const TensorShape& alphaShape) { // NOTE: The inferred PReLU output size will be the maximum size along each dimension |