aboutsummaryrefslogtreecommitdiff
path: root/tests/validation/reference/ElementWiseUnary.cpp
diff options
context:
space:
mode:
Diffstat (limited to 'tests/validation/reference/ElementWiseUnary.cpp')
-rw-r--r--tests/validation/reference/ElementWiseUnary.cpp3
1 files changed, 3 insertions, 0 deletions
diff --git a/tests/validation/reference/ElementWiseUnary.cpp b/tests/validation/reference/ElementWiseUnary.cpp
index dfd4c0600d..06beb2ac09 100644
--- a/tests/validation/reference/ElementWiseUnary.cpp
+++ b/tests/validation/reference/ElementWiseUnary.cpp
@@ -58,6 +58,9 @@ SimpleTensor<T> elementwise_unary(const SimpleTensor<T> &src, ElementWiseUnary o
case ElementWiseUnary::SIN:
dst[i] = std::sin(src[i]);
break;
+ case ElementWiseUnary::ROUND:
+ dst[i] = std::nearbyint(src[i]);
+ break;
default:
ARM_COMPUTE_ERROR("Not implemented");
}