diff options
Diffstat (limited to 'include')
-rw-r--r-- | include/armnnUtils/TensorUtils.hpp | 7 |
1 files changed, 6 insertions, 1 deletions
diff --git a/include/armnnUtils/TensorUtils.hpp b/include/armnnUtils/TensorUtils.hpp index 2d6ec2fea4..a2aa9b0a98 100644 --- a/include/armnnUtils/TensorUtils.hpp +++ b/include/armnnUtils/TensorUtils.hpp @@ -1,5 +1,5 @@ // -// Copyright © 2019,2022 Arm Ltd and Contributors. All rights reserved. +// Copyright © 2018-2023 Arm Ltd and Contributors. All rights reserved. // SPDX-License-Identifier: MIT // @@ -8,6 +8,7 @@ #include <armnn/TypesUtils.hpp> #include <armnn/Tensor.hpp> #include <armnn/Types.hpp> +#include <armnnUtils/TensorUtils.hpp> #include <utility> #include <vector> @@ -41,6 +42,10 @@ armnn::TensorInfo GetTensorInfo(unsigned int numberOfBatches, std::pair<float, float> FindMinMax(armnn::ITensorHandle* tensorHandle); +armnn::TensorShape ReduceDims(const armnn::TensorShape& tensorInfo, unsigned int dimensions); + +armnn::TensorInfo ReduceDims(const armnn::TensorInfo& tensorInfo, unsigned int dimensions); + armnn::TensorShape ExpandDims(const armnn::TensorShape& tensorShape, int axis); std::vector<unsigned int> SqueezeDims(const armnn::TensorShape& tensorShape); |