aboutsummaryrefslogtreecommitdiff
path: root/DriverOptions.hpp
diff options
context:
space:
mode:
authorSadik Armagan <sadik.armagan@arm.com>2021-01-11 16:34:01 +0000
committerSadik Armagan <sadik.armagan@arm.com>2021-01-11 16:34:01 +0000
commitf36e10b8947fe5f0984e7428c2d5d5d7fe18007e (patch)
treec531059e65941665de3931983c81f6ab256c75b7 /DriverOptions.hpp
parent1312959112045a46b1c9d2d5f68c47d8194ce865 (diff)
downloadandroid-nn-driver-f36e10b8947fe5f0984e7428c2d5d5d7fe18007e.tar.gz
IVGCVSW-5485 'Add CacheLoadedNetwork options to Android NN Driver'
* Added Flatbuffers dependency to android-nn-driver * Added CacheLoadedNetwork options to android-nn-driver Options will be android.hardware.neuralnetworks@1.3-service-armnn -v -c GpuAcc -n arm-armnn -s -q <file_path> & Signed-off-by: Sadik Armagan <sadik.armagan@arm.com> Change-Id: I2cdf0b84062cb1c6ad4346cd64b00161cfe85920
Diffstat (limited to 'DriverOptions.hpp')
-rw-r--r--DriverOptions.hpp4
1 files changed, 4 insertions, 0 deletions
diff --git a/DriverOptions.hpp b/DriverOptions.hpp
index 15236521..f3b441ca 100644
--- a/DriverOptions.hpp
+++ b/DriverOptions.hpp
@@ -36,6 +36,8 @@ public:
void SetBackends(const std::vector<armnn::BackendId>& backends) { m_Backends = backends; }
bool ShouldExit() const { return m_ShouldExit; }
int GetExitCode() const { return m_ExitCode; }
+ const std::string& GetCachedNetworkFilePath() const { return m_CachedNetworkFilePath; }
+ bool SaveCachedNetwork() const { return m_SaveCachedNetwork; }
private:
std::vector<armnn::BackendId> m_Backends;
@@ -51,6 +53,8 @@ private:
bool m_FastMathEnabled;
bool m_ShouldExit;
int m_ExitCode;
+ std::string m_CachedNetworkFilePath;
+ bool m_SaveCachedNetwork;
};
} // namespace armnn_driver