aboutsummaryrefslogtreecommitdiff
path: root/tests/TfLiteYoloV3Big-Armnn
diff options
context:
space:
mode:
authorJohn Mcloughlin <john.mcloughlin@arm.com>2023-03-24 12:07:25 +0000
committerFrancis Murtagh <francis.murtagh@arm.com>2023-04-12 18:28:23 +0100
commitc5ee0d7460f1e0ec7e2b0639e3e8962934c4df09 (patch)
tree931f1403589c34fd2de6b94d95e9e172a92424fe /tests/TfLiteYoloV3Big-Armnn
parentca5c82af9269e7fd7ed17c7df9780a75fdaa733e (diff)
downloadarmnn-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')
-rw-r--r--tests/TfLiteYoloV3Big-Armnn/TfLiteYoloV3Big-Armnn.cpp6
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);