diff options
author | Aron Virginas-Tar <Aron.Virginas-Tar@arm.com> | 2019-07-15 14:29:09 +0100 |
---|---|---|
committer | Áron Virginás-Tar <aron.virginas-tar@arm.com> | 2019-07-16 13:31:26 +0000 |
commit | 2b173126319343e49d1f081cfb58eacd96afc715 (patch) | |
tree | b51eaf9d648cb93753c6adc4a075dcb6aea3a68e /OutputShapeUtils.hpp | |
parent | d759323d159a50298af937dfb2c519025efe3900 (diff) | |
download | android-nn-driver-2b173126319343e49d1f081cfb58eacd96afc715.tar.gz |
IVGCVSW-3452 Support dynamic output shape in hal_1_2::HalPolicy::ConvertConv2d
Signed-off-by: Aron Virginas-Tar <Aron.Virginas-Tar@arm.com>
Change-Id: I8694e1f1c62da6f74eb356558b17a63758ccfdad
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 dac4a193..58686954 100644 --- a/OutputShapeUtils.hpp +++ b/OutputShapeUtils.hpp @@ -12,6 +12,10 @@ namespace armnn_driver bool IsDynamicOutput(const armnn::TensorInfo& outputInfo); +armnn::TensorShape InferConvolution2dOutputShape(const armnn::TensorShape& inputShape, + const armnn::TensorShape& kernelShape, + const armnn::Convolution2dDescriptor& descriptor); + armnn::TensorShape InferMaximumOutputShape(const armnn::TensorShape& input0Shape, const armnn::TensorShape& input1Shape); |