Release 18.08

[platform/upstream/armnn.git] / tests / CaffeYolo-Armnn / CaffeYolo-Armnn.cpp

diff --git a/tests/CaffeYolo-Armnn/CaffeYolo-Armnn.cpp b/tests/CaffeYolo-Armnn/CaffeYolo-Armnn.cpp
index ad79d49..7396b76 100644 (file)
--- 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);
                 });