1 // 2 // Copyright © 2019 Arm Ltd. All rights reserved. 3 // SPDX-License-Identifier: MIT 4 // 5 6 #pragma once 7 8 #include <armnn/TypesUtils.hpp> 9 10 namespace armnnUtils 11 { 12 armnn::TensorShape GetTensorShape(unsigned int numberOfBatches, 13 unsigned int numberOfChannels, 14 unsigned int height, 15 unsigned int width, 16 const armnn::DataLayout dataLayout); 17 18 armnn::TensorInfo GetTensorInfo(unsigned int numberOfBatches, 19 unsigned int numberOfChannels, 20 unsigned int height, 21 unsigned int width, 22 const armnn::DataLayout dataLayout, 23 const armnn::DataType dataType); 24 25 std::pair<float, float> FindMinMax(armnn::ITensorHandle* tensorHandle); 26 27 armnn::TensorShape ExpandDims(const armnn::TensorShape& tensorShape, int axis); 28 29 unsigned int GetNumElementsBetween(const armnn::TensorShape& shape, 30 unsigned int firstAxisInclusive, 31 unsigned int lastAxisExclusive); 32 33 unsigned int GetUnsignedAxis(const unsigned int inputDimension, const int axis); 34 35 unsigned int GetNumElementsAfter(const armnn::TensorShape& shape, unsigned int axis); 36 37 std::pair<unsigned int, std::vector<float>> GetPerAxisParams(const armnn::TensorInfo& info); 38 39 } // namespace armnnUtils 40