diff options
Diffstat (limited to 'utils/TypePrinter.h')
-rw-r--r-- | utils/TypePrinter.h | 12 |
1 files changed, 12 insertions, 0 deletions
diff --git a/utils/TypePrinter.h b/utils/TypePrinter.h index 4fd3c0c027..d956d9ca01 100644 --- a/utils/TypePrinter.h +++ b/utils/TypePrinter.h @@ -27,6 +27,7 @@ #include "arm_compute/core/Dimensions.h" #include "arm_compute/core/Error.h" #include "arm_compute/core/Strides.h" +#include "arm_compute/core/TensorInfo.h" #include "arm_compute/core/Types.h" #include "tests/Types.h" @@ -553,6 +554,17 @@ inline ::std::ostream &operator<<(::std::ostream &os, const InterpolationPolicy return os; } +/** Formatted output of the TensorInfo type. */ +inline std::string to_string(const TensorInfo &info) +{ + std::stringstream str; + str << "{Shape=" << info.tensor_shape() << "," + << "Type=" << info.data_type() << "," + << "Channels=" << info.num_channels() << "," + << "FixedPointPos=" << info.fixed_point_position() << "}"; + return str.str(); +} + //FIXME: Check why this doesn't work and the TensorShape and Coordinates overload are needed template <typename T> inline std::string to_string(const Dimensions<T> &dimensions) |