16 using namespace armcomputetensorutils;
20 int CalcAxis(
const unsigned int axis,
const unsigned int inputDimensions)
22 const int intAxis = armnn::numeric_cast<int>(axis);
23 return armnn::numeric_cast<int>(inputDimensions) - intAxis;
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);
58 std::vector<arm_compute::ITensor*> aclInputs;
61 arm_compute::ITensor& aclInput = PolymorphicPointerDowncast<IAclTensorHandle>(input)->GetTensor();
62 aclInputs.emplace_back(&aclInput);
64 arm_compute::ITensor& output = PolymorphicPointerDowncast<IAclTensorHandle>(
67 m_Layer.reset(
new arm_compute::NEStackLayer());
69 m_Layer->configure(aclInputs, aclAxis, &output);