diff options
Diffstat (limited to 'src/backends/reference/workloads/ElementwiseFunction.cpp')
| -rw-r--r-- | src/backends/reference/workloads/ElementwiseFunction.cpp | 7 |
1 files changed, 6 insertions, 1 deletions
diff --git a/src/backends/reference/workloads/ElementwiseFunction.cpp b/src/backends/reference/workloads/ElementwiseFunction.cpp index 7a5c071f70..888037f9a6 100644 --- a/src/backends/reference/workloads/ElementwiseFunction.cpp +++ b/src/backends/reference/workloads/ElementwiseFunction.cpp @@ -32,6 +32,11 @@ template struct armnn::ElementwiseFunction<std::multiplies<float>>; template struct armnn::ElementwiseFunction<std::divides<float>>; template struct armnn::ElementwiseFunction<armnn::maximum<float>>; template struct armnn::ElementwiseFunction<armnn::minimum<float>>; + +// Comparison template struct armnn::ElementwiseFunction<std::equal_to<float>>; template struct armnn::ElementwiseFunction<std::greater<float>>; - +template struct armnn::ElementwiseFunction<std::greater_equal<float>>; +template struct armnn::ElementwiseFunction<std::less<float>>; +template struct armnn::ElementwiseFunction<std::less_equal<float>>; +template struct armnn::ElementwiseFunction<std::not_equal_to<float>>; |