17 template<armnn::DataType DataType>
26 const T* inputData = GetInputTensorData<T>(0, m_Data);
27 T* outputData = GetOutputTensorData<T>(0, m_Data);
31 m_Callback(m_Data.m_Guid, m_Data.m_SlotIndex, m_Data.m_Inputs[0]);
35 Debug(inputInfo, inputData, m_Data.m_Guid, m_Data.m_LayerName, m_Data.m_SlotIndex);
38 std::memcpy(outputData, inputData, inputInfo.
GetNumElements()*
sizeof(T));
41 template<armnn::DataType DataType>
void Execute() const override
CPU Execution: Reference C++ kernels.
const TensorInfo & GetTensorInfo(const ITensorHandle *tensorHandle)
float32 helpers
typename ResolveTypeImpl< DT >::Type ResolveType
Copyright (c) 2020 ARM Limited.
std::function< void(LayerGuid guid, unsigned int slotIndex, ITensorHandle *tensorHandle)> DebugCallbackFunction
Define the type of callback for the Debug layer to call.
#define ARMNN_SCOPED_PROFILING_EVENT(backendId, name)
void Debug(const TensorInfo &inputInfo, const T *inputData, LayerGuid guid, const std::string &layerName, unsigned int slotIndex)
void RegisterDebugCallback(const DebugCallbackFunction &func) override
unsigned int GetNumElements() const