diff options
Diffstat (limited to 'src/armnnUtils/TensorUtils.hpp')
-rw-r--r-- | src/armnnUtils/TensorUtils.hpp | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/src/armnnUtils/TensorUtils.hpp b/src/armnnUtils/TensorUtils.hpp index c273b497b3..03b1c8a2df 100644 --- a/src/armnnUtils/TensorUtils.hpp +++ b/src/armnnUtils/TensorUtils.hpp @@ -24,4 +24,6 @@ armnn::TensorInfo GetTensorInfo(unsigned int numberOfBatches, std::pair<float, float> FindMinMax(armnn::ITensorHandle* tensorHandle); +armnn::TensorShape ExpandDims(const armnn::TensorShape& tensorShape, int axis); + } // namespace armnnUtils |