aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/compiler/xla/primitive_util.cc
diff options
context:
space:
mode:
Diffstat (limited to 'tensorflow/compiler/xla/primitive_util.cc')
-rw-r--r--tensorflow/compiler/xla/primitive_util.cc8
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: