diff options
author | telsoa01 <telmo.soares@arm.com> | 2018-08-31 09:31:35 +0100 |
---|---|---|
committer | telsoa01 <telmo.soares@arm.com> | 2018-08-31 09:31:35 +0100 |
commit | ce3e84a8d449cbf31cee57e30f0eef6a96c0ce94 (patch) | |
tree | 77a769e27879fb712027c990071b061c5e0e3b60 /Utils.hpp | |
parent | deb3bdbe028a59da0759dd7a560387d03a11d322 (diff) | |
download | android-nn-driver-ce3e84a8d449cbf31cee57e30f0eef6a96c0ce94.tar.gz |
Release 18.08
Diffstat (limited to 'Utils.hpp')
-rw-r--r-- | Utils.hpp | 24 |
1 files changed, 15 insertions, 9 deletions
@@ -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); + } |