diff options
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); |