From dd3f71b64072c44cec65a7a883d0c3a29659645c Mon Sep 17 00:00:00 2001 From: Sang-Hoon Park Date: Tue, 18 Feb 2020 11:27:35 +0000 Subject: COMPMID-3060: Add TF Parser support for Transpose Signed-off-by: Sang-Hoon Park Change-Id: I9661787071554b38c5b0ab3c98431f3863b98520 --- src/armnnUtils/PrototxtConversions.hpp | 8 ++++++++ 1 file changed, 8 insertions(+) (limited to 'src/armnnUtils/PrototxtConversions.hpp') 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 +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 -- cgit v1.2.1