18 using namespace armcomputetensorutils;
24 const arm_compute::TensorInfo aclInputInfo = BuildArmComputeTensorInfo(input, descriptor.
m_DataLayout);
25 const arm_compute::TensorInfo aclOutputInfo = BuildArmComputeTensorInfo(output, descriptor.
m_DataLayout);
28 int32_t blockHeight = armnn::numeric_cast<int32_t>(descriptor.
m_BlockShape[0]);
29 int32_t blockWidth = armnn::numeric_cast<int32_t>(descriptor.
m_BlockShape[1]);
31 const arm_compute::Status aclStatus = arm_compute::NEBatchToSpaceLayer::validate(&aclInputInfo,
50 arm_compute::ITensor& input =
51 armnn::PolymorphicPointerDowncast<IAclTensorHandle>(
m_Data.
m_Inputs[0])->GetTensor();
52 arm_compute::ITensor& output =
53 armnn::PolymorphicPointerDowncast<IAclTensorHandle>(
m_Data.
m_Outputs[0])->GetTensor();
56 input.info()->set_data_layout(aclDataLayout);
57 output.info()->set_data_layout(aclDataLayout);
63 m_Layer.reset(
new arm_compute::NEBatchToSpaceLayer());
64 m_Layer->configure(&input, blockWidth, blockHeight, &output);