diff options
author | Sadik Armagan <sadik.armagan@arm.com> | 2019-07-12 13:55:57 +0100 |
---|---|---|
committer | Sadik Armagan <sadik.armagan@arm.com> | 2019-07-12 13:55:57 +0100 |
commit | 5e9521ce5eac717b25844f9da4849b1fc2082f76 (patch) | |
tree | 05f5d7bc6656033784996c7ee08501789c9a5616 /OutputShapeUtils.hpp | |
parent | 6bda94a88536e8abb487a6ab29da4b8cfc8034e1 (diff) | |
download | android-nn-driver-5e9521ce5eac717b25844f9da4849b1fc2082f76.tar.gz |
IVGCVSW-3458 Fix VTS sub dynamic_output_shape test failures
Signed-off-by: Sadik Armagan <sadik.armagan@arm.com>
Change-Id: Ia6e2a082ed4a0255d8c9c10375590553e6ede9f0
Diffstat (limited to 'OutputShapeUtils.hpp')
-rw-r--r-- | OutputShapeUtils.hpp | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/OutputShapeUtils.hpp b/OutputShapeUtils.hpp index d1818db1..be255c28 100644 --- a/OutputShapeUtils.hpp +++ b/OutputShapeUtils.hpp @@ -17,6 +17,8 @@ armnn::TensorShape InferPadOutputShape(const armnn::TensorShape& inputShape, armnn::TensorShape InferPreluOutputShape(const armnn::TensorShape& inputShape, const armnn::TensorShape& alphaShape); +armnn::TensorShape InferSubOutputShape(const armnn::TensorShape& input0Shape, const armnn::TensorShape& input1Shape); + } // namespace armnn_driver |