ArmNN
 25.11
Loading...
Searching...
No Matches
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: