xla_data

package
v0.4.8 Latest Latest
Warning

This package is not in the latest version of its module.

 Go to latest Published: Nov 19, 2024 License: Apache-2.0 Imports: 4 Imported by: 1

Documentation

Index

Constants

This section is empty.

Variables

View Source 
var (
	PrimitiveType_name = map[int32]string{
		0:  "PRIMITIVE_TYPE_INVALID",
		1:  "PRED",
		26: "S2",
		21: "S4",
		2:  "S8",
		3:  "S16",
		4:  "S32",
		5:  "S64",
		27: "U2",
		22: "U4",
		6:  "U8",
		7:  "U16",
		8:  "U32",
		9:  "U64",
		10: "F16",
		11: "F32",
		16: "BF16",
		12: "F64",
		19: "F8E5M2",
		28: "F8E4M3",
		20: "F8E4M3FN",
		23: "F8E4M3B11FNUZ",
		29: "F8E3M4",
		24: "F8E5M2FNUZ",
		25: "F8E4M3FNUZ",
		15: "C64",
		18: "C128",
		13: "TUPLE",
		14: "OPAQUE_TYPE",
		17: "TOKEN",
	}
	PrimitiveType_value = map[string]int32{
		"PRIMITIVE_TYPE_INVALID": 0,
		"PRED":                   1,
		"S2":                     26,
		"S4":                     21,
		"S8":                     2,
		"S16":                    3,
		"S32":                    4,
		"S64":                    5,
		"U2":                     27,
		"U4":                     22,
		"U8":                     6,
		"U16":                    7,
		"U32":                    8,
		"U64":                    9,
		"F16":                    10,
		"F32":                    11,
		"BF16":                   16,
		"F64":                    12,
		"F8E5M2":                 19,
		"F8E4M3":                 28,
		"F8E4M3FN":               20,
		"F8E4M3B11FNUZ":          23,
		"F8E3M4":                 29,
		"F8E5M2FNUZ":             24,
		"F8E4M3FNUZ":             25,
		"C64":                    15,
		"C128":                   18,
		"TUPLE":                  13,
		"OPAQUE_TYPE":            14,
		"TOKEN":                  17,
	}
)

Enum value maps for PrimitiveType.

View Source 
var (
	DimLevelType_name = map[int32]string{
		0: "DIM_DENSE",
		1: "DIM_COMPRESSED",
		2: "DIM_SINGLETON",
		3: "DIM_LOOSE_COMPRESSED",
	}
	DimLevelType_value = map[string]int32{
		"DIM_DENSE":            0,
		"DIM_COMPRESSED":       1,
		"DIM_SINGLETON":        2,
		"DIM_LOOSE_COMPRESSED": 3,
	}
)

Enum value maps for DimLevelType.

View Source 
var (
	ProfileType_name = map[int32]string{
		0: "INVALID",
		1: "WINDOW",
		2: "FLAG",
		3: "INTEGER",
	}
	ProfileType_value = map[string]int32{
		"INVALID": 0,
		"WINDOW":  1,
		"FLAG":    2,
		"INTEGER": 3,
	}
)

Enum value maps for ProfileType.

View Source 
var (
	ProfileSource_name = map[int32]string{
		0: "PROFILE_SOURCE_UNKNOWN_SOURCE",
		1: "PROFILE_SOURCE_EMBEDDED",
		2: "PROFILE_SOURCE_REMOTE",
	}
	ProfileSource_value = map[string]int32{
		"PROFILE_SOURCE_UNKNOWN_SOURCE": 0,
		"PROFILE_SOURCE_EMBEDDED":       1,
		"PROFILE_SOURCE_REMOTE":         2,
	}
)

Enum value maps for ProfileSource.

View Source 
var (
	CompilationEvent_name = map[int32]string{
		0: "COMPILATION_EVENT_UNKNOWN_EVENT",
		1: "COMPILATION_EVENT_FIRST_COMPILATION",
		2: "COMPILATION_EVENT_RECOMPILATION",
	}
	CompilationEvent_value = map[string]int32{
		"COMPILATION_EVENT_UNKNOWN_EVENT":     0,
		"COMPILATION_EVENT_FIRST_COMPILATION": 1,
		"COMPILATION_EVENT_RECOMPILATION":     2,
	}
)

Enum value maps for CompilationEvent.

View Source 
var (
	PaddingType_name = map[int32]string{
		0: "PADDING_INVALID",
		1: "PADDING_VALID",
		2: "PADDING_SAME",
	}
	PaddingType_value = map[string]int32{
		"PADDING_INVALID": 0,
		"PADDING_VALID":   1,
		"PADDING_SAME":    2,
	}
)

Enum value maps for PaddingType.

View Source 
var (
	FftType_name = map[int32]string{
		0: "FFT",
		1: "IFFT",
		2: "RFFT",
		3: "IRFFT",
	}
	FftType_value = map[string]int32{
		"FFT":   0,
		"IFFT":  1,
		"RFFT":  2,
		"IRFFT": 3,
	}
)

Enum value maps for FftType.

View Source 
var (
	SparsityType_name = map[int32]string{
		0: "SPARSITY_INVALID",
		1: "SPARSITY_STRUCTURED_N_M",
	}
	SparsityType_value = map[string]int32{
		"SPARSITY_INVALID":        0,
		"SPARSITY_STRUCTURED_N_M": 1,
	}
)

Enum value maps for SparsityType.

View Source 
var (
	RandomDistribution_name = map[int32]string{
		0: "RNG_INVALID",
		1: "RNG_UNIFORM",
		2: "RNG_NORMAL",
	}
	RandomDistribution_value = map[string]int32{
		"RNG_INVALID": 0,
		"RNG_UNIFORM": 1,
		"RNG_NORMAL":  2,
	}
)

Enum value maps for RandomDistribution.

View Source 
var (
	RandomAlgorithm_name = map[int32]string{
		0: "RNG_DEFAULT",
		1: "RNG_THREE_FRY",
		2: "RNG_PHILOX",
	}
	RandomAlgorithm_value = map[string]int32{
		"RNG_DEFAULT":   0,
		"RNG_THREE_FRY": 1,
		"RNG_PHILOX":    2,
	}
)

Enum value maps for RandomAlgorithm.

View Source 
var (
	ChannelHandle_ChannelType_name = map[int32]string{
		0: "CHANNEL_TYPE_INVALID",
		1: "DEVICE_TO_DEVICE",
		2: "DEVICE_TO_HOST",
		3: "HOST_TO_DEVICE",
	}
	ChannelHandle_ChannelType_value = map[string]int32{
		"CHANNEL_TYPE_INVALID": 0,
		"DEVICE_TO_DEVICE":     1,
		"DEVICE_TO_HOST":       2,
		"HOST_TO_DEVICE":       3,
	}
)

Enum value maps for ChannelHandle_ChannelType.

View Source 
var (
	TriangularSolveOptions_Transpose_name = map[int32]string{
		0: "TRANSPOSE_INVALID",
		1: "NO_TRANSPOSE",
		2: "TRANSPOSE",
		3: "ADJOINT",
	}
	TriangularSolveOptions_Transpose_value = map[string]int32{
		"TRANSPOSE_INVALID": 0,
		"NO_TRANSPOSE":      1,
		"TRANSPOSE":         2,
		"ADJOINT":           3,
	}
)

Enum value maps for TriangularSolveOptions_Transpose.

View Source 
var (
	OpSharding_Type_name = map[int32]string{
		0: "REPLICATED",
		1: "MAXIMAL",
		2: "TUPLE",
		3: "OTHER",
		4: "MANUAL",
		5: "UNKNOWN",
	}
	OpSharding_Type_value = map[string]int32{
		"REPLICATED": 0,
		"MAXIMAL":    1,
		"TUPLE":      2,
		"OTHER":      3,
		"MANUAL":     4,
		"UNKNOWN":    5,
	}
)

Enum value maps for OpSharding_Type.

View Source 
var (
	OpSharding_ShardGroupType_name = map[int32]string{
		0: "AS",
		1: "LIKE",
	}
	OpSharding_ShardGroupType_value = map[string]int32{
		"AS":   0,
		"LIKE": 1,
	}
)

Enum value maps for OpSharding_ShardGroupType.

View Source 
var (
	ResultAccuracy_Mode_name = map[int32]string{
		0: "DEFAULT",
		1: "HIGHEST",
	}
	ResultAccuracy_Mode_value = map[string]int32{
		"DEFAULT": 0,
		"HIGHEST": 1,
	}
)

Enum value maps for ResultAccuracy_Mode.

View Source 
var (
	PrecisionConfig_Precision_name = map[int32]string{
		0: "DEFAULT",
		1: "HIGH",
		2: "HIGHEST",
		3: "PACKED_NIBBLE",
	}
	PrecisionConfig_Precision_value = map[string]int32{
		"DEFAULT":       0,
		"HIGH":          1,
		"HIGHEST":       2,
		"PACKED_NIBBLE": 3,
	}
)

Enum value maps for PrecisionConfig_Precision.

View Source 
var (
	PrecisionConfig_Algorithm_name = map[int32]string{
		0:  "ALG_UNSET",
		1:  "ALG_DOT_ANY_F8_ANY_F8_F32",
		2:  "ALG_DOT_ANY_F8_ANY_F8_F32_FAST_ACCUM",
		3:  "ALG_DOT_F16_F16_F16",
		4:  "ALG_DOT_F16_F16_F32",
		5:  "ALG_DOT_BF16_BF16_BF16",
		6:  "ALG_DOT_BF16_BF16_F32",
		7:  "ALG_DOT_BF16_BF16_F32_X3",
		8:  "ALG_DOT_BF16_BF16_F32_X6",
		9:  "ALG_DOT_TF32_TF32_F32",
		10: "ALG_DOT_TF32_TF32_F32_X3",
		11: "ALG_DOT_F32_F32_F32",
		12: "ALG_DOT_F64_F64_F64",
	}
	PrecisionConfig_Algorithm_value = map[string]int32{
		"ALG_UNSET":                            0,
		"ALG_DOT_ANY_F8_ANY_F8_F32":            1,
		"ALG_DOT_ANY_F8_ANY_F8_F32_FAST_ACCUM": 2,
		"ALG_DOT_F16_F16_F16":                  3,
		"ALG_DOT_F16_F16_F32":                  4,
		"ALG_DOT_BF16_BF16_BF16":               5,
		"ALG_DOT_BF16_BF16_F32":                6,
		"ALG_DOT_BF16_BF16_F32_X3":             7,
		"ALG_DOT_BF16_BF16_F32_X6":             8,
		"ALG_DOT_TF32_TF32_F32":                9,
		"ALG_DOT_TF32_TF32_F32_X3":             10,
		"ALG_DOT_F32_F32_F32":                  11,
		"ALG_DOT_F64_F64_F64":                  12,
	}
)

Enum value maps for PrecisionConfig_Algorithm.

View Source 
var File_xla_xla_data_proto protoreflect.FileDescriptor

Functions

This section is empty.

Types

type ChannelHandle 

type ChannelHandle struct {
	Handle int64                     `protobuf:"varint,1,opt,name=handle,proto3" json:"handle,omitempty"`
	Type   ChannelHandle_ChannelType `protobuf:"varint,2,opt,name=type,proto3,enum=xla.ChannelHandle_ChannelType" json:"type,omitempty"`
	// contains filtered or unexported fields
}

Handle given to a user to represent a channel between two computations via a Send and Recv instruction pair. Channels are unbuffered, so Send Send instructions will be blocked until the data is transferred.

func (*ChannelHandle) Descriptor deprecated 

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

Deprecated: Use ChannelHandle.ProtoReflect.Descriptor instead.

func (*ChannelHandle) GetHandle 

func (x *ChannelHandle) GetHandle() int64

func (*ChannelHandle) GetType 

func (x *ChannelHandle) GetType() ChannelHandle_ChannelType

func (*ChannelHandle) ProtoMessage 

func (*ChannelHandle) ProtoMessage()

func (*ChannelHandle) ProtoReflect 

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

func (*ChannelHandle) Reset 

func (x *ChannelHandle) Reset()

func (*ChannelHandle) String 

func (x *ChannelHandle) String() string

type ChannelHandle_ChannelType 

type ChannelHandle_ChannelType int32
const (
	// Invalid primitive type to serve as default.
	ChannelHandle_CHANNEL_TYPE_INVALID ChannelHandle_ChannelType = 0
	// A channel for sending data between devices.
	ChannelHandle_DEVICE_TO_DEVICE ChannelHandle_ChannelType = 1
	// A channel for sending data from the device to the host. Can only be used
	// with a Send operation.
	ChannelHandle_DEVICE_TO_HOST ChannelHandle_ChannelType = 2
	// A channel for sending data from the host to the device. Can only be used
	// with a Recv operation.
	ChannelHandle_HOST_TO_DEVICE ChannelHandle_ChannelType = 3
)

func (ChannelHandle_ChannelType) Descriptor 

func (ChannelHandle_ChannelType) Descriptor() protoreflect.EnumDescriptor

func (ChannelHandle_ChannelType) Enum 

func (x ChannelHandle_ChannelType) Enum() *ChannelHandle_ChannelType

func (ChannelHandle_ChannelType) EnumDescriptor deprecated 

func (ChannelHandle_ChannelType) EnumDescriptor() ([]byte, []int)

Deprecated: Use ChannelHandle_ChannelType.Descriptor instead.

func (ChannelHandle_ChannelType) Number 

func (x ChannelHandle_ChannelType) Number() protoreflect.EnumNumber

func (ChannelHandle_ChannelType) String 

func (x ChannelHandle_ChannelType) String() string

func (ChannelHandle_ChannelType) Type 

func (ChannelHandle_ChannelType) Type() protoreflect.EnumType

type CholeskyOptions 

type CholeskyOptions struct {

	// If true, uses the lower triangle of `a`. If false, uses the upper triangle
	// of `a`.
	Lower bool `protobuf:"varint,1,opt,name=lower,proto3" json:"lower,omitempty"`
	// contains filtered or unexported fields
}

func (*CholeskyOptions) Descriptor deprecated 

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

Deprecated: Use CholeskyOptions.ProtoReflect.Descriptor instead.

func (*CholeskyOptions) GetLower 

func (x *CholeskyOptions) GetLower() bool

func (*CholeskyOptions) ProtoMessage 

func (*CholeskyOptions) ProtoMessage()

func (*CholeskyOptions) ProtoReflect 

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

func (*CholeskyOptions) Reset 

func (x *CholeskyOptions) Reset()

func (*CholeskyOptions) String 

func (x *CholeskyOptions) String() string

type CollectiveDeviceListProto 

type CollectiveDeviceListProto struct {

	// ReplicaGroupV1: List of replica groups. Legacy way of representing device
	// lists.
	ReplicaGroups []*ReplicaGroup `protobuf:"bytes,1,rep,name=replica_groups,json=replicaGroups,proto3" json:"replica_groups,omitempty"`
	// ReplicaGroupV2: Represents a list of replica groups with reshaping and
	// transposing an iota array.
	IotaReplicaGroupList *IotaReplicaGroupListProto `protobuf:"bytes,2,opt,name=iota_replica_group_list,json=iotaReplicaGroupList,proto3" json:"iota_replica_group_list,omitempty"`
	// contains filtered or unexported fields
}

Represents a series of devices participating in a collective operation (e.g., all-reduce and all-to-all). While this directly translates to a list of replica groups, it may be used to represent these lists in a compact form.

func (*CollectiveDeviceListProto) Descriptor deprecated 

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

Deprecated: Use CollectiveDeviceListProto.ProtoReflect.Descriptor instead.

func (*CollectiveDeviceListProto) GetIotaReplicaGroupList 

func (x *CollectiveDeviceListProto) GetIotaReplicaGroupList() *IotaReplicaGroupListProto

func (*CollectiveDeviceListProto) GetReplicaGroups 

func (x *CollectiveDeviceListProto) GetReplicaGroups() []*ReplicaGroup

func (*CollectiveDeviceListProto) ProtoMessage 

func (*CollectiveDeviceListProto) ProtoMessage()

func (*CollectiveDeviceListProto) ProtoReflect 

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

func (*CollectiveDeviceListProto) Reset 

func (x *CollectiveDeviceListProto) Reset()

func (*CollectiveDeviceListProto) String 

func (x *CollectiveDeviceListProto) String() string

type CompilationEvent 

type CompilationEvent int32

The compilation event that triggered the use of the profile. 

const (
	CompilationEvent_COMPILATION_EVENT_UNKNOWN_EVENT     CompilationEvent = 0
	CompilationEvent_COMPILATION_EVENT_FIRST_COMPILATION CompilationEvent = 1
	CompilationEvent_COMPILATION_EVENT_RECOMPILATION     CompilationEvent = 2
)

func (CompilationEvent) Descriptor 

func (CompilationEvent) Descriptor() protoreflect.EnumDescriptor

func (CompilationEvent) Enum 

func (x CompilationEvent) Enum() *CompilationEvent

func (CompilationEvent) EnumDescriptor deprecated 

func (CompilationEvent) EnumDescriptor() ([]byte, []int)

Deprecated: Use CompilationEvent.Descriptor instead.

func (CompilationEvent) Number 

func (x CompilationEvent) Number() protoreflect.EnumNumber

func (CompilationEvent) String 

func (x CompilationEvent) String() string

func (CompilationEvent) Type 

func (CompilationEvent) Type() protoreflect.EnumType

type ComputationStats 

type ComputationStats struct {

	// The number of floating point operations in the computation.
	FlopCount float64 `protobuf:"fixed64,1,opt,name=flop_count,json=flopCount,proto3" json:"flop_count,omitempty"`
	// The number of transcendental operations (e.g., exp) in the computation.
	TranscendentalCount float64 `protobuf:"fixed64,2,opt,name=transcendental_count,json=transcendentalCount,proto3" json:"transcendental_count,omitempty"`
	// contains filtered or unexported fields
}

Statistics of a computation.

func (*ComputationStats) Descriptor deprecated 

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

Deprecated: Use ComputationStats.ProtoReflect.Descriptor instead.

func (*ComputationStats) GetFlopCount 

func (x *ComputationStats) GetFlopCount() float64

func (*ComputationStats) GetTranscendentalCount 

func (x *ComputationStats) GetTranscendentalCount() float64

func (*ComputationStats) ProtoMessage 

func (*ComputationStats) ProtoMessage()

func (*ComputationStats) ProtoReflect 

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

func (*ComputationStats) Reset 

func (x *ComputationStats) Reset()

func (*ComputationStats) String 

func (x *ComputationStats) String() string

type ConvolutionDimensionNumbers 

type ConvolutionDimensionNumbers struct {

	// The number of the dimension that represents batch in the input.
	InputBatchDimension int64 `protobuf:"varint,7,opt,name=input_batch_dimension,json=inputBatchDimension,proto3" json:"input_batch_dimension,omitempty"`
	// The number of the dimension that represents features in the input.
	InputFeatureDimension int64 `` /* 127-byte string literal not displayed */
	// The dimension numbers for the spatial dimensions that the window
	// moves through in the input.
	InputSpatialDimensions []int64 `` /* 138-byte string literal not displayed */
	// The number of the dimension that represents input features in the
	// convolutional kernel (rhs).
	KernelInputFeatureDimension int64 `` /* 147-byte string literal not displayed */
	// The number of the dimension that represents output features in
	// the convolutional kernel (rhs).
	KernelOutputFeatureDimension int64 `` /* 150-byte string literal not displayed */
	// The dimension numbers for the spatial dimensions that the window
	// moves through in the kernel (rhs). window.strides(0) is the
	// stride in the kernel_spatial_dimensions(0) dimension.
	KernelSpatialDimensions []int64 `` /* 140-byte string literal not displayed */
	// The number of the dimension that represents batch in the output.
	OutputBatchDimension int64 `protobuf:"varint,9,opt,name=output_batch_dimension,json=outputBatchDimension,proto3" json:"output_batch_dimension,omitempty"`
	// The number of the dimension that represents features in the output.
	OutputFeatureDimension int64 `` /* 131-byte string literal not displayed */
	// The dimension numbers for the spatial dimensions that the window
	// moves through in the output.
	OutputSpatialDimensions []int64 `` /* 141-byte string literal not displayed */
	// contains filtered or unexported fields
}

func (*ConvolutionDimensionNumbers) Descriptor deprecated 

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

Deprecated: Use ConvolutionDimensionNumbers.ProtoReflect.Descriptor instead.

func (*ConvolutionDimensionNumbers) GetInputBatchDimension 

func (x *ConvolutionDimensionNumbers) GetInputBatchDimension() int64

func (*ConvolutionDimensionNumbers) GetInputFeatureDimension 

func (x *ConvolutionDimensionNumbers) GetInputFeatureDimension() int64

func (*ConvolutionDimensionNumbers) GetInputSpatialDimensions 

func (x *ConvolutionDimensionNumbers) GetInputSpatialDimensions() []int64

func (*ConvolutionDimensionNumbers) GetKernelInputFeatureDimension 

func (x *ConvolutionDimensionNumbers) GetKernelInputFeatureDimension() int64

func (*ConvolutionDimensionNumbers) GetKernelOutputFeatureDimension 

func (x *ConvolutionDimensionNumbers) GetKernelOutputFeatureDimension() int64

func (*ConvolutionDimensionNumbers) GetKernelSpatialDimensions 

func (x *ConvolutionDimensionNumbers) GetKernelSpatialDimensions() []int64

func (*ConvolutionDimensionNumbers) GetOutputBatchDimension 

func (x *ConvolutionDimensionNumbers) GetOutputBatchDimension() int64

func (*ConvolutionDimensionNumbers) GetOutputFeatureDimension 

func (x *ConvolutionDimensionNumbers) GetOutputFeatureDimension() int64

func (*ConvolutionDimensionNumbers) GetOutputSpatialDimensions 

func (x *ConvolutionDimensionNumbers) GetOutputSpatialDimensions() []int64

func (*ConvolutionDimensionNumbers) ProtoMessage 

func (*ConvolutionDimensionNumbers) ProtoMessage()

func (*ConvolutionDimensionNumbers) ProtoReflect 

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

func (*ConvolutionDimensionNumbers) Reset 

func (x *ConvolutionDimensionNumbers) Reset()

func (*ConvolutionDimensionNumbers) String 

func (x *ConvolutionDimensionNumbers) String() string

type DeviceAssignmentProto 

type DeviceAssignmentProto struct {
	ReplicaCount       int32                                      `protobuf:"varint,1,opt,name=replica_count,json=replicaCount,proto3" json:"replica_count,omitempty"`
	ComputationCount   int32                                      `protobuf:"varint,2,opt,name=computation_count,json=computationCount,proto3" json:"computation_count,omitempty"`
	ComputationDevices []*DeviceAssignmentProto_ComputationDevice `protobuf:"bytes,3,rep,name=computation_devices,json=computationDevices,proto3" json:"computation_devices,omitempty"`
	// contains filtered or unexported fields
}

DeviceAssignmentProto is a serialized form of DeviceAssignment class, which represents the device ids assigned to a set of replicated computations. See xla::DeviceAssignment class comment for more details.

func (*DeviceAssignmentProto) Descriptor deprecated 

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

Deprecated: Use DeviceAssignmentProto.ProtoReflect.Descriptor instead.

func (*DeviceAssignmentProto) GetComputationCount 

func (x *DeviceAssignmentProto) GetComputationCount() int32

func (*DeviceAssignmentProto) GetComputationDevices 

func (x *DeviceAssignmentProto) GetComputationDevices() []*DeviceAssignmentProto_ComputationDevice

func (*DeviceAssignmentProto) GetReplicaCount 

func (x *DeviceAssignmentProto) GetReplicaCount() int32

func (*DeviceAssignmentProto) ProtoMessage 

func (*DeviceAssignmentProto) ProtoMessage()

func (*DeviceAssignmentProto) ProtoReflect 

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

func (*DeviceAssignmentProto) Reset 

func (x *DeviceAssignmentProto) Reset()

func (*DeviceAssignmentProto) String 

func (x *DeviceAssignmentProto) String() string

type DeviceAssignmentProto_ComputationDevice 

type DeviceAssignmentProto_ComputationDevice struct {
	ReplicaDeviceIds []int64 `protobuf:"varint,1,rep,packed,name=replica_device_ids,json=replicaDeviceIds,proto3" json:"replica_device_ids,omitempty"`
	// contains filtered or unexported fields
}

Each logical computation runs on replica_count physical devices. ComputationDevice represents the device ids assinged to the replicas.

func (*DeviceAssignmentProto_ComputationDevice) Descriptor deprecated 

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

Deprecated: Use DeviceAssignmentProto_ComputationDevice.ProtoReflect.Descriptor instead.

func (*DeviceAssignmentProto_ComputationDevice) GetReplicaDeviceIds 

func (x *DeviceAssignmentProto_ComputationDevice) GetReplicaDeviceIds() []int64

func (*DeviceAssignmentProto_ComputationDevice) ProtoMessage 

func (*DeviceAssignmentProto_ComputationDevice) ProtoMessage()

func (*DeviceAssignmentProto_ComputationDevice) ProtoReflect 

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

func (*DeviceAssignmentProto_ComputationDevice) Reset 

func (x *DeviceAssignmentProto_ComputationDevice) Reset()

func (*DeviceAssignmentProto_ComputationDevice) String 

func (x *DeviceAssignmentProto_ComputationDevice) String() string

type DeviceHandle 

type DeviceHandle struct {
	Handle int64 `protobuf:"varint,1,opt,name=handle,proto3" json:"handle,omitempty"`
	// The number of model-parallel virtual devices that communicate via XLA
	// Send/Recv instructions.
	DeviceCount int64 `protobuf:"varint,2,opt,name=device_count,json=deviceCount,proto3" json:"device_count,omitempty"`
	// contains filtered or unexported fields
}

Handle given to a user that represents a replicated virtual device. Each replicated device represents N physical devices for execution where N is the number of replicas.

func (*DeviceHandle) Descriptor deprecated 

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

Deprecated: Use DeviceHandle.ProtoReflect.Descriptor instead.

func (*DeviceHandle) GetDeviceCount 

func (x *DeviceHandle) GetDeviceCount() int64

func (*DeviceHandle) GetHandle 

func (x *DeviceHandle) GetHandle() int64

func (*DeviceHandle) ProtoMessage 

func (*DeviceHandle) ProtoMessage()

func (*DeviceHandle) ProtoReflect 

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

func (*DeviceHandle) Reset 

func (x *DeviceHandle) Reset()

func (*DeviceHandle) String 

func (x *DeviceHandle) String() string

type DimLevelType 

type DimLevelType int32

A DimLevelType indicates the encoding method for a dimension in an array. The semantics of this field are identical to those of the MLIR SparseTensor dialect. This should be kept in sync with the SparseTensor DimLevelType enum: https://github.com/llvm/llvm-project/blob/5674a3c88088e668b684326c2194a6282e8270ff/mlir/include/mlir/Dialect/SparseTensor/IR/SparseTensorAttrDefs.td#L86 

const (
	// The corresponding dimension is Dense, every entry is stored.
	DimLevelType_DIM_DENSE DimLevelType = 0
	// The corresponding dimension is Compressed, only nonzeros are stored.
	DimLevelType_DIM_COMPRESSED DimLevelType = 1
	// The corresponding dimension contains a single coordinate, no sibling
	// elements for each parent.
	DimLevelType_DIM_SINGLETON DimLevelType = 2
	// The corresponding dimension is Compressed, but with potential trailing
	// zeros, thus an extra upper bound (high) is used to exclude those zeros.
	// E.g., indices = [1, 2, 0, 0, 3, 4, 0, 0], position = [(0, 2), (4, 6)].
	DimLevelType_DIM_LOOSE_COMPRESSED DimLevelType = 3
)

func (DimLevelType) Descriptor 

func (DimLevelType) Descriptor() protoreflect.EnumDescriptor

func (DimLevelType) Enum 

func (x DimLevelType) Enum() *DimLevelType

func (DimLevelType) EnumDescriptor deprecated 

func (DimLevelType) EnumDescriptor() ([]byte, []int)

Deprecated: Use DimLevelType.Descriptor instead.

func (DimLevelType) Number 

func (x DimLevelType) Number() protoreflect.EnumNumber

func (DimLevelType) String 

func (x DimLevelType) String() string

func (DimLevelType) Type 

func (DimLevelType) Type() protoreflect.EnumType

type DotDimensionNumbers 

type DotDimensionNumbers struct {

	// The dimension numbers that represent the 'lhs' contracting dimensions.
	LhsContractingDimensions []int64 `` /* 143-byte string literal not displayed */
	// The dimension numbers that represent the 'rhs' contracting dimensions.
	RhsContractingDimensions []int64 `` /* 143-byte string literal not displayed */
	// The dimension numbers that represent the 'lhs' batch dimensions.
	LhsBatchDimensions []int64 `protobuf:"varint,3,rep,packed,name=lhs_batch_dimensions,json=lhsBatchDimensions,proto3" json:"lhs_batch_dimensions,omitempty"`
	// The dimension numbers that represent the 'rhs' batch dimensions.
	RhsBatchDimensions []int64 `protobuf:"varint,4,rep,packed,name=rhs_batch_dimensions,json=rhsBatchDimensions,proto3" json:"rhs_batch_dimensions,omitempty"`
	// contains filtered or unexported fields
}

func (*DotDimensionNumbers) Descriptor deprecated 

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

Deprecated: Use DotDimensionNumbers.ProtoReflect.Descriptor instead.

func (*DotDimensionNumbers) GetLhsBatchDimensions 

func (x *DotDimensionNumbers) GetLhsBatchDimensions() []int64

func (*DotDimensionNumbers) GetLhsContractingDimensions 

func (x *DotDimensionNumbers) GetLhsContractingDimensions() []int64

func (*DotDimensionNumbers) GetRhsBatchDimensions 

func (x *DotDimensionNumbers) GetRhsBatchDimensions() []int64

func (*DotDimensionNumbers) GetRhsContractingDimensions 

func (x *DotDimensionNumbers) GetRhsContractingDimensions() []int64

func (*DotDimensionNumbers) ProtoMessage 

func (*DotDimensionNumbers) ProtoMessage()

func (*DotDimensionNumbers) ProtoReflect 

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

func (*DotDimensionNumbers) Reset 

func (x *DotDimensionNumbers) Reset()

func (*DotDimensionNumbers) String 

func (x *DotDimensionNumbers) String() string

type ExecutionHandle 

type ExecutionHandle struct {
	Handle int64 `protobuf:"varint,1,opt,name=handle,proto3" json:"handle,omitempty"`
	// contains filtered or unexported fields
}

Handle given to a user that represents an execution that the user launched asynchronously on the device.

func (*ExecutionHandle) Descriptor deprecated 

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

Deprecated: Use ExecutionHandle.ProtoReflect.Descriptor instead.

func (*ExecutionHandle) GetHandle 

func (x *ExecutionHandle) GetHandle() int64

func (*ExecutionHandle) ProtoMessage 

func (*ExecutionHandle) ProtoMessage()

func (*ExecutionHandle) ProtoReflect 

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

func (*ExecutionHandle) Reset 

func (x *ExecutionHandle) Reset()

func (*ExecutionHandle) String 

func (x *ExecutionHandle) String() string

type ExecutionProfile 

type ExecutionProfile struct {

	// Whether the executable was read from the compilation cache.
	CompilationCacheHit bool `protobuf:"varint,1,opt,name=compilation_cache_hit,json=compilationCacheHit,proto3" json:"compilation_cache_hit,omitempty"`
	// The time in milliseconds spent to compile the computation. This only set if
	// the executable was not read from the compilation cache
	// (compilation_cache_hit == false).
	CompileTimeMs int64 `protobuf:"varint,2,opt,name=compile_time_ms,json=compileTimeMs,proto3" json:"compile_time_ms,omitempty"`
	// The number of cycles spent for the computation. This does not include the
	// time taken for the data transfers between the host and the device. This is
	// a target-dependent field and only used for debugging purposes.
	ComputeCycleCount int64 `protobuf:"varint,3,opt,name=compute_cycle_count,json=computeCycleCount,proto3" json:"compute_cycle_count,omitempty"`
	// The time in nanoseconds spent for the computation, without data transfer.
	ComputeTimeNs int64 `protobuf:"varint,4,opt,name=compute_time_ns,json=computeTimeNs,proto3" json:"compute_time_ns,omitempty"`
	// The time in nanoseconds spent for the entire computation, including the
	// result data transfer time. Current implementation does not spend any cycles
	// for the input data transfer since the memory is initialized with the proper
	// values before the execution.
	ComputeAndTransferTimeNs int64 `` /* 140-byte string literal not displayed */
	// The size of the binary code in the executable.
	ExecutableSizeInBytes int64 `` /* 129-byte string literal not displayed */
	// Whether this profile was drawn from a cache of profiles instead of from
	// execution on the hardware.
	ProfileCacheHit bool `protobuf:"varint,7,opt,name=profile_cache_hit,json=profileCacheHit,proto3" json:"profile_cache_hit,omitempty"`
	// Whether a warm-up run of the computation was executed before the
	// measured execution.
	WarmupRunExecuted bool `protobuf:"varint,8,opt,name=warmup_run_executed,json=warmupRunExecuted,proto3" json:"warmup_run_executed,omitempty"`
	// contains filtered or unexported fields
}

Profile data from the execution of a computation.

func (*ExecutionProfile) Descriptor deprecated 

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

Deprecated: Use ExecutionProfile.ProtoReflect.Descriptor instead.

func (*ExecutionProfile) GetCompilationCacheHit 

func (x *ExecutionProfile) GetCompilationCacheHit() bool

func (*ExecutionProfile) GetCompileTimeMs 

func (x *ExecutionProfile) GetCompileTimeMs() int64

func (*ExecutionProfile) GetComputeAndTransferTimeNs 

func (x *ExecutionProfile) GetComputeAndTransferTimeNs() int64

func (*ExecutionProfile) GetComputeCycleCount 

func (x *ExecutionProfile) GetComputeCycleCount() int64

func (*ExecutionProfile) GetComputeTimeNs 

func (x *ExecutionProfile) GetComputeTimeNs() int64

func (*ExecutionProfile) GetExecutableSizeInBytes 

func (x *ExecutionProfile) GetExecutableSizeInBytes() int64

func (*ExecutionProfile) GetProfileCacheHit 

func (x *ExecutionProfile) GetProfileCacheHit() bool

func (*ExecutionProfile) GetWarmupRunExecuted 

func (x *ExecutionProfile) GetWarmupRunExecuted() bool

func (*ExecutionProfile) ProtoMessage 

func (*ExecutionProfile) ProtoMessage()

func (*ExecutionProfile) ProtoReflect 

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

func (*ExecutionProfile) Reset 

func (x *ExecutionProfile) Reset()

func (*ExecutionProfile) String 

func (x *ExecutionProfile) String() string

type FftType 

type FftType int32
const (
	FftType_FFT   FftType = 0 // Forward FFT; complex in, complex out.
	FftType_IFFT  FftType = 1 // Inverse FFT; complex in, complex out.
	FftType_RFFT  FftType = 2 // Forward real FFT; real in, fft_length / 2 + 1 complex out
	FftType_IRFFT FftType = 3 // Inverse real FFT; fft_length / 2 + 1 complex in,
)

func (FftType) Descriptor 

func (FftType) Descriptor() protoreflect.EnumDescriptor

func (FftType) Enum 

func (x FftType) Enum() *FftType

func (FftType) EnumDescriptor deprecated 

func (FftType) EnumDescriptor() ([]byte, []int)

Deprecated: Use FftType.Descriptor instead.

func (FftType) Number 

func (x FftType) Number() protoreflect.EnumNumber

func (FftType) String 

func (x FftType) String() string

func (FftType) Type 

func (FftType) Type() protoreflect.EnumType

type FrontendAttributes 

type FrontendAttributes struct {
	Map map[string]string `` /* 147-byte string literal not displayed */
	// contains filtered or unexported fields
}

Generic map of attributes used to pass hints / configuration options from the Python frontend to the XLA backend.

func (*FrontendAttributes) Descriptor deprecated 

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

Deprecated: Use FrontendAttributes.ProtoReflect.Descriptor instead.

func (*FrontendAttributes) GetMap 

func (x *FrontendAttributes) GetMap() map[string]string

func (*FrontendAttributes) ProtoMessage 

func (*FrontendAttributes) ProtoMessage()

func (*FrontendAttributes) ProtoReflect 

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

func (*FrontendAttributes) Reset 

func (x *FrontendAttributes) Reset()

func (*FrontendAttributes) String 

func (x *FrontendAttributes) String() string

type GatherDimensionNumbers 

type GatherDimensionNumbers struct {

	// "Window indices" is a term for a set of indices that index into the
	// interior of a dynamic-slice from the input tensor, the starting indices for
	// which were computed from output_gather_dims (see the operation semantic for
	// how this is defined) and the start_indices tensor.
	//
	// The window indices for a specific output index Out is computed as:
	//
	//	i = 0
	//	for (k : [0, input_tensor_shape.rank))
	//	  window_indices[k] =
	//	    if k in collapsed_slice_dims
	//	    then 0
	//	    else Out[offset_dims[i++]]
	OffsetDims         []int64 `protobuf:"varint,1,rep,packed,name=offset_dims,json=offsetDims,proto3" json:"offset_dims,omitempty"`
	CollapsedSliceDims []int64 `protobuf:"varint,2,rep,packed,name=collapsed_slice_dims,json=collapsedSliceDims,proto3" json:"collapsed_slice_dims,omitempty"`
	// This is interpreted as a map from i to start_index_map[i]. It
	// transforms the gather index looked up from the start_indices tensor into
	// the starting index in the input space.
	StartIndexMap []int64 `protobuf:"varint,3,rep,packed,name=start_index_map,json=startIndexMap,proto3" json:"start_index_map,omitempty"`
	// The dimension in the start_indices input that contains the starting
	// indices.
	IndexVectorDim int64 `protobuf:"varint,4,opt,name=index_vector_dim,json=indexVectorDim,proto3" json:"index_vector_dim,omitempty"`
	// This is the batch dimensions in the operand.
	OperandBatchingDims []int64 `` /* 128-byte string literal not displayed */
	// This is the batch dimensions in the index, and it should be the same size
	// as operand_batching_dims.
	StartIndicesBatchingDims []int64 `` /* 145-byte string literal not displayed */
	// contains filtered or unexported fields
}

Describes the dimension numbers for a gather operation.

See https://www.tensorflow.org/performance/xla/operation_semantics#gather for more details.

func (*GatherDimensionNumbers) Descriptor deprecated 

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

Deprecated: Use GatherDimensionNumbers.ProtoReflect.Descriptor instead.

func (*GatherDimensionNumbers) GetCollapsedSliceDims 

func (x *GatherDimensionNumbers) GetCollapsedSliceDims() []int64

func (*GatherDimensionNumbers) GetIndexVectorDim 

func (x *GatherDimensionNumbers) GetIndexVectorDim() int64

func (*GatherDimensionNumbers) GetOffsetDims 

func (x *GatherDimensionNumbers) GetOffsetDims() []int64

func (*GatherDimensionNumbers) GetOperandBatchingDims 

func (x *GatherDimensionNumbers) GetOperandBatchingDims() []int64

func (*GatherDimensionNumbers) GetStartIndexMap 

func (x *GatherDimensionNumbers) GetStartIndexMap() []int64

func (*GatherDimensionNumbers) GetStartIndicesBatchingDims 

func (x *GatherDimensionNumbers) GetStartIndicesBatchingDims() []int64

func (*GatherDimensionNumbers) ProtoMessage 

func (*GatherDimensionNumbers) ProtoMessage()

func (*GatherDimensionNumbers) ProtoReflect 

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

func (*GatherDimensionNumbers) Reset 

func (x *GatherDimensionNumbers) Reset()

func (*GatherDimensionNumbers) String 

func (x *GatherDimensionNumbers) String() string

type GlobalDataHandle 

type GlobalDataHandle struct {
	Handle int64 `protobuf:"varint,1,opt,name=handle,proto3" json:"handle,omitempty"`
	// contains filtered or unexported fields
}

Handle given to a user that represents a globally accessible allocation. Contrast this against a ComputationDataHandle, which is not globally accessible, since it only exists within a specific computation.

func (*GlobalDataHandle) Descriptor deprecated 

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

Deprecated: Use GlobalDataHandle.ProtoReflect.Descriptor instead.

func (*GlobalDataHandle) GetHandle 

func (x *GlobalDataHandle) GetHandle() int64

func (*GlobalDataHandle) ProtoMessage 

func (*GlobalDataHandle) ProtoMessage()

func (*GlobalDataHandle) ProtoReflect 

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

func (*GlobalDataHandle) Reset 

func (x *GlobalDataHandle) Reset()

func (*GlobalDataHandle) String 

func (x *GlobalDataHandle) String() string

type IotaReplicaGroupListProto 

type IotaReplicaGroupListProto struct {

	// Number of replica groups.
	NumReplicaGroups int64 `protobuf:"varint,1,opt,name=num_replica_groups,json=numReplicaGroups,proto3" json:"num_replica_groups,omitempty"`
	// Number of devices per group.
	NumDevicesPerGroup int64 `protobuf:"varint,2,opt,name=num_devices_per_group,json=numDevicesPerGroup,proto3" json:"num_devices_per_group,omitempty"`
	// The dimensions used to reshape the 1D iota array of device IDs.
	IotaReshapeDims []int64 `protobuf:"varint,3,rep,packed,name=iota_reshape_dims,json=iotaReshapeDims,proto3" json:"iota_reshape_dims,omitempty"`
	// The dimension permutations to transposed the iota array reshaped to
	// iota_reshape_dims. This must have the same size as iota_reshape_dims.
	IotaTransposePerm []int32 `protobuf:"varint,4,rep,packed,name=iota_transpose_perm,json=iotaTransposePerm,proto3" json:"iota_transpose_perm,omitempty"`
	// contains filtered or unexported fields
}

Represents a list of replica groups (a list of list of devices) with reshaping and transposing an iota array (iota tile assignment). Can be used to represent certain common patterns of device lists in a compact, scalable format.

func (*IotaReplicaGroupListProto) Descriptor deprecated 

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

Deprecated: Use IotaReplicaGroupListProto.ProtoReflect.Descriptor instead.

func (*IotaReplicaGroupListProto) GetIotaReshapeDims 

func (x *IotaReplicaGroupListProto) GetIotaReshapeDims() []int64

func (*IotaReplicaGroupListProto) GetIotaTransposePerm 

func (x *IotaReplicaGroupListProto) GetIotaTransposePerm() []int32

func (*IotaReplicaGroupListProto) GetNumDevicesPerGroup 

func (x *IotaReplicaGroupListProto) GetNumDevicesPerGroup() int64

func (*IotaReplicaGroupListProto) GetNumReplicaGroups 

func (x *IotaReplicaGroupListProto) GetNumReplicaGroups() int64

func (*IotaReplicaGroupListProto) ProtoMessage 

func (*IotaReplicaGroupListProto) ProtoMessage()

func (*IotaReplicaGroupListProto) ProtoReflect 

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

func (*IotaReplicaGroupListProto) Reset 

func (x *IotaReplicaGroupListProto) Reset()

func (*IotaReplicaGroupListProto) String 

func (x *IotaReplicaGroupListProto) String() string

type LayoutProto 

type LayoutProto struct {

	// The dimension level type list for this array, specifying the way in which
	// each array dimension is represented in memory. If this list is empty, the
	// array is assumed to be dense.
	DimLevelTypes []DimLevelType `` /* 132-byte string literal not displayed */
	// Whether each dimension is unique or ordered.  Each of the following lists
	// must be empty, or have one entry for each entry of dim_level_types.  If
	// either list is empty, all dimensions are assumed to be unique and ordered,
	// respectively.  Entries in this list may not be false for some DimLevelType
	// values (such as DIM_DENSE in particular).
	DimUnique  []bool `protobuf:"varint,13,rep,packed,name=dim_unique,json=dimUnique,proto3" json:"dim_unique,omitempty"`
	DimOrdered []bool `protobuf:"varint,14,rep,packed,name=dim_ordered,json=dimOrdered,proto3" json:"dim_ordered,omitempty"`
	// Sequence of dimension numbers, from minor (fastest varying index) to major
	// (slowest varying index). This field is required.
	MinorToMajor []int64 `protobuf:"varint,1,rep,packed,name=minor_to_major,json=minorToMajor,proto3" json:"minor_to_major,omitempty"`
	// A sequence of tiles, starting from the tile that's applied first to the
	// Shape.
	//
	// TODO(b/119839262): implement tiling in each backend or add Unimplemented
	// error.
	Tiles []*TileProto `protobuf:"bytes,6,rep,name=tiles,proto3" json:"tiles,omitempty"`
	// The shape is padded at the end to multiple of, in terms of number of
	// elements. This is useful when tiling does not bring the shape to certain
	// desired granules. Tiling effectively pads/reshapes/transposes the shape
	// to another shape. This field pads the total number of elements of that
	// new shape to a multiple of certain number of elements. This is useful such
	// as we want a layout which does not tile the data but still requires it to
	// be padded to certain number of elements.
	TailPaddingAlignmentInElements int64 `` /* 159-byte string literal not displayed */
	// (Optional) Bit size of each element. When unspecified or being 0, default
	// to ShapeUtil::ByteSizeOfPrimitiveType.
	ElementSizeInBits int64 `protobuf:"varint,7,opt,name=element_size_in_bits,json=elementSizeInBits,proto3" json:"element_size_in_bits,omitempty"`
	// Memory space where this array resides. The integer field is interpreted in
	// a backend-specific manner.
	MemorySpace int64 `protobuf:"varint,8,opt,name=memory_space,json=memorySpace,proto3" json:"memory_space,omitempty"`
	// The integer types to be used for indices and pointers.  These fields must
	// not be used unless the layout represents a sparse array.  The PrimitiveType
	// must correspond to an unsigned integer (U8, U16, U32, or U64).
	// If not provided, the compiler will use the largest unsigned integer
	// that is naturally supported by the target device (U32 or U64 in currently
	// supported devices).
	IndexPrimitiveType   PrimitiveType `` /* 142-byte string literal not displayed */
	PointerPrimitiveType PrimitiveType `` /* 148-byte string literal not displayed */
	// The physical, on-device shape used to represent the shape this layout
	// belongs to. Only used for sparse arrays.
	// The layout(s) contained within the physical shape should not also contain
	// a physical shape.
	PhysicalShape *ShapeProto `protobuf:"bytes,10,opt,name=physical_shape,json=physicalShape,proto3" json:"physical_shape,omitempty"`
	// The dynamic shape metadata size in bytes in front of the shape data. The
	// field may be non-zero for a static shape whose associated buffer is for a
	// dynamic shape, e.g. a result of SliceToDynamic.
	DynamicShapeMetadataPrefixBytes int64 `` /* 162-byte string literal not displayed */
	// The split configurations which describe if/how the data is split between
	// different memories.
	SplitConfigs []*SplitConfigProto `protobuf:"bytes,17,rep,name=split_configs,json=splitConfigs,proto3" json:"split_configs,omitempty"`
	// contains filtered or unexported fields
}

A layout describes how the array is placed in (1D) memory space. This includes the minor-to-major ordering of dimensions within a shape.

Clients must specify the layouts of input Literals to the computation. Layouts specified in interior operations which take Shapes (for example, Convert) are ignored.

See the XLA documentation for more information on shapes and layouts.

LINT.IfChange

func (*LayoutProto) Descriptor deprecated 

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

Deprecated: Use LayoutProto.ProtoReflect.Descriptor instead.

func (*LayoutProto) GetDimLevelTypes 

func (x *LayoutProto) GetDimLevelTypes() []DimLevelType

func (*LayoutProto) GetDimOrdered 

func (x *LayoutProto) GetDimOrdered() []bool

func (*LayoutProto) GetDimUnique 

func (x *LayoutProto) GetDimUnique() []bool

func (*LayoutProto) GetDynamicShapeMetadataPrefixBytes 

func (x *LayoutProto) GetDynamicShapeMetadataPrefixBytes() int64

func (*LayoutProto) GetElementSizeInBits 

func (x *LayoutProto) GetElementSizeInBits() int64

func (*LayoutProto) GetIndexPrimitiveType 

func (x *LayoutProto) GetIndexPrimitiveType() PrimitiveType

func (*LayoutProto) GetMemorySpace 

func (x *LayoutProto) GetMemorySpace() int64

func (*LayoutProto) GetMinorToMajor 

func (x *LayoutProto) GetMinorToMajor() []int64

func (*LayoutProto) GetPhysicalShape 

func (x *LayoutProto) GetPhysicalShape() *ShapeProto

func (*LayoutProto) GetPointerPrimitiveType 

func (x *LayoutProto) GetPointerPrimitiveType() PrimitiveType

func (*LayoutProto) GetSplitConfigs 

func (x *LayoutProto) GetSplitConfigs() []*SplitConfigProto

func (*LayoutProto) GetTailPaddingAlignmentInElements 

func (x *LayoutProto) GetTailPaddingAlignmentInElements() int64

func (*LayoutProto) GetTiles 

func (x *LayoutProto) GetTiles() []*TileProto

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 LiteralProto 

type LiteralProto struct {
	Shape         *ShapeProto     `protobuf:"bytes,1,opt,name=shape,proto3" json:"shape,omitempty"`
	Preds         []bool          `protobuf:"varint,2,rep,packed,name=preds,proto3" json:"preds,omitempty"`
	S2S           []byte          `protobuf:"bytes,26,opt,name=s2s,proto3" json:"s2s,omitempty"`
	S4S           []byte          `protobuf:"bytes,21,opt,name=s4s,proto3" json:"s4s,omitempty"`
	S8S           []byte          `protobuf:"bytes,15,opt,name=s8s,proto3" json:"s8s,omitempty"`
	U2S           []byte          `protobuf:"bytes,27,opt,name=u2s,proto3" json:"u2s,omitempty"`
	U4S           []byte          `protobuf:"bytes,22,opt,name=u4s,proto3" json:"u4s,omitempty"`
	U8S           []byte          `protobuf:"bytes,3,opt,name=u8s,proto3" json:"u8s,omitempty"`
	S32S          []int32         `protobuf:"varint,4,rep,packed,name=s32s,proto3" json:"s32s,omitempty"`
	S64S          []int64         `protobuf:"varint,5,rep,packed,name=s64s,proto3" json:"s64s,omitempty"`
	U32S          []uint32        `protobuf:"varint,6,rep,packed,name=u32s,proto3" json:"u32s,omitempty"`
	U64S          []uint64        `protobuf:"varint,7,rep,packed,name=u64s,proto3" json:"u64s,omitempty"`
	F32S          []float32       `protobuf:"fixed32,8,rep,packed,name=f32s,proto3" json:"f32s,omitempty"`
	F64S          []float64       `protobuf:"fixed64,9,rep,packed,name=f64s,proto3" json:"f64s,omitempty"`
	C64S          []float32       `protobuf:"fixed32,12,rep,packed,name=c64s,proto3" json:"c64s,omitempty"`   // Stored as interleaved real, imag floats.
	C128S         []float64       `protobuf:"fixed64,18,rep,packed,name=c128s,proto3" json:"c128s,omitempty"` // Stored as interleaved real, imag doubles.
	TupleLiterals []*LiteralProto `protobuf:"bytes,10,rep,name=tuple_literals,json=tupleLiterals,proto3" json:"tuple_literals,omitempty"`
	// The F16s, BF16s, U16s and S16s are encoded in little endian byte order
	F16S           []byte  `protobuf:"bytes,11,opt,name=f16s,proto3" json:"f16s,omitempty"`
	Bf16S          []byte  `protobuf:"bytes,13,opt,name=bf16s,proto3" json:"bf16s,omitempty"`
	U16S           []byte  `protobuf:"bytes,16,opt,name=u16s,proto3" json:"u16s,omitempty"`
	S16S           []byte  `protobuf:"bytes,17,opt,name=s16s,proto3" json:"s16s,omitempty"`
	F8E5M2S        []byte  `protobuf:"bytes,19,opt,name=f8e5m2s,proto3" json:"f8e5m2s,omitempty"`
	F8E4M3S        []byte  `protobuf:"bytes,28,opt,name=f8e4m3s,proto3" json:"f8e4m3s,omitempty"`
	F8E4M3Fns      []byte  `protobuf:"bytes,20,opt,name=f8e4m3fns,proto3" json:"f8e4m3fns,omitempty"`
	F8E4M3B11Fnuzs []byte  `protobuf:"bytes,23,opt,name=f8e4m3b11fnuzs,proto3" json:"f8e4m3b11fnuzs,omitempty"`
	F8E5M2Fnuzs    []byte  `protobuf:"bytes,24,opt,name=f8e5m2fnuzs,proto3" json:"f8e5m2fnuzs,omitempty"`
	F8E4M3Fnuzs    []byte  `protobuf:"bytes,25,opt,name=f8e4m3fnuzs,proto3" json:"f8e4m3fnuzs,omitempty"`
	F8E3M4S        []byte  `protobuf:"bytes,29,opt,name=f8e3m4s,proto3" json:"f8e3m4s,omitempty"`
	SparseIndices  []int64 `protobuf:"varint,14,rep,packed,name=sparse_indices,json=sparseIndices,proto3" json:"sparse_indices,omitempty"` // Next = 30
	// contains filtered or unexported fields
}

Literals are used when the server and client need to exchange materialized data / results. Literals are also used to describe constants used in computations.

Transfers to/from the client are encoded in literal form, and the structure of the repeated fields is implied by the shape.

func (*LiteralProto) Descriptor deprecated 

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

Deprecated: Use LiteralProto.ProtoReflect.Descriptor instead.

func (*LiteralProto) GetBf16S 

func (x *LiteralProto) GetBf16S() []byte

func (*LiteralProto) GetC128S 

func (x *LiteralProto) GetC128S() []float64

func (*LiteralProto) GetC64S 

func (x *LiteralProto) GetC64S() []float32

func (*LiteralProto) GetF16S 

func (x *LiteralProto) GetF16S() []byte

func (*LiteralProto) GetF32S 

func (x *LiteralProto) GetF32S() []float32

func (*LiteralProto) GetF64S 

func (x *LiteralProto) GetF64S() []float64

func (*LiteralProto) GetF8E3M4S added in v0.4.7 

func (x *LiteralProto) GetF8E3M4S() []byte

func (*LiteralProto) GetF8E4M3B11Fnuzs 

func (x *LiteralProto) GetF8E4M3B11Fnuzs() []byte

func (*LiteralProto) GetF8E4M3Fns 

func (x *LiteralProto) GetF8E4M3Fns() []byte

func (*LiteralProto) GetF8E4M3Fnuzs 

func (x *LiteralProto) GetF8E4M3Fnuzs() []byte

func (*LiteralProto) GetF8E4M3S added in v0.4.7 

func (x *LiteralProto) GetF8E4M3S() []byte

func (*LiteralProto) GetF8E5M2Fnuzs 

func (x *LiteralProto) GetF8E5M2Fnuzs() []byte

func (*LiteralProto) GetF8E5M2S 

func (x *LiteralProto) GetF8E5M2S() []byte

func (*LiteralProto) GetPreds 

func (x *LiteralProto) GetPreds() []bool

func (*LiteralProto) GetS16S 

func (x *LiteralProto) GetS16S() []byte

func (*LiteralProto) GetS2S 

func (x *LiteralProto) GetS2S() []byte

func (*LiteralProto) GetS32S 

func (x *LiteralProto) GetS32S() []int32

func (*LiteralProto) GetS4S 

func (x *LiteralProto) GetS4S() []byte

func (*LiteralProto) GetS64S 

func (x *LiteralProto) GetS64S() []int64

func (*LiteralProto) GetS8S 

func (x *LiteralProto) GetS8S() []byte

func (*LiteralProto) GetShape 

func (x *LiteralProto) GetShape() *ShapeProto

func (*LiteralProto) GetSparseIndices 

func (x *LiteralProto) GetSparseIndices() []int64

func (*LiteralProto) GetTupleLiterals 

func (x *LiteralProto) GetTupleLiterals() []*LiteralProto

func (*LiteralProto) GetU16S 

func (x *LiteralProto) GetU16S() []byte

func (*LiteralProto) GetU2S 

func (x *LiteralProto) GetU2S() []byte

func (*LiteralProto) GetU32S 

func (x *LiteralProto) GetU32S() []uint32

func (*LiteralProto) GetU4S 

func (x *LiteralProto) GetU4S() []byte

func (*LiteralProto) GetU64S 

func (x *LiteralProto) GetU64S() []uint64

func (*LiteralProto) GetU8S 

func (x *LiteralProto) GetU8S() []byte

func (*LiteralProto) ProtoMessage 

func (*LiteralProto) ProtoMessage()

func (*LiteralProto) ProtoReflect 

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

func (*LiteralProto) Reset 

func (x *LiteralProto) Reset()

func (*LiteralProto) String 

func (x *LiteralProto) String() string

type OpMetadata 

type OpMetadata struct {

	// The framework op name that generated this XLA op.
	//
	// Frameworks that build on top of XLA should mirror the names of their ops
	// back to users by specifying the op_type. In this way, even if the
	// framework's "ops" are implemented as multiple XLA HLO Ops, they can be
	// grouped appropriately. (e.g. if a SoftMax layer is emitted into XLA as
	// multiple ops, then each op should have the op_type be "SoftMax".)
	OpType string `protobuf:"bytes,1,opt,name=op_type,json=opType,proto3" json:"op_type,omitempty"`
	// The user-specified name of the op.
	//
	// This name is often unique within a computation. Note: some frameworks
	// add auto-generated names if the user does not provide one.
	OpName string `protobuf:"bytes,2,opt,name=op_name,json=opName,proto3" json:"op_name,omitempty"`
	// Indicate a file and line that this op is associated to in a user's program.
	//
	// e.g. it could be the file and line of user code that generated the op.
	SourceFile string `protobuf:"bytes,3,opt,name=source_file,json=sourceFile,proto3" json:"source_file,omitempty"`
	SourceLine int32  `protobuf:"varint,4,opt,name=source_line,json=sourceLine,proto3" json:"source_line,omitempty"`
	// Deprecated, use [ProfileInfo][profile_type] instead.
	//
	// Deprecated: Marked as deprecated in xla/xla_data.proto.
	ProfileType []ProfileType `protobuf:"varint,5,rep,packed,name=profile_type,json=profileType,proto3,enum=xla.ProfileType" json:"profile_type,omitempty"`
	// The footprint of the generated code for the instruction.
	SizeOfGeneratedCodeInBytes int64 `` /* 148-byte string literal not displayed */
	// The size of the working set, i.e., the amount of memory, used by the
	// instruction in a compiler-managed fast device memory.
	SizeOfMemoryWorkingSetInBytes int64 `` /* 159-byte string literal not displayed */
	// Profile information for the Op.
	ProfileInfo *OpMetadata_ProfileInfo `protobuf:"bytes,10,opt,name=profile_info,json=profileInfo,proto3" json:"profile_info,omitempty"`
	// Deduplicated HLO name for this op. In some cases, we can have multiple
	// instructions (e.g. fusions) that are considered duplicates. We want to
	// group them together under the same name so that we can group them together
	// during analysis (e.g. HLO Op Profile tool in Xprof).
	// E.g. If we have fusion.1, fusion.2, and fusion.3 marked as duplicates,
	// fusion.2 and fusion.3 will have deduplicated_name = fusion.1
	DeduplicatedName string `protobuf:"bytes,12,opt,name=deduplicated_name,json=deduplicatedName,proto3" json:"deduplicated_name,omitempty"`
	// Whether to preserve the layout of the HLO op.
	PreserveLayout bool `protobuf:"varint,13,opt,name=preserve_layout,json=preserveLayout,proto3" json:"preserve_layout,omitempty"`
	// 1-based position of the frame in frames flat array.
	// Ids are 1-based to keep 0 value as representation of non-set property.
	StackFrameId int32 `protobuf:"varint,15,opt,name=stack_frame_id,json=stackFrameId,proto3" json:"stack_frame_id,omitempty"`
	// Instruction name available upon scheduling.
	SchedulingName string `protobuf:"bytes,16,opt,name=scheduling_name,json=schedulingName,proto3" json:"scheduling_name,omitempty"`
	// contains filtered or unexported fields
}

Symbolization metadata for HLO Instructions.

This metadata is used for debugging XLA code generation, as well as performance profiling of XLA-generated executables.

func (*OpMetadata) Descriptor deprecated 

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

Deprecated: Use OpMetadata.ProtoReflect.Descriptor instead.

func (*OpMetadata) GetDeduplicatedName 

func (x *OpMetadata) GetDeduplicatedName() string

func (*OpMetadata) GetOpName 

func (x *OpMetadata) GetOpName() string

func (*OpMetadata) GetOpType 

func (x *OpMetadata) GetOpType() string

func (*OpMetadata) GetPreserveLayout 

func (x *OpMetadata) GetPreserveLayout() bool

func (*OpMetadata) GetProfileInfo 

func (x *OpMetadata) GetProfileInfo() *OpMetadata_ProfileInfo

func (*OpMetadata) GetProfileType deprecated 

func (x *OpMetadata) GetProfileType() []ProfileType

Deprecated: Marked as deprecated in xla/xla_data.proto.

func (*OpMetadata) GetSchedulingName 

func (x *OpMetadata) GetSchedulingName() string

func (*OpMetadata) GetSizeOfGeneratedCodeInBytes 

func (x *OpMetadata) GetSizeOfGeneratedCodeInBytes() int64

func (*OpMetadata) GetSizeOfMemoryWorkingSetInBytes 

func (x *OpMetadata) GetSizeOfMemoryWorkingSetInBytes() int64

func (*OpMetadata) GetSourceFile 

func (x *OpMetadata) GetSourceFile() string

func (*OpMetadata) GetSourceLine 

func (x *OpMetadata) GetSourceLine() int32

func (*OpMetadata) GetStackFrameId 

func (x *OpMetadata) GetStackFrameId() int32

func (*OpMetadata) ProtoMessage 

func (*OpMetadata) ProtoMessage()

func (*OpMetadata) ProtoReflect 

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

func (*OpMetadata) Reset 

func (x *OpMetadata) Reset()

func (*OpMetadata) String 

func (x *OpMetadata) String() string

type OpMetadata_ProfileInfo 

type OpMetadata_ProfileInfo struct {

	// The type of optimization profiles that this operation contains.
	ProfileType []ProfileType `protobuf:"varint,1,rep,packed,name=profile_type,json=profileType,proto3,enum=xla.ProfileType" json:"profile_type,omitempty"`
	// Speedup of tuned config compared to default config.
	// TODO(b/203817882) Set the relative_speedup.
	RelativeSpeedup float64 `protobuf:"fixed64,2,opt,name=relative_speedup,json=relativeSpeedup,proto3" json:"relative_speedup,omitempty"`
	// The source of the optimization profiles that this operation contains.
	ProfileSource ProfileSource `protobuf:"varint,3,opt,name=profile_source,json=profileSource,proto3,enum=xla.ProfileSource" json:"profile_source,omitempty"`
	// The compilation event that triggered the use of the profiles.
	CompilationEvent CompilationEvent `` /* 136-byte string literal not displayed */
	// contains filtered or unexported fields
}

Information about the optimization profile that this operation contains.

func (*OpMetadata_ProfileInfo) Descriptor deprecated 

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

Deprecated: Use OpMetadata_ProfileInfo.ProtoReflect.Descriptor instead.

func (*OpMetadata_ProfileInfo) GetCompilationEvent 

func (x *OpMetadata_ProfileInfo) GetCompilationEvent() CompilationEvent

func (*OpMetadata_ProfileInfo) GetProfileSource 

func (x *OpMetadata_ProfileInfo) GetProfileSource() ProfileSource

func (*OpMetadata_ProfileInfo) GetProfileType 

func (x *OpMetadata_ProfileInfo) GetProfileType() []ProfileType

func (*OpMetadata_ProfileInfo) GetRelativeSpeedup 

func (x *OpMetadata_ProfileInfo) GetRelativeSpeedup() float64

func (*OpMetadata_ProfileInfo) ProtoMessage 

func (*OpMetadata_ProfileInfo) ProtoMessage()

func (*OpMetadata_ProfileInfo) ProtoReflect 

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

func (*OpMetadata_ProfileInfo) Reset 

func (x *OpMetadata_ProfileInfo) Reset()

func (*OpMetadata_ProfileInfo) String 

func (x *OpMetadata_ProfileInfo) String() string

type OpSharding 

type OpSharding struct {
	Type OpSharding_Type `protobuf:"varint,1,opt,name=type,proto3,enum=xla.OpSharding_Type" json:"type,omitempty"`
	// The shape of the sharded tile.
	TileShape *ShapeProto `protobuf:"bytes,2,opt,name=tile_shape,json=tileShape,proto3" json:"tile_shape,omitempty"`
	// The shape of the tile assignment tensor - this must be the same rank as
	// tile_shape and the product of its dimensions must equal
	// tile_assignment_devices.size().
	TileAssignmentDimensions []int64 `` /* 143-byte string literal not displayed */
	// Flattened list of device IDs. The order of flattening is the same as used
	// by IndexUtil::MultiToLinearIndex(tile_assignment_shape).
	// Only one of tile_assignment_devices and iota_dimensions shall be non-empty.
	TileAssignmentDevices []int64 `` /* 134-byte string literal not displayed */
	// If type == TUPLE, the sub-shardings, one per leaf node in the tuple shape,
	// in pre-order. The tuple shape could be nested; here we store just a
	// flattened list of all leaves in the tuple shape. Note that the tuple shape
	// is not stored here; shardings do not store the shapes to which they are
	// applied, this is inferred from the instruction this sharding gets attached
	// to.
	TupleShardings []*OpSharding `protobuf:"bytes,5,rep,name=tuple_shardings,json=tupleShardings,proto3" json:"tuple_shardings,omitempty"`
	// Only used for OTHER type. If true, data is sharded according to other
	// dimensions of tile_assignment(), but replicated across devices along the
	// last dimension. (Experimental)
	ReplicateOnLastTileDim bool `` /* 134-byte string literal not displayed */
	// This field is used to track the source of this sharding, usually derived
	// from instructions. Multple metadata may be populated if sharding is
	// combined with other shardings.  Metadata are to not be populated when
	// type == TUPLE and instead metadata should be set on individual tuple
	// elements.
	Metadata []*OpMetadata `protobuf:"bytes,7,rep,name=metadata,proto3" json:"metadata,omitempty"`
	// This field is used to represented the sharding type of each subgroup.
	// For example, sharding={devices=[2,2,2,2]0,1,2,...,15 last_tile_dims={
	// replicate, manual, unreduced}} means that each of the last 3 dimensions
	// in [2,2,2,2] represents a subgrouping in replicate, manual,
	// unreduced sharding type respectively.
	LastTileDims []OpSharding_Type `` /* 132-byte string literal not displayed */
	// Dimensions used to reshape the 1D iota array of device IDs.
	// Only one of tile_assignment_devices and iota_reshape_dims shall be
	// non-empty.
	IotaReshapeDims []int64 `protobuf:"varint,9,rep,packed,name=iota_reshape_dims,json=iotaReshapeDims,proto3" json:"iota_reshape_dims,omitempty"`
	// Dimension permutations to transposed the iota array reshaped to
	// iota_reshape_dims. This must have the same size as iota_reshape_dims.
	IotaTransposePerm []int32 `protobuf:"varint,10,rep,packed,name=iota_transpose_perm,json=iotaTransposePerm,proto3" json:"iota_transpose_perm,omitempty"`
	// This field decides whether this op is in a shard group.
	IsShardGroup bool `protobuf:"varint,11,opt,name=is_shard_group,json=isShardGroup,proto3" json:"is_shard_group,omitempty"`
	// This field is used to store the unique id of the shard group.
	ShardGroupId   int64                     `protobuf:"varint,12,opt,name=shard_group_id,json=shardGroupId,proto3" json:"shard_group_id,omitempty"`
	ShardGroupType OpSharding_ShardGroupType `` /* 142-byte string literal not displayed */
	// contains filtered or unexported fields
}

LINT.IfChange

func (*OpSharding) Descriptor deprecated 

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

Deprecated: Use OpSharding.ProtoReflect.Descriptor instead.

func (*OpSharding) GetIotaReshapeDims 

func (x *OpSharding) GetIotaReshapeDims() []int64

func (*OpSharding) GetIotaTransposePerm 

func (x *OpSharding) GetIotaTransposePerm() []int32

func (*OpSharding) GetIsShardGroup 

func (x *OpSharding) GetIsShardGroup() bool

func (*OpSharding) GetLastTileDims 

func (x *OpSharding) GetLastTileDims() []OpSharding_Type

func (*OpSharding) GetMetadata 

func (x *OpSharding) GetMetadata() []*OpMetadata

func (*OpSharding) GetReplicateOnLastTileDim 

func (x *OpSharding) GetReplicateOnLastTileDim() bool

func (*OpSharding) GetShardGroupId 

func (x *OpSharding) GetShardGroupId() int64

func (*OpSharding) GetShardGroupType 

func (x *OpSharding) GetShardGroupType() OpSharding_ShardGroupType

func (*OpSharding) GetTileAssignmentDevices 

func (x *OpSharding) GetTileAssignmentDevices() []int64

func (*OpSharding) GetTileAssignmentDimensions 

func (x *OpSharding) GetTileAssignmentDimensions() []int64

func (*OpSharding) GetTileShape 

func (x *OpSharding) GetTileShape() *ShapeProto

func (*OpSharding) GetTupleShardings 

func (x *OpSharding) GetTupleShardings() []*OpSharding

func (*OpSharding) GetType 

func (x *OpSharding) GetType() OpSharding_Type

func (*OpSharding) ProtoMessage 

func (*OpSharding) ProtoMessage()

func (*OpSharding) ProtoReflect 

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

func (*OpSharding) Reset 

func (x *OpSharding) Reset()

func (*OpSharding) String 

func (x *OpSharding) String() string

type OpSharding_ShardGroupType 

type OpSharding_ShardGroupType int32

Used to decide whether this op is to be sharded like some other ops, or to which other ops will be sharded like. 

const (
	// This op will be sharded exactly the same as the other op. (hard
	// restriction)
	OpSharding_AS OpSharding_ShardGroupType = 0
	// This op will try to allow sharding propagation within the same group even
	// there is no data dependencies among them, but there is no guarantee that
	// the final shardings within the same group will be exactly the same. (soft
	// restriction)
	OpSharding_LIKE OpSharding_ShardGroupType = 1
)

func (OpSharding_ShardGroupType) Descriptor 

func (OpSharding_ShardGroupType) Descriptor() protoreflect.EnumDescriptor

func (OpSharding_ShardGroupType) Enum 

func (x OpSharding_ShardGroupType) Enum() *OpSharding_ShardGroupType

func (OpSharding_ShardGroupType) EnumDescriptor deprecated 

func (OpSharding_ShardGroupType) EnumDescriptor() ([]byte, []int)

Deprecated: Use OpSharding_ShardGroupType.Descriptor instead.

func (OpSharding_ShardGroupType) Number 

func (x OpSharding_ShardGroupType) Number() protoreflect.EnumNumber

func (OpSharding_ShardGroupType) String 

func (x OpSharding_ShardGroupType) String() string

func (OpSharding_ShardGroupType) Type 

func (OpSharding_ShardGroupType) Type() protoreflect.EnumType

type OpSharding_Type 

type OpSharding_Type int32
const (
	// This sharding is replicated across all devices (implies maximal,
	// all other fields are unused).
	OpSharding_REPLICATED OpSharding_Type = 0
	// This sharding is maximal - one device runs the entire operation.
	OpSharding_MAXIMAL OpSharding_Type = 1
	// This sharding is a tuple - only the tuple_shardings field is valid.
	OpSharding_TUPLE OpSharding_Type = 2
	// None of the above; tile_shape and tile_assignment are both used.
	OpSharding_OTHER OpSharding_Type = 3
	// This op is manually sharded: the shapes are already partitioned and the
	// partitioner should not change this op.
	OpSharding_MANUAL OpSharding_Type = 4
	// This sharding is a placeholder sharding with lowest precedence, it can be
	// overwriten by any other shardings.
	OpSharding_UNKNOWN OpSharding_Type = 5
)

func (OpSharding_Type) Descriptor 

func (OpSharding_Type) Descriptor() protoreflect.EnumDescriptor

func (OpSharding_Type) Enum 

func (x OpSharding_Type) Enum() *OpSharding_Type

func (OpSharding_Type) EnumDescriptor deprecated 

func (OpSharding_Type) EnumDescriptor() ([]byte, []int)

Deprecated: Use OpSharding_Type.Descriptor instead.

func (OpSharding_Type) Number 

func (x OpSharding_Type) Number() protoreflect.EnumNumber

func (OpSharding_Type) String 

func (x OpSharding_Type) String() string

func (OpSharding_Type) Type 

func (OpSharding_Type) Type() protoreflect.EnumType

type OriginalArrayProto 

type OriginalArrayProto struct {
	LeafShapeIndex  []int64 `protobuf:"varint,1,rep,packed,name=leaf_shape_index,json=leafShapeIndex,proto3" json:"leaf_shape_index,omitempty"`
	InstructionName string  `protobuf:"bytes,2,opt,name=instruction_name,json=instructionName,proto3" json:"instruction_name,omitempty"`
	ShapeIndex      []int64 `protobuf:"varint,3,rep,packed,name=shape_index,json=shapeIndex,proto3" json:"shape_index,omitempty"`
	// contains filtered or unexported fields
}

func (*OriginalArrayProto) Descriptor deprecated 

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

Deprecated: Use OriginalArrayProto.ProtoReflect.Descriptor instead.

func (*OriginalArrayProto) GetInstructionName 

func (x *OriginalArrayProto) GetInstructionName() string

func (*OriginalArrayProto) GetLeafShapeIndex 

func (x *OriginalArrayProto) GetLeafShapeIndex() []int64

func (*OriginalArrayProto) GetShapeIndex 

func (x *OriginalArrayProto) GetShapeIndex() []int64

func (*OriginalArrayProto) ProtoMessage 

func (*OriginalArrayProto) ProtoMessage()

func (*OriginalArrayProto) ProtoReflect 

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

func (*OriginalArrayProto) Reset 

func (x *OriginalArrayProto) Reset()

func (*OriginalArrayProto) String 

func (x *OriginalArrayProto) String() string

type OriginalValueProto 

type OriginalValueProto struct {
	Leaves []*OriginalArrayProto `protobuf:"bytes,1,rep,name=leaves,proto3" json:"leaves,omitempty"`
	// contains filtered or unexported fields
}

func (*OriginalValueProto) Descriptor deprecated 

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

Deprecated: Use OriginalValueProto.ProtoReflect.Descriptor instead.

func (*OriginalValueProto) GetLeaves 

func (x *OriginalValueProto) GetLeaves() []*OriginalArrayProto

func (*OriginalValueProto) ProtoMessage 

func (*OriginalValueProto) ProtoMessage()

func (*OriginalValueProto) ProtoReflect 

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

func (*OriginalValueProto) Reset 

func (x *OriginalValueProto) Reset()

func (*OriginalValueProto) String 

func (x *OriginalValueProto) String() string

type OutputOperandAliasing 

type OutputOperandAliasing struct {
	OutputShapeIndex  []int64 `protobuf:"varint,1,rep,packed,name=output_shape_index,json=outputShapeIndex,proto3" json:"output_shape_index,omitempty"`
	OperandIndex      int64   `protobuf:"varint,2,opt,name=operand_index,json=operandIndex,proto3" json:"operand_index,omitempty"`
	OperandShapeIndex []int64 `protobuf:"varint,3,rep,packed,name=operand_shape_index,json=operandShapeIndex,proto3" json:"operand_shape_index,omitempty"`
	// contains filtered or unexported fields
}

Specifies a pair of output/operand buffers that alias each other for kCustomCall and kFusion

func (*OutputOperandAliasing) Descriptor deprecated 

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

Deprecated: Use OutputOperandAliasing.ProtoReflect.Descriptor instead.

func (*OutputOperandAliasing) GetOperandIndex 

func (x *OutputOperandAliasing) GetOperandIndex() int64

func (*OutputOperandAliasing) GetOperandShapeIndex 

func (x *OutputOperandAliasing) GetOperandShapeIndex() []int64

func (*OutputOperandAliasing) GetOutputShapeIndex 

func (x *OutputOperandAliasing) GetOutputShapeIndex() []int64

func (*OutputOperandAliasing) ProtoMessage 

func (*OutputOperandAliasing) ProtoMessage()

func (*OutputOperandAliasing) ProtoReflect 

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

func (*OutputOperandAliasing) Reset 

func (x *OutputOperandAliasing) Reset()

func (*OutputOperandAliasing) String 

func (x *OutputOperandAliasing) String() string

type PaddingConfig 

type PaddingConfig struct {

	// The padding configuration for all dimensions.
	Dimensions []*PaddingConfig_PaddingConfigDimension `protobuf:"bytes,1,rep,name=dimensions,proto3" json:"dimensions,omitempty"`
	// contains filtered or unexported fields
}

Describes the padding configuration for Pad operation. The padding amount on both edges as well as between the elements are specified for each dimension.

func (*PaddingConfig) Descriptor deprecated 

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

Deprecated: Use PaddingConfig.ProtoReflect.Descriptor instead.

func (*PaddingConfig) GetDimensions 

func (x *PaddingConfig) GetDimensions() []*PaddingConfig_PaddingConfigDimension

func (*PaddingConfig) ProtoMessage 

func (*PaddingConfig) ProtoMessage()

func (*PaddingConfig) ProtoReflect 

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

func (*PaddingConfig) Reset 

func (x *PaddingConfig) Reset()

func (*PaddingConfig) String 

func (x *PaddingConfig) String() string

type PaddingConfig_PaddingConfigDimension 

type PaddingConfig_PaddingConfigDimension struct {

	// Padding amount on the low-end (next to the index 0). May be negative.
	EdgePaddingLow int64 `protobuf:"varint,1,opt,name=edge_padding_low,json=edgePaddingLow,proto3" json:"edge_padding_low,omitempty"`
	// Padding amount on the high-end (next to the highest index). May be
	// negative.
	EdgePaddingHigh int64 `protobuf:"varint,2,opt,name=edge_padding_high,json=edgePaddingHigh,proto3" json:"edge_padding_high,omitempty"`
	// Padding amount between the elements. May not be negative.
	InteriorPadding int64 `protobuf:"varint,3,opt,name=interior_padding,json=interiorPadding,proto3" json:"interior_padding,omitempty"`
	// contains filtered or unexported fields
}

Describes the padding configuration for a dimension.

func (*PaddingConfig_PaddingConfigDimension) Descriptor deprecated 

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

Deprecated: Use PaddingConfig_PaddingConfigDimension.ProtoReflect.Descriptor instead.

func (*PaddingConfig_PaddingConfigDimension) GetEdgePaddingHigh 

func (x *PaddingConfig_PaddingConfigDimension) GetEdgePaddingHigh() int64

func (*PaddingConfig_PaddingConfigDimension) GetEdgePaddingLow 

func (x *PaddingConfig_PaddingConfigDimension) GetEdgePaddingLow() int64

func (*PaddingConfig_PaddingConfigDimension) GetInteriorPadding 

func (x *PaddingConfig_PaddingConfigDimension) GetInteriorPadding() int64

func (*PaddingConfig_PaddingConfigDimension) ProtoMessage 

func (*PaddingConfig_PaddingConfigDimension) ProtoMessage()

func (*PaddingConfig_PaddingConfigDimension) ProtoReflect 

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

func (*PaddingConfig_PaddingConfigDimension) Reset 

func (x *PaddingConfig_PaddingConfigDimension) Reset()

func (*PaddingConfig_PaddingConfigDimension) String 

func (x *PaddingConfig_PaddingConfigDimension) String() string

type PaddingType 

type PaddingType int32
const (
	PaddingType_PADDING_INVALID PaddingType = 0
	PaddingType_PADDING_VALID   PaddingType = 1 // Only valid portion of the base are covered.
	PaddingType_PADDING_SAME    PaddingType = 2 // Extra is added to produce same output size as the input.
)

func (PaddingType) Descriptor 

func (PaddingType) Descriptor() protoreflect.EnumDescriptor

func (PaddingType) Enum 

func (x PaddingType) Enum() *PaddingType

func (PaddingType) EnumDescriptor deprecated 

func (PaddingType) EnumDescriptor() ([]byte, []int)

Deprecated: Use PaddingType.Descriptor instead.

func (PaddingType) Number 

func (x PaddingType) Number() protoreflect.EnumNumber

func (PaddingType) String 

func (x PaddingType) String() string

func (PaddingType) Type 

func (PaddingType) Type() protoreflect.EnumType

type ParameterReplication 

type ParameterReplication struct {

	// A list of boolean values for the flattened leaf buffers. Each value
	// indicates whether the corresponding leaf buffer is replicated.
	//
	// If this field is empty, it means no buffer is replicated. Otherwise, the
	// number of elements in this field must match the number of leaf buffers in
	// the HLO instruction's shape.
	ReplicatedAtLeafBuffers []bool `` /* 142-byte string literal not displayed */
	// contains filtered or unexported fields
}

Describes whether all data-parallelism replicas will receive the same parameter data at each buffer.

func (*ParameterReplication) Descriptor deprecated 

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

Deprecated: Use ParameterReplication.ProtoReflect.Descriptor instead.

func (*ParameterReplication) GetReplicatedAtLeafBuffers 

func (x *ParameterReplication) GetReplicatedAtLeafBuffers() []bool

func (*ParameterReplication) ProtoMessage 

func (*ParameterReplication) ProtoMessage()

func (*ParameterReplication) ProtoReflect 

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

func (*ParameterReplication) Reset 

func (x *ParameterReplication) Reset()

func (*ParameterReplication) String 

func (x *ParameterReplication) String() string

type PrecisionConfig 

type PrecisionConfig struct {
	OperandPrecision []PrecisionConfig_Precision `` /* 152-byte string literal not displayed */
	// Currently doesn't do anything, but we plan to support it for dot and
	// possibly more instructions.
	//
	// TODO(b/316147294): Support this on GPU and add this to StableHLO as well.
	//
	// If this is set, then `operand_precision` should be set to DEFAULT and it
	// will be ignored.
	Algorithm PrecisionConfig_Algorithm `protobuf:"varint,2,opt,name=algorithm,proto3,enum=xla.PrecisionConfig_Algorithm" json:"algorithm,omitempty"`
	// contains filtered or unexported fields
}

Used to indicate the precision configuration. It has backend specific meaning.

func (*PrecisionConfig) Descriptor deprecated 

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

Deprecated: Use PrecisionConfig.ProtoReflect.Descriptor instead.

func (*PrecisionConfig) GetAlgorithm 

func (x *PrecisionConfig) GetAlgorithm() PrecisionConfig_Algorithm

func (*PrecisionConfig) GetOperandPrecision 

func (x *PrecisionConfig) GetOperandPrecision() []PrecisionConfig_Precision

func (*PrecisionConfig) ProtoMessage 

func (*PrecisionConfig) ProtoMessage()

func (*PrecisionConfig) ProtoReflect 

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

func (*PrecisionConfig) Reset 

func (x *PrecisionConfig) Reset()

func (*PrecisionConfig) String 

func (x *PrecisionConfig) String() string

type PrecisionConfig_Algorithm 

type PrecisionConfig_Algorithm int32

The algorithm used to evaluate the instruction.

The naming convention for the dot instruction is ALG_DOT_{A_TYPE}_{B_TYPE}_{ACCUM_TYPE}[_X{NUM_OPS}] where A_TYPE, B_TYPE and ACCUM_TYPE correspond to the types in the "primitive dot operations" (such as TensorCore operations) and NUM_OPS is the number of such operations used per "primitive tile". When the NUM_OPS field is skipped, it is assumed to be 1. The types mentioned in the name are independent of the storage types.

In general ATYPE and BTYPE are the precisions that the LHS and RHS of the operation are rounded to and ACCUMTYPE is the accumulation type. If a backend does not support the given algorithm, an error is raised. The Algorithm enum is intended to eventually replace the Precision enum. 

const (
	// If the algorithm is `ALG_UNSET`, we will decide the algorithm based on
	// the operand_precision values (for now).
	PrecisionConfig_ALG_UNSET PrecisionConfig_Algorithm = 0
	// The storage type can be any 8-bit floating point type.
	PrecisionConfig_ALG_DOT_ANY_F8_ANY_F8_F32 PrecisionConfig_Algorithm = 1
	// The storage type can be any 8-bit floating point type. Intermediate
	// results will not periodically be promoted to a higher precision. This
	// corresponds to CUBLASLT_MATMUL_DESC_FAST_ACCUM. Triton's
	// maxNumImpreciseAcc=32 setting may be similar.
	PrecisionConfig_ALG_DOT_ANY_F8_ANY_F8_F32_FAST_ACCUM PrecisionConfig_Algorithm = 2
	PrecisionConfig_ALG_DOT_F16_F16_F16                  PrecisionConfig_Algorithm = 3
	PrecisionConfig_ALG_DOT_F16_F16_F32                  PrecisionConfig_Algorithm = 4
	PrecisionConfig_ALG_DOT_BF16_BF16_BF16               PrecisionConfig_Algorithm = 5
	PrecisionConfig_ALG_DOT_BF16_BF16_F32                PrecisionConfig_Algorithm = 6
	// An algorithm which uses 3 BF16_BF16_F32 matmuls to achieve better
	// precision.
	PrecisionConfig_ALG_DOT_BF16_BF16_F32_X3 PrecisionConfig_Algorithm = 7
	// An algorithm which uses 6 BF16_BF16_F32 matmuls to achieve better
	// precision (similar to F32).
	PrecisionConfig_ALG_DOT_BF16_BF16_F32_X6 PrecisionConfig_Algorithm = 8
	PrecisionConfig_ALG_DOT_TF32_TF32_F32    PrecisionConfig_Algorithm = 9
	// An algorithm which uses 3 TF32_TF32_F32 matmuls to achieve better
	// precision (similar to F32).
	PrecisionConfig_ALG_DOT_TF32_TF32_F32_X3 PrecisionConfig_Algorithm = 10
	PrecisionConfig_ALG_DOT_F32_F32_F32      PrecisionConfig_Algorithm = 11
	PrecisionConfig_ALG_DOT_F64_F64_F64      PrecisionConfig_Algorithm = 12
)

func (PrecisionConfig_Algorithm) Descriptor 

func (PrecisionConfig_Algorithm) Descriptor() protoreflect.EnumDescriptor

func (PrecisionConfig_Algorithm) Enum 

func (x PrecisionConfig_Algorithm) Enum() *PrecisionConfig_Algorithm

func (PrecisionConfig_Algorithm) EnumDescriptor deprecated 

func (PrecisionConfig_Algorithm) EnumDescriptor() ([]byte, []int)

Deprecated: Use PrecisionConfig_Algorithm.Descriptor instead.

func (PrecisionConfig_Algorithm) Number 

func (x PrecisionConfig_Algorithm) Number() protoreflect.EnumNumber

func (PrecisionConfig_Algorithm) String 

func (x PrecisionConfig_Algorithm) String() string

func (PrecisionConfig_Algorithm) Type 

func (PrecisionConfig_Algorithm) Type() protoreflect.EnumType

type PrecisionConfig_Precision 

type PrecisionConfig_Precision int32
const (
	PrecisionConfig_DEFAULT PrecisionConfig_Precision = 0
	PrecisionConfig_HIGH    PrecisionConfig_Precision = 1
	PrecisionConfig_HIGHEST PrecisionConfig_Precision = 2
	// Each U8/S8 value in a tensor actually represents 2 nibble values.
	PrecisionConfig_PACKED_NIBBLE PrecisionConfig_Precision = 3
)

func (PrecisionConfig_Precision) Descriptor 

func (PrecisionConfig_Precision) Descriptor() protoreflect.EnumDescriptor

func (PrecisionConfig_Precision) Enum 

func (x PrecisionConfig_Precision) Enum() *PrecisionConfig_Precision

func (PrecisionConfig_Precision) EnumDescriptor deprecated 

func (PrecisionConfig_Precision) EnumDescriptor() ([]byte, []int)

Deprecated: Use PrecisionConfig_Precision.Descriptor instead.

func (PrecisionConfig_Precision) Number 

func (x PrecisionConfig_Precision) Number() protoreflect.EnumNumber

func (PrecisionConfig_Precision) String 

func (x PrecisionConfig_Precision) String() string

func (PrecisionConfig_Precision) Type 

func (PrecisionConfig_Precision) Type() protoreflect.EnumType

type PrimitiveType 

type PrimitiveType int32

Primitive types are the individual values that can be held in rectangular multidimensional arrays. A description of the rectangular multidimensional array dimensions / primitive type is given by Shape, below.

LINT.IfChange 

const (
	// Invalid primitive type to serve as default.
	PrimitiveType_PRIMITIVE_TYPE_INVALID PrimitiveType = 0
	// Predicates are two-state booleans.
	PrimitiveType_PRED PrimitiveType = 1
	// Signed integral values of fixed width.
	PrimitiveType_S2  PrimitiveType = 26
	PrimitiveType_S4  PrimitiveType = 21
	PrimitiveType_S8  PrimitiveType = 2
	PrimitiveType_S16 PrimitiveType = 3
	PrimitiveType_S32 PrimitiveType = 4
	PrimitiveType_S64 PrimitiveType = 5
	// Unsigned integral values of fixed width.
	PrimitiveType_U2  PrimitiveType = 27
	PrimitiveType_U4  PrimitiveType = 22
	PrimitiveType_U8  PrimitiveType = 6
	PrimitiveType_U16 PrimitiveType = 7
	PrimitiveType_U32 PrimitiveType = 8
	PrimitiveType_U64 PrimitiveType = 9
	// Floating-point values of fixed width.
	//
	// Note: if f16s are not natively supported on the device, they will be
	// converted to f16 from f32 at arbirary points in the computation.
	PrimitiveType_F16 PrimitiveType = 10
	PrimitiveType_F32 PrimitiveType = 11
	// Truncated 16 bit floating-point format. This is similar to IEEE's 16 bit
	// floating-point format, but uses 1 bit for the sign, 8 bits for the exponent
	// and 7 bits for the mantissa.
	PrimitiveType_BF16 PrimitiveType = 16
	PrimitiveType_F64  PrimitiveType = 12
	// FP8 dtypes, as described in this paper: https://arxiv.org/abs/2209.05433
	//
	// F8E5M2 has 5 exponent bits and 2 mantissa bits, and is similar to the
	// existing IEEE types.
	//
	// F8E4M3 has 4 exponent bits and 3 mantissa bits, and is similar to the
	// existing IEEE types.
	//
	// F8E4M3FN has 4 exponent bits and 3 mantissa bits. The "FN" means only
	// Finite and NaN values are supported. Unlike IEEE types, infinities are not
	// supported.  NaN is represented when the exponent and mantissa bits are all
	// 1s. All other values are finite.
	//
	// F8E4M3B11FNUZ has 4 exponent bits and 3 mantissa bits and a bias of 11. The
	// "FNUZ" means only Finite and NaN values are supported; zero is unsigned.
	// Unlike IEEE types, infinities are not supported.  NaN is represented when
	// the exponent and mantissa bits are all 0s with a sign bit of 1. All other
	// values are finite.
	//
	// F8E3M4 has 3 exponent bits and 4 mantissa bits, and is similar to the
	// existing IEEE types.
	//
	// Support for these dtypes is under development. They do not yet work
	// properly in most cases.
	// TODO(b/259609697): Fully support FP8.
	PrimitiveType_F8E5M2        PrimitiveType = 19
	PrimitiveType_F8E4M3        PrimitiveType = 28
	PrimitiveType_F8E4M3FN      PrimitiveType = 20
	PrimitiveType_F8E4M3B11FNUZ PrimitiveType = 23
	PrimitiveType_F8E3M4        PrimitiveType = 29
	// FP8 dtypes, as described in this paper: https://arxiv.org/abs/2206.02915
	//
	// F8E5M2FNUZ has 5 exponent bits and 2 mantissa bits.
	// F8E4M3FNUZ has 4 exponent bits and 3 mantissa bits.
	//
	// The "FNUZ" means only Finite and NaN values are supported; zero is
	// unsigned. Unlike IEEE types, infinities are not supported.  NaN is
	// represented when the exponent and mantissa bits are all 0s with a sign bit
	// of 1. All other values are finite.
	//
	// These differences mean there's an additional exponent value available. To
	// keep the same dynamic range as an IEEE-like FP8 type, the exponent is
	// biased one more than would be expected given the number of exponent bits
	// (8 for Float8E4M3FNUZ and 16 for Float8E5M2FNUZ).
	PrimitiveType_F8E5M2FNUZ PrimitiveType = 24
	PrimitiveType_F8E4M3FNUZ PrimitiveType = 25
	// Complex values of fixed width.
	PrimitiveType_C64  PrimitiveType = 15 // Paired F32 (real, imag), as in std::complex<float>.
	PrimitiveType_C128 PrimitiveType = 18 // Paired F64 (real, imag), as in std::complex<double>.
	// A tuple is a polymorphic sequence; e.g. a shape that holds different
	// sub-shapes. They are used for things like returning multiple values from a
	// computation; e.g. a computation that returns weights and biases may have a
	// signature that results in a tuple like (f32[784x2000], f32[2000])
	//
	// If a shape proto has the tuple element type, it may not have any entries
	// in the dimensions field.
	PrimitiveType_TUPLE PrimitiveType = 13
	// An opaque type used for passing context-specific data to a custom
	// operation. Shapes of this primitive type will have empty dimensions and
	// tuple_shapes fields.
	//
	// (OPAQUE would be a better name for this identifier, but that conflicts with
	// a macro defined in windows.h.)
	PrimitiveType_OPAQUE_TYPE PrimitiveType = 14
	// A token type threaded between side-effecting operations. Shapes of this
	// primitive type will have empty dimensions and tuple_shapes fields.
	PrimitiveType_TOKEN PrimitiveType = 17
)

func (PrimitiveType) Descriptor 

func (PrimitiveType) Descriptor() protoreflect.EnumDescriptor

func (PrimitiveType) Enum 

func (x PrimitiveType) Enum() *PrimitiveType

func (PrimitiveType) EnumDescriptor deprecated 

func (PrimitiveType) EnumDescriptor() ([]byte, []int)

Deprecated: Use PrimitiveType.Descriptor instead.

func (PrimitiveType) Number 

func (x PrimitiveType) Number() protoreflect.EnumNumber

func (PrimitiveType) String 

func (x PrimitiveType) String() string

func (PrimitiveType) Type 

func (PrimitiveType) Type() protoreflect.EnumType

type ProfileSource 

type ProfileSource int32

The source of the optimization profile. 

const (
	ProfileSource_PROFILE_SOURCE_UNKNOWN_SOURCE ProfileSource = 0
	ProfileSource_PROFILE_SOURCE_EMBEDDED       ProfileSource = 1
	ProfileSource_PROFILE_SOURCE_REMOTE         ProfileSource = 2
)

func (ProfileSource) Descriptor 

func (ProfileSource) Descriptor() protoreflect.EnumDescriptor

func (ProfileSource) Enum 

func (x ProfileSource) Enum() *ProfileSource

func (ProfileSource) EnumDescriptor deprecated 

func (ProfileSource) EnumDescriptor() ([]byte, []int)

Deprecated: Use ProfileSource.Descriptor instead.

func (ProfileSource) Number 

func (x ProfileSource) Number() protoreflect.EnumNumber

func (ProfileSource) String 

func (x ProfileSource) String() string

func (ProfileSource) Type 

func (ProfileSource) Type() protoreflect.EnumType

type ProfileType 

type ProfileType int32

The type optimization profiles in use for Op-level optimizations. 

const (
	ProfileType_INVALID ProfileType = 0
	ProfileType_WINDOW  ProfileType = 1
	ProfileType_FLAG    ProfileType = 2
	ProfileType_INTEGER ProfileType = 3
)

func (ProfileType) Descriptor 

func (ProfileType) Descriptor() protoreflect.EnumDescriptor

func (ProfileType) Enum 

func (x ProfileType) Enum() *ProfileType

func (ProfileType) EnumDescriptor deprecated 

func (ProfileType) EnumDescriptor() ([]byte, []int)

Deprecated: Use ProfileType.Descriptor instead.

func (ProfileType) Number 

func (x ProfileType) Number() protoreflect.EnumNumber

func (ProfileType) String 

func (x ProfileType) String() string

func (ProfileType) Type 

func (ProfileType) Type() protoreflect.EnumType

type ProgramShapeProto 

type ProgramShapeProto struct {
	Parameters     []*ShapeProto `protobuf:"bytes,1,rep,name=parameters,proto3" json:"parameters,omitempty"`
	Result         *ShapeProto   `protobuf:"bytes,2,opt,name=result,proto3" json:"result,omitempty"`
	ParameterNames []string      `protobuf:"bytes,3,rep,name=parameter_names,json=parameterNames,proto3" json:"parameter_names,omitempty"`
	// contains filtered or unexported fields
}

Shape of the parameters and output of a computation (like a traditional function signature).

func (*ProgramShapeProto) Descriptor deprecated 

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

Deprecated: Use ProgramShapeProto.ProtoReflect.Descriptor instead.

func (*ProgramShapeProto) GetParameterNames 

func (x *ProgramShapeProto) GetParameterNames() []string

func (*ProgramShapeProto) GetParameters 

func (x *ProgramShapeProto) GetParameters() []*ShapeProto

func (*ProgramShapeProto) GetResult 

func (x *ProgramShapeProto) GetResult() *ShapeProto

func (*ProgramShapeProto) ProtoMessage 

func (*ProgramShapeProto) ProtoMessage()

func (*ProgramShapeProto) ProtoReflect 

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

func (*ProgramShapeProto) Reset 

func (x *ProgramShapeProto) Reset()

func (*ProgramShapeProto) String 

func (x *ProgramShapeProto) String() string

type RaggedDotDimensionNumbers added in v0.4.7 

type RaggedDotDimensionNumbers struct {

	// The contracting and batch dimensions of the 'lhs' and 'rhs'.
	DotDimensionNumbers *DotDimensionNumbers `protobuf:"bytes,1,opt,name=dot_dimension_numbers,json=dotDimensionNumbers,proto3" json:"dot_dimension_numbers,omitempty"`
	// The dimension numbers that represent the 'lhs' ragged dimensions.
	LhsRaggedDimensions []int64 `` /* 128-byte string literal not displayed */
	// The dimension numbers that represent the 'rhs' group dimensions.
	RhsGroupDimensions []int64 `protobuf:"varint,3,rep,packed,name=rhs_group_dimensions,json=rhsGroupDimensions,proto3" json:"rhs_group_dimensions,omitempty"`
	// contains filtered or unexported fields
}

func (*RaggedDotDimensionNumbers) Descriptor deprecated added in v0.4.7 

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

Deprecated: Use RaggedDotDimensionNumbers.ProtoReflect.Descriptor instead.

func (*RaggedDotDimensionNumbers) GetDotDimensionNumbers added in v0.4.7 

func (x *RaggedDotDimensionNumbers) GetDotDimensionNumbers() *DotDimensionNumbers

func (*RaggedDotDimensionNumbers) GetLhsRaggedDimensions added in v0.4.7 

func (x *RaggedDotDimensionNumbers) GetLhsRaggedDimensions() []int64

func (*RaggedDotDimensionNumbers) GetRhsGroupDimensions added in v0.4.7 

func (x *RaggedDotDimensionNumbers) GetRhsGroupDimensions() []int64

func (*RaggedDotDimensionNumbers) ProtoMessage added in v0.4.7 

func (*RaggedDotDimensionNumbers) ProtoMessage()

func (*RaggedDotDimensionNumbers) ProtoReflect added in v0.4.7 

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

func (*RaggedDotDimensionNumbers) Reset added in v0.4.7 

func (x *RaggedDotDimensionNumbers) Reset()

func (*RaggedDotDimensionNumbers) String added in v0.4.7 

func (x *RaggedDotDimensionNumbers) String() string

type RandomAlgorithm 

type RandomAlgorithm int32
const (
	RandomAlgorithm_RNG_DEFAULT   RandomAlgorithm = 0 // Backend dependent default algorithm.
	RandomAlgorithm_RNG_THREE_FRY RandomAlgorithm = 1
	RandomAlgorithm_RNG_PHILOX    RandomAlgorithm = 2 // Next: 2
)

func (RandomAlgorithm) Descriptor 

func (RandomAlgorithm) Descriptor() protoreflect.EnumDescriptor

func (RandomAlgorithm) Enum 

func (x RandomAlgorithm) Enum() *RandomAlgorithm

func (RandomAlgorithm) EnumDescriptor deprecated 

func (RandomAlgorithm) EnumDescriptor() ([]byte, []int)

Deprecated: Use RandomAlgorithm.Descriptor instead.

func (RandomAlgorithm) Number 

func (x RandomAlgorithm) Number() protoreflect.EnumNumber

func (RandomAlgorithm) String 

func (x RandomAlgorithm) String() string

func (RandomAlgorithm) Type 

func (RandomAlgorithm) Type() protoreflect.EnumType

type RandomDistribution 

type RandomDistribution int32
const (
	RandomDistribution_RNG_INVALID RandomDistribution = 0
	// Creates a uniform-distribution-generated random number on the semi-open
	// interval [parameter[0], parameter[1]).
	RandomDistribution_RNG_UNIFORM RandomDistribution = 1
	// Creates a normal-distribution-generated random number with mean
	// parameter[0] and standard deviation parameter[1].
	RandomDistribution_RNG_NORMAL RandomDistribution = 2
)

func (RandomDistribution) Descriptor 

func (RandomDistribution) Descriptor() protoreflect.EnumDescriptor

func (RandomDistribution) Enum 

func (x RandomDistribution) Enum() *RandomDistribution

func (RandomDistribution) EnumDescriptor deprecated 

func (RandomDistribution) EnumDescriptor() ([]byte, []int)

Deprecated: Use RandomDistribution.Descriptor instead.

func (RandomDistribution) Number 

func (x RandomDistribution) Number() protoreflect.EnumNumber

func (RandomDistribution) String 

func (x RandomDistribution) String() string

func (RandomDistribution) Type 

func (RandomDistribution) Type() protoreflect.EnumType

type ReplicaGroup 

type ReplicaGroup struct {

	// The ids of the replicas that belongs to the same group. The ordering of the
	// ids matters in some ops (e.g., all-to-all).
	ReplicaIds []int64 `protobuf:"varint,1,rep,packed,name=replica_ids,json=replicaIds,proto3" json:"replica_ids,omitempty"`
	// contains filtered or unexported fields
}

Describes the replica groups in a cross replica op (e.g., all-reduce and all-to-all).

func (*ReplicaGroup) Descriptor deprecated 

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

Deprecated: Use ReplicaGroup.ProtoReflect.Descriptor instead.

func (*ReplicaGroup) GetReplicaIds 

func (x *ReplicaGroup) GetReplicaIds() []int64

func (*ReplicaGroup) ProtoMessage 

func (*ReplicaGroup) ProtoMessage()

func (*ReplicaGroup) ProtoReflect 

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

func (*ReplicaGroup) Reset 

func (x *ReplicaGroup) Reset()

func (*ReplicaGroup) String 

func (x *ReplicaGroup) String() string

type ResultAccuracy added in v0.4.7 

type ResultAccuracy struct {

	// Types that are assignable to Specs:
	//
	//	*ResultAccuracy_Mode_
	//	*ResultAccuracy_Tolerance_
	Specs isResultAccuracy_Specs `protobuf_oneof:"specs"`
	// contains filtered or unexported fields
}

Describes the types of accuracy the user can request for unary ops with multiple implementations.

func (*ResultAccuracy) Descriptor deprecated added in v0.4.7 

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

Deprecated: Use ResultAccuracy.ProtoReflect.Descriptor instead.

func (*ResultAccuracy) GetMode added in v0.4.7 

func (x *ResultAccuracy) GetMode() ResultAccuracy_Mode

func (*ResultAccuracy) GetSpecs added in v0.4.7 

func (m *ResultAccuracy) GetSpecs() isResultAccuracy_Specs

func (*ResultAccuracy) GetTolerance added in v0.4.7 

func (x *ResultAccuracy) GetTolerance() *ResultAccuracy_Tolerance

func (*ResultAccuracy) ProtoMessage added in v0.4.7 

func (*ResultAccuracy) ProtoMessage()

func (*ResultAccuracy) ProtoReflect added in v0.4.7 

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

func (*ResultAccuracy) Reset added in v0.4.7 

func (x *ResultAccuracy) Reset()

func (*ResultAccuracy) String added in v0.4.7 

func (x *ResultAccuracy) String() string

type ResultAccuracy_Mode added in v0.4.7 

type ResultAccuracy_Mode int32
const (
	ResultAccuracy_DEFAULT ResultAccuracy_Mode = 0
	ResultAccuracy_HIGHEST ResultAccuracy_Mode = 1
)

func (ResultAccuracy_Mode) Descriptor added in v0.4.7 

func (ResultAccuracy_Mode) Descriptor() protoreflect.EnumDescriptor

func (ResultAccuracy_Mode) Enum added in v0.4.7 

func (x ResultAccuracy_Mode) Enum() *ResultAccuracy_Mode

func (ResultAccuracy_Mode) EnumDescriptor deprecated added in v0.4.7 

func (ResultAccuracy_Mode) EnumDescriptor() ([]byte, []int)

Deprecated: Use ResultAccuracy_Mode.Descriptor instead.

func (ResultAccuracy_Mode) Number added in v0.4.7 

func (x ResultAccuracy_Mode) Number() protoreflect.EnumNumber

func (ResultAccuracy_Mode) String added in v0.4.7 

func (x ResultAccuracy_Mode) String() string

func (ResultAccuracy_Mode) Type added in v0.4.7 

func (ResultAccuracy_Mode) Type() protoreflect.EnumType

type ResultAccuracy_Mode_ added in v0.4.7 

type ResultAccuracy_Mode_ struct {
	// Choose either DEFAULT or HIGHEST precision implementation.
	Mode ResultAccuracy_Mode `protobuf:"varint,1,opt,name=mode,proto3,enum=xla.ResultAccuracy_Mode,oneof"`
}

type ResultAccuracy_Tolerance added in v0.4.7 

type ResultAccuracy_Tolerance struct {

	// Absolute error tolerance for unary instructions.
	Atol float64 `protobuf:"fixed64,1,opt,name=atol,proto3" json:"atol,omitempty"`
	// Relative error tolerance for unary instructions.
	Rtol float64 `protobuf:"fixed64,2,opt,name=rtol,proto3" json:"rtol,omitempty"`
	// The error in ulps (units in the last place) is relative to machine
	// precision.
	Ulps int64 `protobuf:"varint,3,opt,name=ulps,proto3" json:"ulps,omitempty"`
	// contains filtered or unexported fields
}

func (*ResultAccuracy_Tolerance) Descriptor deprecated added in v0.4.7 

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

Deprecated: Use ResultAccuracy_Tolerance.ProtoReflect.Descriptor instead.

func (*ResultAccuracy_Tolerance) GetAtol added in v0.4.7 

func (x *ResultAccuracy_Tolerance) GetAtol() float64

func (*ResultAccuracy_Tolerance) GetRtol added in v0.4.7 

func (x *ResultAccuracy_Tolerance) GetRtol() float64

func (*ResultAccuracy_Tolerance) GetUlps added in v0.4.7 

func (x *ResultAccuracy_Tolerance) GetUlps() int64

func (*ResultAccuracy_Tolerance) ProtoMessage added in v0.4.7 

func (*ResultAccuracy_Tolerance) ProtoMessage()

func (*ResultAccuracy_Tolerance) ProtoReflect added in v0.4.7 

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

func (*ResultAccuracy_Tolerance) Reset added in v0.4.7 

func (x *ResultAccuracy_Tolerance) Reset()

func (*ResultAccuracy_Tolerance) String added in v0.4.7 

func (x *ResultAccuracy_Tolerance) String() string

type ResultAccuracy_Tolerance_ added in v0.4.7 

type ResultAccuracy_Tolerance_ struct {
	Tolerance *ResultAccuracy_Tolerance `protobuf:"bytes,2,opt,name=tolerance,proto3,oneof"`
}

type ScatterDimensionNumbers 

type ScatterDimensionNumbers struct {

	// The set of dimensions in the updates shape that are window dimensions.
	UpdateWindowDims []int64 `protobuf:"varint,1,rep,packed,name=update_window_dims,json=updateWindowDims,proto3" json:"update_window_dims,omitempty"`
	// The set of window dimensions that must be inserted into the updates shape.
	InsertedWindowDims       []int64 `protobuf:"varint,2,rep,packed,name=inserted_window_dims,json=insertedWindowDims,proto3" json:"inserted_window_dims,omitempty"`
	ScatterDimsToOperandDims []int64 `` /* 147-byte string literal not displayed */
	IndexVectorDim           int64   `protobuf:"varint,4,opt,name=index_vector_dim,json=indexVectorDim,proto3" json:"index_vector_dim,omitempty"`
	// This is the batch dimension in the input.
	InputBatchingDims []int64 `protobuf:"varint,5,rep,packed,name=input_batching_dims,json=inputBatchingDims,proto3" json:"input_batching_dims,omitempty"`
	// This is the batch dimension in the index.
	ScatterIndicesBatchingDims []int64 `` /* 151-byte string literal not displayed */
	// contains filtered or unexported fields
}

Describes the dimension numbers for a scatter operation.

All the fields are similar to the corresponding fields in GatherDimensionNumbers. Differences are noted below.

func (*ScatterDimensionNumbers) Descriptor deprecated 

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

Deprecated: Use ScatterDimensionNumbers.ProtoReflect.Descriptor instead.

func (*ScatterDimensionNumbers) GetIndexVectorDim 

func (x *ScatterDimensionNumbers) GetIndexVectorDim() int64

func (*ScatterDimensionNumbers) GetInputBatchingDims 

func (x *ScatterDimensionNumbers) GetInputBatchingDims() []int64

func (*ScatterDimensionNumbers) GetInsertedWindowDims 

func (x *ScatterDimensionNumbers) GetInsertedWindowDims() []int64

func (*ScatterDimensionNumbers) GetScatterDimsToOperandDims 

func (x *ScatterDimensionNumbers) GetScatterDimsToOperandDims() []int64

func (*ScatterDimensionNumbers) GetScatterIndicesBatchingDims 

func (x *ScatterDimensionNumbers) GetScatterIndicesBatchingDims() []int64

func (*ScatterDimensionNumbers) GetUpdateWindowDims 

func (x *ScatterDimensionNumbers) GetUpdateWindowDims() []int64

func (*ScatterDimensionNumbers) ProtoMessage 

func (*ScatterDimensionNumbers) ProtoMessage()

func (*ScatterDimensionNumbers) ProtoReflect 

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

func (*ScatterDimensionNumbers) Reset 

func (x *ScatterDimensionNumbers) Reset()

func (*ScatterDimensionNumbers) String 

func (x *ScatterDimensionNumbers) String() string

type ShapeProto 

type ShapeProto struct {

	// The element type for this shape.
	ElementType PrimitiveType `protobuf:"varint,2,opt,name=element_type,json=elementType,proto3,enum=xla.PrimitiveType" json:"element_type,omitempty"`
	// The size (number of elements) for each dimension, or an upper bound on the
	// size if the dimension is dynamic.  In XLA, dimensions are numbered from 0
	// to N-1 for an N-dimensional array. The first element of 'dimensions' is the
	// size of dimension 0, the second element is the size of dimension 1, and so
	// forth.  Empty list indicates a scalar.
	//
	// If the respective element in 'is_dimension_dynamic' is true then the value
	// in this field represents an upper bound on the size of the dimension.
	Dimensions []int64 `protobuf:"varint,3,rep,packed,name=dimensions,proto3" json:"dimensions,omitempty"`
	// For tuples only, the shapes of constituent shapes in the tuple sequence.
	TupleShapes []*ShapeProto `protobuf:"bytes,4,rep,name=tuple_shapes,json=tupleShapes,proto3" json:"tuple_shapes,omitempty"`
	// The layout used to back this shape.
	Layout *LayoutProto `protobuf:"bytes,5,opt,name=layout,proto3" json:"layout,omitempty"`
	// For arrays, this indicates whether or not each dimension is
	// dynamically-sized. The number of elements in this repeated field should be
	// zero (indicating that no dimensions are dynamic) or equal to the number of
	// elements in the 'dimensions' field.
	IsDynamicDimension []bool `protobuf:"varint,6,rep,packed,name=is_dynamic_dimension,json=isDynamicDimension,proto3" json:"is_dynamic_dimension,omitempty"`
	// contains filtered or unexported fields
}

A shape describes the number of dimensions in the array, the size of each dimension, and the primitive component type.

Tuples are a special case in that they have rank zero and have tuple_shapes defined.

See the XLA documentation for more information on shapes and layouts.

LINT.IfChange

func (*ShapeProto) Descriptor deprecated 

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

Deprecated: Use ShapeProto.ProtoReflect.Descriptor instead.

func (*ShapeProto) GetDimensions 

func (x *ShapeProto) GetDimensions() []int64

func (*ShapeProto) GetElementType 

func (x *ShapeProto) GetElementType() PrimitiveType

func (*ShapeProto) GetIsDynamicDimension 

func (x *ShapeProto) GetIsDynamicDimension() []bool

func (*ShapeProto) GetLayout 

func (x *ShapeProto) GetLayout() *LayoutProto

func (*ShapeProto) GetTupleShapes 

func (x *ShapeProto) GetTupleShapes() []*ShapeProto

func (*ShapeProto) ProtoMessage 

func (*ShapeProto) ProtoMessage()

func (*ShapeProto) ProtoReflect 

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

func (*ShapeProto) Reset 

func (x *ShapeProto) Reset()

func (*ShapeProto) String 

func (x *ShapeProto) String() string

type SortOptions 

type SortOptions struct {
	Descending bool `protobuf:"varint,1,opt,name=descending,proto3" json:"descending,omitempty"`
	// contains filtered or unexported fields
}

Attributes of the sort custom call (cub::DeviceRadixSort).

func (*SortOptions) Descriptor deprecated 

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

Deprecated: Use SortOptions.ProtoReflect.Descriptor instead.

func (*SortOptions) GetDescending 

func (x *SortOptions) GetDescending() bool

func (*SortOptions) ProtoMessage 

func (*SortOptions) ProtoMessage()

func (*SortOptions) ProtoReflect 

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

func (*SortOptions) Reset 

func (x *SortOptions) Reset()

func (*SortOptions) String 

func (x *SortOptions) String() string

type SourceTarget 

type SourceTarget struct {
	Source int64 `protobuf:"varint,1,opt,name=source,proto3" json:"source,omitempty"`
	Target int64 `protobuf:"varint,2,opt,name=target,proto3" json:"target,omitempty"`
	// contains filtered or unexported fields
}

Describes the source target pair in the collective permute op.

func (*SourceTarget) Descriptor deprecated 

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

Deprecated: Use SourceTarget.ProtoReflect.Descriptor instead.

func (*SourceTarget) GetSource 

func (x *SourceTarget) GetSource() int64

func (*SourceTarget) GetTarget 

func (x *SourceTarget) GetTarget() int64

func (*SourceTarget) ProtoMessage 

func (*SourceTarget) ProtoMessage()

func (*SourceTarget) ProtoReflect 

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

func (*SourceTarget) Reset 

func (x *SourceTarget) Reset()

func (*SourceTarget) String 

func (x *SourceTarget) String() string

type SparsityDescriptor 

type SparsityDescriptor struct {
	Type SparsityType `protobuf:"varint,1,opt,name=type,proto3,enum=xla.SparsityType" json:"type,omitempty"`
	// Sparse operand index (0 or 1).
	Index int32 `protobuf:"varint,2,opt,name=index,proto3" json:"index,omitempty"`
	// Sparse dimension number.
	Dimension int32 `protobuf:"varint,3,opt,name=dimension,proto3" json:"dimension,omitempty"`
	// Structured N:M sparsity (N < M).
	N int32 `protobuf:"varint,4,opt,name=n,proto3" json:"n,omitempty"`
	M int32 `protobuf:"varint,5,opt,name=m,proto3" json:"m,omitempty"`
	// contains filtered or unexported fields
}

Contains sparsity metadata for a sparse dot operation. The only supported type atm is structured 2:4 sparsity, which is natively supported on NVidia GPUs. Restrictions: - only one operand of the dot operation may be sparse; - only the contracting dimension may be sparse.

func (*SparsityDescriptor) Descriptor deprecated 

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

Deprecated: Use SparsityDescriptor.ProtoReflect.Descriptor instead.

func (*SparsityDescriptor) GetDimension 

func (x *SparsityDescriptor) GetDimension() int32

func (*SparsityDescriptor) GetIndex 

func (x *SparsityDescriptor) GetIndex() int32

func (*SparsityDescriptor) GetM 

func (x *SparsityDescriptor) GetM() int32

func (*SparsityDescriptor) GetN 

func (x *SparsityDescriptor) GetN() int32

func (*SparsityDescriptor) GetType 

func (x *SparsityDescriptor) GetType() SparsityType

func (*SparsityDescriptor) ProtoMessage 

func (*SparsityDescriptor) ProtoMessage()

func (*SparsityDescriptor) ProtoReflect 

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

func (*SparsityDescriptor) Reset 

func (x *SparsityDescriptor) Reset()

func (*SparsityDescriptor) String 

func (x *SparsityDescriptor) String() string

type SparsityType 

type SparsityType int32
const (
	SparsityType_SPARSITY_INVALID SparsityType = 0
	// Structured N:M sparsity.
	SparsityType_SPARSITY_STRUCTURED_N_M SparsityType = 1
)

func (SparsityType) Descriptor 

func (SparsityType) Descriptor() protoreflect.EnumDescriptor

func (SparsityType) Enum 

func (x SparsityType) Enum() *SparsityType

func (SparsityType) EnumDescriptor deprecated 

func (SparsityType) EnumDescriptor() ([]byte, []int)

Deprecated: Use SparsityType.Descriptor instead.

func (SparsityType) Number 

func (x SparsityType) Number() protoreflect.EnumNumber

func (SparsityType) String 

func (x SparsityType) String() string

func (SparsityType) Type 

func (SparsityType) Type() protoreflect.EnumType

type SplitConfigProto 

type SplitConfigProto struct {

	// The dimension that is split.
	Dimension int64 `protobuf:"varint,1,opt,name=dimension,proto3" json:"dimension,omitempty"`
	// The indices where each split point occurs. For example, if the dimension
	// size is 1024, a split_indices value of {512} indicates a two-way split of
	// data through the middle.
	SplitIndices []int64 `protobuf:"varint,2,rep,packed,name=split_indices,json=splitIndices,proto3" json:"split_indices,omitempty"`
	// contains filtered or unexported fields
}

Describes how data should be split between different memories.

func (*SplitConfigProto) Descriptor deprecated 

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

Deprecated: Use SplitConfigProto.ProtoReflect.Descriptor instead.

func (*SplitConfigProto) GetDimension 

func (x *SplitConfigProto) GetDimension() int64

func (*SplitConfigProto) GetSplitIndices 

func (x *SplitConfigProto) GetSplitIndices() []int64

func (*SplitConfigProto) ProtoMessage 

func (*SplitConfigProto) ProtoMessage()

func (*SplitConfigProto) ProtoReflect 

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

func (*SplitConfigProto) Reset 

func (x *SplitConfigProto) Reset()

func (*SplitConfigProto) String 

func (x *SplitConfigProto) String() string

type Statistic 

type Statistic struct {

	// Must be a single word consisting of any alphanumeric characters
	StatName string `protobuf:"bytes,1,opt,name=stat_name,json=statName,proto3" json:"stat_name,omitempty"`
	// Must be within a range of [0, 100], in order for the graph dumper to
	// properly render the statistic onto the graph.
	StatVal float64 `protobuf:"fixed64,2,opt,name=stat_val,json=statVal,proto3" json:"stat_val,omitempty"`
	// contains filtered or unexported fields
}

Represents a single statistic to track.

func (*Statistic) Descriptor deprecated 

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

Deprecated: Use Statistic.ProtoReflect.Descriptor instead.

func (*Statistic) GetStatName 

func (x *Statistic) GetStatName() string

func (*Statistic) GetStatVal 

func (x *Statistic) GetStatVal() float64

func (*Statistic) ProtoMessage 

func (*Statistic) ProtoMessage()

func (*Statistic) ProtoReflect 

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

func (*Statistic) Reset 

func (x *Statistic) Reset()

func (*Statistic) String 

func (x *Statistic) String() string

type StatisticsViz 

type StatisticsViz struct {
	StatIndexToVisualize int64        `` /* 126-byte string literal not displayed */
	Statistics           []*Statistic `protobuf:"bytes,2,rep,name=statistics,proto3" json:"statistics,omitempty"`
	// contains filtered or unexported fields
}

Represents the information needed to visualize propagation statistics when rendering an HLO graph. This includes an array of statistics as well as the index of the statistic to render.

func (*StatisticsViz) Descriptor deprecated 

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

Deprecated: Use StatisticsViz.ProtoReflect.Descriptor instead.

func (*StatisticsViz) GetStatIndexToVisualize 

func (x *StatisticsViz) GetStatIndexToVisualize() int64

func (*StatisticsViz) GetStatistics 

func (x *StatisticsViz) GetStatistics() []*Statistic

func (*StatisticsViz) ProtoMessage 

func (*StatisticsViz) ProtoMessage()

func (*StatisticsViz) ProtoReflect 

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

func (*StatisticsViz) Reset 

func (x *StatisticsViz) Reset()

func (*StatisticsViz) String 

func (x *StatisticsViz) String() string

type TileProto 

type TileProto struct {

	// Number of elements in each dimension of the tile. It's ordered from the
	// most major dimension of the tile to the most minor dimension of the tile.
	// The dimensions correspond to a suffix of the dimensions of the shape being
	// tiled.
	Dimensions []int64 `protobuf:"varint,1,rep,packed,name=dimensions,proto3" json:"dimensions,omitempty"`
	// contains filtered or unexported fields
}

Describes a tile used in tiling-based layout. Refer to g3doc/third_party/xla/docs/tiled_layout.md for details about tiling-based layout.

func (*TileProto) Descriptor deprecated 

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

Deprecated: Use TileProto.ProtoReflect.Descriptor instead.

func (*TileProto) GetDimensions 

func (x *TileProto) GetDimensions() []int64

func (*TileProto) ProtoMessage 

func (*TileProto) ProtoMessage()

func (*TileProto) ProtoReflect 

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

func (*TileProto) Reset 

func (x *TileProto) Reset()

func (*TileProto) String 

func (x *TileProto) String() string

type TriangularSolveOptions 

type TriangularSolveOptions struct {

	// If true, solves ax = b. If false, solves xa = b.
	LeftSide bool `protobuf:"varint,1,opt,name=left_side,json=leftSide,proto3" json:"left_side,omitempty"`
	// If true, 'a' is lower triangular. If false, 'a' is upper triangular.
	Lower bool `protobuf:"varint,2,opt,name=lower,proto3" json:"lower,omitempty"`
	// If true, the diagonal elements of 'a' are assumed to be 1 and not accessed.
	UnitDiagonal bool                             `protobuf:"varint,3,opt,name=unit_diagonal,json=unitDiagonal,proto3" json:"unit_diagonal,omitempty"`
	TransposeA   TriangularSolveOptions_Transpose `` /* 134-byte string literal not displayed */
	// contains filtered or unexported fields
}

func (*TriangularSolveOptions) Descriptor deprecated 

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

Deprecated: Use TriangularSolveOptions.ProtoReflect.Descriptor instead.

func (*TriangularSolveOptions) GetLeftSide 

func (x *TriangularSolveOptions) GetLeftSide() bool

func (*TriangularSolveOptions) GetLower 

func (x *TriangularSolveOptions) GetLower() bool

func (*TriangularSolveOptions) GetTransposeA 

func (x *TriangularSolveOptions) GetTransposeA() TriangularSolveOptions_Transpose

func (*TriangularSolveOptions) GetUnitDiagonal 

func (x *TriangularSolveOptions) GetUnitDiagonal() bool

func (*TriangularSolveOptions) ProtoMessage 

func (*TriangularSolveOptions) ProtoMessage()

func (*TriangularSolveOptions) ProtoReflect 

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

func (*TriangularSolveOptions) Reset 

func (x *TriangularSolveOptions) Reset()

func (*TriangularSolveOptions) String 

func (x *TriangularSolveOptions) String() string

type TriangularSolveOptions_Transpose 

type TriangularSolveOptions_Transpose int32

Should we transpose or use the adjoint of 'a'? 

const (
	TriangularSolveOptions_TRANSPOSE_INVALID TriangularSolveOptions_Transpose = 0
	TriangularSolveOptions_NO_TRANSPOSE      TriangularSolveOptions_Transpose = 1 // Don't transpose 'a'.
	TriangularSolveOptions_TRANSPOSE         TriangularSolveOptions_Transpose = 2 // Transpose 'a'.
	TriangularSolveOptions_ADJOINT           TriangularSolveOptions_Transpose = 3 // Complex conjugate and transpose 'a'.
)

func (TriangularSolveOptions_Transpose) Descriptor 

func (TriangularSolveOptions_Transpose) Descriptor() protoreflect.EnumDescriptor

func (TriangularSolveOptions_Transpose) Enum 

func (x TriangularSolveOptions_Transpose) Enum() *TriangularSolveOptions_Transpose

func (TriangularSolveOptions_Transpose) EnumDescriptor deprecated 

func (TriangularSolveOptions_Transpose) EnumDescriptor() ([]byte, []int)

Deprecated: Use TriangularSolveOptions_Transpose.Descriptor instead.

func (TriangularSolveOptions_Transpose) Number 

func (x TriangularSolveOptions_Transpose) Number() protoreflect.EnumNumber

func (TriangularSolveOptions_Transpose) String 

func (x TriangularSolveOptions_Transpose) String() string

func (TriangularSolveOptions_Transpose) Type 

func (TriangularSolveOptions_Transpose) Type() protoreflect.EnumType

type WhileLoopBackendConfig 

type WhileLoopBackendConfig struct {

	// This indirection lets us distinguish between known-trip-count == 0 and
	// unknown-trip-count.
	KnownTripCount *WhileLoopBackendConfig_KnownTripCount `protobuf:"bytes,1,opt,name=known_trip_count,json=knownTripCount,proto3" json:"known_trip_count,omitempty"`
	// contains filtered or unexported fields
}

A backend-config for kWhile loops that stores the loop's trip count, if it is known.

This is useful for backends that can implement a `for i in 0..N` loop more efficiently than a `while` loop. For example, on GPUs, we can implement a `for i in 0..N` loop by enqueueing the kernels for the loop body N times, whereas implementing a `while` loop requires a host-device sync on each iteration.

func (*WhileLoopBackendConfig) Descriptor deprecated 

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

Deprecated: Use WhileLoopBackendConfig.ProtoReflect.Descriptor instead.

func (*WhileLoopBackendConfig) GetKnownTripCount 

func (x *WhileLoopBackendConfig) GetKnownTripCount() *WhileLoopBackendConfig_KnownTripCount

func (*WhileLoopBackendConfig) ProtoMessage 

func (*WhileLoopBackendConfig) ProtoMessage()

func (*WhileLoopBackendConfig) ProtoReflect 

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

func (*WhileLoopBackendConfig) Reset 

func (x *WhileLoopBackendConfig) Reset()

func (*WhileLoopBackendConfig) String 

func (x *WhileLoopBackendConfig) String() string

type WhileLoopBackendConfig_KnownTripCount 

type WhileLoopBackendConfig_KnownTripCount struct {
	N int64 `protobuf:"varint,1,opt,name=n,proto3" json:"n,omitempty"`
	// contains filtered or unexported fields
}

func (*WhileLoopBackendConfig_KnownTripCount) Descriptor deprecated 

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

Deprecated: Use WhileLoopBackendConfig_KnownTripCount.ProtoReflect.Descriptor instead.

func (*WhileLoopBackendConfig_KnownTripCount) GetN 

func (x *WhileLoopBackendConfig_KnownTripCount) GetN() int64

func (*WhileLoopBackendConfig_KnownTripCount) ProtoMessage 

func (*WhileLoopBackendConfig_KnownTripCount) ProtoMessage()

func (*WhileLoopBackendConfig_KnownTripCount) ProtoReflect 

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

func (*WhileLoopBackendConfig_KnownTripCount) Reset 

func (x *WhileLoopBackendConfig_KnownTripCount) Reset()

func (*WhileLoopBackendConfig_KnownTripCount) String 

func (x *WhileLoopBackendConfig_KnownTripCount) String() string

type Window 

type Window struct {
	Dimensions []*WindowDimension `protobuf:"bytes,1,rep,name=dimensions,proto3" json:"dimensions,omitempty"`
	// contains filtered or unexported fields
}

Describes the windowing in an operation such as convolution.

The window is moved across a base area and for each position of the window a computation is performed. The field below describes the window and the movement of the window across a base area.

func (*Window) Descriptor deprecated 

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

Deprecated: Use Window.ProtoReflect.Descriptor instead.

func (*Window) GetDimensions 

func (x *Window) GetDimensions() []*WindowDimension

func (*Window) ProtoMessage 

func (*Window) ProtoMessage()

func (*Window) ProtoReflect 

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

func (*Window) Reset 

func (x *Window) Reset()

func (*Window) String 

func (x *Window) String() string

type WindowDimension 

type WindowDimension struct {

	// The size of the window in this dimension. For a rectangle, this would be
	// the width or height.
	Size int64 `protobuf:"varint,1,opt,name=size,proto3" json:"size,omitempty"`
	// The stride at which the window moves across the base area in this
	// dimension. In other words, this is the spacing between different
	// positions of the window in this dimension.
	Stride int64 `protobuf:"varint,2,opt,name=stride,proto3" json:"stride,omitempty"`
	// If positive, means the amount of padding to add to the base area at the low
	// end of this dimension; if negative, its negative means the number of
	// elements removed from the low end of this dimension. For example, in the
	// horizontal dimension of a rectangle, this would be the number of padding
	// values to pad on the left, given that indices increase when going right.
	// The actual padding value depends upon the context. Convolution pads with
	// zeros. ReduceWindow and SelectAndScatter pads with the reduce function's
	// init value.
	PaddingLow int64 `protobuf:"varint,3,opt,name=padding_low,json=paddingLow,proto3" json:"padding_low,omitempty"`
	// As padding_low, but on the high end of this dimension. For example, in the
	// horizontal dimension of a rectangle, this would be the number of values to
	// pad on the right, given that indices increase when going right.
	PaddingHigh int64 `protobuf:"varint,4,opt,name=padding_high,json=paddingHigh,proto3" json:"padding_high,omitempty"`
	// Dilation factor of the sliding window in this dimension. A dilation factor
	// of 1 means no dilation. window_dilation - 1 no-op entries ("holes") are
	// implicitly placed between each kernel element. This value may not be less
	// than 1. See documentation for convolution.
	WindowDilation int64 `protobuf:"varint,5,opt,name=window_dilation,json=windowDilation,proto3" json:"window_dilation,omitempty"`
	// Dilation factor of the base area in this dimension. A dilation factor of 1
	// means no dilation. base_dilation - 1 no-op entries ("holes") are implicitly
	// placed between each base area element. This value may not be less than 1.
	// See documentation for convolution.
	BaseDilation int64 `protobuf:"varint,6,opt,name=base_dilation,json=baseDilation,proto3" json:"base_dilation,omitempty"`
	// Window reversal means that this dimension was logically reversed before the
	// operation.
	WindowReversal bool `protobuf:"varint,7,opt,name=window_reversal,json=windowReversal,proto3" json:"window_reversal,omitempty"`
	// contains filtered or unexported fields
}

func (*WindowDimension) Descriptor deprecated 

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

Deprecated: Use WindowDimension.ProtoReflect.Descriptor instead.

func (*WindowDimension) GetBaseDilation 

func (x *WindowDimension) GetBaseDilation() int64

func (*WindowDimension) GetPaddingHigh 

func (x *WindowDimension) GetPaddingHigh() int64

func (*WindowDimension) GetPaddingLow 

func (x *WindowDimension) GetPaddingLow() int64

func (*WindowDimension) GetSize 

func (x *WindowDimension) GetSize() int64

func (*WindowDimension) GetStride 

func (x *WindowDimension) GetStride() int64

func (*WindowDimension) GetWindowDilation 

func (x *WindowDimension) GetWindowDilation() int64

func (*WindowDimension) GetWindowReversal 

func (x *WindowDimension) GetWindowReversal() bool

func (*WindowDimension) ProtoMessage 

func (*WindowDimension) ProtoMessage()

func (*WindowDimension) ProtoReflect 

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

func (*WindowDimension) Reset 

func (x *WindowDimension) Reset()

func (*WindowDimension) String 

func (x *WindowDimension) String() string

Source Files

View all Source files

Jump to

Keyboard shortcuts

? : This menu
/ : Search site
f or F : Jump to
y or Y : Canonical URL
go.dev uses cookies from Google to deliver and enhance the quality of its services and to analyze traffic. Learn more.