| // |
| // Copyright © 2021, 2023-2024 Arm Ltd and Contributors. All rights reserved. |
| // SPDX-License-Identifier: MIT |
| // |
| |
| #include "ShapeTestHelper.hpp" |
| |
| #include <doctest/doctest.h> |
| |
| namespace armnnDelegate |
| { |
| |
| void ShapeSimpleTest(const std::vector<armnn::BackendId>& backends = {}) |
| { |
| std::vector<int32_t> inputShape{ 1, 3, 2, 3 }; |
| |
| std::vector<int32_t> inputValues{ 1, 1, 1, 1, 1, 1, 1, 1, |
| 1, 1, 1, 1, 1, 1, 1, 1, }; |
| |
| std::vector<int32_t> expectedOutputShape{ 4 }; |
| std::vector<int32_t> expectedOutputValues{ 1, 3, 2, 3 }; |
| |
| ShapeTest<int32_t, int32_t>(::tflite::TensorType_INT32, |
| ::tflite::TensorType_INT32, |
| inputShape, |
| inputValues, |
| expectedOutputValues, |
| expectedOutputShape, |
| backends); |
| } |
| |
| // SHAPE Test Suite |
| TEST_SUITE("SHAPE_CpuRefTests") |
| { |
| |
| TEST_CASE("SHAPE_Simple_CpuRef_Test") |
| { |
| std::vector<armnn::BackendId> backends = { armnn::Compute::CpuRef }; |
| ShapeSimpleTest(backends); |
| } |
| |
| } |
| // End of SHAPE Test Suite |
| |
| } // namespace armnnDelegate |