!3679 fix bug for the decimal place of float16

Merge pull request !3679 from huangbingjian/report_float16
5 years ago · 21810a2e69
parent ab0e39eabf 189ccb86ab
commit 21810a2e69
1 changed files with 7 additions and 2 deletions
--- a/mindspore/core/ir/tensor.cc
+++ b/mindspore/core/ir/tensor.cc
@ -233,8 +233,13 @@ class TensorDataImpl : public TensorData {
        if (isScalar) {
          ss << value;
        } else {
-          ss << std::setw(15) << std::setprecision(8) << std::setiosflags(std::ios::scientific | std::ios::right)
-             << value;
+          if (std::is_same<T, float16>::value) {
+            ss << std::setw(11) << std::setprecision(4) << std::setiosflags(std::ios::scientific | std::ios::right)
+               << value;
+          } else {
+            ss << std::setw(15) << std::setprecision(8) << std::setiosflags(std::ios::scientific | std::ios::right)
+               << value;
+          }
        }
        linefeedThreshold = kThreshold1DFloat;
      } else if (type == kNumberTypeBool) {