aboutsummaryrefslogtreecommitdiff
path: root/1.3/ArmnnDriverImpl.cpp
diff options
context:
space:
mode:
Diffstat (limited to '1.3/ArmnnDriverImpl.cpp')
-rw-r--r--1.3/ArmnnDriverImpl.cpp6
1 files changed, 5 insertions, 1 deletions
diff --git a/1.3/ArmnnDriverImpl.cpp b/1.3/ArmnnDriverImpl.cpp
index c8b1d968..474e1c1f 100644
--- a/1.3/ArmnnDriverImpl.cpp
+++ b/1.3/ArmnnDriverImpl.cpp
@@ -281,7 +281,9 @@ Return<V1_3::ErrorStatus> ArmnnDriverImpl::prepareArmnnModel_1_3(
options.IsGpuProfilingEnabled(),
priority,
options.isAsyncModelExecutionEnabled(),
- options.getNoOfArmnnThreads()));
+ options.getNoOfArmnnThreads(),
+ options.isImportEnabled(),
+ options.isExportEnabled()));
// Run a single 'dummy' inference of the model. This means that CL kernels will get compiled (and tuned if
// this is enabled) before the first 'real' inference which removes the overhead of the first inference.
@@ -645,6 +647,8 @@ Return<V1_3::ErrorStatus> ArmnnDriverImpl::prepareModelFromCache_1_3(
V1_3::Priority::MEDIUM,
options.isAsyncModelExecutionEnabled(),
options.getNoOfArmnnThreads(),
+ options.isImportEnabled(),
+ options.isExportEnabled(),
true));
NotifyCallbackAndCheck(cb, V1_3::ErrorStatus::NONE, preparedModel.release());