From b3021435ad91e494af01ca0778915877dc0780c0 Mon Sep 17 00:00:00 2001 From: Sadik Armagan Date: Wed, 13 Jan 2021 15:56:51 +0000 Subject: 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 Change-Id: I3caf07bd4d1d2a3068c58f0b344303c4cf977ca6 --- Utils.hpp | 12 +++++++++++- 1 file changed, 11 insertions(+), 1 deletion(-) (limited to 'Utils.hpp') 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); -- cgit v1.2.1