18 using namespace armcomputetensorutils;
25 const arm_compute::TensorInfo aclInputInfo = BuildArmComputeTensorInfo(input, 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::TensorInfo aclOutputInfo = BuildArmComputeTensorInfo(output, dataLayout);
33 const arm_compute::CropInfo cropInfo = BuildArmComputeCropInfo(descriptor);
35 const arm_compute::Status aclStatus = arm_compute::CLBatchToSpaceLayer::validate(&aclInputInfo,
45 const arm_compute::CLCompileContext& clCompileContext)
59 input.info()->set_data_layout(aclDataLayout);
66 output.info()->set_data_layout(aclDataLayout);
68 const arm_compute::CropInfo cropInfo = BuildArmComputeCropInfo(descriptor.
m_Parameters);
72 m_Layer.configure(clCompileContext, &input, blockWidth, blockHeight, &output, cropInfo);