aboutsummaryrefslogtreecommitdiff
path: root/Utils.hpp
diff options
context:
space:
mode:
authorSadik Armagan <sadik.armagan@arm.com>2021-01-13 15:56:51 +0000
committerSadik Armagan <sadik.armagan@arm.com>2021-01-13 15:56:51 +0000
commitb3021435ad91e494af01ca0778915877dc0780c0 (patch)
tree176bff8eafd8f6a2679d3569cbd640400b10a9ef /Utils.hpp
parentf36e10b8947fe5f0984e7428c2d5d5d7fe18007e (diff)
downloadandroid-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.hpp12
1 files changed, 11 insertions, 1 deletions
diff --git a/Utils.hpp b/Utils.hpp
index e3b7d822..c4d89f7e 100644
--- a/Utils.hpp
+++ b/Utils.hpp
@@ -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);