1 // 2 // Copyright © 2017 Arm Ltd. All rights reserved. 3 // SPDX-License-Identifier: MIT 4 // 5 6 #pragma once 7 8 #include <armnn/DescriptorsFwd.hpp> 9 #include <armnn/TensorFwd.hpp> 10 11 #include <set> 12 13 namespace armnnUtils 14 { 15 16 void ProcessConcatInputTensorInfo(armnn::TensorInfo& inputTensorInfo, 17 armnn::OriginsDescriptor& concatDescriptor, 18 const unsigned int& concatAxis, 19 unsigned int inputIndex, 20 unsigned int& mergeDimOrigin); 21 22 /// Creates a tensor info after reducing the dimensions mentioned in axisData. 23 void CalculateReducedOutputTensoInfo(const armnn::TensorInfo& inputTensorInfo, 24 const std::set<unsigned int>& axisSet, 25 bool keepDims, 26 armnn::TensorInfo& outputTensorInfo); 27 28 /// Create output tensor info for a StridedSlice operator 29 void CalculateStridedSliceOutputTensorInfo(const armnn::TensorInfo& inputTensorInfo, 30 const armnn::StridedSliceDescriptor& desc, 31 armnn::TensorInfo& outputTensorInfo); 32 33 } // namespace armnnUtils 34