diff options
Diffstat (limited to 'src/backends/reference/workloads/RefConcatWorkload.cpp')
-rw-r--r-- | src/backends/reference/workloads/RefConcatWorkload.cpp | 12 |
1 files changed, 11 insertions, 1 deletions
diff --git a/src/backends/reference/workloads/RefConcatWorkload.cpp b/src/backends/reference/workloads/RefConcatWorkload.cpp index e606649ed0..c04c05354e 100644 --- a/src/backends/reference/workloads/RefConcatWorkload.cpp +++ b/src/backends/reference/workloads/RefConcatWorkload.cpp @@ -14,8 +14,18 @@ namespace armnn void RefConcatWorkload::Execute() const { + Execute(m_Data.m_Inputs, m_Data.m_Outputs); +} + +void RefConcatWorkload::ExecuteAsync(WorkingMemDescriptor &workingMemDescriptor) +{ + Execute(workingMemDescriptor.m_Inputs, workingMemDescriptor.m_Outputs); +} + +void RefConcatWorkload::Execute(std::vector<ITensorHandle*> inputs, std::vector<ITensorHandle*> outputs) const +{ ARMNN_SCOPED_PROFILING_EVENT(Compute::CpuRef, "RefConcatWorkload_Execute"); - Concatenate(m_Data); + Concatenate(m_Data, inputs, outputs); } } //namespace armnn |