IVGCVSW-3306 Add Maximum support in HAL 1.2 driver
Signed-off-by: Narumol Prangnawarat <narumol.prangnawarat@arm.com>
Change-Id: I94e410058ffc1ce39465498c12e10cb5669e7ed3
diff --git a/OutputShapeUtils.cpp b/OutputShapeUtils.cpp
index 739038a..b6cdb31 100644
--- a/OutputShapeUtils.cpp
+++ b/OutputShapeUtils.cpp
@@ -54,6 +54,12 @@
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)
{