diff options
| author | Aron Virginas-Tar <Aron.Virginas-Tar@arm.com> | 2019-07-15 18:04:32 +0100 |
|---|---|---|
| committer | Áron Virginás-Tar <aron.virginas-tar@arm.com> | 2019-07-16 13:31:37 +0000 |
| commit | 9fd373954d64fbae72d1726bbdfc57a18a3a2f6d (patch) | |
| tree | 18aec5d50a59ff103bae3a2c1dd5b475ad7694e3 /OutputShapeUtils.hpp | |
| parent | 2b173126319343e49d1f081cfb58eacd96afc715 (diff) | |
| download | android-nn-driver-9fd373954d64fbae72d1726bbdfc57a18a3a2f6d.tar.gz | |
IVGCVSW-3455 Support dynamic output shape in hal_1_2::HalPolicy::ConvertDepthwiseConv2d
Signed-off-by: Aron Virginas-Tar <Aron.Virginas-Tar@arm.com>
Change-Id: Iba64a674d772a76ca071553cb423ed870fae9bfd
Diffstat (limited to 'OutputShapeUtils.hpp')
| -rw-r--r-- | OutputShapeUtils.hpp | 4 |
1 files changed, 4 insertions, 0 deletions
diff --git a/OutputShapeUtils.hpp b/OutputShapeUtils.hpp index 58686954..bcb43475 100644 --- a/OutputShapeUtils.hpp +++ b/OutputShapeUtils.hpp @@ -16,6 +16,10 @@ armnn::TensorShape InferConvolution2dOutputShape(const armnn::TensorShape& input const armnn::TensorShape& kernelShape, const armnn::Convolution2dDescriptor& descriptor); +armnn::TensorShape InferDepthwiseConvolution2dOutputShape(const armnn::TensorShape& inputShape, + const armnn::TensorShape& kernelShape, + const armnn::DepthwiseConvolution2dDescriptor& descriptor); + armnn::TensorShape InferMaximumOutputShape(const armnn::TensorShape& input0Shape, const armnn::TensorShape& input1Shape); |