diff options
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 |