12 #include <arm_compute/runtime/CL/functions/CLSplit.h> 20 using namespace armcomputetensorutils;
24 unsigned int CalcAclAxis(
unsigned int numDimensions,
unsigned int splitAxis)
26 return (numDimensions - splitAxis) - 1;
32 const std::vector<std::reference_wrapper<TensorInfo>>& outputs,
33 unsigned int splitAxis)
35 const arm_compute::TensorInfo aclInputInfo = BuildArmComputeTensorInfo(input);
37 size_t numOutputs = outputs.size();
39 std::vector<arm_compute::TensorInfo> aclOutputs;
40 aclOutputs.reserve(numOutputs);
42 std::vector<arm_compute::ITensorInfo*> aclOutputPtr;
43 aclOutputPtr.reserve(numOutputs);
45 for (
size_t i = 0u; i < outputs.size(); ++i)
47 aclOutputs.emplace_back(BuildArmComputeTensorInfo(outputs[i]));
48 aclOutputPtr.emplace_back(&aclOutputs.back());
52 return arm_compute::CLSplit::validate(&aclInputInfo, aclOutputPtr, aclAxis);
58 bool allOutputsAreSubtensors =
true;
63 if (output && !output->GetParent())
66 allOutputsAreSubtensors =
false;
71 if (allOutputsAreSubtensors)
77 arm_compute::ICLTensor& input = boost::polymorphic_pointer_downcast<IClTensorHandle>(
80 std::vector<arm_compute::ICLTensor *> aclOutputs;
83 arm_compute::ICLTensor& aclOutput = boost::polymorphic_pointer_downcast<IClTensorHandle>(output)->GetTensor();
84 aclOutputs.emplace_back(&aclOutput);
91 if (splitAxis.size() != 1)
97 auto layer = std::make_unique<arm_compute::CLSplit>();
98 layer->configure(&input, aclOutputs, aclAxis);
103 m_Layer = std::move(layer);
ClSplitterWorkload(const SplitterQueueDescriptor &descriptor, const WorkloadInfo &info)
arm_compute::Status ClSplitterWorkloadValidate(const TensorInfo &input, const std::vector< std::reference_wrapper< TensorInfo >> &outputs, unsigned int splitAxis)
uint32_t GetNumDimensions() const
Get the number of dimensions.
#define ARMNN_SCOPED_PROFILING_EVENT_CL(name)
const SplitterQueueDescriptor m_Data
Copyright (c) 2020 ARM Limited.
std::set< unsigned int > ComputeSplitAxis(const armnn::SplitterDescriptor &desc, const TensorShape &input)
LayerDescriptor m_Parameters
std::vector< ITensorHandle * > m_Outputs
void Execute() const override
Contains information about inputs and outputs to a layer.
std::vector< ITensorHandle * > m_Inputs
unsigned int GetNumDimensions() const