diff options
author | Matthew Sloyan <matthew.sloyan@arm.com> | 2021-01-08 10:30:51 +0000 |
---|---|---|
committer | Kevin May <kevin.may@arm.com> | 2021-01-12 10:38:11 +0000 |
commit | 4243211cb16d1960b2e1423ebdb83d619cc74b37 (patch) | |
tree | 84288d3f689bda2f4ca9ebb0828fe00a7829e81b /tests/InferenceModel.hpp | |
parent | 80fbcd5f4d7b362360963af1df0121aa6b561576 (diff) | |
download | armnn-4243211cb16d1960b2e1423ebdb83d619cc74b37.tar.gz |
IVGCVSW-5484 Add CacheLoadedNetwork options to ExecuteNetwork
* Enable ability to save/load ClContext in ExecuteNetwork.
Signed-off-by: Matthew Sloyan <matthew.sloyan@arm.com>
Change-Id: I58c61a53f6713853eb06520cc372ed47baf7f8c4
Diffstat (limited to 'tests/InferenceModel.hpp')
-rw-r--r-- | tests/InferenceModel.hpp | 8 |
1 files changed, 7 insertions, 1 deletions
diff --git a/tests/InferenceModel.hpp b/tests/InferenceModel.hpp index 1b87a10de1..7daae26bd4 100644 --- a/tests/InferenceModel.hpp +++ b/tests/InferenceModel.hpp @@ -97,6 +97,8 @@ struct Params bool m_ParseUnsupported; bool m_InferOutputShape; bool m_EnableFastMath; + bool m_SaveCachedNetwork; + std::string m_CachedNetworkFilePath; Params() : m_ComputeDevices{} @@ -109,6 +111,8 @@ struct Params , m_ParseUnsupported(false) , m_InferOutputShape(false) , m_EnableFastMath(false) + , m_SaveCachedNetwork(false) + , m_CachedNetworkFilePath("") {} }; @@ -426,7 +430,9 @@ public: armnn::BackendOptions gpuAcc("GpuAcc", { - { "FastMathEnabled", params.m_EnableFastMath } + { "FastMathEnabled", params.m_EnableFastMath }, + { "SaveCachedNetwork", params.m_SaveCachedNetwork }, + { "CachedNetworkFilePath", params.m_CachedNetworkFilePath } }); armnn::BackendOptions cpuAcc("CpuAcc", { |