diff options
Diffstat (limited to 'src/armnn/Layer.cpp')
-rw-r--r-- | src/armnn/Layer.cpp | 5 |
1 files changed, 5 insertions, 0 deletions
diff --git a/src/armnn/Layer.cpp b/src/armnn/Layer.cpp index d06b0459f6..62f861b4a0 100644 --- a/src/armnn/Layer.cpp +++ b/src/armnn/Layer.cpp @@ -242,6 +242,11 @@ void Layer::CollectWorkloadOutputs(WorkloadDataCollector& dataCollector) const } } +void Layer::SetAdditionalInfo(QueueDescriptor& descriptor) const +{ + descriptor.m_AdditionalInfoObject = m_AdditionalInfoObject.get(); +} + void Layer::CreateTensorHandles(const TensorHandleFactoryRegistry& registry, const IWorkloadFactory& workloadFactory, const bool IsMemoryManaged) |