ArmNN
 24.02
WorkloadDataCollector.hpp
Go to the documentation of this file.
1 //
2 // Copyright © 2017 Arm Ltd. All rights reserved.
3 // SPDX-License-Identifier: MIT
4 //
5 #pragma once
6 
7 #include <armnn/Tensor.hpp>
8 
9 #include <vector>
10 
11 namespace armnn
12 {
13 class ITensorHandle;
14 
16 {
17 public:
18  WorkloadDataCollector(std::vector<ITensorHandle*>& handles, std::vector<TensorInfo>& infos)
19  : m_Handles(handles)
20  , m_Infos(infos)
21  {
22  }
23 
24  void Push(ITensorHandle* handle, const TensorInfo& info)
25  {
26  m_Handles.push_back(handle);
27  m_Infos.push_back(info);
28  }
29 
30 private:
31  std::vector<ITensorHandle*>& m_Handles;
32  std::vector<TensorInfo>& m_Infos;
33 };
34 
35 
36 } //namespace armnn
armnn::TensorInfo
Definition: Tensor.hpp:152
armnn::ITensorHandle
Definition: ITensorHandle.hpp:16
armnn::WorkloadDataCollector
Definition: WorkloadDataCollector.hpp:15
armnn::BoostLogSeverityMapping::info
@ info
armnn::WorkloadDataCollector::WorkloadDataCollector
WorkloadDataCollector(std::vector< ITensorHandle * > &handles, std::vector< TensorInfo > &infos)
Definition: WorkloadDataCollector.hpp:18
armnn::WorkloadDataCollector::Push
void Push(ITensorHandle *handle, const TensorInfo &info)
Definition: WorkloadDataCollector.hpp:24
Tensor.hpp
armnn
Copyright (c) 2021 ARM Limited and Contributors.
Definition: 01_00_quick_start.dox:6