20 #include <boost/assert.hpp> 26 class IWorkloadFactory;
28 class WorkloadDataCollector;
53 void SetData(std::unique_ptr<ITensorHandle> data) { m_TensorHandle = std::move(data); }
58 std::unique_ptr<ITensorHandle> m_TensorHandle;
60 bool m_bTensorInfoSet =
false;
void CollectWorkloadOutputs(WorkloadDataCollector &dataCollector) const
Fill the outputs for a given queue descriptor.
const TensorInfo & GetTensorInfo() const
Gets the matching TensorInfo for the output.
void SetData(std::unique_ptr< ITensorHandle > data)
bool IsTensorInfoSet() const
Returns true if SetTensorInfo() has been called at least once on this.
void CreateTensorHandles(const IWorkloadFactory &factory, const bool IsMemoryManaged=true)
Creates tensor handles used by the intermediate tensors. Does not allocate memory.
ITensorHandle * GetData() const
Gets the allocated tensor memory.
void SetTensorInfo(const TensorInfo &tensorInfo)
Sets the TensorInfo used by this output handler.