diff options
author | John Mcloughlin <john.mcloughlin@arm.com> | 2023-03-24 12:07:25 +0000 |
---|---|---|
committer | Francis Murtagh <francis.murtagh@arm.com> | 2023-04-12 18:28:23 +0100 |
commit | c5ee0d7460f1e0ec7e2b0639e3e8962934c4df09 (patch) | |
tree | 931f1403589c34fd2de6b94d95e9e172a92424fe /tests/TfLiteYoloV3Big-Armnn/TfLiteYoloV3Big-Armnn.cpp | |
parent | ca5c82af9269e7fd7ed17c7df9780a75fdaa733e (diff) | |
download | armnn-c5ee0d7460f1e0ec7e2b0639e3e8962934c4df09.tar.gz |
IVGCVSW-7197 Implement Pimpl Idiom for OptimizerOptions
Signed-off-by: John Mcloughlin <john.mcloughlin@arm.com>
Change-Id: Id4bdc31e3e6f18ccaef232c29a2d2825c915b21c
Diffstat (limited to 'tests/TfLiteYoloV3Big-Armnn/TfLiteYoloV3Big-Armnn.cpp')
-rw-r--r-- | tests/TfLiteYoloV3Big-Armnn/TfLiteYoloV3Big-Armnn.cpp | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/tests/TfLiteYoloV3Big-Armnn/TfLiteYoloV3Big-Armnn.cpp b/tests/TfLiteYoloV3Big-Armnn/TfLiteYoloV3Big-Armnn.cpp index 75bc9a3244..3ecd160b48 100644 --- a/tests/TfLiteYoloV3Big-Armnn/TfLiteYoloV3Big-Armnn.cpp +++ b/tests/TfLiteYoloV3Big-Armnn/TfLiteYoloV3Big-Armnn.cpp @@ -128,8 +128,8 @@ int LoadModel(const char* filename, ARMNN_LOG(debug) << "Model loaded ok: " << filename; // Optimize backbone model - OptimizerOptions options; - options.m_ImportEnabled = enableImport != ImportMemory::False; + OptimizerOptionsOpaque options; + options.SetImportEnabled(enableImport != ImportMemory::False); auto optimizedModel = Optimize(*model, backendPreferences, runtime.GetDeviceSpec(), options); if (!optimizedModel) { @@ -149,7 +149,7 @@ int LoadModel(const char* filename, { std::string errorMessage; - armnn::MemorySource memSource = options.m_ImportEnabled ? armnn::MemorySource::Malloc + armnn::MemorySource memSource = options.GetImportEnabled() ? armnn::MemorySource::Malloc : armnn::MemorySource::Undefined; INetworkProperties modelProps(false, memSource, memSource); Status status = runtime.LoadNetwork(networkId, std::move(optimizedModel), errorMessage, modelProps); |