aboutsummaryrefslogtreecommitdiff
path: root/src/armnnUtils/PrototxtConversions.hpp
diff options
context:
space:
mode:
authorSang-Hoon Park <sang-hoon.park@arm.com>2020-02-18 11:27:35 +0000
committerSang-Hoon Park <sang-hoon.park@arm.com>2020-02-24 13:29:57 +0000
commitdd3f71b64072c44cec65a7a883d0c3a29659645c (patch)
treed48b0508811e88d47b4b95a1777742312b2ba4db /src/armnnUtils/PrototxtConversions.hpp
parentcf34f510b891b87b4c7be5edb30272b36cab7b51 (diff)
downloadarmnn-dd3f71b64072c44cec65a7a883d0c3a29659645c.tar.gz
COMPMID-3060: Add TF Parser support for Transpose
Signed-off-by: Sang-Hoon Park <sang-hoon.park@arm.com> Change-Id: I9661787071554b38c5b0ab3c98431f3863b98520
Diffstat (limited to 'src/armnnUtils/PrototxtConversions.hpp')
-rw-r--r--src/armnnUtils/PrototxtConversions.hpp8
1 files changed, 8 insertions, 0 deletions
diff --git a/src/armnnUtils/PrototxtConversions.hpp b/src/armnnUtils/PrototxtConversions.hpp
index c90af9edff..fb066729fe 100644
--- a/src/armnnUtils/PrototxtConversions.hpp
+++ b/src/armnnUtils/PrototxtConversions.hpp
@@ -7,10 +7,18 @@
#include <string>
+namespace armnn
+{
+class TensorShape;
+} // namespace armnn
+
namespace armnnUtils
{
/// Converts an int value into the Prototxt octal representation
std::string ConvertInt32ToOctalString(int value);
+/// Converts an TensorShape into Prototxt representation
+std::string ConvertTensorShapeToString(const armnn::TensorShape& shape);
+
} // namespace armnnUtils