proto

Documentation

Index

• Variables
• type LayoutProto
  • func (*LayoutProto) Descriptor() ([]byte, []int)
  • func (x *LayoutProto) GetMeshConfig() *MeshProto
  • func (x *LayoutProto) GetShardingSpecs() []*ShardingSpec
  • func (*LayoutProto) ProtoMessage()
  • func (x *LayoutProto) ProtoReflect() protoreflect.Message
  • func (x *LayoutProto) Reset()
  • func (x *LayoutProto) String() string
• type MeshDimensionProto
  • func (*MeshDimensionProto) Descriptor() ([]byte, []int)
  • func (x *MeshDimensionProto) GetName() string
  • func (x *MeshDimensionProto) GetSize() int64
  • func (*MeshDimensionProto) ProtoMessage()
  • func (x *MeshDimensionProto) ProtoReflect() protoreflect.Message
  • func (x *MeshDimensionProto) Reset()
  • func (x *MeshDimensionProto) String() string
• type MeshProto
  • func (*MeshProto) Descriptor() ([]byte, []int)
  • func (x *MeshProto) GetGlobalDeviceIds() []int64
  • func (x *MeshProto) GetGlobalDevices() []string
  • func (x *MeshProto) GetLocalDeviceIds() []int64
  • func (x *MeshProto) GetLocalDevices() []string
  • func (x *MeshProto) GetMeshDimensions() []*MeshDimensionProto
  • func (x *MeshProto) GetName() string
  • func (*MeshProto) ProtoMessage()
  • func (x *MeshProto) ProtoReflect() protoreflect.Message
  • func (x *MeshProto) Reset()
  • func (x *MeshProto) String() string
• type ShardingSpec
  • func (*ShardingSpec) Descriptor() ([]byte, []int)
  • func (x *ShardingSpec) GetShardingSpec() string
  • func (*ShardingSpec) ProtoMessage()
  • func (x *ShardingSpec) ProtoReflect() protoreflect.Message
  • func (x *ShardingSpec) Reset()
  • func (x *ShardingSpec) String() string

Variables

var File_tensorflow_dtensor_proto_layout_proto protoreflect.FileDescriptor

Types

type LayoutProto

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
}

Proto representation of a Layout.

func (*LayoutProto) Descriptor

func (*LayoutProto) Descriptor() ([]byte, []int)

Deprecated: Use LayoutProto.ProtoReflect.Descriptor instead.

func (*LayoutProto) GetMeshConfig

func (x *LayoutProto) GetMeshConfig() *MeshProto

func (*LayoutProto) GetShardingSpecs

func (x *LayoutProto) GetShardingSpecs() []*ShardingSpec

func (*LayoutProto) ProtoMessage

func (*LayoutProto) ProtoMessage()

func (*LayoutProto) ProtoReflect

func (x *LayoutProto) ProtoReflect() protoreflect.Message

func (*LayoutProto) Reset

func (x *LayoutProto) Reset()

func (*LayoutProto) String

func (x *LayoutProto) String() string

type MeshDimensionProto

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
}

func (*MeshDimensionProto) Descriptor

func (*MeshDimensionProto) Descriptor() ([]byte, []int)

Deprecated: Use MeshDimensionProto.ProtoReflect.Descriptor instead.

func (*MeshDimensionProto) GetName

func (x *MeshDimensionProto) GetName() string

func (*MeshDimensionProto) GetSize

func (x *MeshDimensionProto) GetSize() int64

func (*MeshDimensionProto) ProtoMessage

func (*MeshDimensionProto) ProtoMessage()

func (*MeshDimensionProto) ProtoReflect

func (x *MeshDimensionProto) ProtoReflect() protoreflect.Message

func (*MeshDimensionProto) Reset

func (x *MeshDimensionProto) Reset()

func (*MeshDimensionProto) String

func (x *MeshDimensionProto) String() string

type MeshProto

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
}

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

func (*MeshProto) Descriptor

func (*MeshProto) Descriptor() ([]byte, []int)

Deprecated: Use MeshProto.ProtoReflect.Descriptor instead.

func (*MeshProto) GetGlobalDeviceIds

func (x *MeshProto) GetGlobalDeviceIds() []int64

func (*MeshProto) GetGlobalDevices

func (x *MeshProto) GetGlobalDevices() []string

func (*MeshProto) GetLocalDeviceIds

func (x *MeshProto) GetLocalDeviceIds() []int64

func (*MeshProto) GetLocalDevices

func (x *MeshProto) GetLocalDevices() []string

func (*MeshProto) GetMeshDimensions

func (x *MeshProto) GetMeshDimensions() []*MeshDimensionProto

func (*MeshProto) GetName

func (x *MeshProto) GetName() string

func (*MeshProto) ProtoMessage

func (*MeshProto) ProtoMessage()

func (*MeshProto) ProtoReflect

func (x *MeshProto) ProtoReflect() protoreflect.Message

func (*MeshProto) Reset

func (x *MeshProto) Reset()

func (*MeshProto) String

func (x *MeshProto) String() string

type ShardingSpec

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
}

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.

func (*ShardingSpec) Descriptor

func (*ShardingSpec) Descriptor() ([]byte, []int)

Deprecated: Use ShardingSpec.ProtoReflect.Descriptor instead.

func (*ShardingSpec) GetShardingSpec

func (x *ShardingSpec) GetShardingSpec() string

func (*ShardingSpec) ProtoMessage

func (*ShardingSpec) ProtoMessage()

func (*ShardingSpec) ProtoReflect

func (x *ShardingSpec) ProtoReflect() protoreflect.Message

func (*ShardingSpec) Reset

func (x *ShardingSpec) Reset()

func (*ShardingSpec) String

func (x *ShardingSpec) String() string