diff options
author | Narumol Prangnawarat <narumol.prangnawarat@arm.com> | 2019-09-11 16:43:09 +0100 |
---|---|---|
committer | Áron Virginás-Tar <aron.virginas-tar@arm.com> | 2019-09-12 12:05:42 +0000 |
commit | 0280785b00cd84f14249d3545a15b93627acf57b (patch) | |
tree | aebfb054051d130b8dca735e3cd6b73934cda816 /src/armnnUtils/TensorUtils.hpp | |
parent | b0b802f979a647ace8676dfdda53f06416d3700b (diff) | |
download | armnn-0280785b00cd84f14249d3545a15b93627acf57b.tar.gz |
IVGCVSW-3663 Add utility function to expand tensor dimension
Signed-off-by: Narumol Prangnawarat <narumol.prangnawarat@arm.com>
Change-Id: I54d2416be305db7d3fc15fb3123694ad6740b27d
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 |