diff options
| author | Conor Kennedy <conor.kennedy@arm.com> | 2019-01-09 07:57:38 +0000 |
|---|---|---|
| committer | Matteo Martincigh <matteo.martincigh@arm.com> | 2019-01-09 11:50:23 +0000 |
| commit | 54b2169989fa9916565fb8a2edf717c6c93923d9 (patch) | |
| tree | 81c6940cd33ac268a8ff6ed29532fa102d7c9fc4 /src/backends/neon/NeonLayerSupport.cpp | |
| parent | d80a7126b0abdd532a9f731559827a23f2e565e0 (diff) | |
| download | armnn-54b2169989fa9916565fb8a2edf717c6c93923d9.tar.gz | |
IVGCVSW-2441 Add Minimum operator support (NEON)
Change-Id: I0bc48f118aba419abb659657b83224401fac0715
Diffstat (limited to 'src/backends/neon/NeonLayerSupport.cpp')
| -rw-r--r-- | src/backends/neon/NeonLayerSupport.cpp | 13 |
1 files changed, 13 insertions, 0 deletions
diff --git a/src/backends/neon/NeonLayerSupport.cpp b/src/backends/neon/NeonLayerSupport.cpp index 3846801ac1..cc8f973510 100644 --- a/src/backends/neon/NeonLayerSupport.cpp +++ b/src/backends/neon/NeonLayerSupport.cpp @@ -26,6 +26,7 @@ #include "workloads/NeonMaximumWorkload.hpp" #include "workloads/NeonMeanWorkload.hpp" #include "workloads/NeonMergerWorkload.hpp" +#include "workloads/NeonMinimumWorkload.hpp" #include "workloads/NeonMultiplicationFloatWorkload.hpp" #include "workloads/NeonNormalizationFloatWorkload.hpp" #include "workloads/NeonFullyConnectedWorkload.hpp" @@ -292,6 +293,18 @@ bool NeonLayerSupport::IsMergerSupported(const std::vector<const TensorInfo*> in } } +bool NeonLayerSupport::IsMinimumSupported(const TensorInfo& input0, + const TensorInfo& input1, + const TensorInfo& output, + Optional<std::string&> reasonIfUnsupported) const +{ + FORWARD_WORKLOAD_VALIDATE_FUNC(NeonMinimumWorkloadValidate, + reasonIfUnsupported, + input0, + input1, + output); +} + bool NeonLayerSupport::IsMultiplicationSupported(const TensorInfo& input0, const TensorInfo& input1, const TensorInfo& output, |