Sadik Armagan | 788e2c6 | 2021-02-10 16:26:44 +0000 | [diff] [blame] | 1 | // |
Colm Donelan | 7bcae3c | 2024-01-22 10:07:14 +0000 | [diff] [blame] | 2 | // Copyright © 2021, 2023-2024 Arm Ltd and Contributors. All rights reserved. |
Sadik Armagan | 788e2c6 | 2021-02-10 16:26:44 +0000 | [diff] [blame] | 3 | // SPDX-License-Identifier: MIT |
| 4 | // |
| 5 | |
| 6 | #include "RoundTestHelper.hpp" |
| 7 | |
Sadik Armagan | 788e2c6 | 2021-02-10 16:26:44 +0000 | [diff] [blame] | 8 | #include <doctest/doctest.h> |
| 9 | |
| 10 | namespace armnnDelegate |
| 11 | { |
| 12 | |
Colm Donelan | 7bcae3c | 2024-01-22 10:07:14 +0000 | [diff] [blame] | 13 | void FloorFp32Test(const std::vector<armnn::BackendId>& backends = {}) |
Sadik Armagan | 788e2c6 | 2021-02-10 16:26:44 +0000 | [diff] [blame] | 14 | { |
| 15 | std::vector<int32_t> inputShape {1, 3, 2, 3}; |
| 16 | std::vector<int32_t> outputShape {1, 3, 2, 3}; |
| 17 | |
| 18 | std::vector<float> inputValues { -37.5f, -15.2f, -8.76f, -2.0f, -1.5f, -1.3f, -0.5f, -0.4f, 0.0f, |
| 19 | 1.0f, 0.4f, 0.5f, 1.3f, 1.5f, 2.0f, 8.76f, 15.2f, 37.5f }; |
| 20 | |
| 21 | std::vector<float> expectedOutputValues { -38.0f, -16.0f, -9.0f, -2.0f, -2.0f, -2.0f, -1.0f, -1.0f, 0.0f, |
| 22 | 1.0f, 0.0f, 0.0f, 1.0f, 1.0f, 2.0f, 8.0f, 15.0f, 37.0f }; |
| 23 | |
| 24 | RoundTest<float>(tflite::BuiltinOperator_FLOOR, |
| 25 | ::tflite::TensorType_FLOAT32, |
Sadik Armagan | 788e2c6 | 2021-02-10 16:26:44 +0000 | [diff] [blame] | 26 | inputShape, |
| 27 | inputValues, |
| 28 | expectedOutputValues); |
| 29 | } |
| 30 | |
| 31 | // FLOOR Test Suite |
Colm Donelan | 7bcae3c | 2024-01-22 10:07:14 +0000 | [diff] [blame] | 32 | TEST_SUITE("FLOORTests") |
Sadik Armagan | 788e2c6 | 2021-02-10 16:26:44 +0000 | [diff] [blame] | 33 | { |
| 34 | |
Colm Donelan | 7bcae3c | 2024-01-22 10:07:14 +0000 | [diff] [blame] | 35 | TEST_CASE ("FLOOR_Fp32_Test") |
Sadik Armagan | 788e2c6 | 2021-02-10 16:26:44 +0000 | [diff] [blame] | 36 | { |
Colm Donelan | 7bcae3c | 2024-01-22 10:07:14 +0000 | [diff] [blame] | 37 | FloorFp32Test(); |
Sadik Armagan | 788e2c6 | 2021-02-10 16:26:44 +0000 | [diff] [blame] | 38 | } |
| 39 | |
| 40 | } |
| 41 | // End of FLOOR Test Suite |
| 42 | |
| 43 | } // namespace armnnDelegate |