diff options
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 2a832618..7452ced9 100644 --- a/OutputShapeUtils.hpp +++ b/OutputShapeUtils.hpp @@ -23,6 +23,9 @@ armnn::TensorShape InferDepthwiseConvolution2dOutputShape(const armnn::TensorSha armnn::TensorShape InferMaximumOutputShape(const armnn::TensorShape& input0Shape, const armnn::TensorShape& input1Shape); +armnn::TensorShape InferMinimumOutputShape(const armnn::TensorShape& input0Shape, + const armnn::TensorShape& input1Shape); + armnn::TensorShape InferPadOutputShape(const armnn::TensorShape& inputShape, const std::vector<std::pair<unsigned int, unsigned int>>& padList); |