13 #include <arm_compute/core/Types.h> 14 #include <arm_compute/runtime/CL/functions/CLConcatenateLayer.h> 18 using namespace armcomputetensorutils;
22 size_t CalcAxis(
const OriginsDescriptor& desc)
24 return (desc.GetNumDimensions() - desc.GetConcatAxis()) - 1;
32 std::vector<arm_compute::TensorInfo> aclInputs;
36 aclInputs.emplace_back(aclInputInfo);
38 const arm_compute::TensorInfo aclOutputInfo = BuildArmComputeTensorInfo(output);
39 std::vector<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::CLConcatenateLayer::validate(aclInputPtrs, &aclOutputInfo, aclAxis);
52 bool allInputsAreSubtensors =
true;
55 for (
auto input : descriptor.
m_Inputs)
57 if (!input->GetParent())
60 allInputsAreSubtensors =
false;
65 if (allInputsAreSubtensors)
71 std::vector<arm_compute::ICLTensor *> aclInputs;
74 arm_compute::ICLTensor& aclInput = armnn::PolymorphicPointerDowncast<IClTensorHandle>(input)->GetTensor();
75 aclInputs.emplace_back(&aclInput);
78 arm_compute::ICLTensor& output =
79 armnn::PolymorphicPointerDowncast<IClTensorHandle>(
m_Data.
m_Outputs[0])->GetTensor();
82 auto layer = std::make_unique<arm_compute::CLConcatenateLayer>();
86 layer->configure(aclInputs, &output, aclAxis);
90 m_Layer = std::move(layer);
#define ARMNN_SCOPED_PROFILING_EVENT_CL(name)
const ConcatQueueDescriptor m_Data
Copyright (c) 2020 ARM Limited.
LayerDescriptor m_Parameters
An OriginsDescriptor for the ConcatLayer.
void Execute() const override
std::vector< ITensorHandle * > m_Outputs
ClConcatWorkload(const ConcatQueueDescriptor &descriptor, const WorkloadInfo &info)
arm_compute::Status ClConcatWorkloadValidate(const std::vector< const TensorInfo *> &inputs, const TensorInfo &output, const OriginsDescriptor &descriptor)
Contains information about inputs and outputs to a layer.
std::vector< ITensorHandle * > m_Inputs