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