From bc5f784d34a5315748b56cecb4674f4e546da2cd Mon Sep 17 00:00:00 2001 From: kevmay01 Date: Thu, 30 Aug 2018 12:34:39 +0100 Subject: IVGCVSW-1770: Refactor ModelToINetworkConverter to allow conversion of HAL1.1 operators Change-Id: I9b10f0a9c88344df108b2325c0233f9660fa6b7c --- Utils.hpp | 35 +++++++++++++++++++++++++++++++++-- 1 file changed, 33 insertions(+), 2 deletions(-) (limited to 'Utils.hpp') diff --git a/Utils.hpp b/Utils.hpp index e805f260..5d9f7003 100644 --- a/Utils.hpp +++ b/Utils.hpp @@ -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 +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 -- cgit v1.2.1