Return the default mesh under the current dtensor device context.
tf.experimental.dtensor.get_default_mesh() -> Optional[tf.experimental.dtensor.Mesh]
In the case that dtensor device system is not initialized, this function will return None.
| Returns | |
|---|---|
| The current default mesh for the dtensor device context. |
© 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/experimental/dtensor/get_default_mesh