diff options
Diffstat (limited to 'src/backends/reference/workloads/Debug.cpp')
-rw-r--r-- | src/backends/reference/workloads/Debug.cpp | 24 |
1 files changed, 16 insertions, 8 deletions
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 <typename T> 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<float>(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<uint8_t>(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 |