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.cpp | 14 ++++++++++++++ 1 file changed, 14 insertions(+) (limited to 'src/armnnUtils/PrototxtConversions.cpp') diff --git a/src/armnnUtils/PrototxtConversions.cpp b/src/armnnUtils/PrototxtConversions.cpp index 1a6c053355..cc86321a27 100644 --- a/src/armnnUtils/PrototxtConversions.cpp +++ b/src/armnnUtils/PrototxtConversions.cpp @@ -4,6 +4,7 @@ // #include "PrototxtConversions.hpp" +#include "armnn/Tensor.hpp" #include @@ -29,4 +30,17 @@ std::string ConvertInt32ToOctalString(int value) return returnString; } +/// Converts an TensorShape into Prototxt representation +std::string ConvertTensorShapeToString(const armnn::TensorShape& shape) +{ + std::stringstream ss; + for (unsigned int i = 0 ; i < shape.GetNumDimensions() ; i++) + { + ss << "dim {\n"; + ss << "size: " << std::to_string(shape[i]) << "\n"; + ss << "}\n"; + } + return ss.str(); + +} } // namespace armnnUtils -- cgit v1.2.1