diff options
Diffstat (limited to 'src/armnn/OutputHandler.cpp')
-rw-r--r-- | src/armnn/OutputHandler.cpp | 9 |
1 files changed, 9 insertions, 0 deletions
diff --git a/src/armnn/OutputHandler.cpp b/src/armnn/OutputHandler.cpp index 807262e482..8827d8ac58 100644 --- a/src/armnn/OutputHandler.cpp +++ b/src/armnn/OutputHandler.cpp @@ -35,4 +35,13 @@ void OutputHandler::CollectWorkloadOutputs(WorkloadDataCollector& dataCollector) dataCollector.Push(m_TensorHandle.get(), m_TensorInfo); } +void OutputHandler::SetAllocatedData() +{ + // Set allocated data only once + if (!m_AllocatedTensorHandle) + { + m_AllocatedTensorHandle = std::move(m_TensorHandle); + } +} + } // namespace armnn |