commit 0280785b00cd84f14249d3545a15b93627acf57b
author Narumol Prangnawarat <narumol.prangnawarat@arm.com> Wed Sep 11 16:43:09 2019 +0100
committer Áron Virginás-Tar <aron.virginas-tar@arm.com> Thu Sep 12 12:05:42 2019 +0000
tree aebfb054051d130b8dca735e3cd6b73934cda816
parent b0b802f979a647ace8676dfdda53f06416d3700b

IVGCVSW-3663 Add utility function to expand tensor dimension

Signed-off-by: Narumol Prangnawarat <narumol.prangnawarat@arm.com>
Change-Id: I54d2416be305db7d3fc15fb3123694ad6740b27d

src/armnnUtils/TensorUtils.hpp

diff --git a/src/armnnUtils/TensorUtils.hpp b/src/armnnUtils/TensorUtils.hpp
index c273b49..03b1c8a 100644
--- a/src/armnnUtils/TensorUtils.hpp
+++ b/src/armnnUtils/TensorUtils.hpp
@@ -24,4 +24,6 @@
 
 std::pair<float, float> FindMinMax(armnn::ITensorHandle* tensorHandle);
 
+armnn::TensorShape ExpandDims(const armnn::TensorShape& tensorShape, int axis);
+
 } // namespace armnnUtils