16 using namespace armcomputetensorutils;
20 int CalcAxis(
const unsigned int axis,
const unsigned int inputDimensions)
31 std::vector<arm_compute::TensorInfo> aclInputs;
35 aclInputs.emplace_back(aclInputInfo);
38 std::vector<arm_compute::ITensorInfo*> aclInputPtrs;
39 for (arm_compute::ITensorInfo& input : aclInputs)
41 aclInputPtrs.emplace_back(&input);
44 const arm_compute::TensorInfo aclOutputInfo = BuildArmComputeTensorInfo(output);
46 return arm_compute::NEStackLayer::validate(aclInputPtrs, aclAxis, &aclOutputInfo);
52 std::vector<arm_compute::ITensor*> aclInputs;
55 arm_compute::ITensor& aclInput = PolymorphicPointerDowncast<IAclTensorHandle>(input)->GetTensor();
56 aclInputs.emplace_back(&aclInput);
58 arm_compute::ITensor& output = PolymorphicPointerDowncast<IAclTensorHandle>(
61 m_Layer.reset(
new arm_compute::NEStackLayer());
63 m_Layer->configure(aclInputs, aclAxis, &output);
uint32_t m_Axis
0-based axis along which to stack the input tensors.
TensorShape m_InputShape
Required shape of all input tensors.
#define ARMNN_SCOPED_PROFILING_EVENT_NEON(name)
void Execute() const override
Copyright (c) 2021 ARM Limited and Contributors.
LayerDescriptor m_Parameters
A StackDescriptor for the StackLayer.
arm_compute::Status NeonStackWorkloadValidate(const std::vector< const TensorInfo *> &inputs, const TensorInfo &output, const StackDescriptor &descriptor)
NeonStackWorkload(const StackQueueDescriptor &descriptor, const WorkloadInfo &info)
StackQueueDescriptor m_Data
std::vector< ITensorHandle * > m_Outputs
unsigned int GetNumDimensions() const
Function that returns the tensor rank.
std::enable_if_t< std::is_unsigned< Source >::value &&std::is_unsigned< Dest >::value, Dest > numeric_cast(Source source)
Contains information about inputs and outputs to a layer.
std::vector< ITensorHandle * > m_Inputs