diff options
Diffstat (limited to 'src/backends/backendsCommon/OutputHandler.cpp')
-rw-r--r-- | src/backends/backendsCommon/OutputHandler.cpp | 8 |
1 files changed, 4 insertions, 4 deletions
diff --git a/src/backends/backendsCommon/OutputHandler.cpp b/src/backends/backendsCommon/OutputHandler.cpp index 8f4942d8ba..e3a1b276ea 100644 --- a/src/backends/backendsCommon/OutputHandler.cpp +++ b/src/backends/backendsCommon/OutputHandler.cpp @@ -22,14 +22,14 @@ void OutputHandler::SetTensorInfo(const TensorInfo& tensorInfo) m_bTensorInfoSet = true; } -void OutputHandler::CreateTensorHandles(const IWorkloadFactory& factory) +void OutputHandler::CreateTensorHandles(const IWorkloadFactory& factory, const bool IsMemoryManaged) { - m_TensorHandle = factory.CreateTensorHandle(m_TensorInfo); + m_TensorHandle = factory.CreateTensorHandle(m_TensorInfo, IsMemoryManaged); } -void OutputHandler::CreateTensorHandles(const ITensorHandleFactory& factory) +void OutputHandler::CreateTensorHandles(const ITensorHandleFactory& factory, const bool IsMemoryManaged) { - m_TensorHandle = factory.CreateTensorHandle(m_TensorInfo); + m_TensorHandle = factory.CreateTensorHandle(m_TensorInfo, IsMemoryManaged); } void OutputHandler::CollectWorkloadOutputs(WorkloadDataCollector& dataCollector) const |