17 using namespace armcomputetensorutils;
32 std::vector<arm_compute::TensorInfo> aclInputs;
36 aclInputs.emplace_back(aclInputInfo);
38 const arm_compute::TensorInfo aclOutputInfo = BuildArmComputeTensorInfo(output);
39 std::vector<const arm_compute::ITensorInfo*> aclInputPtrs;
40 for (arm_compute::ITensorInfo& input : aclInputs)
42 aclInputPtrs.emplace_back(&input);
45 size_t aclAxis = CalcAxis(descriptor);
46 return arm_compute::NEConcatenateLayer::validate(aclInputPtrs, &aclOutputInfo, aclAxis);
53 bool allInputsAreSubtensors =
true;
56 for (
auto input : descriptor.
m_Inputs)
58 if (!input->GetParent())
61 allInputsAreSubtensors =
false;
66 if (allInputsAreSubtensors)
72 std::vector<const arm_compute::ITensor *> aclInputs;
75 arm_compute::ITensor& aclInput = armnn::PolymorphicPointerDowncast<IAclTensorHandle>(input)->GetTensor();
76 aclInputs.emplace_back(&aclInput);
78 arm_compute::ITensor& output = armnn::PolymorphicPointerDowncast<IAclTensorHandle>(
82 m_Layer.reset(
new arm_compute::NEConcatenateLayer());
86 m_Layer->configure(aclInputs, &output, aclAxis);
const ConcatQueueDescriptor m_Data
#define ARMNN_SCOPED_PROFILING_EVENT_NEON(name)
NeonConcatWorkload(const ConcatQueueDescriptor &descriptor, const WorkloadInfo &info)
Copyright (c) 2020 ARM Limited.
LayerDescriptor m_Parameters
An OriginsDescriptor for the ConcatLayer.
void Execute() const override
arm_compute::Status NeonConcatWorkloadValidate(const std::vector< const TensorInfo *> &inputs, const TensorInfo &output, const OriginsDescriptor &descriptor)
std::vector< ITensorHandle * > m_Outputs
uint32_t GetNumDimensions() const
Get the number of dimensions.
Contains information about inputs and outputs to a layer.
std::vector< ITensorHandle * > m_Inputs
unsigned int GetConcatAxis() const
Get the concatenation axis value.