diff options
Diffstat (limited to 'arm_compute')
-rw-r--r-- | arm_compute/core/Utils.h | 6 | ||||
-rw-r--r-- | arm_compute/runtime/TensorAllocator.h | 2 |
2 files changed, 7 insertions, 1 deletions
diff --git a/arm_compute/core/Utils.h b/arm_compute/core/Utils.h index a37559d269..b0e26328ed 100644 --- a/arm_compute/core/Utils.h +++ b/arm_compute/core/Utils.h @@ -32,6 +32,7 @@ #include <algorithm> #include <cstdint> #include <cstdlib> +#include <iomanip> #include <numeric> #include <sstream> #include <string> @@ -1150,6 +1151,8 @@ template <typename T> void print_consecutive_elements_impl(std::ostream &s, const T *ptr, unsigned int n, int stream_width = 0, const std::string &element_delim = " ") { using print_type = typename std::conditional<std::is_floating_point<T>::value, T, int>::type; + std::ios stream_status(nullptr); + stream_status.copyfmt(s); for(unsigned int i = 0; i < n; ++i) { @@ -1169,6 +1172,9 @@ void print_consecutive_elements_impl(std::ostream &s, const T *ptr, unsigned int s << std::right << static_cast<print_type>(ptr[i]) << element_delim; } } + + // Restore output stream flags + s.copyfmt(stream_status); } /** Identify the maximum width of n consecutive elements. diff --git a/arm_compute/runtime/TensorAllocator.h b/arm_compute/runtime/TensorAllocator.h index bf574fa0c9..c2a370fb58 100644 --- a/arm_compute/runtime/TensorAllocator.h +++ b/arm_compute/runtime/TensorAllocator.h @@ -72,7 +72,7 @@ public: * @param[in] coords The starting coordinates of the new tensor inside the parent tensor. * @param[in] sub_info The new tensor information (e.g. shape etc) */ - void init(const TensorAllocator &allocator, const Coordinates &coords, TensorInfo sub_info); + void init(const TensorAllocator &allocator, const Coordinates &coords, TensorInfo &sub_info); /** Returns the pointer to the allocated data. * |