diff --git a/tensorflow/compiler/mlir/xla/hlo_utils.cc b/tensorflow/compiler/mlir/xla/hlo_utils.cc index dfed190ba1e..962e18b297b 100644 --- a/tensorflow/compiler/mlir/xla/hlo_utils.cc +++ b/tensorflow/compiler/mlir/xla/hlo_utils.cc @@ -83,12 +83,11 @@ StatusOr CreateDenseElementsAttrFromLiteral( ConvertTensorShapeToType( literal.shape(), builder)); + // TODO(hinsu): Support remaining XLA primitive types. auto element_type = literal.shape().element_type(); switch (element_type) { case PrimitiveType::PRED: return CreateDenseAttrFromLiteral(type, literal); - case PrimitiveType::F16: - return CreateDenseAttrFromLiteral(type, literal); case PrimitiveType::F32: return CreateDenseAttrFromLiteral(type, literal); case PrimitiveType::F64: