1 // 2 // Copyright © 2022 Arm Ltd and Contributors. All rights reserved. 3 // SPDX-License-Identifier: MIT 4 // 5 6 #include "RefSplitterWorkload.hpp" 7 #include "Splitter.hpp" 8 #include "RefWorkloadUtils.hpp" 9 #include "Profiling.hpp" 10 11 namespace armnn 12 { 13 Execute() const14void RefSplitterWorkload::Execute() const 15 { 16 Execute(m_Data.m_Inputs, m_Data.m_Outputs); 17 } 18 ExecuteAsync(ExecutionData & executionData)19void RefSplitterWorkload::ExecuteAsync(ExecutionData& executionData) 20 { 21 WorkingMemDescriptor* workingMemDescriptor = static_cast<WorkingMemDescriptor*>(executionData.m_Data); 22 Execute(workingMemDescriptor->m_Inputs, workingMemDescriptor->m_Outputs); 23 } 24 Execute(std::vector<ITensorHandle * > inputs,std::vector<ITensorHandle * > outputs) const25void RefSplitterWorkload::Execute(std::vector<ITensorHandle*> inputs, std::vector<ITensorHandle*> outputs) const 26 { 27 ARMNN_SCOPED_PROFILING_EVENT(Compute::CpuRef, "RefSplitterWorkload_Execute"); 28 Split(m_Data, inputs, outputs); 29 } 30 31 } //namespace armnn 32