12 #include <boost/numeric/conversion/cast.hpp> 13 #include <boost/polymorphic_pointer_cast.hpp> 17 using namespace armcomputetensorutils;
21 int CalcAxis(
const unsigned int axis,
const unsigned int inputDimensions)
23 const int intAxis = boost::numeric_cast<
int>(axis);
24 return boost::numeric_cast<
int>(inputDimensions) - intAxis;
32 std::vector<arm_compute::TensorInfo> aclInputs;
36 aclInputs.emplace_back(aclInputInfo);
39 std::vector<arm_compute::ITensorInfo*> aclInputPtrs;
40 for (arm_compute::ITensorInfo& input : aclInputs)
42 aclInputPtrs.emplace_back(&input);
45 const arm_compute::TensorInfo aclOutputInfo = BuildArmComputeTensorInfo(output);
47 return arm_compute::NEStackLayer::validate(aclInputPtrs, aclAxis, &aclOutputInfo);
53 std::vector<arm_compute::ITensor*> aclInputs;
56 arm_compute::ITensor& aclInput = boost::polymorphic_pointer_downcast<IAclTensorHandle>(input)->GetTensor();
57 aclInputs.emplace_back(&aclInput);
59 arm_compute::ITensor& output = boost::polymorphic_pointer_downcast<IAclTensorHandle>(
62 m_Layer.reset(
new arm_compute::NEStackLayer());
64 m_Layer->configure(aclInputs, aclAxis, &output);
uint32_t m_Axis
0-based axis along which to stack the input tensors.
unsigned int GetNumDimensions() const
#define ARMNN_SCOPED_PROFILING_EVENT_NEON(name)
LayerDescriptor m_Parameters
arm_compute::Status NeonStackWorkloadValidate(const std::vector< const TensorInfo *> &inputs, const TensorInfo &output, const StackDescriptor &descriptor)
TensorShape m_InputShape
Required shape of all input tensors.
void Execute() const override
const StackQueueDescriptor m_Data
A StackDescriptor for the StackLayer.
std::vector< ITensorHandle * > m_Outputs
std::vector< ITensorHandle * > m_Inputs
NeonStackWorkload(const StackQueueDescriptor &descriptor, const WorkloadInfo &info)