tf.compat.v1.to_bfloat16

Casts a tensor to type bfloat16. (deprecated)

tf.compat.v1.to_bfloat16(
    x, name='ToBFloat16'
)

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

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

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

Raises
`TypeError`	If `x` cannot be cast to the `bfloat16`.

© 2022 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 4.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/api_docs/python/tf/compat/v1/to_bfloat16