diff options
Diffstat (limited to 'src/backends/reference/workloads')
-rw-r--r-- | src/backends/reference/workloads/Debug.cpp | 24 | ||||
-rw-r--r-- | src/backends/reference/workloads/Debug.hpp | 7 | ||||
-rw-r--r-- | src/backends/reference/workloads/RefDebugWorkload.cpp | 2 |
3 files changed, 21 insertions, 12 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 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 <armnn/Descriptors.hpp> #include <armnn/Tensor.hpp> namespace armnn @@ -13,8 +12,10 @@ 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); } //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<DataType>::Execute() const const T* inputData = GetInputTensorData<T>(0, m_Data); T* outputData = GetOutputTensorData<T>(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<DataType::Float32>; |