diff options
Diffstat (limited to 'include/armnnUtils/TensorUtils.hpp')
| -rw-r--r-- | include/armnnUtils/TensorUtils.hpp | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/include/armnnUtils/TensorUtils.hpp b/include/armnnUtils/TensorUtils.hpp index 6a975255c6..fc2f51061c 100644 --- a/include/armnnUtils/TensorUtils.hpp +++ b/include/armnnUtils/TensorUtils.hpp @@ -34,6 +34,8 @@ std::pair<float, float> FindMinMax(armnn::ITensorHandle* tensorHandle); armnn::TensorShape ExpandDims(const armnn::TensorShape& tensorShape, int axis); +std::vector<unsigned int> SqueezeDims(const armnn::TensorShape& tensorShape); + unsigned int GetNumElementsBetween(const armnn::TensorShape& shape, unsigned int firstAxisInclusive, unsigned int lastAxisExclusive); |