diff options
author | Usama Arif <usama.arif@arm.com> | 2019-05-23 14:20:33 +0100 |
---|---|---|
committer | Usama Arif <usama.arif@arm.com> | 2019-05-24 16:11:16 +0000 |
commit | 0a5a57a3f794de851408bae1c63b1660b4c5cbe7 (patch) | |
tree | 5d49d85165a3d9fd257c5676691fc5ae24fda4f3 /tests/validation/fixtures/ElementWiseUnaryFixture.h | |
parent | 43c1bf8e47098d6ab07be5b5e685bda6d8b80344 (diff) | |
download | ComputeLibrary-0a5a57a3f794de851408bae1c63b1660b4c5cbe7.tar.gz |
COMPMID-2160: Implement Round for NEON
Change-Id: Ie80e2ad294eaf95bc823d979842c320e8fb41f67
Signed-off-by: Usama Arif <usama.arif@arm.com>
Reviewed-on: https://review.mlplatform.org/c/1215
Comments-Addressed: Arm Jenkins <bsgcomp@arm.com>
Reviewed-by: Georgios Pinitas <georgios.pinitas@arm.com>
Tested-by: Arm Jenkins <bsgcomp@arm.com>
Diffstat (limited to 'tests/validation/fixtures/ElementWiseUnaryFixture.h')
-rw-r--r-- | tests/validation/fixtures/ElementWiseUnaryFixture.h | 17 |
1 files changed, 17 insertions, 0 deletions
diff --git a/tests/validation/fixtures/ElementWiseUnaryFixture.h b/tests/validation/fixtures/ElementWiseUnaryFixture.h index 7837b085fa..fd66f630ba 100644 --- a/tests/validation/fixtures/ElementWiseUnaryFixture.h +++ b/tests/validation/fixtures/ElementWiseUnaryFixture.h @@ -104,6 +104,12 @@ protected: library->fill(tensor, distribution, i); break; } + case ElementWiseUnary::ROUND: + { + std::uniform_real_distribution<> distribution(100.0f, -100.0f); + library->fill(tensor, distribution, i); + break; + } default: ARM_COMPUTE_ERROR("Not implemented"); } @@ -219,6 +225,17 @@ public: ElementWiseUnaryValidationFixture<TensorType, AccessorType, FunctionType, T>::setup(shape, data_type, ElementWiseUnary::SIN); } }; + +template <typename TensorType, typename AccessorType, typename FunctionType, typename T> +class RoundValidationFixture : public ElementWiseUnaryValidationFixture<TensorType, AccessorType, FunctionType, T> +{ +public: + template <typename...> + void setup(const TensorShape &shape, DataType data_type) + { + ElementWiseUnaryValidationFixture<TensorType, AccessorType, FunctionType, T>::setup(shape, data_type, ElementWiseUnary::ROUND); + } +}; } // namespace validation } // namespace test } // namespace arm_compute |