diff options
Diffstat (limited to 'OutputShapeUtils.cpp')
| -rw-r--r-- | OutputShapeUtils.cpp | 6 |
1 files changed, 6 insertions, 0 deletions
diff --git a/OutputShapeUtils.cpp b/OutputShapeUtils.cpp index 739038ac..b6cdb316 100644 --- a/OutputShapeUtils.cpp +++ b/OutputShapeUtils.cpp @@ -54,6 +54,12 @@ bool IsDynamicOutput(const TensorInfo& outputInfo) return outputInfo.GetNumElements() == 0u; } +TensorShape InferMaximumOutputShape(const armnn::TensorShape& input0Shape, + const armnn::TensorShape& input1Shape) +{ + return CalculateMaxShape(input0Shape, input1Shape); +} + TensorShape InferPadOutputShape(const TensorShape& inputShape, const std::vector<std::pair<unsigned int, unsigned int>>& padList) { |