experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Add explicit support for bfloat16 in tensorflow/core/platform/strcat.h

Browse Source 
PiperOrigin-RevId: 326729760
Change-Id: I36f681b20f91f3c08f4c5b795e488f90d17194aa
This commit is contained in:
A. Unique TensorFlower 2020-08-14 14:25:29 -07:00 committed by TensorFlower Gardener
parent 63dbe41c4c
commit dab490a8bf
2 changed files with 17 additions and 0 deletions
tensorflow/core/platform
strcat.hstrcat_test.cc

2
tensorflow/core/platform/strcat.h
View File

@ -117,6 +117,8 @@ class AlphaNum {
: piece_(digits_, FloatToBuffer(f, digits_)) {}
AlphaNum(double f) // NOLINT(runtime/explicit)
: piece_(digits_, DoubleToBuffer(f, digits_)) {}
AlphaNum(bfloat16 bf) // NOLINT(runtime/explicit)
: piece_(digits_, FloatToBuffer(static_cast<float>(bf), digits_)) {}
AlphaNum(Hex hex); // NOLINT(runtime/explicit)

15
tensorflow/core/platform/strcat_test.cc
View File

@ -61,6 +61,21 @@ TEST(StrCat, Ints) {
EXPECT_EQ(answer, "130");
}
TEST(StrCat, Floats) {
const int s = 0;
const float f = 1.5f;
const double d = 1.5;
const bfloat16 bf(1.5f);
string answer;
answer = tensorflow::strings::StrCat(s, f);
EXPECT_EQ(answer, "01.5");
answer = tensorflow::strings::StrCat(s, d);
EXPECT_EQ(answer, "01.5");
answer = tensorflow::strings::StrCat(s, bf);
EXPECT_EQ(answer, "01.5");
}
TEST(StrCat, Basics) {
string result;