ArmNN
 24.02
WorkloadDataCollector Class Reference

#include <WorkloadDataCollector.hpp>

Public Member Functions

 WorkloadDataCollector (std::vector< ITensorHandle * > &handles, std::vector< TensorInfo > &infos)
 
void Push (ITensorHandle *handle, const TensorInfo &info)
 

Detailed Description

Definition at line 15 of file WorkloadDataCollector.hpp.

Constructor & Destructor Documentation

◆ WorkloadDataCollector()

WorkloadDataCollector ( std::vector< ITensorHandle * > &  handles,
std::vector< TensorInfo > &  infos 
)
inline

Definition at line 18 of file WorkloadDataCollector.hpp.

19  : m_Handles(handles)
20  , m_Infos(infos)
21  {
22  }

Member Function Documentation

◆ Push()

void Push ( ITensorHandle handle,
const TensorInfo info 
)
inline

Definition at line 24 of file WorkloadDataCollector.hpp.

25  {
26  m_Handles.push_back(handle);
27  m_Infos.push_back(info);
28  }

References armnn::info.

Referenced by OutputHandler::CollectWorkloadOutputs().


The documentation for this class was generated from the following file: