# tf.experimental.dtensor.enable_save_as_bf16

<br />

|---|
| [![](https://www.tensorflow.org/images/GitHub-Mark-32px.png) View source on GitHub](https://github.com/tensorflow/tensorflow/blob/v2.14.0/tensorflow/dtensor/python/save_restore.py#L83-L97) |

Allows float32 DVariables to be checkpointed and restored as bfloat16.

    tf.experimental.dtensor.enable_save_as_bf16(
        variables: List[https://www.tensorflow.org/versions/r2.14/api_docs/python/tf/Variable]
    )

The method only affects the DVariable part inside the model and leaves
non-DTensor Variables/Tensors untouched.

<br />

<br />

<br />

| ## Args ||
|---|---|
| `variables` | A list of tf.Variable to be enabled with bfloat16 save/restore. Only has effect on DTensor Variables as they go through d_variables with DTensor Specific logis. |

<br />