diff options
Diffstat (limited to 'src/core/Utils.cpp')
| -rw-r--r-- | src/core/Utils.cpp | 13 |
1 files changed, 9 insertions, 4 deletions
diff --git a/src/core/Utils.cpp b/src/core/Utils.cpp index bca5a31914..fb86d78cb7 100644 --- a/src/core/Utils.cpp +++ b/src/core/Utils.cpp @@ -510,12 +510,15 @@ void arm_compute::print_consecutive_elements(std::ostream &s, DataType dt, const case DataType::S32: print_consecutive_elements_impl<int32_t>(s, reinterpret_cast<const int32_t *>(ptr), n, stream_width, element_delim); break; - case DataType::F32: - print_consecutive_elements_impl<float>(s, reinterpret_cast<const float *>(ptr), n, stream_width, element_delim); + case DataType::BFLOAT16: + print_consecutive_elements_impl<bfloat16>(s, reinterpret_cast<const bfloat16 *>(ptr), n, stream_width, element_delim); break; case DataType::F16: print_consecutive_elements_impl<half>(s, reinterpret_cast<const half *>(ptr), n, stream_width, element_delim); break; + case DataType::F32: + print_consecutive_elements_impl<float>(s, reinterpret_cast<const float *>(ptr), n, stream_width, element_delim); + break; default: ARM_COMPUTE_ERROR("Undefined element size for given data type"); } @@ -542,10 +545,12 @@ int arm_compute::max_consecutive_elements_display_width(std::ostream &s, DataTyp return max_consecutive_elements_display_width_impl<uint32_t>(s, reinterpret_cast<const uint32_t *>(ptr), n); case DataType::S32: return max_consecutive_elements_display_width_impl<int32_t>(s, reinterpret_cast<const int32_t *>(ptr), n); - case DataType::F32: - return max_consecutive_elements_display_width_impl<float>(s, reinterpret_cast<const float *>(ptr), n); + case DataType::BFLOAT16: + return max_consecutive_elements_display_width_impl<bfloat16>(s, reinterpret_cast<const bfloat16 *>(ptr), n); case DataType::F16: return max_consecutive_elements_display_width_impl<half>(s, reinterpret_cast<const half *>(ptr), n); + case DataType::F32: + return max_consecutive_elements_display_width_impl<float>(s, reinterpret_cast<const float *>(ptr), n); default: ARM_COMPUTE_ERROR("Undefined element size for given data type"); } |