1 files changed, 4 insertions, 8 deletions
diff --git a/delegate/test/TileTestHelper.hpp b/delegate/test/TileTestHelper.hpp
index 0bdee9d7d4..880d01b41c 100644
--- a/delegate/test/TileTestHelper.hpp
+++ b/delegate/test/TileTestHelper.hpp
@@ -1,5 +1,5 @@
 //
-// Copyright © 2023 Arm Ltd and Contributors. All rights reserved.
+// Copyright © 2023-2024 Arm Ltd and Contributors. All rights reserved.
 // SPDX-License-Identifier: MIT
 //
 
@@ -10,12 +10,8 @@
 #include <armnn_delegate.hpp>
 #include <DelegateTestInterpreter.hpp>
 
-#include <flatbuffers/flatbuffers.h>
-#include <tensorflow/lite/kernels/register.h>
 #include <tensorflow/lite/version.h>
 
-#include <doctest/doctest.h>
-
 namespace
 {
 std::vector<char> CreateTileTfLiteModel(tflite::BuiltinOperator operatorCode,
@@ -102,13 +98,13 @@ std::vector<char> CreateTileTfLiteModel(tflite::BuiltinOperator operatorCode,
 }
 
 void TileFP32TestImpl(tflite::BuiltinOperator operatorCode,
-                      std::vector<armnn::BackendId>& backends,
                       std::vector<float>& inputValues,
                       std::vector<int32_t> inputShape,
                       std::vector<int32_t> multiplesValues,
                       std::vector<int32_t> multiplesShapes,
                       std::vector<float>& expectedOutputValues,
-                      std::vector<int32_t> expectedOutputShape)
+                      std::vector<int32_t> expectedOutputShape,
+                      const std::vector<armnn::BackendId>& backends = {})
 {
     using namespace delegateTestInterpreter;
 
@@ -129,7 +125,7 @@ void TileFP32TestImpl(tflite::BuiltinOperator operatorCode,
     std::vector<int32_t> tfLiteOutputShape  = tfLiteInterpreter.GetOutputShape(0);
 
     // Setup interpreter with Arm NN Delegate applied.
-    auto armnnInterpreter = DelegateTestInterpreter(modelBuffer, backends);
+    auto armnnInterpreter = DelegateTestInterpreter(modelBuffer, CaptureAvailableBackends(backends));
     CHECK(armnnInterpreter.AllocateTensors() == kTfLiteOk);
     CHECK(armnnInterpreter.FillInputTensor<float>(inputValues, 0) == kTfLiteOk);
     CHECK(armnnInterpreter.FillInputTensor<int32_t>(multiplesValues, 1) == kTfLiteOk);