diff options
Diffstat (limited to 'tensorflow/compiler/xla/primitive_util.cc')
-rw-r--r-- | tensorflow/compiler/xla/primitive_util.cc | 8 |
1 files changed, 1 insertions, 7 deletions
diff --git a/tensorflow/compiler/xla/primitive_util.cc b/tensorflow/compiler/xla/primitive_util.cc index 2bce56b7bd..2113b5e06f 100644 --- a/tensorflow/compiler/xla/primitive_util.cc +++ b/tensorflow/compiler/xla/primitive_util.cc @@ -79,11 +79,6 @@ PrimitiveType NativeToPrimitiveType<double>() { } template <> -PrimitiveType NativeToPrimitiveType<bfloat16>() { - return BF16; -} - -template <> PrimitiveType NativeToPrimitiveType<half>() { return F16; } @@ -94,7 +89,7 @@ PrimitiveType NativeToPrimitiveType<complex64>() { } bool IsFloatingPointType(PrimitiveType type) { - return type == F16 || type == F32 || type == F64 || type == BF16; + return type == F16 || type == F32 || type == F64; } bool IsComplexType(PrimitiveType type) { return type == C64; } @@ -123,7 +118,6 @@ int BitWidth(PrimitiveType type) { case S16: case U16: case F16: - case BF16: return 16; case U32: |