diff options
author | A. Unique TensorFlower <gardener@tensorflow.org> | 2018-01-16 15:52:12 -0800 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-01-16 15:55:47 -0800 |
commit | ccbd14b741e6efbe51769f0f1b9cb3719c42c23b (patch) | |
tree | bcc79094e48982780b084cd19110d903186eea8f /tensorflow/core/lib/strings/strcat.h | |
parent | 287fe4f2404a7b69ffce89cc41ff3f049ca7a08b (diff) |
Enable bfloat16 for CPU kernels
PiperOrigin-RevId: 182124532
Diffstat (limited to 'tensorflow/core/lib/strings/strcat.h')
-rw-r--r-- | tensorflow/core/lib/strings/strcat.h | 3 |
1 files changed, 3 insertions, 0 deletions
diff --git a/tensorflow/core/lib/strings/strcat.h b/tensorflow/core/lib/strings/strcat.h index 8e35549ed4..5835b0101d 100644 --- a/tensorflow/core/lib/strings/strcat.h +++ b/tensorflow/core/lib/strings/strcat.h @@ -119,6 +119,9 @@ class AlphaNum { AlphaNum(float f) // NOLINT(runtime/explicit) : piece_(digits_, strlen(FloatToBuffer(f, digits_))) {} + AlphaNum(bfloat16 f) // NOLINT(runtime/explicit) + : piece_(digits_, strlen(FloatToBuffer(static_cast<float>(f), digits_))) { + } AlphaNum(double f) // NOLINT(runtime/explicit) : piece_(digits_, strlen(DoubleToBuffer(f, digits_))) {} |