<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/api.py#L89-L112) |

Sets the default DTensor device mesh to use for enclosed functions.

```python
@contextlib.contextmanager
tf.experimental.dtensor.default_mesh(
    mesh: `tf.experimental.dtensor.Mesh`
)
```

This function returns a scope. All the ops and tf.functions in this scope will
default to this DTensor mesh if a mesh cannot be inferred from any of the
inputs
This is useful for wrapping any tf.function that doesn't take a DTensor as
input but would like to produce DTensor as result. The scope will also make
sure all small constants are replicated as DTensors.

<br />

<br />

<br />

| ## Args ||
|---|---|
| `mesh` | A Mesh instance to extract a default mesh from. |

<br />

<br />

<br />

<br />

| ## Yields ||
|---|---|
| A context in which all ops and tf.functions will run on the given mesh. ||

<br />