package tensorflow.dtensor

Get desktop application:
View/edit binary Protocol Buffers messages

Proto representation of a Layout.

• repeated ShardingSpec sharding_specs = 1

• optional MeshProto mesh_config = 2

Used in: MeshProto

• string name = 1

  Name of mesh dimension, required to create Mesh.

• int64 size = 2

  Length of mesh dimension.

Proto representation of a Mesh. TODO(allenl): Add a unique mesh ID so we have an efficient way to identify meshes in mappings.

Used in: LayoutProto

• repeated MeshDimensionProto mesh_dimensions = 1

• repeated int64 global_device_ids = 2

  A list of global devices ids.

• repeated int64 local_device_ids = 4

  Devices ids handled by the local process.

• repeated string local_devices = 5

  Devices handled by the local process.

• repeated string global_devices = 6

  Global device names (Optional). Set when multiple device meshes are in use.

• string name = 3

  Required name which uniquely identifies this mesh in a program.

Defines a sharding spec over a tensor. A sharding spec can either be a mesh dimension of the associated mesh or the special values scalar, or not_sharded.

Used in: LayoutProto

• string sharding_spec = 2

  If set, the mesh dimension this tensor dimension is sharded over.