From 9fd373954d64fbae72d1726bbdfc57a18a3a2f6d Mon Sep 17 00:00:00 2001 From: Aron Virginas-Tar Date: Mon, 15 Jul 2019 18:04:32 +0100 Subject: IVGCVSW-3455 Support dynamic output shape in hal_1_2::HalPolicy::ConvertDepthwiseConv2d Signed-off-by: Aron Virginas-Tar Change-Id: Iba64a674d772a76ca071553cb423ed870fae9bfd --- OutputShapeUtils.hpp | 4 ++++ 1 file changed, 4 insertions(+) (limited to 'OutputShapeUtils.hpp') 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); -- cgit v1.2.1