ML Community Day is November 9! Join us for updates from TensorFlow, JAX, and more Learn more

tf.compat.v1.to_bfloat16

Casts a tensor to type bfloat16. (deprecated)

Migrate to TF2

This name was deprecated and removed in TF2, but has an exact replacement tf.cast(..., tf.bfloat16). There are no further issues with eager execution or tf.function.

Before:

tf.compat.v1.to_bfloat16(tf.constant(3.14, dtype=tf.float32))
<tf.Tensor: shape=(), dtype=bfloat16, numpy=3.14>

After:

tf.cast(tf.constant(3.14, dtype=tf.float32), tf.bfloat16)
<tf.Tensor: shape=(), dtype=bfloat16, numpy=3.14>

Description

x A Tensor or SparseTensor or IndexedSlices.
name A name for the operation (optional).

A Tensor or SparseTensor or IndexedSlices with same shape as x with type bfloat16.

TypeError If x cannot be cast to the bfloat16.