aboutsummaryrefslogtreecommitdiff
path: root/delegate/test/TileTest.cpp
diff options
context:
space:
mode:
authorTianle Cheng <tianle.cheng@arm.com>2023-07-25 16:41:00 +0100
committerTianle Cheng <tianle.cheng@arm.com>2023-07-28 08:11:56 +0000
commit92ce35cda7c5e97eff05d6f37dc86990386309bb (patch)
treebd868f94a9ff0a126b626f94f4129736af927cfa /delegate/test/TileTest.cpp
parent35bae832a88dc06555492e7fe1ccb36a47594c21 (diff)
downloadarmnn-92ce35cda7c5e97eff05d6f37dc86990386309bb.tar.gz
IVGCVSW-7886 Add TILE to delegate and opaque delegate
* Adding support for Tile in classic and opaque delegates * CMake files updated * Tests added Signed-off-by: Tianle Cheng <tianle.cheng@arm.com> Change-Id: I9b52cea3480eb71961cbccb1a346805f73b5661a
Diffstat (limited to 'delegate/test/TileTest.cpp')
-rw-r--r--delegate/test/TileTest.cpp91
1 files changed, 91 insertions, 0 deletions
diff --git a/delegate/test/TileTest.cpp b/delegate/test/TileTest.cpp
new file mode 100644
index 0000000000..2e20859f5b
--- /dev/null
+++ b/delegate/test/TileTest.cpp
@@ -0,0 +1,91 @@
+//
+// Copyright © 2023 Arm Ltd and Contributors. All rights reserved.
+// SPDX-License-Identifier: MIT
+//
+
+#include "TileTestHelper.hpp"
+
+#include <armnn_delegate.hpp>
+#include <flatbuffers/flatbuffers.h>
+#include <tensorflow/lite/interpreter.h>
+#include <tensorflow/lite/kernels/register.h>
+#include <tensorflow/lite/model.h>
+#include <schema_generated.h>
+#include <tensorflow/lite/version.h>
+#include <doctest/doctest.h>
+
+namespace armnnDelegate
+{
+void TileFloat32Test(std::vector<armnn::BackendId>& backends)
+{
+ // Set input data
+ std::vector<float> inputValues =
+ {
+ 0.f, 1.f, 2.f,
+ 3.f, 4.f, 5.f
+ };
+
+ // Set output data
+ std::vector<float> expectedOutputValues =
+ {
+ 0.f, 1.f, 2.f, 0.f, 1.f, 2.f,
+ 3.f, 4.f, 5.f, 3.f, 4.f, 5.f,
+
+ 0.f, 1.f, 2.f, 0.f, 1.f, 2.f,
+ 3.f, 4.f, 5.f, 3.f, 4.f, 5.f
+ };
+
+ // The multiples
+ const std::vector<int32_t> multiplesValues = { 2, 2 };
+
+ // Set shapes
+ const std::vector<int32_t> inputShape = { 2, 3 };
+ const std::vector<int32_t> multiplesShape = { 2 };
+ const std::vector<int32_t> expectedOutputShape = { 4, 6 };
+
+ TileFP32TestImpl(tflite::BuiltinOperator_TILE,
+ backends,
+ inputValues,
+ inputShape,
+ multiplesValues,
+ multiplesShape,
+ expectedOutputValues,
+ expectedOutputShape);
+}
+
+#if defined(TILE_GPUACC)
+TEST_SUITE("TileTests_GpuAccTests")
+{
+
+ TEST_CASE ("Tile_Float32_GpuAcc_Test")
+ {
+ std::vector<armnn::BackendId> backends = { armnn::Compute::GpuAcc };
+ TileFloat32Test(backends);
+ }
+
+} // TEST_SUITE("Tile_Float32_GpuAcc_Test")
+#endif
+
+TEST_SUITE("TileTests_CpuAccTests")
+{
+
+ TEST_CASE ("Tile_Float32_CpuAcc_Test")
+ {
+ std::vector<armnn::BackendId> backends = { armnn::Compute::CpuAcc };
+ TileFloat32Test(backends);
+ }
+
+} // TEST_SUITE("Tile_Float32_CpuAcc_Test")
+
+TEST_SUITE("TileTests_CpuRefTests")
+{
+
+ TEST_CASE ("Tile_Float32_CpuRef_Test")
+ {
+ std::vector<armnn::BackendId> backends = { armnn::Compute::CpuRef };
+ TileFloat32Test(backends);
+ }
+
+} // TEST_SUITE("Tile_Float32_CpuRef_Test")
+
+} // namespace armnnDelegate \ No newline at end of file