aboutsummaryrefslogtreecommitdiff
path: root/tests/TfLiteMobileNetSsd-Armnn
diff options
context:
space:
mode:
Diffstat (limited to 'tests/TfLiteMobileNetSsd-Armnn')
-rw-r--r--tests/TfLiteMobileNetSsd-Armnn/TfLiteMobileNetSsd-Armnn.cpp2
1 files changed, 1 insertions, 1 deletions
diff --git a/tests/TfLiteMobileNetSsd-Armnn/TfLiteMobileNetSsd-Armnn.cpp b/tests/TfLiteMobileNetSsd-Armnn/TfLiteMobileNetSsd-Armnn.cpp
index b1bc0f6120..3328339318 100644
--- a/tests/TfLiteMobileNetSsd-Armnn/TfLiteMobileNetSsd-Armnn.cpp
+++ b/tests/TfLiteMobileNetSsd-Armnn/TfLiteMobileNetSsd-Armnn.cpp
@@ -59,7 +59,7 @@ int main(int argc, char* argv[])
modelParams.m_InputShapes = { inputTensorShape };
modelParams.m_IsModelBinary = true;
- modelParams.m_ComputeDevice = modelOptions.m_ComputeDevice;
+ modelParams.m_ComputeDevices = modelOptions.GetComputeDevicesAsBackendIds();
modelParams.m_VisualizePostOptimizationModel = modelOptions.m_VisualizePostOptimizationModel;
modelParams.m_EnableFp16TurboMode = modelOptions.m_EnableFp16TurboMode;