aboutsummaryrefslogtreecommitdiff
path: root/src/backends/reference/RefLayerSupport.cpp
diff options
context:
space:
mode:
authorsaoste01 <saoirse.stewart@arm.com>2018-11-28 16:57:20 +0000
committerLes Bell <les.bell@arm.com>2018-11-29 10:28:22 +0000
commit2df12b3abf2171869582581dd275d556bad87411 (patch)
tree6c6345b45bd04dd7d4c29d889bf1cfe9428a1c0c /src/backends/reference/RefLayerSupport.cpp
parent90539697433b507e20cf0d56f6c7f18614e03973 (diff)
downloadarmnn-2df12b3abf2171869582581dd275d556bad87411.tar.gz
IVGCVSW-2254 Add Reference workload for Maximum
Change-Id: Id7302c6b1df995ebe6eb8eb94bab38bee1b31b0b
Diffstat (limited to 'src/backends/reference/RefLayerSupport.cpp')
-rw-r--r--src/backends/reference/RefLayerSupport.cpp13
1 files changed, 13 insertions, 0 deletions
diff --git a/src/backends/reference/RefLayerSupport.cpp b/src/backends/reference/RefLayerSupport.cpp
index 00e4c5c09c..7222af6402 100644
--- a/src/backends/reference/RefLayerSupport.cpp
+++ b/src/backends/reference/RefLayerSupport.cpp
@@ -308,6 +308,19 @@ bool RefLayerSupport::IsLstmSupported(const TensorInfo& input,
&FalseFuncU8<>);
}
+bool RefLayerSupport::IsMaximumSupported(const TensorInfo& input0,
+ const TensorInfo& input1,
+ const TensorInfo& output,
+ Optional<std::string&> reasonIfUnsupported) const
+{
+ ignore_unused(input1);
+ ignore_unused(output);
+ return IsSupportedForDataTypeRef(reasonIfUnsupported,
+ input0.GetDataType(),
+ &TrueFunc<>,
+ &TrueFunc<>);
+}
+
bool RefLayerSupport::IsMeanSupported(const TensorInfo& input,
const TensorInfo& output,
const MeanDescriptor& descriptor,