diff options
author | Matteo Martincigh <matteo.martincigh@arm.com> | 2019-11-28 11:35:47 +0000 |
---|---|---|
committer | Matteo Martincigh <matteo.martincigh@arm.com> | 2019-11-29 10:30:58 +0000 |
commit | e011d20f279e6e67e899bc6930b0266fc357bc1c (patch) | |
tree | 4a190b08df3c63c4d540506491d31bee7c8dcd46 /src/armnnUtils/TensorUtils.hpp | |
parent | 2e259276fba9fa5c6c2e146de3b26e3d6c6cccc6 (diff) | |
download | armnn-e011d20f279e6e67e899bc6930b0266fc357bc1c.tar.gz |
IVGCVSW-4209 Create a public API for the ArmNN Utils
* Moved the relevant armnnUtils headers to the new location:
include/armnnUtils
* Update the header usage throughout the source code
!android-nn-driver:2387
Signed-off-by: Matteo Martincigh <matteo.martincigh@arm.com>
Change-Id: I2ba15cebcacafad2b5a1a7b9c3312ffc585e09d6
Diffstat (limited to 'src/armnnUtils/TensorUtils.hpp')
-rw-r--r-- | src/armnnUtils/TensorUtils.hpp | 41 |
1 files changed, 0 insertions, 41 deletions
diff --git a/src/armnnUtils/TensorUtils.hpp b/src/armnnUtils/TensorUtils.hpp deleted file mode 100644 index b67431d757..0000000000 --- a/src/armnnUtils/TensorUtils.hpp +++ /dev/null @@ -1,41 +0,0 @@ -// -// Copyright © 2017 Arm Ltd. All rights reserved. -// SPDX-License-Identifier: MIT -// - -#pragma once - -#include <armnn/TypesUtils.hpp> - -#include <boost/assert.hpp> - -namespace armnnUtils -{ -armnn::TensorShape GetTensorShape(unsigned int numberOfBatches, - unsigned int numberOfChannels, - unsigned int height, - unsigned int width, - const armnn::DataLayout dataLayout); - -armnn::TensorInfo GetTensorInfo(unsigned int numberOfBatches, - unsigned int numberOfChannels, - unsigned int height, - unsigned int width, - const armnn::DataLayout dataLayout, - const armnn::DataType dataType); - -std::pair<float, float> FindMinMax(armnn::ITensorHandle* tensorHandle); - -armnn::TensorShape ExpandDims(const armnn::TensorShape& tensorShape, int axis); - -unsigned int GetNumElementsBetween(const armnn::TensorShape& shape, - unsigned int firstAxisInclusive, - unsigned int lastAxisExclusive); - -unsigned int GetUnsignedAxis(const unsigned int inputDimension, const int axis); - -unsigned int GetNumElementsAfter(const armnn::TensorShape& shape, unsigned int axis); - -std::pair<unsigned int, std::vector<float>> GetPerAxisParams(const armnn::TensorInfo& info); - -} // namespace armnnUtils |