aboutsummaryrefslogtreecommitdiff
path: root/tests/CaffeYolo-Armnn/CaffeYolo-Armnn.cpp
diff options
context:
space:
mode:
Diffstat (limited to 'tests/CaffeYolo-Armnn/CaffeYolo-Armnn.cpp')
-rw-r--r--tests/CaffeYolo-Armnn/CaffeYolo-Armnn.cpp1
1 files changed, 1 insertions, 0 deletions
diff --git a/tests/CaffeYolo-Armnn/CaffeYolo-Armnn.cpp b/tests/CaffeYolo-Armnn/CaffeYolo-Armnn.cpp
index ad79d49f0c..7396b7672c 100644
--- a/tests/CaffeYolo-Armnn/CaffeYolo-Armnn.cpp
+++ b/tests/CaffeYolo-Armnn/CaffeYolo-Armnn.cpp
@@ -37,6 +37,7 @@ int main(int argc, char* argv[])
modelParams.m_IsModelBinary = true;
modelParams.m_ComputeDevice = modelOptions.m_ComputeDevice;
modelParams.m_VisualizePostOptimizationModel = modelOptions.m_VisualizePostOptimizationModel;
+ modelParams.m_EnableFp16TurboMode = modelOptions.m_EnableFp16TurboMode;
return std::make_unique<YoloInferenceModel>(modelParams);
});