diff options
Diffstat (limited to 'tests/validation/reference')
-rw-r--r-- | tests/validation/reference/ElementWiseUnary.cpp | 3 |
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"); } |