diff options
Diffstat (limited to 'include')
-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 a2aa9b0a98..7bf41c1ac8 100644 --- a/include/armnnUtils/TensorUtils.hpp +++ b/include/armnnUtils/TensorUtils.hpp @@ -48,6 +48,8 @@ armnn::TensorInfo ReduceDims(const armnn::TensorInfo& tensorInfo, unsigned int d armnn::TensorShape ExpandDims(const armnn::TensorShape& tensorShape, int axis); +armnn::TensorShape ExpandDimsToRank(const armnn::TensorShape& tensorShape, unsigned int rank); + std::vector<unsigned int> SqueezeDims(const armnn::TensorShape& tensorShape); unsigned int GetNumElementsBetween(const armnn::TensorShape& shape, |