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 15 class WorkloadDataCollector 16 { 17 public: WorkloadDataCollector(std::vector<ITensorHandle * > & handles,std::vector<TensorInfo> & infos)18 WorkloadDataCollector(std::vector<ITensorHandle*>& handles, std::vector<TensorInfo>& infos) 19 : m_Handles(handles) 20 , m_Infos(infos) 21 { 22 } 23 Push(ITensorHandle * handle,const TensorInfo & info)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 37