diff options
author | Sadik Armagan <sadik.armagan@arm.com> | 2019-07-11 10:53:38 +0100 |
---|---|---|
committer | Sadik Armagan <sadik.armagan@arm.com> | 2019-07-11 10:53:38 +0100 |
commit | 310d8ff46f26b6c656b417ac6dc59ef610aaa513 (patch) | |
tree | 5d7f85ce31e7cd0c4fcb5427addf32775de58733 /OutputShapeUtils.hpp | |
parent | 366e0a66f4566cf71dff3f850556350709ee66a8 (diff) | |
download | android-nn-driver-310d8ff46f26b6c656b417ac6dc59ef610aaa513.tar.gz |
IVGCVSW-3457 Fix VTS pad dynamic_output_shape test failures
Signed-off-by: Sadik Armagan <sadik.armagan@arm.com>
Change-Id: Icb9a308f19ac6adbcf0d1de4d8733b95bab3bf0a
Diffstat (limited to 'OutputShapeUtils.hpp')
-rw-r--r-- | OutputShapeUtils.hpp | 3 |
1 files changed, 3 insertions, 0 deletions
diff --git a/OutputShapeUtils.hpp b/OutputShapeUtils.hpp index 6e2a76db..d1818db1 100644 --- a/OutputShapeUtils.hpp +++ b/OutputShapeUtils.hpp @@ -12,6 +12,9 @@ namespace armnn_driver 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 |