diff options
author | Aron Virginas-Tar <Aron.Virginas-Tar@arm.com> | 2019-07-25 11:24:42 +0100 |
---|---|---|
committer | Aron Virginas-Tar <Aron.Virginas-Tar@arm.com> | 2019-07-25 11:31:48 +0100 |
commit | ad1ab53f2898862e82f9b354853764fdcd1df97d (patch) | |
tree | 9df7f2c13686473a84d85988bcc21cc938002ad3 /OutputShapeUtils.hpp | |
parent | 6111316eb609bd71589b963cf6fc56b18ba3d241 (diff) | |
download | android-nn-driver-ad1ab53f2898862e82f9b354853764fdcd1df97d.tar.gz |
IVGCVSW-3569 Fix conversion of HAL1.2 SpaceToDepth
Signed-off-by: Aron Virginas-Tar <Aron.Virginas-Tar@arm.com>
Change-Id: I82e38c8a9e44e773c099e347f8ce0070bb5f8662
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 222c1235..85cafbf6 100644 --- a/OutputShapeUtils.hpp +++ b/OutputShapeUtils.hpp @@ -32,6 +32,9 @@ armnn::TensorShape InferPreluOutputShape(const armnn::TensorShape& inputShape, c armnn::TensorShape InferResizeOutputShape(const armnn::TensorShape& inputShape, const armnn::ResizeDescriptor& descriptor); +armnn::TensorShape InferSpaceToDepthOutputShape(const armnn::TensorShape& inputShape, + const armnn::SpaceToDepthDescriptor& descriptor); + armnn::TensorShape InferSubOutputShape(const armnn::TensorShape& input0Shape, const armnn::TensorShape& input1Shape); } // namespace armnn_driver |