17 template<armnn::DataType DataType>
20 Execute(m_Data.m_Inputs);
23 template<armnn::DataType DataType>
26 Execute(workingMemDescriptor.
m_Inputs);
29 template<armnn::DataType DataType>
38 const T* inputData = GetInputTensorData<T>(0, m_Data);
39 T* outputData = GetOutputTensorData<T>(0, m_Data);
43 m_Callback(m_Data.m_Guid, m_Data.m_SlotIndex, inputs[0]);
47 Debug(inputInfo, inputData, m_Data.m_Guid, m_Data.m_LayerName, m_Data.m_SlotIndex);
50 std::memcpy(outputData, inputData, inputInfo.
GetNumElements()*
sizeof(T));
53 template<armnn::DataType DataType>
void Execute() const override
CPU Execution: Reference C++ kernels.
typename ResolveTypeImpl< DT >::Type ResolveType
void ExecuteAsync(WorkingMemDescriptor &workingMemDescriptor) override
Copyright (c) 2021 ARM Limited and Contributors.
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)
std::vector< ITensorHandle * > m_Inputs
void Debug(const TensorInfo &inputInfo, const T *inputData, LayerGuid guid, const std::string &layerName, unsigned int slotIndex)
const TensorInfo & GetTensorInfo(const ITensorHandle *tensorHandle)
float32 helpers
void RegisterDebugCallback(const DebugCallbackFunction &func) override
unsigned int GetNumElements() const