View source on GitHub |
Creates a single-client mesh.
tf.experimental.dtensor.create_mesh(
mesh_dims: Optional[List[Tuple[str, int]]] = None,
mesh_name: str = '',
devices: Optional[List[str]] = None,
device_type: Optional[str] = None
) -> tf.experimental.dtensor.Mesh
If both mesh_dims
and devices
are specified, they must match each otehr.
As a special case, when all arguments are missing, this creates a 1D CPU mesh
with an empty name, assigning all available devices to that dimension.
Returns | |
---|---|
A single-client mesh created from specified or default arguments. |