diff options
| author | Narumol Prangnawarat <narumol.prangnawarat@arm.com> | 2020-08-19 14:39:07 +0100 |
|---|---|---|
| committer | Narumol Prangnawarat <narumol.prangnawarat@arm.com> | 2020-08-19 14:43:09 +0100 |
| commit | a2493a0483f19fe9654be63a15badfb0834aaff6 (patch) | |
| tree | a526ee366356b173947d13162ba4b34a965f23ed /tests | |
| parent | 37c8197c9153924ebe934d5b521c0985eab9e477 (diff) | |
| download | armnn-a2493a0483f19fe9654be63a15badfb0834aaff6.tar.gz | |
IVGCVSW-5012 Add importEnabled option for OptimizerOptions
* Default importEnabled to false
* Improve error messages
Signed-off-by: Narumol Prangnawarat <narumol.prangnawarat@arm.com>
Change-Id: I17f78986aa1d23e48b0844297a52029b1a9bbe3e
Diffstat (limited to 'tests')
| -rw-r--r-- | tests/TfLiteYoloV3Big-Armnn/TfLiteYoloV3Big-Armnn.cpp | 4 |
1 files changed, 3 insertions, 1 deletions
diff --git a/tests/TfLiteYoloV3Big-Armnn/TfLiteYoloV3Big-Armnn.cpp b/tests/TfLiteYoloV3Big-Armnn/TfLiteYoloV3Big-Armnn.cpp index 93982fdb98..2d373cd1dd 100644 --- a/tests/TfLiteYoloV3Big-Armnn/TfLiteYoloV3Big-Armnn.cpp +++ b/tests/TfLiteYoloV3Big-Armnn/TfLiteYoloV3Big-Armnn.cpp @@ -119,7 +119,9 @@ int LoadModel(const char* filename, ARMNN_LOG(debug) << "Model loaded ok: " << filename; // Optimize backbone model - auto optimizedModel = Optimize(*model, backendPreferences, runtime.GetDeviceSpec()); + OptimizerOptions options; + options.m_ImportEnabled = enableImport; + auto optimizedModel = Optimize(*model, backendPreferences, runtime.GetDeviceSpec(), options); if (!optimizedModel) { ARMNN_LOG(fatal) << "Could not optimize the model:" << filename; |