diff options
author | Narumol Prangnawarat <narumol.prangnawarat@arm.com> | 2019-07-15 12:02:20 +0100 |
---|---|---|
committer | Narumol Prangnawarat <narumol.prangnawarat@arm.com> | 2019-07-15 16:40:01 +0000 |
commit | 95b1ef62568cc71b4bcdde6af02b26687d020a1e (patch) | |
tree | dcba796eb09f9fc75cb53b93c0bd5d601ee2f84c /OutputShapeUtils.cpp | |
parent | 2b6e275e354e9b28a12573aad3bdd4f2a80d52bf (diff) | |
download | android-nn-driver-95b1ef62568cc71b4bcdde6af02b26687d020a1e.tar.gz |
IVGCVSW-3306 Add Maximum support in HAL 1.2 driver
Signed-off-by: Narumol Prangnawarat <narumol.prangnawarat@arm.com>
Change-Id: I94e410058ffc1ce39465498c12e10cb5669e7ed3
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) { |