diff options
Diffstat (limited to 'tests/InferenceTest.inl')
-rw-r--r-- | tests/InferenceTest.inl | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/tests/InferenceTest.inl b/tests/InferenceTest.inl index 83a99459e3..a36e231e76 100644 --- a/tests/InferenceTest.inl +++ b/tests/InferenceTest.inl @@ -307,6 +307,7 @@ int ClassifierInferenceTestMain(int argc, char* argv[], const char* modelFilenam modelParams.m_InputTensorShape = inputTensorShape; modelParams.m_IsModelBinary = isModelBinary; modelParams.m_ComputeDevice = modelOptions.m_ComputeDevice; + modelParams.m_VisualizePostOptimizationModel = modelOptions.m_VisualizePostOptimizationModel; return std::make_unique<InferenceModel>(modelParams); }); |