24 class IWorkloadFactory;
26 class WorkloadDataCollector;
51 void SetData(std::unique_ptr<ITensorHandle> data) { m_TensorHandle = std::move(data); }
56 std::unique_ptr<ITensorHandle> m_TensorHandle;
58 bool m_bTensorInfoSet =
false;
void SetData(std::unique_ptr< ITensorHandle > data)
Copyright (c) 2021 ARM Limited and Contributors.
void CreateTensorHandles(const IWorkloadFactory &factory, const bool IsMemoryManaged=true)
Creates tensor handles used by the intermediate tensors.
void CollectWorkloadOutputs(WorkloadDataCollector &dataCollector) const
Fill the outputs for a given queue descriptor.
ITensorHandle * GetData() const
Gets the allocated tensor memory.
bool IsTensorInfoSet() const
Returns true if SetTensorInfo() has been called at least once on this.
void SetTensorInfo(const TensorInfo &tensorInfo)
Sets the TensorInfo used by this output handler.
const TensorInfo & GetTensorInfo() const
Gets the matching TensorInfo for the output.