You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
45 lines
1.2 KiB
45 lines
1.2 KiB
//
|
|
// Copyright © 2021, 2023 Arm Ltd and Contributors. All rights reserved.
|
|
// SPDX-License-Identifier: MIT
|
|
//
|
|
|
|
#include "ShapeTestHelper.hpp"
|
|
|
|
#include <doctest/doctest.h>
|
|
|
|
namespace armnnDelegate
|
|
{
|
|
|
|
void ShapeSimpleTest(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,
|
|
backends,
|
|
inputShape,
|
|
inputValues,
|
|
expectedOutputValues,
|
|
expectedOutputShape);
|
|
}
|
|
|
|
// 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
|