8 #include <arm_compute/runtime/NEON/functions/NESplit.h> 21 using namespace armcomputetensorutils;
25 unsigned int CalcAclAxis(
unsigned int numDimensions,
unsigned int splitAxis)
27 return (numDimensions - splitAxis) - 1;
33 const std::vector<std::reference_wrapper<TensorInfo>>& outputs,
34 unsigned int splitAxis)
36 const arm_compute::TensorInfo aclInputInfo = BuildArmComputeTensorInfo(input);
38 size_t numOutputs = outputs.size();
40 std::vector<arm_compute::TensorInfo> aclOutputs;
41 aclOutputs.reserve(numOutputs);
43 std::vector<arm_compute::ITensorInfo*> aclOutputPtr;
44 aclOutputPtr.reserve(numOutputs);
46 for (
size_t i = 0u; i < outputs.size(); ++i)
48 aclOutputs.emplace_back(BuildArmComputeTensorInfo(outputs[i]));
49 aclOutputPtr.emplace_back(&aclOutputs.back());
53 return arm_compute::NESplit::validate(&aclInputInfo, aclOutputPtr, aclAxis);
59 bool allOutputsAreSubtensors =
true;
64 if (output && !output->GetParent())
67 allOutputsAreSubtensors =
false;
72 if (allOutputsAreSubtensors)
78 arm_compute::ITensor& input = PolymorphicDowncast<IAclTensorHandle*>(
m_Data.
m_Inputs[0])->GetTensor();
80 std::vector<arm_compute::ITensor *> aclOutputs;
83 arm_compute::ITensor& aclOutput = PolymorphicPointerDowncast<IAclTensorHandle>(output)->GetTensor();
84 aclOutputs.emplace_back(&aclOutput);
88 std::unique_ptr<arm_compute::NESplit> layer(
new arm_compute::NESplit());
92 if (splitAxis.size() != 1)
98 layer->configure(&input, aclOutputs, aclAxis);
102 m_Layer.reset(layer.release());
arm_compute::Status NeonSplitterWorkloadValidate(const TensorInfo &input, const std::vector< std::reference_wrapper< TensorInfo >> &outputs, unsigned int splitAxis)
uint32_t GetNumDimensions() const
Get the number of dimensions.
const SplitterQueueDescriptor m_Data
#define ARMNN_SCOPED_PROFILING_EVENT_NEON(name)
void Execute() const override
NeonSplitterWorkload(const SplitterQueueDescriptor &descriptor, const WorkloadInfo &info)
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
Contains information about inputs and outputs to a layer.
std::vector< ITensorHandle * > m_Inputs
unsigned int GetNumDimensions() const