aboutsummaryrefslogtreecommitdiff
path: root/Utils.hpp
diff options
context:
space:
mode:
authortelsoa01 <telmo.soares@arm.com>2018-08-31 09:31:35 +0100
committertelsoa01 <telmo.soares@arm.com>2018-08-31 09:31:35 +0100
commitce3e84a8d449cbf31cee57e30f0eef6a96c0ce94 (patch)
tree77a769e27879fb712027c990071b061c5e0e3b60 /Utils.hpp
parentdeb3bdbe028a59da0759dd7a560387d03a11d322 (diff)
downloadandroid-nn-driver-ce3e84a8d449cbf31cee57e30f0eef6a96c0ce94.tar.gz
Release 18.08
Diffstat (limited to 'Utils.hpp')
-rw-r--r--Utils.hpp24
1 files changed, 15 insertions, 9 deletions
diff --git a/Utils.hpp b/Utils.hpp
index 4b5066ee..e805f260 100644
--- a/Utils.hpp
+++ b/Utils.hpp
@@ -5,13 +5,13 @@
#pragma once
-#include "HalInterfaces.h"
-#include "NeuralNetworks.h"
+#include "ArmnnDriver.hpp"
+
+#include <NeuralNetworks.h>
+
#include <armnn/ArmNN.hpp>
#include <CpuExecutor.h>
-#include "ArmnnDriver.hpp"
-
#include <vector>
#include <string>
@@ -43,14 +43,20 @@ void* GetMemoryFromPool(DataLocation location,
armnn::TensorInfo GetTensorInfoForOperand(const Operand& operand);
std::string GetOperandSummary(const Operand& operand);
-std::string GetModelSummary(const V1_0::Model& model);
+std::string GetModelSummary(const ::android::hardware::neuralnetworks::V1_0::Model& model);
void DumpTensor(const std::string& dumpDir,
- const std::string& requestName,
- const std::string& tensorName,
- const armnn::ConstTensor& tensor);
+ const std::string& requestName,
+ const std::string& tensorName,
+ const armnn::ConstTensor& tensor);
+
+void DumpJsonProfilingIfRequired(bool gpuProfilingEnabled,
+ const std::string& dumpDir,
+ armnn::NetworkId networkId,
+ const armnn::IProfiler* profiler);
void ExportNetworkGraphToDotFile(const armnn::IOptimizedNetwork& optimizedNetwork,
const std::string& dumpDir,
- const V1_0::Model& model);
+ const ::android::hardware::neuralnetworks::V1_0::Model& model);
+
}