aboutsummaryrefslogtreecommitdiff
path: root/delegate/test/BroadcastToTestHelper.hpp
diff options
context:
space:
mode:
Diffstat (limited to 'delegate/test/BroadcastToTestHelper.hpp')
-rw-r--r--delegate/test/BroadcastToTestHelper.hpp6
1 files changed, 3 insertions, 3 deletions
diff --git a/delegate/test/BroadcastToTestHelper.hpp b/delegate/test/BroadcastToTestHelper.hpp
index 6d05863b0c..8fcb762474 100644
--- a/delegate/test/BroadcastToTestHelper.hpp
+++ b/delegate/test/BroadcastToTestHelper.hpp
@@ -119,13 +119,13 @@ namespace
template<typename T>
void BroadcastToTestImpl(tflite::TensorType inputTensorType,
tflite::BuiltinOperator operatorCode,
- std::vector<armnn::BackendId>& backends,
std::vector<T>& inputValues,
std::vector<int32_t> inputShape,
std::vector<int32_t> shapeShapes,
std::vector<int32_t> shapeData,
std::vector<T>& expectedOutputValues,
- std::vector<int32_t> expectedOutputShape)
+ std::vector<int32_t> expectedOutputShape,
+ const std::vector<armnn::BackendId>& backends)
{
using namespace delegateTestInterpreter;
@@ -147,7 +147,7 @@ namespace
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<T>(inputValues, 0) == kTfLiteOk);
CHECK(armnnInterpreter.FillInputTensor<int32_t>(shapeData, 1) == kTfLiteOk);