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/Descriptors.hpp> 13 #include "armnn/Tensor.hpp" 14 15 namespace armnn 16 { 17 18 void SpaceToBatchNd(const TensorInfo& inputInfo, 19 const TensorInfo& outputInfo, 20 const SpaceToBatchNdDescriptor& params, 21 Decoder<float>& inputData, 22 Encoder<float>& outputData); 23 24 } //namespace armnn 25