diff options
author | Sadik Armagan <sadik.armagan@arm.com> | 2021-01-13 15:56:51 +0000 |
---|---|---|
committer | Sadik Armagan <sadik.armagan@arm.com> | 2021-01-13 15:56:51 +0000 |
commit | b3021435ad91e494af01ca0778915877dc0780c0 (patch) | |
tree | 176bff8eafd8f6a2679d3569cbd640400b10a9ef /Utils.hpp | |
parent | f36e10b8947fe5f0984e7428c2d5d5d7fe18007e (diff) | |
download | android-nn-driver-b3021435ad91e494af01ca0778915877dc0780c0.tar.gz |
IVGCVSW-4417 'Serialise ArmNN Model on android-nn-driver'
* Implemented serialization of the network on android-nn-driver
!armnn:4850
Signed-off-by: Sadik Armagan <sadik.armagan@arm.com>
Change-Id: I3caf07bd4d1d2a3068c58f0b344303c4cf977ca6
Diffstat (limited to 'Utils.hpp')
-rw-r--r-- | Utils.hpp | 12 |
1 files changed, 11 insertions, 1 deletions
@@ -135,7 +135,17 @@ void DumpJsonProfilingIfRequired(bool gpuProfilingEnabled, std::string ExportNetworkGraphToDotFile(const armnn::IOptimizedNetwork& optimizedNetwork, const std::string& dumpDir); -void RenameGraphDotFile(const std::string& oldName, const std::string& dumpDir, const armnn::NetworkId networkId); +std::string SerializeNetwork(const armnn::INetwork& network, const std::string& dumpDir); + +void RenameExportedFiles(const std::string& existingSerializedFileName, + const std::string& existingDotFileName, + const std::string& dumpDir, + const armnn::NetworkId networkId); + +void RenameFile(const std::string& existingName, + const std::string& extension, + const std::string& dumpDir, + const armnn::NetworkId networkId); /// Checks if a tensor info represents a dynamic tensor bool IsDynamicTensor(const armnn::TensorInfo& outputInfo); |