diff options
author | A. Unique TensorFlower <gardener@tensorflow.org> | 2018-03-22 21:46:31 -0700 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-03-22 21:48:55 -0700 |
commit | ed0defc1ffef77fa27bf360100d983eeee3359db (patch) | |
tree | dc01d7d8edc0a77652939eee1013ee0152fe07bd /tensorflow/core/kernels/cast_op_test.cc | |
parent | 0191d264a6e3da12ff7db5ba8002fed6356f071b (diff) |
Add cast op support for bfloat16 in CPU.
PiperOrigin-RevId: 190173501
Diffstat (limited to 'tensorflow/core/kernels/cast_op_test.cc')
-rw-r--r-- | tensorflow/core/kernels/cast_op_test.cc | 4 |
1 files changed, 3 insertions, 1 deletions
diff --git a/tensorflow/core/kernels/cast_op_test.cc b/tensorflow/core/kernels/cast_op_test.cc index 057e209a71..7da9d28a3d 100644 --- a/tensorflow/core/kernels/cast_op_test.cc +++ b/tensorflow/core/kernels/cast_op_test.cc @@ -75,7 +75,8 @@ class CastOpTest : public OpsTestBase { TEST_CAST(in, int64); \ TEST_CAST(in, half); \ TEST_CAST(in, float); \ - TEST_CAST(in, double) + TEST_CAST(in, double); \ + TEST_CAST(in, bfloat16); TEST_ALL_CASTS_FROM(uint8) TEST_ALL_CASTS_FROM(uint16) @@ -85,6 +86,7 @@ TEST_ALL_CASTS_FROM(int64) TEST_ALL_CASTS_FROM(half) TEST_ALL_CASTS_FROM(float) TEST_ALL_CASTS_FROM(double) +TEST_ALL_CASTS_FROM(bfloat16) #undef TEST_ALL_CASTS_FROM #undef TEST_CAST |