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::CropInfo cropInfo = BuildArmComputeCropInfo(descriptor);
33 const arm_compute::Status aclStatus = arm_compute::NEBatchToSpaceLayer::validate(&aclInputInfo,
53 arm_compute::ITensor& input =
54 armnn::PolymorphicPointerDowncast<IAclTensorHandle>(
m_Data.
m_Inputs[0])->GetTensor();
55 arm_compute::ITensor& output =
56 armnn::PolymorphicPointerDowncast<IAclTensorHandle>(
m_Data.
m_Outputs[0])->GetTensor();
59 input.info()->set_data_layout(aclDataLayout);
60 output.info()->set_data_layout(aclDataLayout);
66 const arm_compute::CropInfo cropInfo = BuildArmComputeCropInfo(descriptor.
m_Parameters);
68 m_Layer.reset(
new arm_compute::NEBatchToSpaceLayer());
69 m_Layer->configure(&input, blockWidth, blockHeight, &output, cropInfo);