ArmNN
 24.08
RefDebugWorkload.hpp
Go to the documentation of this file.
1 //
2 // Copyright © 2018-2024 Arm Ltd and Contributors. All rights reserved.
3 // SPDX-License-Identifier: MIT
4 //
5 
6 #pragma once
7 
8 #include <armnn/TypesUtils.hpp>
9 
10 #include "RefBaseWorkload.hpp"
11 
12 namespace armnn
13 {
14 
15 template <armnn::DataType DataType>
16 class RefDebugWorkload : public TypedWorkload<DebugQueueDescriptor, DataType>
17 {
18 public:
19  RefDebugWorkload(const DebugQueueDescriptor& descriptor, const WorkloadInfo& info)
21  , m_Callback(nullptr) {}
22 
23  virtual const std::string& GetName() const override
24  {
25  static const std::string name = std::string("RefDebug") + GetDataTypeName(DataType) + "Workload";
26  return name;
27  }
28 
31 
32  void Execute() const override;
33  void ExecuteAsync(ExecutionData& executionData) override;
34 
35  void RegisterDebugCallback(const DebugCallbackFunction& func) override;
36 
37 private:
38  void Execute(std::vector<ITensorHandle*> inputs) const;
39  DebugCallbackFunction m_Callback;
40 };
41 
52 
53 } // namespace armnn
armnn::DebugQueueDescriptor
Definition: WorkloadData.hpp:484
armnn::RefDebugWorkload::RefDebugWorkload
RefDebugWorkload(const DebugQueueDescriptor &descriptor, const WorkloadInfo &info)
Definition: RefDebugWorkload.hpp:19
TypesUtils.hpp
armnn::RefDebugWorkload::GetName
virtual const std::string & GetName() const override
Definition: RefDebugWorkload.hpp:23
armnn::RefDebugWorkload::RegisterDebugCallback
void RegisterDebugCallback(const DebugCallbackFunction &func) override
Definition: RefDebugWorkload.cpp:55
armnn::RefDebugWorkload::ExecuteAsync
void ExecuteAsync(ExecutionData &executionData) override
Definition: RefDebugWorkload.cpp:24
armnn::GetDataTypeName
constexpr const char * GetDataTypeName(DataType dataType)
Definition: TypesUtils.hpp:233
armnn::RefDebugWorkload
Definition: RefDebugWorkload.hpp:16
armnn::TypedWorkload
Definition: Workload.hpp:101
armnn::WorkloadInfo
Contains information about TensorInfos of a layer.
Definition: WorkloadInfo.hpp:16
armnn::DataType
DataType
Definition: Types.hpp:48
armnn::RefDebugWorkload::Execute
void Execute() const override
Definition: RefDebugWorkload.cpp:18
armnn::BoostLogSeverityMapping::info
@ info
armnn::DebugCallbackFunction
std::function< void(LayerGuid guid, unsigned int slotIndex, ITensorHandle *tensorHandle)> DebugCallbackFunction
Define the type of callback for the Debug layer to call.
Definition: Types.hpp:398
armnn
Copyright (c) 2021 ARM Limited and Contributors.
Definition: 01_00_quick_start.dox:6
RefBaseWorkload.hpp
armnn::experimental::ExecutionData
Definition: ExecutionData.hpp:14