1 // 2 // Copyright © 2017 Arm Ltd. All rights reserved. 3 // SPDX-License-Identifier: MIT 4 // 5 6 #pragma once 7 8 #include "BaseIterator.hpp" 9 #include "Decoders.hpp" 10 #include "Encoders.hpp" 11 12 #include <armnn/Types.hpp> 13 14 #include <armnnUtils/DataLayoutIndexed.hpp> 15 16 #include <armnn/backends/Workload.hpp> 17 #include <armnn/backends/WorkloadData.hpp> 18 19 namespace armnn 20 { 21 22 void BatchToSpaceNd(const armnnUtils::DataLayoutIndexed& dataLayout, 23 const TensorInfo& inputTensorInfo, 24 const TensorInfo& outputTensorInfo, 25 const std::vector<unsigned int>& blockShape, 26 const std::vector<std::pair<unsigned int, unsigned int>>& cropsData, 27 Decoder<float>& inputDecoder, 28 Encoder<float>& outputEncoder); 29 } // namespace armnn 30