diff options
author | Sang-Hoon Park <sang-hoon.park@arm.com> | 2020-02-18 11:27:35 +0000 |
---|---|---|
committer | Sang-Hoon Park <sang-hoon.park@arm.com> | 2020-02-24 13:29:57 +0000 |
commit | dd3f71b64072c44cec65a7a883d0c3a29659645c (patch) | |
tree | d48b0508811e88d47b4b95a1777742312b2ba4db /src/armnnUtils/PrototxtConversions.hpp | |
parent | cf34f510b891b87b4c7be5edb30272b36cab7b51 (diff) | |
download | armnn-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.hpp | 8 |
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 |