<br />

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

Allows float32 DVariables to be checkpointed and restored as bfloat16.

```python
tf.experimental.dtensor.enable_save_as_bf16(
    variables: List[`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 />