proto

type LayoutProto struct {
	ShardingSpecs []*ShardingSpec `protobuf:"bytes,1,rep,name=sharding_specs,json=shardingSpecs,proto3" json:"sharding_specs,omitempty"`
	MeshConfig    *MeshProto      `protobuf:"bytes,2,opt,name=mesh_config,json=meshConfig,proto3" json:"mesh_config,omitempty"`
	// contains filtered or unexported fields
}

type MeshDimensionProto struct {

	// Name of mesh dimension, required to create Mesh.
	Name string `protobuf:"bytes,1,opt,name=name,proto3" json:"name,omitempty"`
	// Length of mesh dimension.
	Size int64 `protobuf:"varint,2,opt,name=size,proto3" json:"size,omitempty"`
	// contains filtered or unexported fields
}

type MeshProto struct {
	MeshDimensions []*MeshDimensionProto `protobuf:"bytes,1,rep,name=mesh_dimensions,json=meshDimensions,proto3" json:"mesh_dimensions,omitempty"`
	// A list of global devices ids.
	GlobalDeviceIds []int64 `protobuf:"varint,2,rep,packed,name=global_device_ids,json=globalDeviceIds,proto3" json:"global_device_ids,omitempty"`
	// Devices ids handled by the local process.
	LocalDeviceIds []int64 `protobuf:"varint,4,rep,packed,name=local_device_ids,json=localDeviceIds,proto3" json:"local_device_ids,omitempty"`
	// Devices handled by the local process.
	LocalDevices []string `protobuf:"bytes,5,rep,name=local_devices,json=localDevices,proto3" json:"local_devices,omitempty"`
	// Global device names (Optional). Set when multiple device meshes are in use.
	GlobalDevices []string `protobuf:"bytes,6,rep,name=global_devices,json=globalDevices,proto3" json:"global_devices,omitempty"`
	// Required name which uniquely identifies this mesh in a program.
	Name string `protobuf:"bytes,3,opt,name=name,proto3" json:"name,omitempty"`
	// contains filtered or unexported fields
}

type ShardingSpec struct {

	// If set, the mesh dimension this tensor dimension is sharded over.
	ShardingSpec string `protobuf:"bytes,2,opt,name=sharding_spec,json=shardingSpec,proto3" json:"sharding_spec,omitempty"`
	// contains filtered or unexported fields
}