From 964e955d8962590d3ccbaba9784962e895ca656f Mon Sep 17 00:00:00 2001 From: Nattapat Chaimanowong Date: Tue, 26 Mar 2019 11:03:26 +0000 Subject: IVGCVSW-2881 Remove DebugDescriptor * Also update Debug layer to use layer guid information Change-Id: I9ec1f639299c3f855b670ff031a0e88d685cfc6b Signed-off-by: Nattapat Chaimanowong --- src/backends/reference/workloads/Debug.cpp | 24 ++++++++++++++-------- src/backends/reference/workloads/Debug.hpp | 7 ++++--- .../reference/workloads/RefDebugWorkload.cpp | 2 +- 3 files changed, 21 insertions(+), 12 deletions(-) (limited to 'src/backends/reference/workloads') diff --git a/src/backends/reference/workloads/Debug.cpp b/src/backends/reference/workloads/Debug.cpp index cc83c7b4ee..b263db67cf 100644 --- a/src/backends/reference/workloads/Debug.cpp +++ b/src/backends/reference/workloads/Debug.cpp @@ -16,9 +16,11 @@ namespace armnn template void Debug(const TensorInfo& inputInfo, const TensorInfo& outputInfo, - const DebugDescriptor& descriptor, const T* inputData, - T* outputData) + T* outputData, + LayerGuid guid, + const std::string& layerName, + unsigned int slotIndex) { const unsigned int numDims = inputInfo.GetNumDimensions(); const unsigned int numElements = inputInfo.GetNumElements(); @@ -33,8 +35,9 @@ void Debug(const TensorInfo& inputInfo, } std::cout << "{ "; - std::cout << "\"layer\": \"" << descriptor.m_LayerName << "\", "; - std::cout << "\"outputSlot\": " << descriptor.m_SlotIndex << ", "; + std::cout << "\"layerGuid\": " << guid << ", "; + std::cout << "\"layerName\": \"" << layerName << "\", "; + std::cout << "\"outputSlot\": " << slotIndex << ", "; std::cout << "\"shape\": "; std::cout << "["; @@ -89,13 +92,18 @@ void Debug(const TensorInfo& inputInfo, template void Debug(const TensorInfo& inputInfo, const TensorInfo& outputInfo, - const DebugDescriptor& descriptor, const float* inputData, - float* outputData); + float* outputData, + LayerGuid guid, + const std::string& layerName, + unsigned int slotIndex); template void Debug(const TensorInfo& inputInfo, const TensorInfo& outputInfo, - const DebugDescriptor& descriptor, const uint8_t* inputData, - uint8_t* outputData); + uint8_t* outputData, + LayerGuid guid, + const std::string& layerName, + unsigned int slotIndex); + } // namespace armnn diff --git a/src/backends/reference/workloads/Debug.hpp b/src/backends/reference/workloads/Debug.hpp index 682f0bd31b..29a7d40662 100644 --- a/src/backends/reference/workloads/Debug.hpp +++ b/src/backends/reference/workloads/Debug.hpp @@ -4,7 +4,6 @@ // #pragma once -#include #include namespace armnn @@ -13,8 +12,10 @@ namespace armnn template void Debug(const TensorInfo& inputInfo, const TensorInfo& outputInfo, - const DebugDescriptor& descriptor, const T* inputData, - T* outputData); + T* outputData, + LayerGuid guid, + const std::string& layerName, + unsigned int slotIndex); } //namespace armnn diff --git a/src/backends/reference/workloads/RefDebugWorkload.cpp b/src/backends/reference/workloads/RefDebugWorkload.cpp index d9a47c0596..412d399adc 100644 --- a/src/backends/reference/workloads/RefDebugWorkload.cpp +++ b/src/backends/reference/workloads/RefDebugWorkload.cpp @@ -25,7 +25,7 @@ void RefDebugWorkload::Execute() const const T* inputData = GetInputTensorData(0, m_Data); T* outputData = GetOutputTensorData(0, m_Data); - Debug(inputInfo, outputInfo, m_Data.m_Parameters, inputData, outputData); + Debug(inputInfo, outputInfo, inputData, outputData, m_Data.m_Guid, m_Data.m_LayerName, m_Data.m_SlotIndex); } template class RefDebugWorkload; -- cgit v1.2.1