diff options
author | kevmay01 <kevin.may@arm.com> | 2018-08-30 12:34:39 +0100 |
---|---|---|
committer | Matthew Bentham <matthew.bentham@arm.com> | 2018-09-18 12:40:38 +0100 |
commit | bc5f784d34a5315748b56cecb4674f4e546da2cd (patch) | |
tree | 70f300c7932301802cc8937531b5cf189e3bd536 /Utils.hpp | |
parent | 69558df6f149647888267caabf72de6b05fb6d53 (diff) | |
download | android-nn-driver-bc5f784d34a5315748b56cecb4674f4e546da2cd.tar.gz |
IVGCVSW-1770: Refactor ModelToINetworkConverter to allow conversion of HAL1.1 operators
Change-Id: I9b10f0a9c88344df108b2325c0233f9660fa6b7c
Diffstat (limited to 'Utils.hpp')
-rw-r--r-- | Utils.hpp | 35 |
1 files changed, 33 insertions, 2 deletions
@@ -43,7 +43,38 @@ void* GetMemoryFromPool(DataLocation location, armnn::TensorInfo GetTensorInfoForOperand(const Operand& operand); std::string GetOperandSummary(const Operand& operand); -std::string GetModelSummary(const ::android::hardware::neuralnetworks::V1_0::Model& model); + +template <typename Model> +std::string GetModelSummary(const Model& model) +{ + std::stringstream result; + + result << model.inputIndexes.size() << " input(s), " << model.operations.size() << " operation(s), " << + model.outputIndexes.size() << " output(s), " << model.operands.size() << " operand(s)" << std::endl; + + result << "Inputs: "; + for (uint32_t i = 0; i < model.inputIndexes.size(); i++) + { + result << GetOperandSummary(model.operands[model.inputIndexes[i]]) << ", "; + } + result << std::endl; + + result << "Operations: "; + for (uint32_t i = 0; i < model.operations.size(); i++) + { + result << toString(model.operations[i].type).c_str() << ", "; + } + result << std::endl; + + result << "Outputs: "; + for (uint32_t i = 0; i < model.outputIndexes.size(); i++) + { + result << GetOperandSummary(model.operands[model.outputIndexes[i]]) << ", "; + } + result << std::endl; + + return result.str(); +} void DumpTensor(const std::string& dumpDir, const std::string& requestName, @@ -59,4 +90,4 @@ void ExportNetworkGraphToDotFile(const armnn::IOptimizedNetwork& optimizedNetwor const std::string& dumpDir, const ::android::hardware::neuralnetworks::V1_0::Model& model); -} +}
\ No newline at end of file |