protos

package
v0.2.0 Latest Latest
Warning

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

 Go to latest Published: Jul 18, 2024 License: Apache-2.0 Imports: 7 Imported by: 1

Documentation

Overview

Package proto includes the Go bindings for several XLA protos needed for the pjrt package.

They are all auto-generated. See docs/devel.md for details on the tools needed to generate these.

Index

Constants

This section is empty.

Variables

View Source 
var (
	AutotuneResult_FailureKind_name = map[int32]string{
		0: "UNKNOWN",
		1: "REDZONE_MODIFIED",
		2: "WRONG_RESULT",
		3: "DISQUALIFIED",
	}
	AutotuneResult_FailureKind_value = map[string]int32{
		"UNKNOWN":          0,
		"REDZONE_MODIFIED": 1,
		"WRONG_RESULT":     2,
		"DISQUALIFIED":     3,
	}
)

Enum value maps for AutotuneResult_FailureKind.

View Source 
var (
	DataType_name = map[int32]string{
		0:  "kFloat",
		1:  "kDouble",
		2:  "kHalf",
		3:  "kInt8",
		4:  "kInt32",
		5:  "kComplexFloat",
		6:  "kComplexDouble",
		7:  "kBF16",
		8:  "kF8E5M2",
		9:  "kF8E4M3FN",
		10: "kF8E5M2FNUZ",
		11: "kF8E4M3FNUZ",
		12: "kInt64",
	}
	DataType_value = map[string]int32{
		"kFloat":         0,
		"kDouble":        1,
		"kHalf":          2,
		"kInt8":          3,
		"kInt32":         4,
		"kComplexFloat":  5,
		"kComplexDouble": 6,
		"kBF16":          7,
		"kF8E5M2":        8,
		"kF8E4M3FN":      9,
		"kF8E5M2FNUZ":    10,
		"kF8E4M3FNUZ":    11,
		"kInt64":         12,
	}
)

Enum value maps for DataType.

View Source 
var (
	DataLayout_name = map[int32]string{
		0: "kYXDepthBatch",
		1: "kYXBatchDepth",
		2: "kBatchYXDepth",
		3: "kBatchDepthYX",
		4: "kBatchDepthYX4",
		5: "kBatchDepthYX32",
	}
	DataLayout_value = map[string]int32{
		"kYXDepthBatch":   0,
		"kYXBatchDepth":   1,
		"kBatchYXDepth":   2,
		"kBatchDepthYX":   3,
		"kBatchDepthYX4":  4,
		"kBatchDepthYX32": 5,
	}
)

Enum value maps for DataLayout.

View Source 
var (
	FilterLayout_name = map[int32]string{
		0: "kOutputInputYX",
		1: "kOutputYXInput",
		2: "kOutputInputYX4",
		5: "kOutputInputYX32",
		6: "kOutputInputYX32_CudnnReordered",
		3: "kInputYXOutput",
		4: "kYXInputOutput",
	}
	FilterLayout_value = map[string]int32{
		"kOutputInputYX":                  0,
		"kOutputYXInput":                  1,
		"kOutputInputYX4":                 2,
		"kOutputInputYX32":                5,
		"kOutputInputYX32_CudnnReordered": 6,
		"kInputYXOutput":                  3,
		"kYXInputOutput":                  4,
	}
)

Enum value maps for FilterLayout.

View Source 
var (
	ActivationMode_name = map[int32]string{
		0: "kNone",
		1: "kSigmoid",
		2: "kRelu",
		3: "kRelu6",
		4: "kReluX",
		5: "kTanh",
		6: "kBandPass",
		7: "kElu",
		8: "kLeakyRelu",
		9: "kGeluExact",
	}
	ActivationMode_value = map[string]int32{
		"kNone":      0,
		"kSigmoid":   1,
		"kRelu":      2,
		"kRelu6":     3,
		"kReluX":     4,
		"kTanh":      5,
		"kBandPass":  6,
		"kElu":       7,
		"kLeakyRelu": 8,
		"kGeluExact": 9,
	}
)

Enum value maps for ActivationMode.

View Source 
var (
	ConvolutionMode_name = map[int32]string{
		0: "CROSS_CORRELATION",
		1: "CONVOLUTION",
	}
	ConvolutionMode_value = map[string]int32{
		"CROSS_CORRELATION": 0,
		"CONVOLUTION":       1,
	}
)

Enum value maps for ConvolutionMode.

View Source 
var (
	ConvolutionKind_name = map[int32]string{
		0: "INVALID",
		1: "FORWARD",
		2: "BACKWARD_FILTER",
		3: "BACKWARD_DATA",
		4: "FORWARD_BIAS_ACTIVATION",
		5: "FORWARD_GRAPH",
	}
	ConvolutionKind_value = map[string]int32{
		"INVALID":                 0,
		"FORWARD":                 1,
		"BACKWARD_FILTER":         2,
		"BACKWARD_DATA":           3,
		"FORWARD_BIAS_ACTIVATION": 4,
		"FORWARD_GRAPH":           5,
	}
)

Enum value maps for ConvolutionKind.

View Source 
var (
	NormKind_name = map[int32]string{
		0: "LAYER_FWD_INFER",
		1: "LAYER_FWD_TRAIN",
		2: "LAYER_BWD",
	}
	NormKind_value = map[string]int32{
		"LAYER_FWD_INFER": 0,
		"LAYER_FWD_TRAIN": 1,
		"LAYER_BWD":       2,
	}
)

Enum value maps for NormKind.

View Source 
var (
	FusedMHAKind_name = map[int32]string{
		0: "BMM1_OUTPUT_UNKNOWN",
		1: "BMM1_OUTPUT_INPUT_TYPE",
		2: "BMM1_OUTPUT_FLOAT",
	}
	FusedMHAKind_value = map[string]int32{
		"BMM1_OUTPUT_UNKNOWN":    0,
		"BMM1_OUTPUT_INPUT_TYPE": 1,
		"BMM1_OUTPUT_FLOAT":      2,
	}
)

Enum value maps for FusedMHAKind.

View Source 
var (
	FMHAMaskKind_name = map[int32]string{
		0: "NO_MASK",
		1: "PADDING",
		2: "CAUSAL",
		3: "PADDING_CAUSAL",
		4: "ALIBI",
	}
	FMHAMaskKind_value = map[string]int32{
		"NO_MASK":        0,
		"PADDING":        1,
		"CAUSAL":         2,
		"PADDING_CAUSAL": 3,
		"ALIBI":          4,
	}
)

Enum value maps for FMHAMaskKind.

View Source 
var (
	AlgorithmProto_MathType_name = map[int32]string{
		0: "DEFAULT_MATH",
		1: "TENSOR_OP_MATH",
	}
	AlgorithmProto_MathType_value = map[string]int32{
		"DEFAULT_MATH":   0,
		"TENSOR_OP_MATH": 1,
	}
)

Enum value maps for AlgorithmProto_MathType.

View Source 
var (
	CustomCallSchedule_name = map[int32]string{
		0: "SCHEDULE_NONE",
		1: "SCHEDULE_LATEST",
		2: "SCHEDULE_EARLIEST",
	}
	CustomCallSchedule_value = map[string]int32{
		"SCHEDULE_NONE":     0,
		"SCHEDULE_LATEST":   1,
		"SCHEDULE_EARLIEST": 2,
	}
)

Enum value maps for CustomCallSchedule.

View Source 
var (
	CustomCallApiVersion_name = map[int32]string{
		0: "API_VERSION_UNSPECIFIED",
		1: "API_VERSION_ORIGINAL",
		2: "API_VERSION_STATUS_RETURNING",
		3: "API_VERSION_STATUS_RETURNING_UNIFIED",
		4: "API_VERSION_TYPED_FFI",
	}
	CustomCallApiVersion_value = map[string]int32{
		"API_VERSION_UNSPECIFIED":              0,
		"API_VERSION_ORIGINAL":                 1,
		"API_VERSION_STATUS_RETURNING":         2,
		"API_VERSION_STATUS_RETURNING_UNIFIED": 3,
		"API_VERSION_TYPED_FFI":                4,
	}
)

Enum value maps for CustomCallApiVersion.

View Source 
var (
	Kind_name = map[int32]string{
		0: "UNDEFINED_ALIAS",
		1: "MAY_ALIAS",
		2: "MUST_ALIAS",
	}
	Kind_value = map[string]int32{
		"UNDEFINED_ALIAS": 0,
		"MAY_ALIAS":       1,
		"MUST_ALIAS":      2,
	}
)

Enum value maps for Kind.

View Source 
var (
	HloModuleProto_ProfileType_name = map[int32]string{
		0: "INVALID",
		1: "FLAG",
		2: "FUSION",
		3: "LAYOUT",
		4: "DOT",
	}
	HloModuleProto_ProfileType_value = map[string]int32{
		"INVALID": 0,
		"FLAG":    1,
		"FUSION":  2,
		"LAYOUT":  3,
		"DOT":     4,
	}
)

Enum value maps for HloModuleProto_ProfileType.

View Source 
var (
	HeapSimulatorTrace_Event_Kind_name = map[int32]string{
		0: "ALLOC",
		1: "FREE",
		2: "SHARE_WITH",
	}
	HeapSimulatorTrace_Event_Kind_value = map[string]int32{
		"ALLOC":      0,
		"FREE":       1,
		"SHARE_WITH": 2,
	}
)

Enum value maps for HeapSimulatorTrace_Event_Kind.

View Source 
var (
	DebugOptions_ShapeChecks_name = map[int32]string{
		0: "IGNORE",
		1: "RUNTIME",
		2: "COMPILE_TIME",
	}
	DebugOptions_ShapeChecks_value = map[string]int32{
		"IGNORE":       0,
		"RUNTIME":      1,
		"COMPILE_TIME": 2,
	}
)

Enum value maps for DebugOptions_ShapeChecks.

View Source 
var (
	DebugOptions_StepMarkerLocation_name = map[int32]string{
		0: "STEP_MARK_AT_ENTRY",
		1: "STEP_MARK_AT_TOP_LEVEL_WHILE_LOOP",
		3: "STEP_MARK_AT_SECOND_LEVEL_WHILE_LOOP",
		2: "STEP_MARK_NONE",
	}
	DebugOptions_StepMarkerLocation_value = map[string]int32{
		"STEP_MARK_AT_ENTRY":                   0,
		"STEP_MARK_AT_TOP_LEVEL_WHILE_LOOP":    1,
		"STEP_MARK_AT_SECOND_LEVEL_WHILE_LOOP": 3,
		"STEP_MARK_NONE":                       2,
	}
)

Enum value maps for DebugOptions_StepMarkerLocation.

View Source 
var (
	DebugOptions_CollectiveOpType_name = map[int32]string{
		0: "NOOP",
		1: "ALLREDUCE",
		2: "ALLGATHER",
		3: "REDUCESCATTER",
		4: "COLLECTIVEBROADCAST",
		5: "ALLTOALL",
		6: "COLLECTIVEPERMUTE",
	}
	DebugOptions_CollectiveOpType_value = map[string]int32{
		"NOOP":                0,
		"ALLREDUCE":           1,
		"ALLGATHER":           2,
		"REDUCESCATTER":       3,
		"COLLECTIVEBROADCAST": 4,
		"ALLTOALL":            5,
		"COLLECTIVEPERMUTE":   6,
	}
)

Enum value maps for DebugOptions_CollectiveOpType.

View Source 
var (
	DebugOptions_CommandBufferCmdType_name = map[int32]string{
		0: "INVALID",
		1: "FUSION",
		2: "CUBLAS",
		3: "CUDNN",
		4: "COLLECTIVES",
		5: "CONDITIONALS",
		6: "CUSTOM_CALL",
		7: "CUBLASLT",
	}
	DebugOptions_CommandBufferCmdType_value = map[string]int32{
		"INVALID":      0,
		"FUSION":       1,
		"CUBLAS":       2,
		"CUDNN":        3,
		"COLLECTIVES":  4,
		"CONDITIONALS": 5,
		"CUSTOM_CALL":  6,
		"CUBLASLT":     7,
	}
)

Enum value maps for DebugOptions_CommandBufferCmdType.

View Source 
var (
	DebugOptions_PartitioningAlgorithm_name = map[int32]string{
		0: "PARTITIONING_ALGORITHM_NOOP",
		1: "PARTITIONING_ALGORITHM_EXP0",
		2: "PARTITIONING_ALGORITHM_EXP1",
		3: "PARTITIONING_ALGORITHM_EXP2",
	}
	DebugOptions_PartitioningAlgorithm_value = map[string]int32{
		"PARTITIONING_ALGORITHM_NOOP": 0,
		"PARTITIONING_ALGORITHM_EXP0": 1,
		"PARTITIONING_ALGORITHM_EXP1": 2,
		"PARTITIONING_ALGORITHM_EXP2": 3,
	}
)

Enum value maps for DebugOptions_PartitioningAlgorithm.

View Source 
var (
	DebugOptions_WhileLoopUnrolling_name = map[int32]string{
		0: "WHILE_LOOP_UNROLLING_NO_UNROLL",
		1: "WHILE_LOOP_UNROLLING_DOUBLE_BUFFER",
		2: "WHILE_LOOP_UNROLLING_FULL_UNROLL",
	}
	DebugOptions_WhileLoopUnrolling_value = map[string]int32{
		"WHILE_LOOP_UNROLLING_NO_UNROLL":     0,
		"WHILE_LOOP_UNROLLING_DOUBLE_BUFFER": 1,
		"WHILE_LOOP_UNROLLING_FULL_UNROLL":   2,
	}
)

Enum value maps for DebugOptions_WhileLoopUnrolling.

View Source 
var (
	HloModuleConfigProto_FusionConfigCollection_name = map[int32]string{
		0: "OFF",
		1: "PER_EDGE",
		2: "PER_NODE",
	}
	HloModuleConfigProto_FusionConfigCollection_value = map[string]int32{
		"OFF":      0,
		"PER_EDGE": 1,
		"PER_NODE": 2,
	}
)

Enum value maps for HloModuleConfigProto_FusionConfigCollection.

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",
		20: "F8E4M3FN",
		23: "F8E4M3B11FNUZ",
		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,
		"F8E4M3FN":               20,
		"F8E4M3B11FNUZ":          23,
		"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 (
	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_tsl_protobuf_dnn_proto protoreflect.FileDescriptor
View Source 
var File_xla_autotune_results_proto protoreflect.FileDescriptor
View Source 
var File_xla_autotuning_proto protoreflect.FileDescriptor
View Source 
var File_xla_pjrt_compile_options_proto protoreflect.FileDescriptor
View Source 
var File_xla_service_hlo_proto protoreflect.FileDescriptor
View Source 
var File_xla_stream_executor_device_description_proto protoreflect.FileDescriptor
View Source 
var File_xla_xla_data_proto protoreflect.FileDescriptor
View Source 
var File_xla_xla_proto protoreflect.FileDescriptor

Functions

This section is empty.

Types

type ActivationMode 

type ActivationMode int32

Describes a kind of non-linearity (threshold-like mathematical function). 

const (
	ActivationMode_kNone    ActivationMode = 0
	ActivationMode_kSigmoid ActivationMode = 1
	// Rectified linear activation: f(x) = x < 0 ? 0 : x
	ActivationMode_kRelu ActivationMode = 2
	// Rectified linear activation; where upper maximum is 6.0.
	ActivationMode_kRelu6 ActivationMode = 3
	// Rectified linear activation; where upper maximum specified by
	// BatchDescriptor::value_max().
	ActivationMode_kReluX ActivationMode = 4
	ActivationMode_kTanh  ActivationMode = 5
	// Like ReluX; but passes all values in the range [-X,X].
	ActivationMode_kBandPass ActivationMode = 6
	// Exponential linear activation: f(x) = x < 0 ? e^x - 1 : x
	ActivationMode_kElu ActivationMode = 7
	// Leaky Rectified linear activation: f(x) = x < 0 ? alpha * x : x
	ActivationMode_kLeakyRelu ActivationMode = 8
	// Gaussian Error linear unit activation:
	//
	//	x * P(X <= x) = 0.5 * x * (1 + erf(x / sqrt(2))), where P(X) ~ N(0, 1).
	ActivationMode_kGeluExact ActivationMode = 9
)

func (ActivationMode) Descriptor 

func (ActivationMode) Descriptor() protoreflect.EnumDescriptor

func (ActivationMode) Enum 

func (x ActivationMode) Enum() *ActivationMode

func (ActivationMode) EnumDescriptor deprecated 

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

Deprecated: Use ActivationMode.Descriptor instead.

func (ActivationMode) Number 

func (x ActivationMode) Number() protoreflect.EnumNumber

func (ActivationMode) String 

func (x ActivationMode) String() string

func (ActivationMode) Type 

func (ActivationMode) Type() protoreflect.EnumType

type AlgorithmConfigProto 

type AlgorithmConfigProto struct {

	// Use oneof to emulate optional semantics in proto2 since older
	// version of proto3 cannot distinguish "unset field" and "default field".
	//
	// Types that are assignable to OptionalAlgorithm:
	//
	//	*AlgorithmConfigProto_Algorithm
	OptionalAlgorithm isAlgorithmConfigProto_OptionalAlgorithm `protobuf_oneof:"optional_algorithm"`
	// Types that are assignable to OptionalAlgorithmNoScratch:
	//
	//	*AlgorithmConfigProto_AlgorithmNoScratch
	OptionalAlgorithmNoScratch isAlgorithmConfigProto_OptionalAlgorithmNoScratch `protobuf_oneof:"optional_algorithm_no_scratch"`
	// Types that are assignable to OptionalScratchSize:
	//
	//	*AlgorithmConfigProto_ScratchSize
	OptionalScratchSize isAlgorithmConfigProto_OptionalScratchSize `protobuf_oneof:"optional_scratch_size"`
	// contains filtered or unexported fields
}

Proto definition of AlgorithmConfig in "dnn.h". TODO(ruochengw): After cl/380702564 is submitted, add support for algorithm configs with cuDNN Frontend APIs.

func (*AlgorithmConfigProto) Descriptor deprecated 

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

Deprecated: Use AlgorithmConfigProto.ProtoReflect.Descriptor instead.

func (*AlgorithmConfigProto) GetAlgorithm 

func (x *AlgorithmConfigProto) GetAlgorithm() *AlgorithmProto

func (*AlgorithmConfigProto) GetAlgorithmNoScratch 

func (x *AlgorithmConfigProto) GetAlgorithmNoScratch() *AlgorithmProto

func (*AlgorithmConfigProto) GetOptionalAlgorithm 

func (m *AlgorithmConfigProto) GetOptionalAlgorithm() isAlgorithmConfigProto_OptionalAlgorithm

func (*AlgorithmConfigProto) GetOptionalAlgorithmNoScratch 

func (m *AlgorithmConfigProto) GetOptionalAlgorithmNoScratch() isAlgorithmConfigProto_OptionalAlgorithmNoScratch

func (*AlgorithmConfigProto) GetOptionalScratchSize 

func (m *AlgorithmConfigProto) GetOptionalScratchSize() isAlgorithmConfigProto_OptionalScratchSize

func (*AlgorithmConfigProto) GetScratchSize 

func (x *AlgorithmConfigProto) GetScratchSize() int64

func (*AlgorithmConfigProto) ProtoMessage 

func (*AlgorithmConfigProto) ProtoMessage()

func (*AlgorithmConfigProto) ProtoReflect 

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

func (*AlgorithmConfigProto) Reset 

func (x *AlgorithmConfigProto) Reset()

func (*AlgorithmConfigProto) String 

func (x *AlgorithmConfigProto) String() string

type AlgorithmConfigProto_Algorithm 

type AlgorithmConfigProto_Algorithm struct {
	Algorithm *AlgorithmProto `protobuf:"bytes,1,opt,name=algorithm,proto3,oneof"`
}

type AlgorithmConfigProto_AlgorithmNoScratch 

type AlgorithmConfigProto_AlgorithmNoScratch struct {
	AlgorithmNoScratch *AlgorithmProto `protobuf:"bytes,2,opt,name=algorithm_no_scratch,json=algorithmNoScratch,proto3,oneof"`
}

type AlgorithmConfigProto_ScratchSize 

type AlgorithmConfigProto_ScratchSize struct {
	ScratchSize int64 `protobuf:"varint,3,opt,name=scratch_size,json=scratchSize,proto3,oneof"`
}

type AlgorithmProto 

type AlgorithmProto struct {
	AlgoId      int64                   `protobuf:"varint,1,opt,name=algo_id,json=algoId,proto3" json:"algo_id,omitempty"`
	MathType    AlgorithmProto_MathType `` /* 135-byte string literal not displayed */
	TuningKnobs map[int64]int64         `` /* 184-byte string literal not displayed */
	// Legacy algorithm enums and cuDNN Frontend engine numbers need to coexist in
	// the same proto medium-term, until we can be confident of no longer needing
	// the legacy cuDNN convolution API.  Once the migration is complete, we can
	// stop producing legacy algorithm enums and remove this field.
	IsCudnnFrontend bool `protobuf:"varint,5,opt,name=is_cudnn_frontend,json=isCudnnFrontend,proto3" json:"is_cudnn_frontend,omitempty"`
	// For ROCm only, it's impossible to re-query the required workspace size
	// after running the algorithm search, so we must store the workspace size
	// along with the choice of algorithm.  For consistency and convenience,
	// cuDNN uses this field in the same way, even though it would be possible to
	// re-query the workspace size from cuDNN at each use.
	//
	// Since this message is persisted in files, we need to be able to distinguish
	// 0 workspace size from unknown workspace size in an old message, so this is
	// a message field.
	WorkspaceSize *wrapperspb.UInt64Value `protobuf:"bytes,6,opt,name=workspace_size,json=workspaceSize,proto3" json:"workspace_size,omitempty"`
	// contains filtered or unexported fields
}

Generic algorithm representation.

func (*AlgorithmProto) Descriptor deprecated 

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

Deprecated: Use AlgorithmProto.ProtoReflect.Descriptor instead.

func (*AlgorithmProto) GetAlgoId 

func (x *AlgorithmProto) GetAlgoId() int64

func (*AlgorithmProto) GetIsCudnnFrontend 

func (x *AlgorithmProto) GetIsCudnnFrontend() bool

func (*AlgorithmProto) GetMathType 

func (x *AlgorithmProto) GetMathType() AlgorithmProto_MathType

func (*AlgorithmProto) GetTuningKnobs 

func (x *AlgorithmProto) GetTuningKnobs() map[int64]int64

func (*AlgorithmProto) GetWorkspaceSize 

func (x *AlgorithmProto) GetWorkspaceSize() *wrapperspb.UInt64Value

func (*AlgorithmProto) ProtoMessage 

func (*AlgorithmProto) ProtoMessage()

func (*AlgorithmProto) ProtoReflect 

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

func (*AlgorithmProto) Reset 

func (x *AlgorithmProto) Reset()

func (*AlgorithmProto) String 

func (x *AlgorithmProto) String() string

type AlgorithmProto_MathType 

type AlgorithmProto_MathType int32
const (
	AlgorithmProto_DEFAULT_MATH AlgorithmProto_MathType = 0
	// The GPU may operate 4x4 matrix FMA.
	// See cuDNN's documentation for CUDNN_TENSOR_OP_MATH.
	AlgorithmProto_TENSOR_OP_MATH AlgorithmProto_MathType = 1
)

func (AlgorithmProto_MathType) Descriptor 

func (AlgorithmProto_MathType) Descriptor() protoreflect.EnumDescriptor

func (AlgorithmProto_MathType) Enum 

func (x AlgorithmProto_MathType) Enum() *AlgorithmProto_MathType

func (AlgorithmProto_MathType) EnumDescriptor deprecated 

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

Deprecated: Use AlgorithmProto_MathType.Descriptor instead.

func (AlgorithmProto_MathType) Number 

func (x AlgorithmProto_MathType) Number() protoreflect.EnumNumber

func (AlgorithmProto_MathType) String 

func (x AlgorithmProto_MathType) String() string

func (AlgorithmProto_MathType) Type 

func (AlgorithmProto_MathType) Type() protoreflect.EnumType

type AutotuneResult 

type AutotuneResult struct {
	ScratchBytes int64                         `protobuf:"varint,8,opt,name=scratch_bytes,json=scratchBytes,proto3" json:"scratch_bytes,omitempty"`
	RunTime      *durationpb.Duration          `protobuf:"bytes,9,opt,name=run_time,json=runTime,proto3" json:"run_time,omitempty"`
	Failure      *AutotuneResult_FailureResult `protobuf:"bytes,7,opt,name=failure,proto3" json:"failure,omitempty"`
	// Types that are assignable to Key:
	//
	//	*AutotuneResult_Conv
	//	*AutotuneResult_Gemm
	//	*AutotuneResult_Triton
	//	*AutotuneResult_CudaConvPlan
	//	*AutotuneResult_Algorithm
	Key isAutotuneResult_Key `protobuf_oneof:"key"`
	// contains filtered or unexported fields
}

func (*AutotuneResult) Descriptor deprecated 

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

Deprecated: Use AutotuneResult.ProtoReflect.Descriptor instead.

func (*AutotuneResult) GetAlgorithm 

func (x *AutotuneResult) GetAlgorithm() *AlgorithmProto

func (*AutotuneResult) GetConv 

func (x *AutotuneResult) GetConv() *AutotuneResult_ConvKey

func (*AutotuneResult) GetCudaConvPlan 

func (x *AutotuneResult) GetCudaConvPlan() *AutotuneResult_CudaConvPlanKey

func (*AutotuneResult) GetFailure 

func (x *AutotuneResult) GetFailure() *AutotuneResult_FailureResult

func (*AutotuneResult) GetGemm 

func (x *AutotuneResult) GetGemm() *AutotuneResult_GemmKey

func (*AutotuneResult) GetKey 

func (m *AutotuneResult) GetKey() isAutotuneResult_Key

func (*AutotuneResult) GetRunTime 

func (x *AutotuneResult) GetRunTime() *durationpb.Duration

func (*AutotuneResult) GetScratchBytes 

func (x *AutotuneResult) GetScratchBytes() int64

func (*AutotuneResult) GetTriton 

func (x *AutotuneResult) GetTriton() *AutotuneResult_TritonGemmKey

func (*AutotuneResult) ProtoMessage 

func (*AutotuneResult) ProtoMessage()

func (*AutotuneResult) ProtoReflect 

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

func (*AutotuneResult) Reset 

func (x *AutotuneResult) Reset()

func (*AutotuneResult) String 

func (x *AutotuneResult) String() string

type AutotuneResult_Algorithm 

type AutotuneResult_Algorithm struct {
	Algorithm *AlgorithmProto `protobuf:"bytes,16,opt,name=algorithm,proto3,oneof"`
}

type AutotuneResult_Conv 

type AutotuneResult_Conv struct {
	Conv *AutotuneResult_ConvKey `protobuf:"bytes,5,opt,name=conv,proto3,oneof"`
}

type AutotuneResult_ConvKey 

type AutotuneResult_ConvKey struct {
	Algorithm        int64 `protobuf:"varint,1,opt,name=algorithm,proto3" json:"algorithm,omitempty"`
	TensorOpsEnabled bool  `protobuf:"varint,2,opt,name=tensor_ops_enabled,json=tensorOpsEnabled,proto3" json:"tensor_ops_enabled,omitempty"`
	// contains filtered or unexported fields
}

Legacy and unused in new data; superseded by AlgorithmProto.

func (*AutotuneResult_ConvKey) Descriptor deprecated 

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

Deprecated: Use AutotuneResult_ConvKey.ProtoReflect.Descriptor instead.

func (*AutotuneResult_ConvKey) GetAlgorithm 

func (x *AutotuneResult_ConvKey) GetAlgorithm() int64

func (*AutotuneResult_ConvKey) GetTensorOpsEnabled 

func (x *AutotuneResult_ConvKey) GetTensorOpsEnabled() bool

func (*AutotuneResult_ConvKey) ProtoMessage 

func (*AutotuneResult_ConvKey) ProtoMessage()

func (*AutotuneResult_ConvKey) ProtoReflect 

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

func (*AutotuneResult_ConvKey) Reset 

func (x *AutotuneResult_ConvKey) Reset()

func (*AutotuneResult_ConvKey) String 

func (x *AutotuneResult_ConvKey) String() string

type AutotuneResult_CudaConvPlan 

type AutotuneResult_CudaConvPlan struct {
	CudaConvPlan *AutotuneResult_CudaConvPlanKey `protobuf:"bytes,15,opt,name=cuda_conv_plan,json=cudaConvPlan,proto3,oneof"`
}

type AutotuneResult_CudaConvPlanKey 

type AutotuneResult_CudaConvPlanKey struct {
	ExecPlanId string `protobuf:"bytes,1,opt,name=exec_plan_id,json=execPlanId,proto3" json:"exec_plan_id,omitempty"`
	// contains filtered or unexported fields
}

Legacy and unused in new data; superseded by AlgorithmProto.

func (*AutotuneResult_CudaConvPlanKey) Descriptor deprecated 

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

Deprecated: Use AutotuneResult_CudaConvPlanKey.ProtoReflect.Descriptor instead.

func (*AutotuneResult_CudaConvPlanKey) GetExecPlanId 

func (x *AutotuneResult_CudaConvPlanKey) GetExecPlanId() string

func (*AutotuneResult_CudaConvPlanKey) ProtoMessage 

func (*AutotuneResult_CudaConvPlanKey) ProtoMessage()

func (*AutotuneResult_CudaConvPlanKey) ProtoReflect 

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

func (*AutotuneResult_CudaConvPlanKey) Reset 

func (x *AutotuneResult_CudaConvPlanKey) Reset()

func (*AutotuneResult_CudaConvPlanKey) String 

func (x *AutotuneResult_CudaConvPlanKey) String() string

type AutotuneResult_FailureKind 

type AutotuneResult_FailureKind int32
const (
	AutotuneResult_UNKNOWN AutotuneResult_FailureKind = 0
	// Algorithm wrote memory outside its output buffers.
	AutotuneResult_REDZONE_MODIFIED AutotuneResult_FailureKind = 1
	// Algorithm gave a different result from a reference algorithm.
	AutotuneResult_WRONG_RESULT AutotuneResult_FailureKind = 2
	// Algorithm was rejected for failing to run or for known bugs.
	AutotuneResult_DISQUALIFIED AutotuneResult_FailureKind = 3
)

func (AutotuneResult_FailureKind) Descriptor 

func (AutotuneResult_FailureKind) Descriptor() protoreflect.EnumDescriptor

func (AutotuneResult_FailureKind) Enum 

func (x AutotuneResult_FailureKind) Enum() *AutotuneResult_FailureKind

func (AutotuneResult_FailureKind) EnumDescriptor deprecated 

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

Deprecated: Use AutotuneResult_FailureKind.Descriptor instead.

func (AutotuneResult_FailureKind) Number 

func (x AutotuneResult_FailureKind) Number() protoreflect.EnumNumber

func (AutotuneResult_FailureKind) String 

func (x AutotuneResult_FailureKind) String() string

func (AutotuneResult_FailureKind) Type 

func (AutotuneResult_FailureKind) Type() protoreflect.EnumType

type AutotuneResult_FailureResult 

type AutotuneResult_FailureResult struct {
	Kind AutotuneResult_FailureKind `protobuf:"varint,1,opt,name=kind,proto3,enum=xla.AutotuneResult_FailureKind" json:"kind,omitempty"`
	Msg  string                     `protobuf:"bytes,2,opt,name=msg,proto3" json:"msg,omitempty"`
	// For failure_kind == WRONG_RESULT, this field indicates the reference
	// configuration that we compared against.
	//
	// Note that the reference algorithm isn't always correct.  However,
	// empirically it's more correct, as it's "algo 0", less fancy than the
	// compared one.
	//
	// Types that are assignable to Key:
	//
	//	*AutotuneResult_FailureResult_ReferenceConv
	//	*AutotuneResult_FailureResult_ReferenceGemm
	//	*AutotuneResult_FailureResult_ReferenceCudaConvPlan
	//	*AutotuneResult_FailureResult_ReferenceAlgorithm
	Key           isAutotuneResult_FailureResult_Key `protobuf_oneof:"key"`
	BufferAddress int64                              `protobuf:"varint,13,opt,name=buffer_address,json=bufferAddress,proto3" json:"buffer_address,omitempty"`
	// contains filtered or unexported fields
}

func (*AutotuneResult_FailureResult) Descriptor deprecated 

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

Deprecated: Use AutotuneResult_FailureResult.ProtoReflect.Descriptor instead.

func (*AutotuneResult_FailureResult) GetBufferAddress 

func (x *AutotuneResult_FailureResult) GetBufferAddress() int64

func (*AutotuneResult_FailureResult) GetKey 

func (m *AutotuneResult_FailureResult) GetKey() isAutotuneResult_FailureResult_Key

func (*AutotuneResult_FailureResult) GetKind 

func (x *AutotuneResult_FailureResult) GetKind() AutotuneResult_FailureKind

func (*AutotuneResult_FailureResult) GetMsg 

func (x *AutotuneResult_FailureResult) GetMsg() string

func (*AutotuneResult_FailureResult) GetReferenceAlgorithm 

func (x *AutotuneResult_FailureResult) GetReferenceAlgorithm() *AlgorithmProto

func (*AutotuneResult_FailureResult) GetReferenceConv 

func (x *AutotuneResult_FailureResult) GetReferenceConv() *AutotuneResult_ConvKey

func (*AutotuneResult_FailureResult) GetReferenceCudaConvPlan 

func (x *AutotuneResult_FailureResult) GetReferenceCudaConvPlan() *AutotuneResult_CudaConvPlanKey

func (*AutotuneResult_FailureResult) GetReferenceGemm 

func (x *AutotuneResult_FailureResult) GetReferenceGemm() *AutotuneResult_GemmKey

func (*AutotuneResult_FailureResult) ProtoMessage 

func (*AutotuneResult_FailureResult) ProtoMessage()

func (*AutotuneResult_FailureResult) ProtoReflect 

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

func (*AutotuneResult_FailureResult) Reset 

func (x *AutotuneResult_FailureResult) Reset()

func (*AutotuneResult_FailureResult) String 

func (x *AutotuneResult_FailureResult) String() string

type AutotuneResult_FailureResult_ReferenceAlgorithm 

type AutotuneResult_FailureResult_ReferenceAlgorithm struct {
	ReferenceAlgorithm *AlgorithmProto `protobuf:"bytes,15,opt,name=reference_algorithm,json=referenceAlgorithm,proto3,oneof"`
}

type AutotuneResult_FailureResult_ReferenceConv 

type AutotuneResult_FailureResult_ReferenceConv struct {
	ReferenceConv *AutotuneResult_ConvKey `protobuf:"bytes,11,opt,name=reference_conv,json=referenceConv,proto3,oneof"`
}

type AutotuneResult_FailureResult_ReferenceCudaConvPlan 

type AutotuneResult_FailureResult_ReferenceCudaConvPlan struct {
	ReferenceCudaConvPlan *AutotuneResult_CudaConvPlanKey `protobuf:"bytes,14,opt,name=reference_cuda_conv_plan,json=referenceCudaConvPlan,proto3,oneof"`
}

type AutotuneResult_FailureResult_ReferenceGemm 

type AutotuneResult_FailureResult_ReferenceGemm struct {
	ReferenceGemm *AutotuneResult_GemmKey `protobuf:"bytes,12,opt,name=reference_gemm,json=referenceGemm,proto3,oneof"`
}

type AutotuneResult_Gemm 

type AutotuneResult_Gemm struct {
	Gemm *AutotuneResult_GemmKey `protobuf:"bytes,6,opt,name=gemm,proto3,oneof"`
}

type AutotuneResult_GemmKey 

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

func (*AutotuneResult_GemmKey) Descriptor deprecated 

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

Deprecated: Use AutotuneResult_GemmKey.ProtoReflect.Descriptor instead.

func (*AutotuneResult_GemmKey) GetAlgorithm 

func (x *AutotuneResult_GemmKey) GetAlgorithm() int64

func (*AutotuneResult_GemmKey) ProtoMessage 

func (*AutotuneResult_GemmKey) ProtoMessage()

func (*AutotuneResult_GemmKey) ProtoReflect 

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

func (*AutotuneResult_GemmKey) Reset 

func (x *AutotuneResult_GemmKey) Reset()

func (*AutotuneResult_GemmKey) String 

func (x *AutotuneResult_GemmKey) String() string

type AutotuneResult_Triton 

type AutotuneResult_Triton struct {
	Triton *AutotuneResult_TritonGemmKey `protobuf:"bytes,17,opt,name=triton,proto3,oneof"`
}

type AutotuneResult_TritonGemmKey 

type AutotuneResult_TritonGemmKey struct {
	BlockM    int64 `protobuf:"varint,1,opt,name=block_m,json=blockM,proto3" json:"block_m,omitempty"`
	BlockN    int64 `protobuf:"varint,2,opt,name=block_n,json=blockN,proto3" json:"block_n,omitempty"`
	BlockK    int64 `protobuf:"varint,3,opt,name=block_k,json=blockK,proto3" json:"block_k,omitempty"`
	SplitK    int64 `protobuf:"varint,4,opt,name=split_k,json=splitK,proto3" json:"split_k,omitempty"`
	NumStages int64 `protobuf:"varint,5,opt,name=num_stages,json=numStages,proto3" json:"num_stages,omitempty"`
	NumWarps  int64 `protobuf:"varint,6,opt,name=num_warps,json=numWarps,proto3" json:"num_warps,omitempty"`
	NumCtas   int64 `protobuf:"varint,7,opt,name=num_ctas,json=numCtas,proto3" json:"num_ctas,omitempty"`
	// contains filtered or unexported fields
}

If you don't need a proto in your code, please use TritonGemmConfig instead of using this proto directly.

func (*AutotuneResult_TritonGemmKey) Descriptor deprecated 

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

Deprecated: Use AutotuneResult_TritonGemmKey.ProtoReflect.Descriptor instead.

func (*AutotuneResult_TritonGemmKey) GetBlockK 

func (x *AutotuneResult_TritonGemmKey) GetBlockK() int64

func (*AutotuneResult_TritonGemmKey) GetBlockM 

func (x *AutotuneResult_TritonGemmKey) GetBlockM() int64

func (*AutotuneResult_TritonGemmKey) GetBlockN 

func (x *AutotuneResult_TritonGemmKey) GetBlockN() int64

func (*AutotuneResult_TritonGemmKey) GetNumCtas 

func (x *AutotuneResult_TritonGemmKey) GetNumCtas() int64

func (*AutotuneResult_TritonGemmKey) GetNumStages 

func (x *AutotuneResult_TritonGemmKey) GetNumStages() int64

func (*AutotuneResult_TritonGemmKey) GetNumWarps 

func (x *AutotuneResult_TritonGemmKey) GetNumWarps() int64

func (*AutotuneResult_TritonGemmKey) GetSplitK 

func (x *AutotuneResult_TritonGemmKey) GetSplitK() int64

func (*AutotuneResult_TritonGemmKey) ProtoMessage 

func (*AutotuneResult_TritonGemmKey) ProtoMessage()

func (*AutotuneResult_TritonGemmKey) ProtoReflect 

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

func (*AutotuneResult_TritonGemmKey) Reset 

func (x *AutotuneResult_TritonGemmKey) Reset()

func (*AutotuneResult_TritonGemmKey) String 

func (x *AutotuneResult_TritonGemmKey) String() string

type AutotuneResults 

type AutotuneResults struct {
	Version int32                    `protobuf:"varint,1,opt,name=version,proto3" json:"version,omitempty"`
	Results []*AutotuneResults_Entry `protobuf:"bytes,4,rep,name=results,proto3" json:"results,omitempty"`
	// contains filtered or unexported fields
}

A collection of algorithms for particular dot/convs. Usually this is "the best" algorithm for the particular dot/conv, although that's not strictly required.

Users don't interact with this proto directly. It's used internally to facilitate ahead-of-time autotuning -- The string used by xla::{Serialize,Load}AutotuneResults is, internally, a serialization of this proto.

LINT.IfChange

func (*AutotuneResults) Descriptor deprecated 

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

Deprecated: Use AutotuneResults.ProtoReflect.Descriptor instead.

func (*AutotuneResults) GetResults 

func (x *AutotuneResults) GetResults() []*AutotuneResults_Entry

func (*AutotuneResults) GetVersion 

func (x *AutotuneResults) GetVersion() int32

func (*AutotuneResults) ProtoMessage 

func (*AutotuneResults) ProtoMessage()

func (*AutotuneResults) ProtoReflect 

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

func (*AutotuneResults) Reset 

func (x *AutotuneResults) Reset()

func (*AutotuneResults) String 

func (x *AutotuneResults) String() string

type AutotuneResults_Entry 

type AutotuneResults_Entry struct {
	Device string          `protobuf:"bytes,1,opt,name=device,proto3" json:"device,omitempty"`
	Hlo    string          `protobuf:"bytes,2,opt,name=hlo,proto3" json:"hlo,omitempty"`
	Result *AutotuneResult `protobuf:"bytes,3,opt,name=result,proto3" json:"result,omitempty"`
	// contains filtered or unexported fields
}

func (*AutotuneResults_Entry) Descriptor deprecated 

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

Deprecated: Use AutotuneResults_Entry.ProtoReflect.Descriptor instead.

func (*AutotuneResults_Entry) GetDevice 

func (x *AutotuneResults_Entry) GetDevice() string

func (*AutotuneResults_Entry) GetHlo 

func (x *AutotuneResults_Entry) GetHlo() string

func (*AutotuneResults_Entry) GetResult 

func (x *AutotuneResults_Entry) GetResult() *AutotuneResult

func (*AutotuneResults_Entry) ProtoMessage 

func (*AutotuneResults_Entry) ProtoMessage()

func (*AutotuneResults_Entry) ProtoReflect 

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

func (*AutotuneResults_Entry) Reset 

func (x *AutotuneResults_Entry) Reset()

func (*AutotuneResults_Entry) String 

func (x *AutotuneResults_Entry) String() string

type AutotuningLog 

type AutotuningLog struct {
	Instr *anypb.Any `protobuf:"bytes,1,opt,name=instr,proto3" json:"instr,omitempty"`
	// Records all auto-tuning results per algorithm.
	Results           []*AutotuneResult  `protobuf:"bytes,2,rep,name=results,proto3" json:"results,omitempty"`
	CudnnVersion      *CudnnVersion      `protobuf:"bytes,3,opt,name=cudnn_version,json=cudnnVersion,proto3" json:"cudnn_version,omitempty"`
	ComputeCapability *ComputeCapability `protobuf:"bytes,4,opt,name=compute_capability,json=computeCapability,proto3" json:"compute_capability,omitempty"`
	// stream_executor::DeviceDescription::pci_bus_id.
	DevicePciBusId string `protobuf:"bytes,5,opt,name=device_pci_bus_id,json=devicePciBusId,proto3" json:"device_pci_bus_id,omitempty"`
	BlasVersion    string `protobuf:"bytes,6,opt,name=blas_version,json=blasVersion,proto3" json:"blas_version,omitempty"`
	FusionName     string `protobuf:"bytes,7,opt,name=fusion_name,json=fusionName,proto3" json:"fusion_name,omitempty"`
	FusionCount    int64  `protobuf:"varint,8,opt,name=fusion_count,json=fusionCount,proto3" json:"fusion_count,omitempty"`
	// contains filtered or unexported fields
}

func (*AutotuningLog) Descriptor deprecated 

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

Deprecated: Use AutotuningLog.ProtoReflect.Descriptor instead.

func (*AutotuningLog) GetBlasVersion 

func (x *AutotuningLog) GetBlasVersion() string

func (*AutotuningLog) GetComputeCapability 

func (x *AutotuningLog) GetComputeCapability() *ComputeCapability

func (*AutotuningLog) GetCudnnVersion 

func (x *AutotuningLog) GetCudnnVersion() *CudnnVersion

func (*AutotuningLog) GetDevicePciBusId 

func (x *AutotuningLog) GetDevicePciBusId() string

func (*AutotuningLog) GetFusionCount 

func (x *AutotuningLog) GetFusionCount() int64

func (*AutotuningLog) GetFusionName 

func (x *AutotuningLog) GetFusionName() string

func (*AutotuningLog) GetInstr 

func (x *AutotuningLog) GetInstr() *anypb.Any

func (*AutotuningLog) GetResults 

func (x *AutotuningLog) GetResults() []*AutotuneResult

func (*AutotuningLog) ProtoMessage 

func (*AutotuningLog) ProtoMessage()

func (*AutotuningLog) ProtoReflect 

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

func (*AutotuningLog) Reset 

func (x *AutotuningLog) Reset()

func (*AutotuningLog) String 

func (x *AutotuningLog) String() string

type AutotuningLogs 

type AutotuningLogs struct {
	Logs []*AutotuningLog `protobuf:"bytes,1,rep,name=logs,proto3" json:"logs,omitempty"`
	// contains filtered or unexported fields
}

func (*AutotuningLogs) Descriptor deprecated 

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

Deprecated: Use AutotuningLogs.ProtoReflect.Descriptor instead.

func (*AutotuningLogs) GetLogs 

func (x *AutotuningLogs) GetLogs() []*AutotuningLog

func (*AutotuningLogs) ProtoMessage 

func (*AutotuningLogs) ProtoMessage()

func (*AutotuningLogs) ProtoReflect 

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

func (*AutotuningLogs) Reset 

func (x *AutotuningLogs) Reset()

func (*AutotuningLogs) String 

func (x *AutotuningLogs) String() string

type BufferAllocationProto 

type BufferAllocationProto struct {
	Index                       int64                             `protobuf:"varint,1,opt,name=index,proto3" json:"index,omitempty"`
	Size                        int64                             `protobuf:"varint,2,opt,name=size,proto3" json:"size,omitempty"`
	IsThreadLocal               bool                              `protobuf:"varint,3,opt,name=is_thread_local,json=isThreadLocal,proto3" json:"is_thread_local,omitempty"`
	IsTuple                     bool                              `protobuf:"varint,11,opt,name=is_tuple,json=isTuple,proto3" json:"is_tuple,omitempty"`
	IsEntryComputationParameter bool                              `` /* 147-byte string literal not displayed */
	IsConstant                  bool                              `protobuf:"varint,12,opt,name=is_constant,json=isConstant,proto3" json:"is_constant,omitempty"`
	ParameterNumber             int64                             `protobuf:"varint,6,opt,name=parameter_number,json=parameterNumber,proto3" json:"parameter_number,omitempty"`
	ParameterShapeIndex         []int64                           `` /* 129-byte string literal not displayed */
	MaybeLiveOut                bool                              `protobuf:"varint,7,opt,name=maybe_live_out,json=maybeLiveOut,proto3" json:"maybe_live_out,omitempty"`
	Color                       int64                             `protobuf:"varint,8,opt,name=color,proto3" json:"color,omitempty"`
	Assigned                    []*BufferAllocationProto_Assigned `protobuf:"bytes,9,rep,name=assigned,proto3" json:"assigned,omitempty"`
	// contains filtered or unexported fields
}

Serialization of BufferAllocation.

func (*BufferAllocationProto) Descriptor deprecated 

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

Deprecated: Use BufferAllocationProto.ProtoReflect.Descriptor instead.

func (*BufferAllocationProto) GetAssigned 

func (x *BufferAllocationProto) GetAssigned() []*BufferAllocationProto_Assigned

func (*BufferAllocationProto) GetColor 

func (x *BufferAllocationProto) GetColor() int64

func (*BufferAllocationProto) GetIndex 

func (x *BufferAllocationProto) GetIndex() int64

func (*BufferAllocationProto) GetIsConstant 

func (x *BufferAllocationProto) GetIsConstant() bool

func (*BufferAllocationProto) GetIsEntryComputationParameter 

func (x *BufferAllocationProto) GetIsEntryComputationParameter() bool

func (*BufferAllocationProto) GetIsThreadLocal 

func (x *BufferAllocationProto) GetIsThreadLocal() bool

func (*BufferAllocationProto) GetIsTuple 

func (x *BufferAllocationProto) GetIsTuple() bool

func (*BufferAllocationProto) GetMaybeLiveOut 

func (x *BufferAllocationProto) GetMaybeLiveOut() bool

func (*BufferAllocationProto) GetParameterNumber 

func (x *BufferAllocationProto) GetParameterNumber() int64

func (*BufferAllocationProto) GetParameterShapeIndex 

func (x *BufferAllocationProto) GetParameterShapeIndex() []int64

func (*BufferAllocationProto) GetSize 

func (x *BufferAllocationProto) GetSize() int64

func (*BufferAllocationProto) ProtoMessage 

func (*BufferAllocationProto) ProtoMessage()

func (*BufferAllocationProto) ProtoReflect 

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

func (*BufferAllocationProto) Reset 

func (x *BufferAllocationProto) Reset()

func (*BufferAllocationProto) String 

func (x *BufferAllocationProto) String() string

type BufferAllocationProto_Assigned 

type BufferAllocationProto_Assigned struct {
	LogicalBufferId int64 `protobuf:"varint,1,opt,name=logical_buffer_id,json=logicalBufferId,proto3" json:"logical_buffer_id,omitempty"`
	Offset          int64 `protobuf:"varint,2,opt,name=offset,proto3" json:"offset,omitempty"`
	Size            int64 `protobuf:"varint,3,opt,name=size,proto3" json:"size,omitempty"`
	// contains filtered or unexported fields
}

Assigned represents a single LogicalBuffer that is assigned to this BufferAllocation.

func (*BufferAllocationProto_Assigned) Descriptor deprecated 

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

Deprecated: Use BufferAllocationProto_Assigned.ProtoReflect.Descriptor instead.

func (*BufferAllocationProto_Assigned) GetLogicalBufferId 

func (x *BufferAllocationProto_Assigned) GetLogicalBufferId() int64

func (*BufferAllocationProto_Assigned) GetOffset 

func (x *BufferAllocationProto_Assigned) GetOffset() int64

func (*BufferAllocationProto_Assigned) GetSize 

func (x *BufferAllocationProto_Assigned) GetSize() int64

func (*BufferAllocationProto_Assigned) ProtoMessage 

func (*BufferAllocationProto_Assigned) ProtoMessage()

func (*BufferAllocationProto_Assigned) ProtoReflect 

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

func (*BufferAllocationProto_Assigned) Reset 

func (x *BufferAllocationProto_Assigned) Reset()

func (*BufferAllocationProto_Assigned) String 

func (x *BufferAllocationProto_Assigned) String() string

type BufferAssignmentProto 

type BufferAssignmentProto struct {
	LogicalBuffers      []*LogicalBufferProto                `protobuf:"bytes,1,rep,name=logical_buffers,json=logicalBuffers,proto3" json:"logical_buffers,omitempty"`
	BufferAliases       []*BufferAssignmentProto_BufferAlias `protobuf:"bytes,2,rep,name=buffer_aliases,json=bufferAliases,proto3" json:"buffer_aliases,omitempty"`
	BufferAllocations   []*BufferAllocationProto             `protobuf:"bytes,3,rep,name=buffer_allocations,json=bufferAllocations,proto3" json:"buffer_allocations,omitempty"`
	HeapSimulatorTraces []*HeapSimulatorTrace                `protobuf:"bytes,4,rep,name=heap_simulator_traces,json=heapSimulatorTraces,proto3" json:"heap_simulator_traces,omitempty"`
	// contains filtered or unexported fields
}

Serialization of BufferAssignment.

func (*BufferAssignmentProto) Descriptor deprecated 

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

Deprecated: Use BufferAssignmentProto.ProtoReflect.Descriptor instead.

func (*BufferAssignmentProto) GetBufferAliases 

func (x *BufferAssignmentProto) GetBufferAliases() []*BufferAssignmentProto_BufferAlias

func (*BufferAssignmentProto) GetBufferAllocations 

func (x *BufferAssignmentProto) GetBufferAllocations() []*BufferAllocationProto

func (*BufferAssignmentProto) GetHeapSimulatorTraces 

func (x *BufferAssignmentProto) GetHeapSimulatorTraces() []*HeapSimulatorTrace

func (*BufferAssignmentProto) GetLogicalBuffers 

func (x *BufferAssignmentProto) GetLogicalBuffers() []*LogicalBufferProto

func (*BufferAssignmentProto) ProtoMessage 

func (*BufferAssignmentProto) ProtoMessage()

func (*BufferAssignmentProto) ProtoReflect 

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

func (*BufferAssignmentProto) Reset 

func (x *BufferAssignmentProto) Reset()

func (*BufferAssignmentProto) String 

func (x *BufferAssignmentProto) String() string

type BufferAssignmentProto_BufferAlias 

type BufferAssignmentProto_BufferAlias struct {
	SourceBufferId int64                        `protobuf:"varint,1,opt,name=source_buffer_id,json=sourceBufferId,proto3" json:"source_buffer_id,omitempty"`
	Location       *LogicalBufferProto_Location `protobuf:"bytes,2,opt,name=location,proto3" json:"location,omitempty"`
	// contains filtered or unexported fields
}

Alias represents a source LogicalBuffer, and the buffer location that aliases it.

func (*BufferAssignmentProto_BufferAlias) Descriptor deprecated 

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

Deprecated: Use BufferAssignmentProto_BufferAlias.ProtoReflect.Descriptor instead.

func (*BufferAssignmentProto_BufferAlias) GetLocation 

func (x *BufferAssignmentProto_BufferAlias) GetLocation() *LogicalBufferProto_Location

func (*BufferAssignmentProto_BufferAlias) GetSourceBufferId 

func (x *BufferAssignmentProto_BufferAlias) GetSourceBufferId() int64

func (*BufferAssignmentProto_BufferAlias) ProtoMessage 

func (*BufferAssignmentProto_BufferAlias) ProtoMessage()

func (*BufferAssignmentProto_BufferAlias) ProtoReflect 

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

func (*BufferAssignmentProto_BufferAlias) Reset 

func (x *BufferAssignmentProto_BufferAlias) Reset()

func (*BufferAssignmentProto_BufferAlias) String 

func (x *BufferAssignmentProto_BufferAlias) String() string

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 CompilationEnvironmentsProto 

type CompilationEnvironmentsProto struct {
	Environments []*anypb.Any `protobuf:"bytes,1,rep,name=environments,proto3" json:"environments,omitempty"`
	// contains filtered or unexported fields
}

Proto version of `xla::CompilationEnvironments`.

func (*CompilationEnvironmentsProto) Descriptor deprecated 

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

Deprecated: Use CompilationEnvironmentsProto.ProtoReflect.Descriptor instead.

func (*CompilationEnvironmentsProto) GetEnvironments 

func (x *CompilationEnvironmentsProto) GetEnvironments() []*anypb.Any

func (*CompilationEnvironmentsProto) ProtoMessage 

func (*CompilationEnvironmentsProto) ProtoMessage()

func (*CompilationEnvironmentsProto) ProtoReflect 

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

func (*CompilationEnvironmentsProto) Reset 

func (x *CompilationEnvironmentsProto) Reset()

func (*CompilationEnvironmentsProto) String 

func (x *CompilationEnvironmentsProto) 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 CompileOptionsProto 

type CompileOptionsProto struct {

	// Refer CompileOptions for documentation of fields.
	ArgumentLayouts            []*ShapeProto                   `protobuf:"bytes,1,rep,name=argument_layouts,json=argumentLayouts,proto3" json:"argument_layouts,omitempty"`
	ParameterIsTupledArguments bool                            `` /* 144-byte string literal not displayed */
	ExecutableBuildOptions     *ExecutableBuildOptionsProto    `` /* 129-byte string literal not displayed */
	CompilePortableExecutable  bool                            `` /* 139-byte string literal not displayed */
	ProfileVersion             int64                           `protobuf:"varint,5,opt,name=profile_version,json=profileVersion,proto3" json:"profile_version,omitempty"`
	SerializedMultiSliceConfig []byte                          `` /* 143-byte string literal not displayed */
	EnvOptionOverrides         map[string]*OptionOverrideProto `` /* 205-byte string literal not displayed */
	TargetConfig               *GpuTargetConfigProto           `protobuf:"bytes,8,opt,name=target_config,json=targetConfig,proto3" json:"target_config,omitempty"`
	// contains filtered or unexported fields
}

func (*CompileOptionsProto) Descriptor deprecated 

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

Deprecated: Use CompileOptionsProto.ProtoReflect.Descriptor instead.

func (*CompileOptionsProto) GetArgumentLayouts 

func (x *CompileOptionsProto) GetArgumentLayouts() []*ShapeProto

func (*CompileOptionsProto) GetCompilePortableExecutable 

func (x *CompileOptionsProto) GetCompilePortableExecutable() bool

func (*CompileOptionsProto) GetEnvOptionOverrides 

func (x *CompileOptionsProto) GetEnvOptionOverrides() map[string]*OptionOverrideProto

func (*CompileOptionsProto) GetExecutableBuildOptions 

func (x *CompileOptionsProto) GetExecutableBuildOptions() *ExecutableBuildOptionsProto

func (*CompileOptionsProto) GetParameterIsTupledArguments 

func (x *CompileOptionsProto) GetParameterIsTupledArguments() bool

func (*CompileOptionsProto) GetProfileVersion 

func (x *CompileOptionsProto) GetProfileVersion() int64

func (*CompileOptionsProto) GetSerializedMultiSliceConfig 

func (x *CompileOptionsProto) GetSerializedMultiSliceConfig() []byte

func (*CompileOptionsProto) GetTargetConfig 

func (x *CompileOptionsProto) GetTargetConfig() *GpuTargetConfigProto

func (*CompileOptionsProto) ProtoMessage 

func (*CompileOptionsProto) ProtoMessage()

func (*CompileOptionsProto) ProtoReflect 

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

func (*CompileOptionsProto) Reset 

func (x *CompileOptionsProto) Reset()

func (*CompileOptionsProto) String 

func (x *CompileOptionsProto) String() string

type CompileRequest 

type CompileRequest struct {

	// The graph to be compiled.
	Computation *HloModuleProto `protobuf:"bytes,1,opt,name=computation,proto3" json:"computation,omitempty"`
	// Options that affect how XLA compiles code to service this request.
	ExecutionOptions *ExecutionOptions `protobuf:"bytes,2,opt,name=execution_options,json=executionOptions,proto3" json:"execution_options,omitempty"`
	// The layouts of the input arguments. If not set, the default layout will be
	// used. Although the real arguments are not needed in compilation, the
	// layouts of the arguments can affect the compilation.
	InputShapeWithLayout []*ShapeProto `protobuf:"bytes,3,rep,name=input_shape_with_layout,json=inputShapeWithLayout,proto3" json:"input_shape_with_layout,omitempty"`
	// contains filtered or unexported fields
}

func (*CompileRequest) Descriptor deprecated 

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

Deprecated: Use CompileRequest.ProtoReflect.Descriptor instead.

func (*CompileRequest) GetComputation 

func (x *CompileRequest) GetComputation() *HloModuleProto

func (*CompileRequest) GetExecutionOptions 

func (x *CompileRequest) GetExecutionOptions() *ExecutionOptions

func (*CompileRequest) GetInputShapeWithLayout 

func (x *CompileRequest) GetInputShapeWithLayout() []*ShapeProto

func (*CompileRequest) ProtoMessage 

func (*CompileRequest) ProtoMessage()

func (*CompileRequest) ProtoReflect 

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

func (*CompileRequest) Reset 

func (x *CompileRequest) Reset()

func (*CompileRequest) String 

func (x *CompileRequest) String() string

type CompileResponse 

type CompileResponse struct {

	// The handle to the executable.
	Handle *ExecutionHandle `protobuf:"bytes,1,opt,name=handle,proto3" json:"handle,omitempty"`
	// contains filtered or unexported fields
}

func (*CompileResponse) Descriptor deprecated 

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

Deprecated: Use CompileResponse.ProtoReflect.Descriptor instead.

func (*CompileResponse) GetHandle 

func (x *CompileResponse) GetHandle() *ExecutionHandle

func (*CompileResponse) ProtoMessage 

func (*CompileResponse) ProtoMessage()

func (*CompileResponse) ProtoReflect 

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

func (*CompileResponse) Reset 

func (x *CompileResponse) Reset()

func (*CompileResponse) String 

func (x *CompileResponse) String() string

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 ComputeCapability 

type ComputeCapability struct {
	Major int32 `protobuf:"varint,1,opt,name=major,proto3" json:"major,omitempty"`
	Minor int32 `protobuf:"varint,2,opt,name=minor,proto3" json:"minor,omitempty"`
	// contains filtered or unexported fields
}

func (*ComputeCapability) Descriptor deprecated 

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

Deprecated: Use ComputeCapability.ProtoReflect.Descriptor instead.

func (*ComputeCapability) GetMajor 

func (x *ComputeCapability) GetMajor() int32

func (*ComputeCapability) GetMinor 

func (x *ComputeCapability) GetMinor() int32

func (*ComputeCapability) ProtoMessage 

func (*ComputeCapability) ProtoMessage()

func (*ComputeCapability) ProtoReflect 

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

func (*ComputeCapability) Reset 

func (x *ComputeCapability) Reset()

func (*ComputeCapability) String 

func (x *ComputeCapability) String() string

type ComputeConstantGraphRequest 

type ComputeConstantGraphRequest struct {
	Computation  *HloModuleProto `protobuf:"bytes,1,opt,name=computation,proto3" json:"computation,omitempty"`
	OutputLayout *LayoutProto    `protobuf:"bytes,2,opt,name=output_layout,json=outputLayout,proto3" json:"output_layout,omitempty"`
	// contains filtered or unexported fields
}

func (*ComputeConstantGraphRequest) Descriptor deprecated 

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

Deprecated: Use ComputeConstantGraphRequest.ProtoReflect.Descriptor instead.

func (*ComputeConstantGraphRequest) GetComputation 

func (x *ComputeConstantGraphRequest) GetComputation() *HloModuleProto

func (*ComputeConstantGraphRequest) GetOutputLayout 

func (x *ComputeConstantGraphRequest) GetOutputLayout() *LayoutProto

func (*ComputeConstantGraphRequest) ProtoMessage 

func (*ComputeConstantGraphRequest) ProtoMessage()

func (*ComputeConstantGraphRequest) ProtoReflect 

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

func (*ComputeConstantGraphRequest) Reset 

func (x *ComputeConstantGraphRequest) Reset()

func (*ComputeConstantGraphRequest) String 

func (x *ComputeConstantGraphRequest) String() string

type ComputeConstantResponse 

type ComputeConstantResponse struct {

	// A LiteralProto is returned directly for this request.
	Literal *LiteralProto `protobuf:"bytes,1,opt,name=literal,proto3" json:"literal,omitempty"`
	// contains filtered or unexported fields
}

func (*ComputeConstantResponse) Descriptor deprecated 

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

Deprecated: Use ComputeConstantResponse.ProtoReflect.Descriptor instead.

func (*ComputeConstantResponse) GetLiteral 

func (x *ComputeConstantResponse) GetLiteral() *LiteralProto

func (*ComputeConstantResponse) ProtoMessage 

func (*ComputeConstantResponse) ProtoMessage()

func (*ComputeConstantResponse) ProtoReflect 

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

func (*ComputeConstantResponse) Reset 

func (x *ComputeConstantResponse) Reset()

func (*ComputeConstantResponse) String 

func (x *ComputeConstantResponse) String() string

type ConvolutionDescriptorProto 

type ConvolutionDescriptorProto struct {
	Paddings  []int64 `protobuf:"varint,1,rep,packed,name=paddings,proto3" json:"paddings,omitempty"`
	Strides   []int64 `protobuf:"varint,2,rep,packed,name=strides,proto3" json:"strides,omitempty"`
	Dilations []int64 `protobuf:"varint,3,rep,packed,name=dilations,proto3" json:"dilations,omitempty"`
	// The "accumulator" type. For example, use F32 as an accumulator for F16
	// convolutions.
	// See cuDNN's cudnnConvolutionMode_t.
	ComputeMode DataType `` /* 129-byte string literal not displayed */
	// See cuDNN's group count.
	GroupCount      int32           `protobuf:"varint,5,opt,name=group_count,json=groupCount,proto3" json:"group_count,omitempty"`
	ConvolutionMode ConvolutionMode `` /* 148-byte string literal not displayed */
	// Tensorflow node name, same as in NodeDef, for debugging purposes.
	Name string `protobuf:"bytes,7,opt,name=name,proto3" json:"name,omitempty"`
	// contains filtered or unexported fields
}

Convolution-specific parameters.

func (*ConvolutionDescriptorProto) Descriptor deprecated 

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

Deprecated: Use ConvolutionDescriptorProto.ProtoReflect.Descriptor instead.

func (*ConvolutionDescriptorProto) GetComputeMode 

func (x *ConvolutionDescriptorProto) GetComputeMode() DataType

func (*ConvolutionDescriptorProto) GetConvolutionMode 

func (x *ConvolutionDescriptorProto) GetConvolutionMode() ConvolutionMode

func (*ConvolutionDescriptorProto) GetDilations 

func (x *ConvolutionDescriptorProto) GetDilations() []int64

func (*ConvolutionDescriptorProto) GetGroupCount 

func (x *ConvolutionDescriptorProto) GetGroupCount() int32

func (*ConvolutionDescriptorProto) GetName 

func (x *ConvolutionDescriptorProto) GetName() string

func (*ConvolutionDescriptorProto) GetPaddings 

func (x *ConvolutionDescriptorProto) GetPaddings() []int64

func (*ConvolutionDescriptorProto) GetStrides 

func (x *ConvolutionDescriptorProto) GetStrides() []int64

func (*ConvolutionDescriptorProto) ProtoMessage 

func (*ConvolutionDescriptorProto) ProtoMessage()

func (*ConvolutionDescriptorProto) ProtoReflect 

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

func (*ConvolutionDescriptorProto) Reset 

func (x *ConvolutionDescriptorProto) Reset()

func (*ConvolutionDescriptorProto) String 

func (x *ConvolutionDescriptorProto) 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 ConvolutionKind 

type ConvolutionKind int32
const (
	ConvolutionKind_INVALID                 ConvolutionKind = 0
	ConvolutionKind_FORWARD                 ConvolutionKind = 1
	ConvolutionKind_BACKWARD_FILTER         ConvolutionKind = 2
	ConvolutionKind_BACKWARD_DATA           ConvolutionKind = 3
	ConvolutionKind_FORWARD_BIAS_ACTIVATION ConvolutionKind = 4
	ConvolutionKind_FORWARD_GRAPH           ConvolutionKind = 5
)

func (ConvolutionKind) Descriptor 

func (ConvolutionKind) Descriptor() protoreflect.EnumDescriptor

func (ConvolutionKind) Enum 

func (x ConvolutionKind) Enum() *ConvolutionKind

func (ConvolutionKind) EnumDescriptor deprecated 

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

Deprecated: Use ConvolutionKind.Descriptor instead.

func (ConvolutionKind) Number 

func (x ConvolutionKind) Number() protoreflect.EnumNumber

func (ConvolutionKind) String 

func (x ConvolutionKind) String() string

func (ConvolutionKind) Type 

func (ConvolutionKind) Type() protoreflect.EnumType

type ConvolutionMode 

type ConvolutionMode int32

Describe the math definition for the conv op. The popular behavior is actually called cross-correlation in math, despite the operation is often referred as convolution. See cuDNN cudnnConvolutionMode_t. 

const (
	ConvolutionMode_CROSS_CORRELATION ConvolutionMode = 0
	ConvolutionMode_CONVOLUTION       ConvolutionMode = 1
)

func (ConvolutionMode) Descriptor 

func (ConvolutionMode) Descriptor() protoreflect.EnumDescriptor

func (ConvolutionMode) Enum 

func (x ConvolutionMode) Enum() *ConvolutionMode

func (ConvolutionMode) EnumDescriptor deprecated 

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

Deprecated: Use ConvolutionMode.Descriptor instead.

func (ConvolutionMode) Number 

func (x ConvolutionMode) Number() protoreflect.EnumNumber

func (ConvolutionMode) String 

func (x ConvolutionMode) String() string

func (ConvolutionMode) Type 

func (ConvolutionMode) Type() protoreflect.EnumType

type CreateChannelHandleRequest 

type CreateChannelHandleRequest struct {
	ChannelType ChannelHandle_ChannelType `` /* 130-byte string literal not displayed */
	// contains filtered or unexported fields
}

func (*CreateChannelHandleRequest) Descriptor deprecated 

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

Deprecated: Use CreateChannelHandleRequest.ProtoReflect.Descriptor instead.

func (*CreateChannelHandleRequest) GetChannelType 

func (x *CreateChannelHandleRequest) GetChannelType() ChannelHandle_ChannelType

func (*CreateChannelHandleRequest) ProtoMessage 

func (*CreateChannelHandleRequest) ProtoMessage()

func (*CreateChannelHandleRequest) ProtoReflect 

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

func (*CreateChannelHandleRequest) Reset 

func (x *CreateChannelHandleRequest) Reset()

func (*CreateChannelHandleRequest) String 

func (x *CreateChannelHandleRequest) String() string

type CreateChannelHandleResponse 

type CreateChannelHandleResponse struct {
	Channel *ChannelHandle `protobuf:"bytes,1,opt,name=channel,proto3" json:"channel,omitempty"`
	// contains filtered or unexported fields
}

func (*CreateChannelHandleResponse) Descriptor deprecated 

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

Deprecated: Use CreateChannelHandleResponse.ProtoReflect.Descriptor instead.

func (*CreateChannelHandleResponse) GetChannel 

func (x *CreateChannelHandleResponse) GetChannel() *ChannelHandle

func (*CreateChannelHandleResponse) ProtoMessage 

func (*CreateChannelHandleResponse) ProtoMessage()

func (*CreateChannelHandleResponse) ProtoReflect 

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

func (*CreateChannelHandleResponse) Reset 

func (x *CreateChannelHandleResponse) Reset()

func (*CreateChannelHandleResponse) String 

func (x *CreateChannelHandleResponse) String() string

type CrossProgramPrefetch 

type CrossProgramPrefetch struct {
	Parameter int64   `protobuf:"varint,1,opt,name=parameter,proto3" json:"parameter,omitempty"`
	Index     []int64 `protobuf:"varint,2,rep,packed,name=index,proto3" json:"index,omitempty"`
	Offset    int64   `protobuf:"varint,3,opt,name=offset,proto3" json:"offset,omitempty"`
	// contains filtered or unexported fields
}

func (*CrossProgramPrefetch) Descriptor deprecated 

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

Deprecated: Use CrossProgramPrefetch.ProtoReflect.Descriptor instead.

func (*CrossProgramPrefetch) GetIndex 

func (x *CrossProgramPrefetch) GetIndex() []int64

func (*CrossProgramPrefetch) GetOffset 

func (x *CrossProgramPrefetch) GetOffset() int64

func (*CrossProgramPrefetch) GetParameter 

func (x *CrossProgramPrefetch) GetParameter() int64

func (*CrossProgramPrefetch) ProtoMessage 

func (*CrossProgramPrefetch) ProtoMessage()

func (*CrossProgramPrefetch) ProtoReflect 

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

func (*CrossProgramPrefetch) Reset 

func (x *CrossProgramPrefetch) Reset()

func (*CrossProgramPrefetch) String 

func (x *CrossProgramPrefetch) String() string

type CudaComputeCapabilityProto 

type CudaComputeCapabilityProto struct {
	Major int32 `protobuf:"varint,1,opt,name=major,proto3" json:"major,omitempty"`
	Minor int32 `protobuf:"varint,2,opt,name=minor,proto3" json:"minor,omitempty"`
	// contains filtered or unexported fields
}

func (*CudaComputeCapabilityProto) Descriptor deprecated 

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

Deprecated: Use CudaComputeCapabilityProto.ProtoReflect.Descriptor instead.

func (*CudaComputeCapabilityProto) GetMajor 

func (x *CudaComputeCapabilityProto) GetMajor() int32

func (*CudaComputeCapabilityProto) GetMinor 

func (x *CudaComputeCapabilityProto) GetMinor() int32

func (*CudaComputeCapabilityProto) ProtoMessage 

func (*CudaComputeCapabilityProto) ProtoMessage()

func (*CudaComputeCapabilityProto) ProtoReflect 

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

func (*CudaComputeCapabilityProto) Reset 

func (x *CudaComputeCapabilityProto) Reset()

func (*CudaComputeCapabilityProto) String 

func (x *CudaComputeCapabilityProto) String() string

type CudnnVersion 

type CudnnVersion struct {
	Major int32 `protobuf:"varint,1,opt,name=major,proto3" json:"major,omitempty"`
	Minor int32 `protobuf:"varint,2,opt,name=minor,proto3" json:"minor,omitempty"`
	Patch int32 `protobuf:"varint,3,opt,name=patch,proto3" json:"patch,omitempty"`
	// contains filtered or unexported fields
}

func (*CudnnVersion) Descriptor deprecated 

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

Deprecated: Use CudnnVersion.ProtoReflect.Descriptor instead.

func (*CudnnVersion) GetMajor 

func (x *CudnnVersion) GetMajor() int32

func (*CudnnVersion) GetMinor 

func (x *CudnnVersion) GetMinor() int32

func (*CudnnVersion) GetPatch 

func (x *CudnnVersion) GetPatch() int32

func (*CudnnVersion) ProtoMessage 

func (*CudnnVersion) ProtoMessage()

func (*CudnnVersion) ProtoReflect 

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

func (*CudnnVersion) Reset 

func (x *CudnnVersion) Reset()

func (*CudnnVersion) String 

func (x *CudnnVersion) String() string

type CustomCallApiVersion 

type CustomCallApiVersion int32

The version of the API used by the custom call function. The signatures for each version are given below. TODO(b/189822916): Remove this enum when all clients are migrated to the status-returning API. 

const (
	CustomCallApiVersion_API_VERSION_UNSPECIFIED CustomCallApiVersion = 0
	// The first version of the API, with the following signatures:
	//
	// CPU:
	//
	//	void do_custom_call(void* out, const void** in);
	//
	// GPU:
	//
	//	void do_custom_call(CUstream stream, void** buffers,
	//	                    const char* opaque, size_t opaque_len);
	CustomCallApiVersion_API_VERSION_ORIGINAL CustomCallApiVersion = 1
	// When the ability to return success/failure status was added:
	//
	// CPU:
	//
	//	void do_custom_call(void* out, const void** in,
	//	                    XlaCustomCallStatus* status);
	//
	// GPU:
	//
	//	void do_custom_call(CUstream stream, void** buffers,
	//	                    const char* opaque, size_t opaque_len,
	//	                    XlaCustomCallStatus* status);
	CustomCallApiVersion_API_VERSION_STATUS_RETURNING CustomCallApiVersion = 2
	// Fixes the API signatures on the CPU side of the version STATUS_RETURNING by
	// adding the opaque string so that the custom call API is consistent across
	// CPUs and GPUs. For GPUs, the behaviors invoked by
	// API_VERSION_STATUS_RETURNING and API_VERSION_STATUS_RETURNING_UNIFIED are
	// the same.
	//
	// CPU:
	//
	//	void do_custom_call(void* out, const void** in,
	//	                    const char* opaque, size_t opaque_len,
	//	                    XlaCustomCallStatus* status);
	//
	// GPU:
	//
	//	void do_custom_call(CUstream stream, void** buffers,
	//	                    const char* opaque, size_t opaque_len,
	//	                    XlaCustomCallStatus* status);
	CustomCallApiVersion_API_VERSION_STATUS_RETURNING_UNIFIED CustomCallApiVersion = 3
	// Api version implementing XLA runtime custom call calling convention. These
	// custom calls can be registered as an XLA runtime custom call (1) or as XLA
	// runtime FFI binding (2).
	//
	// This type of custom call uses custom ABI to pass type information along
	// with custom call arguments. Also it passes buffer arguments together with
	// data type, sizes and strides.
	//
	// Example: (XLA runtime custom call)
	//
	//	absl::Status DoCustomCall(StridedMemrefView arg, float attr);
	//
	//	CustomCall::Bind("custom_call")
	//	  .Arg<StridedMemrefView>()
	//	  .Attr<float>("attr")
	//	  .To(DoCustomCall);
	//
	// (1) xla/runtime/custom_call.h
	// (2) xla/runtime/ffi/ffi.h
	CustomCallApiVersion_API_VERSION_TYPED_FFI CustomCallApiVersion = 4
)

func (CustomCallApiVersion) Descriptor 

func (CustomCallApiVersion) Descriptor() protoreflect.EnumDescriptor

func (CustomCallApiVersion) Enum 

func (x CustomCallApiVersion) Enum() *CustomCallApiVersion

func (CustomCallApiVersion) EnumDescriptor deprecated 

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

Deprecated: Use CustomCallApiVersion.Descriptor instead.

func (CustomCallApiVersion) Number 

func (x CustomCallApiVersion) Number() protoreflect.EnumNumber

func (CustomCallApiVersion) String 

func (x CustomCallApiVersion) String() string

func (CustomCallApiVersion) Type 

func (CustomCallApiVersion) Type() protoreflect.EnumType

type CustomCallSchedule 

type CustomCallSchedule int32
const (
	CustomCallSchedule_SCHEDULE_NONE     CustomCallSchedule = 0
	CustomCallSchedule_SCHEDULE_LATEST   CustomCallSchedule = 1
	CustomCallSchedule_SCHEDULE_EARLIEST CustomCallSchedule = 2
)

func (CustomCallSchedule) Descriptor 

func (CustomCallSchedule) Descriptor() protoreflect.EnumDescriptor

func (CustomCallSchedule) Enum 

func (x CustomCallSchedule) Enum() *CustomCallSchedule

func (CustomCallSchedule) EnumDescriptor deprecated 

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

Deprecated: Use CustomCallSchedule.Descriptor instead.

func (CustomCallSchedule) Number 

func (x CustomCallSchedule) Number() protoreflect.EnumNumber

func (CustomCallSchedule) String 

func (x CustomCallSchedule) String() string

func (CustomCallSchedule) Type 

func (CustomCallSchedule) Type() protoreflect.EnumType

type DataLayout 

type DataLayout int32

Describes how a convolution input or output layer's data is formatted. 

const (
	// Naming convention:
	// Y <-> row or height
	// X <-> column or width
	// Batch <-> batch, or N
	// Depth <-> feature, or channel
	// TODO(timshen): turn them into cuDNN names, e.g. kNCHW.
	//
	// Note: In cudnn, kBatchDepthYX4 and kBatchDepthYX32 are the same layout
	// (namely, NCHW_VECT_C).  It differentiates between these two by using a
	// different data type (int8x4 vs int8x32).  In StreamExecutor we use
	// different layouts for these, because we don't usually pass an explicit data
	// type to StreamExecutor functions.
	DataLayout_kYXDepthBatch   DataLayout = 0
	DataLayout_kYXBatchDepth   DataLayout = 1
	DataLayout_kBatchYXDepth   DataLayout = 2 // cuDNN's NHWC layout
	DataLayout_kBatchDepthYX   DataLayout = 3 // cuDNN's NCHW layout
	DataLayout_kBatchDepthYX4  DataLayout = 4 // cuDNN's NCHW_VECT_C with 4-elem vectors (e.g. int8x4)
	DataLayout_kBatchDepthYX32 DataLayout = 5 // cuDNN's NCHW_VECT_C with 32-elem vects (e.g. int8x32)
)

func (DataLayout) Descriptor 

func (DataLayout) Descriptor() protoreflect.EnumDescriptor

func (DataLayout) Enum 

func (x DataLayout) Enum() *DataLayout

func (DataLayout) EnumDescriptor deprecated 

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

Deprecated: Use DataLayout.Descriptor instead.

func (DataLayout) Number 

func (x DataLayout) Number() protoreflect.EnumNumber

func (DataLayout) String 

func (x DataLayout) String() string

func (DataLayout) Type 

func (DataLayout) Type() protoreflect.EnumType

type DataType 

type DataType int32

Specifies the data type used by an operation. 

const (
	DataType_kFloat         DataType = 0
	DataType_kDouble        DataType = 1
	DataType_kHalf          DataType = 2
	DataType_kInt8          DataType = 3
	DataType_kInt32         DataType = 4
	DataType_kComplexFloat  DataType = 5
	DataType_kComplexDouble DataType = 6
	DataType_kBF16          DataType = 7
	DataType_kF8E5M2        DataType = 8
	DataType_kF8E4M3FN      DataType = 9
	DataType_kF8E5M2FNUZ    DataType = 10
	DataType_kF8E4M3FNUZ    DataType = 11
	DataType_kInt64         DataType = 12
)

func (DataType) Descriptor 

func (DataType) Descriptor() protoreflect.EnumDescriptor

func (DataType) Enum 

func (x DataType) Enum() *DataType

func (DataType) EnumDescriptor deprecated 

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

Deprecated: Use DataType.Descriptor instead.

func (DataType) Number 

func (x DataType) Number() protoreflect.EnumNumber

func (DataType) String 

func (x DataType) String() string

func (DataType) Type 

func (DataType) Type() protoreflect.EnumType

type DebugOptions 

type DebugOptions struct {

	// Show addresses of HLO ops in graph dump.
	XlaHloGraphAddresses bool `` /* 126-byte string literal not displayed */
	// Instrument the computation to collect per-HLO cycle counts.
	XlaHloProfile bool `protobuf:"varint,9,opt,name=xla_hlo_profile,json=xlaHloProfile,proto3" json:"xla_hlo_profile,omitempty"`
	// List of HLO passes to disable/enable. These names must exactly match the
	// pass names as specified by the HloPassInterface::name() method.
	//
	// At least one of xla_disable_hlo_passes and xla_enable_hlo_passes_only must
	// be empty.
	XlaDisableHloPasses    []string `protobuf:"bytes,30,rep,name=xla_disable_hlo_passes,json=xlaDisableHloPasses,proto3" json:"xla_disable_hlo_passes,omitempty"`
	XlaEnableHloPassesOnly []string `` /* 135-byte string literal not displayed */
	// Disables all HLO passes.  Notes that some passes are necessary for
	// correctness and the invariants that must be satisfied by "fully optimized"
	// HLO are different for different devices and may change over time.  The only
	// "guarantee", such as it is, is that if you compile XLA and dump the
	// optimized HLO for some graph, you should be able to run it again on the
	// same device with the same build of XLA.
	XlaDisableAllHloPasses bool `` /* 136-byte string literal not displayed */
	// Numerical optimization level for the XLA compiler backend; the specific
	// interpretation of this value is left to the backends.
	XlaBackendOptimizationLevel int32 `` /* 148-byte string literal not displayed */
	// Embed the compiler IR as a string in the executable.
	XlaEmbedIrInExecutable bool `` /* 135-byte string literal not displayed */
	// Eliminate implicit broadcasts when lowering user computations to HLO
	// instructions; use explicit broadcast instead.
	XlaEliminateHloImplicitBroadcast bool `` /* 165-byte string literal not displayed */
	// When generating calls to Eigen in the CPU backend, use multi-threaded Eigen
	// mode.
	XlaCpuMultiThreadEigen bool `` /* 135-byte string literal not displayed */
	// Path to directory with cuda/ptx tools and libraries.
	XlaGpuCudaDataDir string `protobuf:"bytes,61,opt,name=xla_gpu_cuda_data_dir,json=xlaGpuCudaDataDir,proto3" json:"xla_gpu_cuda_data_dir,omitempty"`
	// Enable flush-to-zero semantics in the GPU backend.
	XlaGpuFtz bool `protobuf:"varint,62,opt,name=xla_gpu_ftz,json=xlaGpuFtz,proto3" json:"xla_gpu_ftz,omitempty"`
	// If true, in LLVM-based backends, emit !alias.scope metadata in
	// generated IR.
	XlaLlvmEnableAliasScopeMetadata bool `` /* 164-byte string literal not displayed */
	// If true, in LLVM-based backends, emit !noalias metadata in the
	// generated IR.
	XlaLlvmEnableNoaliasMetadata bool `` /* 153-byte string literal not displayed */
	// If true, in LLVM-based backends, emit !invariant.load metadata in
	// the generated IR.
	XlaLlvmEnableInvariantLoadMetadata bool `` /* 173-byte string literal not displayed */
	// If true, a set of expensive LLVM optimization passes will not be run.
	XlaLlvmDisableExpensivePasses bool `` /* 156-byte string literal not displayed */
	// This is used by ClientLibraryTestBase::ComputeAndCompare*. If true, the
	// computation will run n! times with all permunations of layouts for the
	// output shape in rank n. For example, with a 3D shape, all permutations of
	// the set {0, 1, 2} are tried.
	XlaTestAllOutputLayouts bool `` /* 138-byte string literal not displayed */
	// This is used by ClientLibraryTestBase::ComputeAndCompare*. If true, the
	// computation will run for all permunations of layouts of all input
	// arguments. For example, with 2 input arguments in 2D and 4D shapes, the
	// computation will run 2! * 4! times.
	XlaTestAllInputLayouts bool `` /* 135-byte string literal not displayed */
	// Assign colors based on sharding information when generating the Graphviz
	// HLO graph.
	XlaHloGraphShardingColor bool `` /* 141-byte string literal not displayed */
	// Generate calls to MKL-DNN in the CPU backend.
	XlaCpuUseMklDnn       bool `protobuf:"varint,97,opt,name=xla_cpu_use_mkl_dnn,json=xlaCpuUseMklDnn,proto3" json:"xla_cpu_use_mkl_dnn,omitempty"`
	XlaCpuUseThunkRuntime bool `` /* 133-byte string literal not displayed */
	// When true, XLA:CPU uses HLO module scheduler that is optimized for
	// extracting concurrency at the cost of extra memory: we extend the live
	// ranges of temporaries to allow XLA runtime to schedule independent
	// operations in parallel on separate threads.
	XlaCpuEnableConcurrencyOptimizedScheduler bool `` /* 195-byte string literal not displayed */
	// A `prefer-vector-width` value that is passed to the LLVM backend. Default
	// value is `256` (AVX2 on x86 platforms).
	XlaCpuPreferVectorWidth int32 `` /* 139-byte string literal not displayed */
	// When true, "unsafe" mathematical optimizations are enabled. These
	// transformations include but are not limited to:
	//
	//   - Reducing the precision of operations (e.g. using an approximate sin
	//     function, or transforming x/y into x * (1/y)).
	//   - Assuming that operations never produce or consume NaN or +/- Inf (this
	//     behavior can be adjusted using xla_cpu_fast_math_allow_{nans|infs}).
	//   - Assuming that +0 and -0 are indistinguishable.
	XlaCpuEnableFastMath bool `` /* 129-byte string literal not displayed */
	// When xla_cpu_enable_fast_math is true then this controls whether we allow
	// operations to produce NaNs.  Ignored when xla_cpu_enable_fast_math is
	// false.
	XlaCpuFastMathHonorNans bool `` /* 141-byte string literal not displayed */
	// When xla_cpu_enable_fast_math is true then this controls whether we allow
	// operations to produce infinites. Ignored when xla_cpu_enable_fast_math is
	// false.
	XlaCpuFastMathHonorInfs bool `` /* 141-byte string literal not displayed */
	// When xla_cpu_enable_fast_math is true then this controls whether we forbid
	// to use the reciprocal of an argument instead of division. Ignored when
	// xla_cpu_enable_fast_math is false.
	XlaCpuFastMathHonorDivision bool `` /* 153-byte string literal not displayed */
	// When xla_cpu_enable_fast_math is true then this controls whether we forbid
	// to approximate calculations for functions. Ignored when
	// xla_cpu_enable_fast_math is false.
	XlaCpuFastMathHonorFunctions bool `` /* 156-byte string literal not displayed */
	// When false we lower the Minimum and Maximum hlos in the CPU backend such
	// that Min(NotNaN, NaN) = Min(NaN, NotNaN) = NaN.  In other words, if flag
	// this is false we always propagate NaNs through Min and Max.
	//
	// Note, this does not correspond to the exact same behavior as the gpu flag
	// below!
	XlaCpuEnableFastMinMax bool `` /* 138-byte string literal not displayed */
	// When true we lower the Minimum and Maximum hlos in the GPU backend such
	// that Min(NotNaN, NaN) = Min(NaN, NotNaN) = NotNaN.  In other words, if flag
	// this is true we don't propagate NaNs through Min and Max.
	//
	// Note, this does not correspond to the exact same behavior as the cpu flag
	// above!
	XlaGpuEnableFastMinMax bool `` /* 138-byte string literal not displayed */
	// Allows xla to increase the output precision of floating point operations
	// and all floating-point conversions to be simplified, including those
	// that affect the numerics. The `FloatNormalization` pass inserts many
	// `f32 -> bf16 -> f32` conversion pairs. These are not removed by the
	// `AlgebraicSimplifier`, as that will only simplify conversions that are
	// no-ops, e.g. `bf16 -> f32 -> bf16`. Removing these improves accuracy.
	XlaAllowExcessPrecision bool `` /* 137-byte string literal not displayed */
	// Crashes the program when any kind of verification fails, instead of just
	// logging the failures. One example is cross checking of convolution results
	// among different algorithms.
	XlaGpuCrashOnVerificationFailures bool `` /* 171-byte string literal not displayed */
	// 0:   Disable gemm and convolution autotuning.
	// 1:   Enable autotuning, but disable correctness checking.
	// 2:   Also set output buffers to random numbers during autotuning.
	// 3:   Also reset output buffers to random numbers after autotuning each
	//
	//	algorithm.
	//
	// 4+:  Also check for correct outputs and for out-of-bounds reads/writes.
	//
	// Default: 4.
	XlaGpuAutotuneLevel int32 `protobuf:"varint,123,opt,name=xla_gpu_autotune_level,json=xlaGpuAutotuneLevel,proto3" json:"xla_gpu_autotune_level,omitempty"`
	// Force the host platform to pretend that there are these many host
	// "devices".  All these devices are backed by the same threadpool.  Defaults
	// to 1.
	//
	// Setting this to anything other than 1 can increase overhead from context
	// switching but we let the user override this behavior to help run tests on
	// the host that run models in parallel across multiple devices.
	XlaForceHostPlatformDeviceCount int32 `` /* 165-byte string literal not displayed */
	// If set to true XLA:GPU invokes `ptxas` with -O0 (default is -O3).
	XlaGpuDisableGpuasmOptimizations bool                     `` /* 166-byte string literal not displayed */
	XlaGpuShapeChecks                DebugOptions_ShapeChecks `` /* 153-byte string literal not displayed */
	// Enable fast math with eigen in the HLO evaluator.
	XlaHloEvaluatorUseFastPath bool `` /* 150-byte string literal not displayed */
	// Temporary option to allow support for both the R1 and the scalar index
	// versions of DynamicSlice and DynamicUpdateSlice. Only used for testing.
	XlaAllowScalarIndexDynamicOps bool `` /* 159-byte string literal not displayed */
	// Option to emit a target-specific marker to indicate the start of a training
	// step. The location of the marker (if any) is determined by the option
	// value.
	XlaStepMarkerLocation DebugOptions_StepMarkerLocation `` /* 172-byte string literal not displayed */
	// Directory to dump into.
	XlaDumpTo string `protobuf:"bytes,109,opt,name=xla_dump_to,json=xlaDumpTo,proto3" json:"xla_dump_to,omitempty"`
	// If specified, will only dump modules which match this regexp.
	XlaDumpHloModuleRe string `protobuf:"bytes,110,opt,name=xla_dump_hlo_module_re,json=xlaDumpHloModuleRe,proto3" json:"xla_dump_hlo_module_re,omitempty"`
	// If this flag is specified, will also dump HLO before and after passes that
	// match this regular expression.  Set to .* to dump before/after all passes.
	XlaDumpHloPassRe string `protobuf:"bytes,111,opt,name=xla_dump_hlo_pass_re,json=xlaDumpHloPassRe,proto3" json:"xla_dump_hlo_pass_re,omitempty"`
	// Specifies the format that HLO is dumped in.  Multiple of these may be
	// specified.
	XlaDumpHloAsText  bool `protobuf:"varint,112,opt,name=xla_dump_hlo_as_text,json=xlaDumpHloAsText,proto3" json:"xla_dump_hlo_as_text,omitempty"`
	XlaDumpHloAsProto bool `protobuf:"varint,113,opt,name=xla_dump_hlo_as_proto,json=xlaDumpHloAsProto,proto3" json:"xla_dump_hlo_as_proto,omitempty"`
	XlaDumpHloAsDot   bool `protobuf:"varint,114,opt,name=xla_dump_hlo_as_dot,json=xlaDumpHloAsDot,proto3" json:"xla_dump_hlo_as_dot,omitempty"`
	XlaDumpHloAsUrl   bool `protobuf:"varint,115,opt,name=xla_dump_hlo_as_url,json=xlaDumpHloAsUrl,proto3" json:"xla_dump_hlo_as_url,omitempty"`
	// Dump HLO graphs as an HTML (DOT -> SVG inlined in HTML)
	XlaDumpHloAsHtml bool `protobuf:"varint,116,opt,name=xla_dump_hlo_as_html,json=xlaDumpHloAsHtml,proto3" json:"xla_dump_hlo_as_html,omitempty"`
	// Dump the visualization of the fusion progress.
	XlaDumpFusionVisualization bool `` /* 146-byte string literal not displayed */
	// If true, every time an HLO module is run, we will dump an HloSnapshot
	// (essentially, a serialized module plus its inputs) to the --xla_dump_to
	// directory.
	XlaDumpHloSnapshots bool `protobuf:"varint,118,opt,name=xla_dump_hlo_snapshots,json=xlaDumpHloSnapshots,proto3" json:"xla_dump_hlo_snapshots,omitempty"`
	// Include a timestamp in the dumped filenames.
	XlaDumpIncludeTimestamp bool `` /* 137-byte string literal not displayed */
	// Max number of hlo module dumps in a directory. Set to < 0 for unbounded.
	XlaDumpMaxHloModules int32 `` /* 130-byte string literal not displayed */
	// Dump HloModuleMetadata as a text proto for each HLO module.
	XlaDumpModuleMetadata bool `` /* 131-byte string literal not displayed */
	// GZip-compress protos dumped via --xla_dump_hlo_as_proto.
	XlaDumpCompressProtos bool `` /* 131-byte string literal not displayed */
	// Dump HLO in long text format. Ignored unless xla_dump_hlo_as_text is true.
	XlaDumpHloAsLongText bool `` /* 132-byte string literal not displayed */
	// Overrides for XLA GPU's convolution layout heuristic.
	XlaGpuForceConvNchw bool `` /* 127-byte string literal not displayed */
	XlaGpuForceConvNhwc bool `` /* 127-byte string literal not displayed */
	// Paths to files with ptx code.
	XlaGpuPtxFile []string `protobuf:"bytes,127,rep,name=xla_gpu_ptx_file,json=xlaGpuPtxFile,proto3" json:"xla_gpu_ptx_file,omitempty"`
	// Whether to dump llvm ir when compiling to ptx.
	XlaGpuDumpLlvmir bool `protobuf:"varint,155,opt,name=xla_gpu_dump_llvmir,json=xlaGpuDumpLlvmir,proto3" json:"xla_gpu_dump_llvmir,omitempty"`
	// Whether to dump mlir using pretty print form.
	XlaDumpEnableMlirPrettyForm bool `` /* 153-byte string literal not displayed */
	// Denylist for cuDNN convolutions.
	XlaGpuAlgorithmDenylistPath string `` /* 150-byte string literal not displayed */
	// Debug options that trigger execution errors when NaN or Inf are detected.
	XlaTpuDetectNan bool `protobuf:"varint,135,opt,name=xla_tpu_detect_nan,json=xlaTpuDetectNan,proto3" json:"xla_tpu_detect_nan,omitempty"`
	XlaTpuDetectInf bool `protobuf:"varint,136,opt,name=xla_tpu_detect_inf,json=xlaTpuDetectInf,proto3" json:"xla_tpu_detect_inf,omitempty"`
	// True if TraceMe annotations are enabled for XLA:CPU.
	XlaCpuEnableXprofTraceme bool `` /* 142-byte string literal not displayed */
	// It is usually preferable to not fallback to the driver; it can consume more
	// memory, or have bugs.
	XlaGpuUnsafeFallbackToDriverOnPtxasNotFound bool `` /* 209-byte string literal not displayed */
	// Extra parameters to pass the GPU assembler.
	XlaGpuAsmExtraFlags string `` /* 126-byte string literal not displayed */
	// Per-heap size constraint. New heaps will be created if per-heap max size is
	// reached.
	XlaMultiheapSizeConstraintPerHeap int32 `` /* 171-byte string literal not displayed */
	// Enable detailed logging into vlog. If this is disabled, no
	// compilation summary will be printed in the end of computation.
	XlaDetailedLogging bool `protobuf:"varint,252,opt,name=xla_detailed_logging,json=xlaDetailedLogging,proto3" json:"xla_detailed_logging,omitempty"`
	// Enable HLO dumping. If this is disabled, no HLO modules will be dumped.
	XlaEnableDumping bool `protobuf:"varint,253,opt,name=xla_enable_dumping,json=xlaEnableDumping,proto3" json:"xla_enable_dumping,omitempty"`
	// Overrides normal multi-threaded compilation setting to use this many
	// threads. Setting to 0 (the default value) means no enforcement.
	XlaGpuForceCompilationParallelism            int32 `` /* 169-byte string literal not displayed */
	XlaGpuEnableLlvmModuleCompilationParallelism bool  `` /* 206-byte string literal not displayed */
	// Guarantees run-to-run determinism.
	// This flag implies --xla_gpu_exclude_nondeterministic_ops and in addition
	// disables autotuning.
	XlaGpuDeterministicOps bool `` /* 134-byte string literal not displayed */
	// Paths to files with LLVM code.
	XlaGpuLlvmIrFile              []string                        `protobuf:"bytes,150,rep,name=xla_gpu_llvm_ir_file,json=xlaGpuLlvmIrFile,proto3" json:"xla_gpu_llvm_ir_file,omitempty"`
	XlaGpuDisableAsyncCollectives []DebugOptions_CollectiveOpType `` /* 203-byte string literal not displayed */
	// Size threshold (in bytes) for the GPU collective combiners.
	XlaGpuAllReduceCombineThresholdBytes     int64 `` /* 182-byte string literal not displayed */
	XlaGpuAllGatherCombineThresholdBytes     int64 `` /* 182-byte string literal not displayed */
	XlaGpuReduceScatterCombineThresholdBytes int64 `` /* 194-byte string literal not displayed */
	// Combine all-gather/scatter-reduce ops with the same dimension or
	// irrespective of their dimension.
	XlaGpuEnableAllGatherCombineByDim     bool `` /* 175-byte string literal not displayed */
	XlaGpuEnableReduceScatterCombineByDim bool `` /* 187-byte string literal not displayed */
	// Split AR across all partitions into logical RS followed by AR.
	XlaGpuEnableAllReduceSplitter bool `` /* 159-byte string literal not displayed */
	// Enable allreduce reassociation on allreduces that are converted to a wider
	// type. The resulting allreduce will be promoted to a wider-typed allreduce.
	XlaGpuEnableReassociationForConvertedAr bool `` /* 191-byte string literal not displayed */
	// Number of devices per host for first stage of BlueConnect decomposition
	// pass. The pass will attempt to decompose all-reduces ops into a
	// ReduceScatter-AllReduce-AllGather sequence, with the initial ReduceScatter
	// being performed over all of the devices in the same host. Set to < 1 to
	// disable all-reduce decomposition.
	XlaGpuAllReduceBlueconnectNumDevicesPerHost int32 `` /* 207-byte string literal not displayed */
	// Enable hoisting of reduce-scatter out of while loops.
	XlaGpuEnableWhileLoopReduceScatterCodeMotion bool `` /* 210-byte string literal not displayed */
	// Inflate collective cost by running each collective multiple times.
	XlaGpuCollectiveInflationFactor int32 `` /* 163-byte string literal not displayed */
	// Whether to force inline before llvm module split to get a more balanced
	// splits for parallel compilation.
	XlaLlvmForceInlineBeforeSplit bool `` /* 159-byte string literal not displayed */
	// Whether to use the cuDNN frontend API for convolutions when possible.
	XlaGpuEnableCudnnFrontend       bool `` /* 145-byte string literal not displayed */
	XlaGpuEnableCudnnFmha           bool `` /* 133-byte string literal not displayed */
	XlaGpuFusedAttentionUseCudnnRng bool `` /* 167-byte string literal not displayed */
	// Rewrite layer norm patterns into cuDNN library calls.
	XlaGpuEnableCudnnLayerNorm bool `` /* 150-byte string literal not displayed */
	// Disable dumping metadata in HLO dumps.
	XlaDumpDisableMetadata bool `` /* 134-byte string literal not displayed */
	// If this flag is specified, will only dump HLO before and after passes in
	// the pass pipeline that matches this regular expression. Default empty value
	// enables dumping in all pipelines.
	XlaDumpHloPipelineRe string `` /* 129-byte string literal not displayed */
	// If true, abort immediately when conv algorithm picker fails, rather than
	// logging a warning and proceeding with fallback.
	XlaGpuStrictConvAlgorithmPicker bool `` /* 165-byte string literal not displayed */
	// If true, XLA will try to pattern match subgraphs of HLO operations into
	// custom fusions registered in the current process (pre-compiled hand written
	// kernels, e.g. various GEMM fusions writtent in CUTLASS).
	XlaGpuEnableCustomFusions bool `` /* 145-byte string literal not displayed */
	// A regular expression enabling only a subset of custom fusions. Enabled only
	// if `xla_gpu_enable_custom_fusion` set to true.
	XlaGpuEnableCustomFusionsRe string `` /* 152-byte string literal not displayed */
	// Enables address computation fusion to optimize dynamic-slice and
	// dynamic-update-slice operations around library calls.
	XlaGpuEnableAddressComputationFusion bool `` /* 180-byte string literal not displayed */
	// Timeout in seconds before terminating jobs that are stuck in a NCCL
	// Rendezvous. Negative value disables the timeout and will not terminate.
	XlaGpuNcclTerminationTimeoutSeconds int64 `` /* 177-byte string literal not displayed */
	// Enables shared constants for XLA/GPU. This allows large constants to be
	// shared among multiple GPU executables.
	XlaGpuEnableSharedConstants bool `` /* 151-byte string literal not displayed */
	// Whether to use cuBLASLt for GEMMs on GPUs.
	XlaGpuEnableCublaslt bool `` /* 128-byte string literal not displayed */
	// Determine the types of commands that are recorded into command buffers.
	XlaGpuEnableCommandBuffer []DebugOptions_CommandBufferCmdType `` /* 195-byte string literal not displayed */
	// This number determines how many moved instructions like fusion kernels are
	// required for a region to be captured as a function to be launched as a GPU
	// graph.
	XlaGpuGraphMinGraphSize int32 `` /* 141-byte string literal not displayed */
	// Identify concurrent regions in GPU graphs and execute them concurrently.
	XlaGpuGraphEnableConcurrentRegion bool `` /* 171-byte string literal not displayed */
	// Size threshold (in megabytes) for the GPU redzone scratch allocator.
	XlaGpuRedzoneScratchMaxMegabytes int64 `` /* 168-byte string literal not displayed */
	// Amount of padding the redzone allocator will put on one side of each buffer
	// it allocates.  (So the buffer's total size will be increased by 2x this
	// value.)
	//
	// Higher values make it more likely that we'll catch an out-of-bounds read or
	// write.  Smaller values consume less memory during autotuning.  Note that a
	// fused cudnn conv has up to 6 total buffers (4 inputs, 1 output, and 1
	// scratch), so this can be multiplied by quite a lot.
	XlaGpuRedzonePaddingBytes int64 `` /* 145-byte string literal not displayed */
	// Generate calls to Arm Compute Library in the CPU backend.
	XlaCpuUseAcl bool `protobuf:"varint,174,opt,name=xla_cpu_use_acl,json=xlaCpuUseAcl,proto3" json:"xla_cpu_use_acl,omitempty"`
	// By default, XLA:CPU will run fp16 dot/conv as fp32, as this is generally
	// (much) faster on our hardware.  Set this flag to disable this behavior.
	XlaCpuStrictDotConvMath bool `` /* 141-byte string literal not displayed */
	// An option to enable using cuDNN runtime compiled fusion kernels which is
	// available and recommended for Ampere+ GPUs.
	XlaGpuUseRuntimeFusion       bool `` /* 136-byte string literal not displayed */
	XlaDumpLatencyHidingSchedule bool `` /* 154-byte string literal not displayed */
	// By default, MLIR lowering will use Linalg elementwise fusion. If this flag
	// is enabled, the pipeline will use tiling, fusion, peeling, vectorization
	// instead.
	XlaCpuEnableMlirTilingAndFusion bool `` /* 167-byte string literal not displayed */
	// XLA:CPU-Next tiling parameters for matmul.
	XlaCpuEnableCustomMatmulTiling  bool  `` /* 162-byte string literal not displayed */
	XlaCpuMatmulTilingMDim          int64 `` /* 138-byte string literal not displayed */
	XlaCpuMatmulTilingNDim          int64 `` /* 138-byte string literal not displayed */
	XlaCpuMatmulTilingKDim          int64 `` /* 138-byte string literal not displayed */
	XlaCpuEnableMlirFusionOutlining bool  `` /* 165-byte string literal not displayed */
	// If set, use the experimental deallocation pass from mlir-hlo.
	XlaCpuEnableExperimentalDeallocation   bool   `` /* 178-byte string literal not displayed */
	XlaGpuEnableLatencyHidingScheduler     bool   `` /* 174-byte string literal not displayed */
	XlaGpuEnableHighestPriorityAsyncStream bool   `` /* 188-byte string literal not displayed */
	XlaGpuEnableAnalyticalLatencyEstimator bool   `` /* 186-byte string literal not displayed */
	XlaGpuLhsEnableGpuAsyncTracker         bool   `` /* 164-byte string literal not displayed */
	XlaGpuPgleProfileFileOrDirectoryPath   string `` /* 183-byte string literal not displayed */
	XlaGpuMemoryLimitSlopFactor            int32  `` /* 153-byte string literal not displayed */
	XlaGpuEnablePipelinedCollectives       bool   `` /* 166-byte string literal not displayed */
	XlaGpuEnablePipelinedAllReduce         bool   `` /* 162-byte string literal not displayed */
	XlaGpuEnablePipelinedAllGather         bool   `` /* 162-byte string literal not displayed */
	XlaGpuEnablePipelinedReduceScatter     bool   `` /* 174-byte string literal not displayed */
	XlaGpuEnablePipelinedP2P               bool   `` /* 142-byte string literal not displayed */
	// The minimum data size in bytes to trigger collective-permute-decomposer
	// transformation.
	XlaGpuCollectivePermuteDecomposerThreshold int64 `` /* 198-byte string literal not displayed */
	// The partitioning algorithm to be used in the PartitionAssignment pass.
	XlaPartitioningAlgorithm                      DebugOptions_PartitioningAlgorithm `` /* 182-byte string literal not displayed */
	XlaGpuEnableTritonGemm                        bool                               `` /* 136-byte string literal not displayed */
	XlaGpuEnableCudnnInt8X32ConvolutionReordering bool                               `` /* 209-byte string literal not displayed */
	// Creates triton fusion for all supported gemms.
	// To make sure only triton gemm is chosen by the autotuner run with
	// `xla_gpu_cublas_fallback` set to false.
	XlaGpuTritonGemmAny                     bool `` /* 127-byte string literal not displayed */
	XlaGpuExhaustiveTilingSearch            bool `` /* 154-byte string literal not displayed */
	XlaGpuEnableTritonSoftmaxFusion         bool `` /* 165-byte string literal not displayed */
	XlaGpuEnablePriorityFusion              bool `` /* 148-byte string literal not displayed */
	XlaGpuEnableTritonSoftmaxPriorityFusion bool `` /* 191-byte string literal not displayed */
	// File to write autotune results to. It will be a binary file unless the name
	// ends with .txt or .textproto. Warning: The results are written at every
	// compilation, possibly multiple times per process. This only works on CUDA.
	XlaGpuDumpAutotuneResultsTo string `` /* 152-byte string literal not displayed */
	// File to load autotune results from. It will be considered a binary file
	// unless the name ends with .txt or .textproto. At most one loading will
	// happen during the lifetime of one process, even if the first one is
	// unsuccessful or different file paths are passed here. This only works on
	// CUDA.
	XlaGpuLoadAutotuneResultsFrom string `` /* 158-byte string literal not displayed */
	// Description of the target platform in GpuTargetConfigProto format; if
	// provided, deviceless compilation is assumed, and the current device is
	// ignored.
	XlaGpuTargetConfigFilename string `` /* 147-byte string literal not displayed */
	// Memory budget in GB per device for AutoSharding.
	XlaGpuAutoSpmdPartitioningMemoryBudgetGb int32 `` /* 196-byte string literal not displayed */
	// See the definition of the
	// xla_gpu_auto_spmd_partitioning_memory_budget_ratio flag for the meaning of
	// this field.
	XlaGpuAutoSpmdPartitioningMemoryBudgetRatio      float32 `` /* 206-byte string literal not displayed */
	XlaGpuTritonGemmDisableReducedPrecisionReduction bool    `` /* 220-byte string literal not displayed */
	XlaGpuTritonFusionLevel                          int32   `` /* 139-byte string literal not displayed */
	XlaGpuDumpAutotunedGemmFusions                   bool    `` /* 162-byte string literal not displayed */
	XlaGpuOverrideGemmAutotuner                      string  `` /* 150-byte string literal not displayed */
	XlaGpuCopyInsertionUseRegionAnalysis             bool    `` /* 182-byte string literal not displayed */
	// If true, each fusion instruction will have a cost model runtime estimate in
	// backend config after compilation.
	XlaGpuCollectCostModelStats  bool `` /* 153-byte string literal not displayed */
	XlaGpuEnableSplitKAutotuning bool `` /* 156-byte string literal not displayed */
	// Whether reduction epilogue fusion is enabled in fusion passes.
	XlaGpuEnableReductionEpilogueFusion bool `` /* 177-byte string literal not displayed */
	// Allow early return when acquiring NCCL cliques.
	XlaGpuEnableNcclCliqueOptimization bool `` /* 174-byte string literal not displayed */
	// Replace custom calls with noop operations.
	XlaGpuMockCustomCalls bool `` /* 133-byte string literal not displayed */
	// Allow Triton GEMM autotuning to fall back to cuBLAS when that is
	// faster.
	XlaGpuCublasFallback bool `` /* 128-byte string literal not displayed */
	// Enable double buffering for loops.
	XlaGpuEnableWhileLoopDoubleBuffering bool `` /* 182-byte string literal not displayed */
	// Determine the while loop unrolling scheme.
	XlaGpuEnableWhileLoopUnrolling DebugOptions_WhileLoopUnrolling `` /* 203-byte string literal not displayed */
	// Change the layout of the second triton dot operand to be column major.
	// Only works for (bf16 x bf16) -> bf16.
	XlaGpuEnsureMinorDotContractionDims bool `` /* 179-byte string literal not displayed */
	// Filter out kernels that spill registers during autotuning.
	XlaGpuFilterKernelsSpillingRegistersOnAutotuning bool `` /* 220-byte string literal not displayed */
	// Maximum number of buffers to print when debugging buffer assignment.
	XlaDebugBufferAssignmentShowMax int64 `` /* 165-byte string literal not displayed */
	XlaGpuLlvmVerificationLevel     int32 `` /* 151-byte string literal not displayed */
	// Enable radix sort using CUB.
	XlaGpuEnableCubRadixSort bool `` /* 144-byte string literal not displayed */
	// Threshold to enable windowed einsum (collective matmul) in MB.
	XlaGpuThresholdForWindowedEinsumMib int64 `` /* 179-byte string literal not displayed */
	// Enables currently disabled features within Triton for Hopper.
	XlaGpuEnableTritonHopper bool `` /* 142-byte string literal not displayed */
	// Enable NCCL user buffers.
	XlaGpuEnableNcclUserBuffers bool `` /* 153-byte string literal not displayed */
	// Enable NCCL communicator splitting.
	XlaGpuEnableNcclCommSplitting bool `` /* 159-byte string literal not displayed */
	// Enable NCCL per stream communicators.
	XlaGpuEnableNcclPerStreamComms bool `` /* 164-byte string literal not displayed */
	// If enabled, uses the libnvptxcompiler library to compile PTX to cuBIN.
	XlaGpuEnableLibnvptxcompiler     bool `` /* 152-byte string literal not displayed */
	XlaGpuEnableDotStrengthReduction bool `` /* 168-byte string literal not displayed */
	// Whether to use multiple compute streams to run windowed einsum.
	XlaGpuMultiStreamedWindowedEinsum bool `` /* 171-byte string literal not displayed */
	// If enabled, uses bf16_6way gemm to compute F32 gemm.
	XlaGpuEnableBf16_6WayGemm bool `` /* 144-byte string literal not displayed */
	// If enabled, uses bf16_3way gemm to compute F32 gemm.
	XlaGpuEnableBf16_3WayGemm bool `` /* 144-byte string literal not displayed */
	// Specify the maximum number of channels(SMs) NCCL
	// will use for collective operations.
	XlaGpuNcclCollectiveMaxNchannels int64 `` /* 168-byte string literal not displayed */
	// Specify the maximum number of channels(SMs) NCCL
	// will use for p2p operations.
	XlaGpuNcclP2PMaxNchannels int64 `` /* 147-byte string literal not displayed */
	// Choose the level of mlir emitters that are enabled.
	// Current levels:
	// 0: Disabled.
	// 1: Loop emitter
	// 2: + Loop-like emitters
	// 3: + Transpose
	// 4: + Reduce
	XlaGpuMlirEmitterLevel int64 `` /* 136-byte string literal not displayed */
	// The maximum number of kernels to emit with MLIR. Unlimited if 0.
	XlaGpuMaxMlirKernels int64 `` /* 130-byte string literal not displayed */
	// The number of initial kernels to not emit with MLIR. Only supported kernels
	// are counted.
	XlaGpuSkipMlirKernels int64 `` /* 133-byte string literal not displayed */
	// Threshold to rewrite matmul to cuBLAS or Triton (minumum combined number of
	// elements of both matrices in non-batch dimensions to be considered for a
	// rewrite).
	XlaGpuGemmRewriteSizeThreshold int64 `` /* 162-byte string literal not displayed */
	// If true, will require complete AOT autotuning results; in the case of
	// missing AOT result, the model will not be compiled or executed, a
	// `NotFound` error will be returned.
	XlaGpuRequireCompleteAotAutotuneResults bool `` /* 191-byte string literal not displayed */
	// Let GEMM fusion autotuning probe cuDNN as a backend.
	// Current levels:
	// 0: Disabled.
	// 1: Fusions of GEMM, elementwise, transpose/reshape operations.
	// 2: + Broadcasts.
	// 3: + Nontrivial noncontracting dimension reshapes/transposes.
	XlaGpuCudnnGemmFusionLevel int32 `` /* 150-byte string literal not displayed */
	// This instructs the runtime whether to use
	// memcpy for p2p communication when source and
	// target are located within a node(nvlink).
	XlaGpuUseMemcpyLocalP2P bool `` /* 141-byte string literal not displayed */
	// If non-zero, limits the number of solutions to be used by GEMM autotuner.
	// This might be useful if underlying math library returns too many GEMM
	// solutions.
	XlaGpuAutotuneMaxSolutions int64 `` /* 148-byte string literal not displayed */
	// If true, large constants will be printed out when dumping HLOs.
	XlaDumpLargeConstants bool `` /* 131-byte string literal not displayed */
	// If true, will verify that the numerical results of Triton fusions match
	// the results of regular emitters.
	XlaGpuVerifyTritonFusionNumerics bool `` /* 168-byte string literal not displayed */
	// File to write autotune logs to. It will stored in txt format.
	XlaGpuDumpAutotuneLogsTo string `` /* 143-byte string literal not displayed */
	// Base length to rewrite the reduce window to, no rewrite if set to 0.
	XlaReduceWindowRewriteBaseLength int64 `` /* 168-byte string literal not displayed */
	// If true, will enable host memory offloading on a device.
	XlaGpuEnableHostMemoryOffloading bool `` /* 168-byte string literal not displayed */
	// Excludes non-deterministic ops from compiled executables.
	// Unlike --xla_gpu_deterministic_ops does not disable autotuning - the
	// compilation itself can be non-deterministic.
	// At present, the HLO op SelectAndScatter does not have a
	// deterministic XLA:GPU implementation.
	// Compilation errors out if SelectAndScatter is encountered.
	// Scatter ops can non-deterministic by default; these get converted to
	// a deterministic implementation.
	XlaGpuExcludeNondeterministicOps bool `` /* 166-byte string literal not displayed */
	// If true, Nccl errors will terminate the process.
	XlaGpuNcclTerminateOnError bool `` /* 150-byte string literal not displayed */
	// Use Shardonnay, a new partitioner, to replace the existing
	// ShardingPropagation and SpmdPartitioner. See go/xla-sdy-pipeline for
	// details.
	XlaUseShardonnay                    bool   `protobuf:"varint,302,opt,name=xla_use_shardonnay,json=xlaUseShardonnay,proto3" json:"xla_use_shardonnay,omitempty"`
	XlaGpuShardAutotuning               bool   `` /* 131-byte string literal not displayed */
	XlaGpuEnableApproxCostlyCollectives bool   `` /* 177-byte string literal not displayed */
	XlaGpuKernelCacheFile               string `` /* 132-byte string literal not displayed */
	// Extra options to pass to the compilation backend (e.g. LLVM); specific
	// interpretation of these values is left to the backend.
	XlaBackendExtraOptions map[string]string `` /* 221-byte string literal not displayed */
	// contains filtered or unexported fields
}

Debugging options for XLA. These options may change at any time - there are no guarantees about backward or forward compatibility for these fields.

func (*DebugOptions) Descriptor deprecated 

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

Deprecated: Use DebugOptions.ProtoReflect.Descriptor instead.

func (*DebugOptions) GetXlaAllowExcessPrecision 

func (x *DebugOptions) GetXlaAllowExcessPrecision() bool

func (*DebugOptions) GetXlaAllowScalarIndexDynamicOps 

func (x *DebugOptions) GetXlaAllowScalarIndexDynamicOps() bool

func (*DebugOptions) GetXlaBackendExtraOptions 

func (x *DebugOptions) GetXlaBackendExtraOptions() map[string]string

func (*DebugOptions) GetXlaBackendOptimizationLevel 

func (x *DebugOptions) GetXlaBackendOptimizationLevel() int32

func (*DebugOptions) GetXlaCpuEnableConcurrencyOptimizedScheduler 

func (x *DebugOptions) GetXlaCpuEnableConcurrencyOptimizedScheduler() bool

func (*DebugOptions) GetXlaCpuEnableCustomMatmulTiling 

func (x *DebugOptions) GetXlaCpuEnableCustomMatmulTiling() bool

func (*DebugOptions) GetXlaCpuEnableExperimentalDeallocation 

func (x *DebugOptions) GetXlaCpuEnableExperimentalDeallocation() bool

func (*DebugOptions) GetXlaCpuEnableFastMath 

func (x *DebugOptions) GetXlaCpuEnableFastMath() bool

func (*DebugOptions) GetXlaCpuEnableFastMinMax 

func (x *DebugOptions) GetXlaCpuEnableFastMinMax() bool

func (*DebugOptions) GetXlaCpuEnableMlirFusionOutlining 

func (x *DebugOptions) GetXlaCpuEnableMlirFusionOutlining() bool

func (*DebugOptions) GetXlaCpuEnableMlirTilingAndFusion 

func (x *DebugOptions) GetXlaCpuEnableMlirTilingAndFusion() bool

func (*DebugOptions) GetXlaCpuEnableXprofTraceme 

func (x *DebugOptions) GetXlaCpuEnableXprofTraceme() bool

func (*DebugOptions) GetXlaCpuFastMathHonorDivision 

func (x *DebugOptions) GetXlaCpuFastMathHonorDivision() bool

func (*DebugOptions) GetXlaCpuFastMathHonorFunctions 

func (x *DebugOptions) GetXlaCpuFastMathHonorFunctions() bool

func (*DebugOptions) GetXlaCpuFastMathHonorInfs 

func (x *DebugOptions) GetXlaCpuFastMathHonorInfs() bool

func (*DebugOptions) GetXlaCpuFastMathHonorNans 

func (x *DebugOptions) GetXlaCpuFastMathHonorNans() bool

func (*DebugOptions) GetXlaCpuMatmulTilingKDim 

func (x *DebugOptions) GetXlaCpuMatmulTilingKDim() int64

func (*DebugOptions) GetXlaCpuMatmulTilingMDim 

func (x *DebugOptions) GetXlaCpuMatmulTilingMDim() int64

func (*DebugOptions) GetXlaCpuMatmulTilingNDim 

func (x *DebugOptions) GetXlaCpuMatmulTilingNDim() int64

func (*DebugOptions) GetXlaCpuMultiThreadEigen 

func (x *DebugOptions) GetXlaCpuMultiThreadEigen() bool

func (*DebugOptions) GetXlaCpuPreferVectorWidth 

func (x *DebugOptions) GetXlaCpuPreferVectorWidth() int32

func (*DebugOptions) GetXlaCpuStrictDotConvMath 

func (x *DebugOptions) GetXlaCpuStrictDotConvMath() bool

func (*DebugOptions) GetXlaCpuUseAcl 

func (x *DebugOptions) GetXlaCpuUseAcl() bool

func (*DebugOptions) GetXlaCpuUseMklDnn 

func (x *DebugOptions) GetXlaCpuUseMklDnn() bool

func (*DebugOptions) GetXlaCpuUseThunkRuntime 

func (x *DebugOptions) GetXlaCpuUseThunkRuntime() bool

func (*DebugOptions) GetXlaDebugBufferAssignmentShowMax 

func (x *DebugOptions) GetXlaDebugBufferAssignmentShowMax() int64

func (*DebugOptions) GetXlaDetailedLogging 

func (x *DebugOptions) GetXlaDetailedLogging() bool

func (*DebugOptions) GetXlaDisableAllHloPasses 

func (x *DebugOptions) GetXlaDisableAllHloPasses() bool

func (*DebugOptions) GetXlaDisableHloPasses 

func (x *DebugOptions) GetXlaDisableHloPasses() []string

func (*DebugOptions) GetXlaDumpCompressProtos 

func (x *DebugOptions) GetXlaDumpCompressProtos() bool

func (*DebugOptions) GetXlaDumpDisableMetadata 

func (x *DebugOptions) GetXlaDumpDisableMetadata() bool

func (*DebugOptions) GetXlaDumpEnableMlirPrettyForm 

func (x *DebugOptions) GetXlaDumpEnableMlirPrettyForm() bool

func (*DebugOptions) GetXlaDumpFusionVisualization 

func (x *DebugOptions) GetXlaDumpFusionVisualization() bool

func (*DebugOptions) GetXlaDumpHloAsDot 

func (x *DebugOptions) GetXlaDumpHloAsDot() bool

func (*DebugOptions) GetXlaDumpHloAsHtml 

func (x *DebugOptions) GetXlaDumpHloAsHtml() bool

func (*DebugOptions) GetXlaDumpHloAsLongText 

func (x *DebugOptions) GetXlaDumpHloAsLongText() bool

func (*DebugOptions) GetXlaDumpHloAsProto 

func (x *DebugOptions) GetXlaDumpHloAsProto() bool

func (*DebugOptions) GetXlaDumpHloAsText 

func (x *DebugOptions) GetXlaDumpHloAsText() bool

func (*DebugOptions) GetXlaDumpHloAsUrl 

func (x *DebugOptions) GetXlaDumpHloAsUrl() bool

func (*DebugOptions) GetXlaDumpHloModuleRe 

func (x *DebugOptions) GetXlaDumpHloModuleRe() string

func (*DebugOptions) GetXlaDumpHloPassRe 

func (x *DebugOptions) GetXlaDumpHloPassRe() string

func (*DebugOptions) GetXlaDumpHloPipelineRe 

func (x *DebugOptions) GetXlaDumpHloPipelineRe() string

func (*DebugOptions) GetXlaDumpHloSnapshots 

func (x *DebugOptions) GetXlaDumpHloSnapshots() bool

func (*DebugOptions) GetXlaDumpIncludeTimestamp 

func (x *DebugOptions) GetXlaDumpIncludeTimestamp() bool

func (*DebugOptions) GetXlaDumpLargeConstants 

func (x *DebugOptions) GetXlaDumpLargeConstants() bool

func (*DebugOptions) GetXlaDumpLatencyHidingSchedule 

func (x *DebugOptions) GetXlaDumpLatencyHidingSchedule() bool

func (*DebugOptions) GetXlaDumpMaxHloModules 

func (x *DebugOptions) GetXlaDumpMaxHloModules() int32

func (*DebugOptions) GetXlaDumpModuleMetadata 

func (x *DebugOptions) GetXlaDumpModuleMetadata() bool

func (*DebugOptions) GetXlaDumpTo 

func (x *DebugOptions) GetXlaDumpTo() string

func (*DebugOptions) GetXlaEliminateHloImplicitBroadcast 

func (x *DebugOptions) GetXlaEliminateHloImplicitBroadcast() bool

func (*DebugOptions) GetXlaEmbedIrInExecutable 

func (x *DebugOptions) GetXlaEmbedIrInExecutable() bool

func (*DebugOptions) GetXlaEnableDumping 

func (x *DebugOptions) GetXlaEnableDumping() bool

func (*DebugOptions) GetXlaEnableHloPassesOnly 

func (x *DebugOptions) GetXlaEnableHloPassesOnly() []string

func (*DebugOptions) GetXlaForceHostPlatformDeviceCount 

func (x *DebugOptions) GetXlaForceHostPlatformDeviceCount() int32

func (*DebugOptions) GetXlaGpuAlgorithmDenylistPath 

func (x *DebugOptions) GetXlaGpuAlgorithmDenylistPath() string

func (*DebugOptions) GetXlaGpuAllGatherCombineThresholdBytes 

func (x *DebugOptions) GetXlaGpuAllGatherCombineThresholdBytes() int64

func (*DebugOptions) GetXlaGpuAllReduceBlueconnectNumDevicesPerHost 

func (x *DebugOptions) GetXlaGpuAllReduceBlueconnectNumDevicesPerHost() int32

func (*DebugOptions) GetXlaGpuAllReduceCombineThresholdBytes 

func (x *DebugOptions) GetXlaGpuAllReduceCombineThresholdBytes() int64

func (*DebugOptions) GetXlaGpuAsmExtraFlags 

func (x *DebugOptions) GetXlaGpuAsmExtraFlags() string

func (*DebugOptions) GetXlaGpuAutoSpmdPartitioningMemoryBudgetGb 

func (x *DebugOptions) GetXlaGpuAutoSpmdPartitioningMemoryBudgetGb() int32

func (*DebugOptions) GetXlaGpuAutoSpmdPartitioningMemoryBudgetRatio 

func (x *DebugOptions) GetXlaGpuAutoSpmdPartitioningMemoryBudgetRatio() float32

func (*DebugOptions) GetXlaGpuAutotuneLevel 

func (x *DebugOptions) GetXlaGpuAutotuneLevel() int32

func (*DebugOptions) GetXlaGpuAutotuneMaxSolutions 

func (x *DebugOptions) GetXlaGpuAutotuneMaxSolutions() int64

func (*DebugOptions) GetXlaGpuCollectCostModelStats 

func (x *DebugOptions) GetXlaGpuCollectCostModelStats() bool

func (*DebugOptions) GetXlaGpuCollectiveInflationFactor 

func (x *DebugOptions) GetXlaGpuCollectiveInflationFactor() int32

func (*DebugOptions) GetXlaGpuCollectivePermuteDecomposerThreshold 

func (x *DebugOptions) GetXlaGpuCollectivePermuteDecomposerThreshold() int64

func (*DebugOptions) GetXlaGpuCopyInsertionUseRegionAnalysis 

func (x *DebugOptions) GetXlaGpuCopyInsertionUseRegionAnalysis() bool

func (*DebugOptions) GetXlaGpuCrashOnVerificationFailures 

func (x *DebugOptions) GetXlaGpuCrashOnVerificationFailures() bool

func (*DebugOptions) GetXlaGpuCublasFallback 

func (x *DebugOptions) GetXlaGpuCublasFallback() bool

func (*DebugOptions) GetXlaGpuCudaDataDir 

func (x *DebugOptions) GetXlaGpuCudaDataDir() string

func (*DebugOptions) GetXlaGpuCudnnGemmFusionLevel 

func (x *DebugOptions) GetXlaGpuCudnnGemmFusionLevel() int32

func (*DebugOptions) GetXlaGpuDeterministicOps 

func (x *DebugOptions) GetXlaGpuDeterministicOps() bool

func (*DebugOptions) GetXlaGpuDisableAsyncCollectives 

func (x *DebugOptions) GetXlaGpuDisableAsyncCollectives() []DebugOptions_CollectiveOpType

func (*DebugOptions) GetXlaGpuDisableGpuasmOptimizations 

func (x *DebugOptions) GetXlaGpuDisableGpuasmOptimizations() bool

func (*DebugOptions) GetXlaGpuDumpAutotuneLogsTo 

func (x *DebugOptions) GetXlaGpuDumpAutotuneLogsTo() string

func (*DebugOptions) GetXlaGpuDumpAutotuneResultsTo 

func (x *DebugOptions) GetXlaGpuDumpAutotuneResultsTo() string

func (*DebugOptions) GetXlaGpuDumpAutotunedGemmFusions 

func (x *DebugOptions) GetXlaGpuDumpAutotunedGemmFusions() bool

func (*DebugOptions) GetXlaGpuDumpLlvmir 

func (x *DebugOptions) GetXlaGpuDumpLlvmir() bool

func (*DebugOptions) GetXlaGpuEnableAddressComputationFusion 

func (x *DebugOptions) GetXlaGpuEnableAddressComputationFusion() bool

func (*DebugOptions) GetXlaGpuEnableAllGatherCombineByDim 

func (x *DebugOptions) GetXlaGpuEnableAllGatherCombineByDim() bool

func (*DebugOptions) GetXlaGpuEnableAllReduceSplitter 

func (x *DebugOptions) GetXlaGpuEnableAllReduceSplitter() bool

func (*DebugOptions) GetXlaGpuEnableAnalyticalLatencyEstimator 

func (x *DebugOptions) GetXlaGpuEnableAnalyticalLatencyEstimator() bool

func (*DebugOptions) GetXlaGpuEnableApproxCostlyCollectives 

func (x *DebugOptions) GetXlaGpuEnableApproxCostlyCollectives() bool

func (*DebugOptions) GetXlaGpuEnableBf16_3WayGemm 

func (x *DebugOptions) GetXlaGpuEnableBf16_3WayGemm() bool

func (*DebugOptions) GetXlaGpuEnableBf16_6WayGemm 

func (x *DebugOptions) GetXlaGpuEnableBf16_6WayGemm() bool

func (*DebugOptions) GetXlaGpuEnableCommandBuffer 

func (x *DebugOptions) GetXlaGpuEnableCommandBuffer() []DebugOptions_CommandBufferCmdType

func (*DebugOptions) GetXlaGpuEnableCubRadixSort 

func (x *DebugOptions) GetXlaGpuEnableCubRadixSort() bool

func (*DebugOptions) GetXlaGpuEnableCublaslt 

func (x *DebugOptions) GetXlaGpuEnableCublaslt() bool

func (*DebugOptions) GetXlaGpuEnableCudnnFmha 

func (x *DebugOptions) GetXlaGpuEnableCudnnFmha() bool

func (*DebugOptions) GetXlaGpuEnableCudnnFrontend 

func (x *DebugOptions) GetXlaGpuEnableCudnnFrontend() bool

func (*DebugOptions) GetXlaGpuEnableCudnnInt8X32ConvolutionReordering 

func (x *DebugOptions) GetXlaGpuEnableCudnnInt8X32ConvolutionReordering() bool

func (*DebugOptions) GetXlaGpuEnableCudnnLayerNorm 

func (x *DebugOptions) GetXlaGpuEnableCudnnLayerNorm() bool

func (*DebugOptions) GetXlaGpuEnableCustomFusions 

func (x *DebugOptions) GetXlaGpuEnableCustomFusions() bool

func (*DebugOptions) GetXlaGpuEnableCustomFusionsRe 

func (x *DebugOptions) GetXlaGpuEnableCustomFusionsRe() string

func (*DebugOptions) GetXlaGpuEnableDotStrengthReduction 

func (x *DebugOptions) GetXlaGpuEnableDotStrengthReduction() bool

func (*DebugOptions) GetXlaGpuEnableFastMinMax 

func (x *DebugOptions) GetXlaGpuEnableFastMinMax() bool

func (*DebugOptions) GetXlaGpuEnableHighestPriorityAsyncStream 

func (x *DebugOptions) GetXlaGpuEnableHighestPriorityAsyncStream() bool

func (*DebugOptions) GetXlaGpuEnableHostMemoryOffloading 

func (x *DebugOptions) GetXlaGpuEnableHostMemoryOffloading() bool

func (*DebugOptions) GetXlaGpuEnableLatencyHidingScheduler 

func (x *DebugOptions) GetXlaGpuEnableLatencyHidingScheduler() bool

func (*DebugOptions) GetXlaGpuEnableLibnvptxcompiler 

func (x *DebugOptions) GetXlaGpuEnableLibnvptxcompiler() bool

func (*DebugOptions) GetXlaGpuEnableLlvmModuleCompilationParallelism 

func (x *DebugOptions) GetXlaGpuEnableLlvmModuleCompilationParallelism() bool

func (*DebugOptions) GetXlaGpuEnableNcclCliqueOptimization 

func (x *DebugOptions) GetXlaGpuEnableNcclCliqueOptimization() bool

func (*DebugOptions) GetXlaGpuEnableNcclCommSplitting 

func (x *DebugOptions) GetXlaGpuEnableNcclCommSplitting() bool

func (*DebugOptions) GetXlaGpuEnableNcclPerStreamComms 

func (x *DebugOptions) GetXlaGpuEnableNcclPerStreamComms() bool

func (*DebugOptions) GetXlaGpuEnableNcclUserBuffers 

func (x *DebugOptions) GetXlaGpuEnableNcclUserBuffers() bool

func (*DebugOptions) GetXlaGpuEnablePipelinedAllGather 

func (x *DebugOptions) GetXlaGpuEnablePipelinedAllGather() bool

func (*DebugOptions) GetXlaGpuEnablePipelinedAllReduce 

func (x *DebugOptions) GetXlaGpuEnablePipelinedAllReduce() bool

func (*DebugOptions) GetXlaGpuEnablePipelinedCollectives 

func (x *DebugOptions) GetXlaGpuEnablePipelinedCollectives() bool

func (*DebugOptions) GetXlaGpuEnablePipelinedP2P 

func (x *DebugOptions) GetXlaGpuEnablePipelinedP2P() bool

func (*DebugOptions) GetXlaGpuEnablePipelinedReduceScatter 

func (x *DebugOptions) GetXlaGpuEnablePipelinedReduceScatter() bool

func (*DebugOptions) GetXlaGpuEnablePriorityFusion 

func (x *DebugOptions) GetXlaGpuEnablePriorityFusion() bool

func (*DebugOptions) GetXlaGpuEnableReassociationForConvertedAr 

func (x *DebugOptions) GetXlaGpuEnableReassociationForConvertedAr() bool

func (*DebugOptions) GetXlaGpuEnableReduceScatterCombineByDim 

func (x *DebugOptions) GetXlaGpuEnableReduceScatterCombineByDim() bool

func (*DebugOptions) GetXlaGpuEnableReductionEpilogueFusion 

func (x *DebugOptions) GetXlaGpuEnableReductionEpilogueFusion() bool

func (*DebugOptions) GetXlaGpuEnableSharedConstants 

func (x *DebugOptions) GetXlaGpuEnableSharedConstants() bool

func (*DebugOptions) GetXlaGpuEnableSplitKAutotuning 

func (x *DebugOptions) GetXlaGpuEnableSplitKAutotuning() bool

func (*DebugOptions) GetXlaGpuEnableTritonGemm 

func (x *DebugOptions) GetXlaGpuEnableTritonGemm() bool

func (*DebugOptions) GetXlaGpuEnableTritonHopper 

func (x *DebugOptions) GetXlaGpuEnableTritonHopper() bool

func (*DebugOptions) GetXlaGpuEnableTritonSoftmaxFusion 

func (x *DebugOptions) GetXlaGpuEnableTritonSoftmaxFusion() bool

func (*DebugOptions) GetXlaGpuEnableTritonSoftmaxPriorityFusion 

func (x *DebugOptions) GetXlaGpuEnableTritonSoftmaxPriorityFusion() bool

func (*DebugOptions) GetXlaGpuEnableWhileLoopDoubleBuffering 

func (x *DebugOptions) GetXlaGpuEnableWhileLoopDoubleBuffering() bool

func (*DebugOptions) GetXlaGpuEnableWhileLoopReduceScatterCodeMotion 

func (x *DebugOptions) GetXlaGpuEnableWhileLoopReduceScatterCodeMotion() bool

func (*DebugOptions) GetXlaGpuEnableWhileLoopUnrolling 

func (x *DebugOptions) GetXlaGpuEnableWhileLoopUnrolling() DebugOptions_WhileLoopUnrolling

func (*DebugOptions) GetXlaGpuEnsureMinorDotContractionDims 

func (x *DebugOptions) GetXlaGpuEnsureMinorDotContractionDims() bool

func (*DebugOptions) GetXlaGpuExcludeNondeterministicOps 

func (x *DebugOptions) GetXlaGpuExcludeNondeterministicOps() bool

func (*DebugOptions) GetXlaGpuExhaustiveTilingSearch 

func (x *DebugOptions) GetXlaGpuExhaustiveTilingSearch() bool

func (*DebugOptions) GetXlaGpuFilterKernelsSpillingRegistersOnAutotuning 

func (x *DebugOptions) GetXlaGpuFilterKernelsSpillingRegistersOnAutotuning() bool

func (*DebugOptions) GetXlaGpuForceCompilationParallelism 

func (x *DebugOptions) GetXlaGpuForceCompilationParallelism() int32

func (*DebugOptions) GetXlaGpuForceConvNchw 

func (x *DebugOptions) GetXlaGpuForceConvNchw() bool

func (*DebugOptions) GetXlaGpuForceConvNhwc 

func (x *DebugOptions) GetXlaGpuForceConvNhwc() bool

func (*DebugOptions) GetXlaGpuFtz 

func (x *DebugOptions) GetXlaGpuFtz() bool

func (*DebugOptions) GetXlaGpuFusedAttentionUseCudnnRng 

func (x *DebugOptions) GetXlaGpuFusedAttentionUseCudnnRng() bool

func (*DebugOptions) GetXlaGpuGemmRewriteSizeThreshold 

func (x *DebugOptions) GetXlaGpuGemmRewriteSizeThreshold() int64

func (*DebugOptions) GetXlaGpuGraphEnableConcurrentRegion 

func (x *DebugOptions) GetXlaGpuGraphEnableConcurrentRegion() bool

func (*DebugOptions) GetXlaGpuGraphMinGraphSize 

func (x *DebugOptions) GetXlaGpuGraphMinGraphSize() int32

func (*DebugOptions) GetXlaGpuKernelCacheFile 

func (x *DebugOptions) GetXlaGpuKernelCacheFile() string

func (*DebugOptions) GetXlaGpuLhsEnableGpuAsyncTracker 

func (x *DebugOptions) GetXlaGpuLhsEnableGpuAsyncTracker() bool

func (*DebugOptions) GetXlaGpuLlvmIrFile 

func (x *DebugOptions) GetXlaGpuLlvmIrFile() []string

func (*DebugOptions) GetXlaGpuLlvmVerificationLevel 

func (x *DebugOptions) GetXlaGpuLlvmVerificationLevel() int32

func (*DebugOptions) GetXlaGpuLoadAutotuneResultsFrom 

func (x *DebugOptions) GetXlaGpuLoadAutotuneResultsFrom() string

func (*DebugOptions) GetXlaGpuMaxMlirKernels 

func (x *DebugOptions) GetXlaGpuMaxMlirKernels() int64

func (*DebugOptions) GetXlaGpuMemoryLimitSlopFactor 

func (x *DebugOptions) GetXlaGpuMemoryLimitSlopFactor() int32

func (*DebugOptions) GetXlaGpuMlirEmitterLevel 

func (x *DebugOptions) GetXlaGpuMlirEmitterLevel() int64

func (*DebugOptions) GetXlaGpuMockCustomCalls 

func (x *DebugOptions) GetXlaGpuMockCustomCalls() bool

func (*DebugOptions) GetXlaGpuMultiStreamedWindowedEinsum 

func (x *DebugOptions) GetXlaGpuMultiStreamedWindowedEinsum() bool

func (*DebugOptions) GetXlaGpuNcclCollectiveMaxNchannels 

func (x *DebugOptions) GetXlaGpuNcclCollectiveMaxNchannels() int64

func (*DebugOptions) GetXlaGpuNcclP2PMaxNchannels 

func (x *DebugOptions) GetXlaGpuNcclP2PMaxNchannels() int64

func (*DebugOptions) GetXlaGpuNcclTerminateOnError 

func (x *DebugOptions) GetXlaGpuNcclTerminateOnError() bool

func (*DebugOptions) GetXlaGpuNcclTerminationTimeoutSeconds 

func (x *DebugOptions) GetXlaGpuNcclTerminationTimeoutSeconds() int64

func (*DebugOptions) GetXlaGpuOverrideGemmAutotuner 

func (x *DebugOptions) GetXlaGpuOverrideGemmAutotuner() string

func (*DebugOptions) GetXlaGpuPgleProfileFileOrDirectoryPath 

func (x *DebugOptions) GetXlaGpuPgleProfileFileOrDirectoryPath() string

func (*DebugOptions) GetXlaGpuPtxFile 

func (x *DebugOptions) GetXlaGpuPtxFile() []string

func (*DebugOptions) GetXlaGpuReduceScatterCombineThresholdBytes 

func (x *DebugOptions) GetXlaGpuReduceScatterCombineThresholdBytes() int64

func (*DebugOptions) GetXlaGpuRedzonePaddingBytes 

func (x *DebugOptions) GetXlaGpuRedzonePaddingBytes() int64

func (*DebugOptions) GetXlaGpuRedzoneScratchMaxMegabytes 

func (x *DebugOptions) GetXlaGpuRedzoneScratchMaxMegabytes() int64

func (*DebugOptions) GetXlaGpuRequireCompleteAotAutotuneResults 

func (x *DebugOptions) GetXlaGpuRequireCompleteAotAutotuneResults() bool

func (*DebugOptions) GetXlaGpuShapeChecks 

func (x *DebugOptions) GetXlaGpuShapeChecks() DebugOptions_ShapeChecks

func (*DebugOptions) GetXlaGpuShardAutotuning 

func (x *DebugOptions) GetXlaGpuShardAutotuning() bool

func (*DebugOptions) GetXlaGpuSkipMlirKernels 

func (x *DebugOptions) GetXlaGpuSkipMlirKernels() int64

func (*DebugOptions) GetXlaGpuStrictConvAlgorithmPicker 

func (x *DebugOptions) GetXlaGpuStrictConvAlgorithmPicker() bool

func (*DebugOptions) GetXlaGpuTargetConfigFilename 

func (x *DebugOptions) GetXlaGpuTargetConfigFilename() string

func (*DebugOptions) GetXlaGpuThresholdForWindowedEinsumMib 

func (x *DebugOptions) GetXlaGpuThresholdForWindowedEinsumMib() int64

func (*DebugOptions) GetXlaGpuTritonFusionLevel 

func (x *DebugOptions) GetXlaGpuTritonFusionLevel() int32

func (*DebugOptions) GetXlaGpuTritonGemmAny 

func (x *DebugOptions) GetXlaGpuTritonGemmAny() bool

func (*DebugOptions) GetXlaGpuTritonGemmDisableReducedPrecisionReduction 

func (x *DebugOptions) GetXlaGpuTritonGemmDisableReducedPrecisionReduction() bool

func (*DebugOptions) GetXlaGpuUnsafeFallbackToDriverOnPtxasNotFound 

func (x *DebugOptions) GetXlaGpuUnsafeFallbackToDriverOnPtxasNotFound() bool

func (*DebugOptions) GetXlaGpuUseMemcpyLocalP2P 

func (x *DebugOptions) GetXlaGpuUseMemcpyLocalP2P() bool

func (*DebugOptions) GetXlaGpuUseRuntimeFusion 

func (x *DebugOptions) GetXlaGpuUseRuntimeFusion() bool

func (*DebugOptions) GetXlaGpuVerifyTritonFusionNumerics 

func (x *DebugOptions) GetXlaGpuVerifyTritonFusionNumerics() bool

func (*DebugOptions) GetXlaHloEvaluatorUseFastPath 

func (x *DebugOptions) GetXlaHloEvaluatorUseFastPath() bool

func (*DebugOptions) GetXlaHloGraphAddresses 

func (x *DebugOptions) GetXlaHloGraphAddresses() bool

func (*DebugOptions) GetXlaHloGraphShardingColor 

func (x *DebugOptions) GetXlaHloGraphShardingColor() bool

func (*DebugOptions) GetXlaHloProfile 

func (x *DebugOptions) GetXlaHloProfile() bool

func (*DebugOptions) GetXlaLlvmDisableExpensivePasses 

func (x *DebugOptions) GetXlaLlvmDisableExpensivePasses() bool

func (*DebugOptions) GetXlaLlvmEnableAliasScopeMetadata 

func (x *DebugOptions) GetXlaLlvmEnableAliasScopeMetadata() bool

func (*DebugOptions) GetXlaLlvmEnableInvariantLoadMetadata 

func (x *DebugOptions) GetXlaLlvmEnableInvariantLoadMetadata() bool

func (*DebugOptions) GetXlaLlvmEnableNoaliasMetadata 

func (x *DebugOptions) GetXlaLlvmEnableNoaliasMetadata() bool

func (*DebugOptions) GetXlaLlvmForceInlineBeforeSplit 

func (x *DebugOptions) GetXlaLlvmForceInlineBeforeSplit() bool

func (*DebugOptions) GetXlaMultiheapSizeConstraintPerHeap 

func (x *DebugOptions) GetXlaMultiheapSizeConstraintPerHeap() int32

func (*DebugOptions) GetXlaPartitioningAlgorithm 

func (x *DebugOptions) GetXlaPartitioningAlgorithm() DebugOptions_PartitioningAlgorithm

func (*DebugOptions) GetXlaReduceWindowRewriteBaseLength 

func (x *DebugOptions) GetXlaReduceWindowRewriteBaseLength() int64

func (*DebugOptions) GetXlaStepMarkerLocation 

func (x *DebugOptions) GetXlaStepMarkerLocation() DebugOptions_StepMarkerLocation

func (*DebugOptions) GetXlaTestAllInputLayouts 

func (x *DebugOptions) GetXlaTestAllInputLayouts() bool

func (*DebugOptions) GetXlaTestAllOutputLayouts 

func (x *DebugOptions) GetXlaTestAllOutputLayouts() bool

func (*DebugOptions) GetXlaTpuDetectInf 

func (x *DebugOptions) GetXlaTpuDetectInf() bool

func (*DebugOptions) GetXlaTpuDetectNan 

func (x *DebugOptions) GetXlaTpuDetectNan() bool

func (*DebugOptions) GetXlaUseShardonnay 

func (x *DebugOptions) GetXlaUseShardonnay() bool

func (*DebugOptions) ProtoMessage 

func (*DebugOptions) ProtoMessage()

func (*DebugOptions) ProtoReflect 

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

func (*DebugOptions) Reset 

func (x *DebugOptions) Reset()

func (*DebugOptions) String 

func (x *DebugOptions) String() string

type DebugOptions_CollectiveOpType 

type DebugOptions_CollectiveOpType int32

Enum to define all collective ops that xla supports. 

const (
	DebugOptions_NOOP                DebugOptions_CollectiveOpType = 0
	DebugOptions_ALLREDUCE           DebugOptions_CollectiveOpType = 1
	DebugOptions_ALLGATHER           DebugOptions_CollectiveOpType = 2
	DebugOptions_REDUCESCATTER       DebugOptions_CollectiveOpType = 3
	DebugOptions_COLLECTIVEBROADCAST DebugOptions_CollectiveOpType = 4
	DebugOptions_ALLTOALL            DebugOptions_CollectiveOpType = 5
	DebugOptions_COLLECTIVEPERMUTE   DebugOptions_CollectiveOpType = 6
)

func (DebugOptions_CollectiveOpType) Descriptor 

func (DebugOptions_CollectiveOpType) Descriptor() protoreflect.EnumDescriptor

func (DebugOptions_CollectiveOpType) Enum 

func (x DebugOptions_CollectiveOpType) Enum() *DebugOptions_CollectiveOpType

func (DebugOptions_CollectiveOpType) EnumDescriptor deprecated 

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

Deprecated: Use DebugOptions_CollectiveOpType.Descriptor instead.

func (DebugOptions_CollectiveOpType) Number 

func (x DebugOptions_CollectiveOpType) Number() protoreflect.EnumNumber

func (DebugOptions_CollectiveOpType) String 

func (x DebugOptions_CollectiveOpType) String() string

func (DebugOptions_CollectiveOpType) Type 

func (DebugOptions_CollectiveOpType) Type() protoreflect.EnumType

type DebugOptions_CommandBufferCmdType 

type DebugOptions_CommandBufferCmdType int32

Commands are categorized into 5 types: FUSION represents regular fusion kernels. CUBLAS/CUBLASLT, CUDNN, and COLLECTIVES represent library calls. CONDITIONALS represents control flow. 

const (
	DebugOptions_INVALID      DebugOptions_CommandBufferCmdType = 0
	DebugOptions_FUSION       DebugOptions_CommandBufferCmdType = 1
	DebugOptions_CUBLAS       DebugOptions_CommandBufferCmdType = 2
	DebugOptions_CUDNN        DebugOptions_CommandBufferCmdType = 3
	DebugOptions_COLLECTIVES  DebugOptions_CommandBufferCmdType = 4
	DebugOptions_CONDITIONALS DebugOptions_CommandBufferCmdType = 5
	DebugOptions_CUSTOM_CALL  DebugOptions_CommandBufferCmdType = 6
	DebugOptions_CUBLASLT     DebugOptions_CommandBufferCmdType = 7
)

func (DebugOptions_CommandBufferCmdType) Descriptor 

func (DebugOptions_CommandBufferCmdType) Descriptor() protoreflect.EnumDescriptor

func (DebugOptions_CommandBufferCmdType) Enum 

func (x DebugOptions_CommandBufferCmdType) Enum() *DebugOptions_CommandBufferCmdType

func (DebugOptions_CommandBufferCmdType) EnumDescriptor deprecated 

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

Deprecated: Use DebugOptions_CommandBufferCmdType.Descriptor instead.

func (DebugOptions_CommandBufferCmdType) Number 

func (x DebugOptions_CommandBufferCmdType) Number() protoreflect.EnumNumber

func (DebugOptions_CommandBufferCmdType) String 

func (x DebugOptions_CommandBufferCmdType) String() string

func (DebugOptions_CommandBufferCmdType) Type 

func (DebugOptions_CommandBufferCmdType) Type() protoreflect.EnumType

type DebugOptions_PartitioningAlgorithm 

type DebugOptions_PartitioningAlgorithm int32
const (
	DebugOptions_PARTITIONING_ALGORITHM_NOOP DebugOptions_PartitioningAlgorithm = 0
	DebugOptions_PARTITIONING_ALGORITHM_EXP0 DebugOptions_PartitioningAlgorithm = 1
	DebugOptions_PARTITIONING_ALGORITHM_EXP1 DebugOptions_PartitioningAlgorithm = 2
	DebugOptions_PARTITIONING_ALGORITHM_EXP2 DebugOptions_PartitioningAlgorithm = 3
)

func (DebugOptions_PartitioningAlgorithm) Descriptor 

func (DebugOptions_PartitioningAlgorithm) Descriptor() protoreflect.EnumDescriptor

func (DebugOptions_PartitioningAlgorithm) Enum 

func (x DebugOptions_PartitioningAlgorithm) Enum() *DebugOptions_PartitioningAlgorithm

func (DebugOptions_PartitioningAlgorithm) EnumDescriptor deprecated 

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

Deprecated: Use DebugOptions_PartitioningAlgorithm.Descriptor instead.

func (DebugOptions_PartitioningAlgorithm) Number 

func (x DebugOptions_PartitioningAlgorithm) Number() protoreflect.EnumNumber

func (DebugOptions_PartitioningAlgorithm) String 

func (x DebugOptions_PartitioningAlgorithm) String() string

func (DebugOptions_PartitioningAlgorithm) Type 

func (DebugOptions_PartitioningAlgorithm) Type() protoreflect.EnumType

type DebugOptions_ShapeChecks 

type DebugOptions_ShapeChecks int32
const (
	// Do not insert any shape checks for dynamically shaped operations; output
	// buffers might contain garbage data if shapes don't match.
	DebugOptions_IGNORE DebugOptions_ShapeChecks = 0
	// Check shapes at runtime, will insert an extra synchronization if shapes
	// cannot be proven correct at compile time.
	DebugOptions_RUNTIME DebugOptions_ShapeChecks = 1
	// Will refuse to compile any program where shape correctness can not be
	// established at compile time.
	DebugOptions_COMPILE_TIME DebugOptions_ShapeChecks = 2
)

func (DebugOptions_ShapeChecks) Descriptor 

func (DebugOptions_ShapeChecks) Descriptor() protoreflect.EnumDescriptor

func (DebugOptions_ShapeChecks) Enum 

func (x DebugOptions_ShapeChecks) Enum() *DebugOptions_ShapeChecks

func (DebugOptions_ShapeChecks) EnumDescriptor deprecated 

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

Deprecated: Use DebugOptions_ShapeChecks.Descriptor instead.

func (DebugOptions_ShapeChecks) Number 

func (x DebugOptions_ShapeChecks) Number() protoreflect.EnumNumber

func (DebugOptions_ShapeChecks) String 

func (x DebugOptions_ShapeChecks) String() string

func (DebugOptions_ShapeChecks) Type 

func (DebugOptions_ShapeChecks) Type() protoreflect.EnumType

type DebugOptions_StepMarkerLocation 

type DebugOptions_StepMarkerLocation int32
const (
	// Generate a step marker at the program entry. This handles the case where
	// each step is done by one or multiple program execution(s). Only the first
	// program will be tagged for generating a step marker at the program entry.
	// This is the default.
	DebugOptions_STEP_MARK_AT_ENTRY DebugOptions_StepMarkerLocation = 0
	// Generate a step marker at each iteration of the top level while loop,
	// which is assumed to be a training loop.
	DebugOptions_STEP_MARK_AT_TOP_LEVEL_WHILE_LOOP DebugOptions_StepMarkerLocation = 1
	// Generate a step marker at each iteration of the second level while loops,
	// which is assumed to be a training or eval loop.
	DebugOptions_STEP_MARK_AT_SECOND_LEVEL_WHILE_LOOP DebugOptions_StepMarkerLocation = 3
	// No step marker generated.
	DebugOptions_STEP_MARK_NONE DebugOptions_StepMarkerLocation = 2
)

func (DebugOptions_StepMarkerLocation) Descriptor 

func (DebugOptions_StepMarkerLocation) Descriptor() protoreflect.EnumDescriptor

func (DebugOptions_StepMarkerLocation) Enum 

func (x DebugOptions_StepMarkerLocation) Enum() *DebugOptions_StepMarkerLocation

func (DebugOptions_StepMarkerLocation) EnumDescriptor deprecated 

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

Deprecated: Use DebugOptions_StepMarkerLocation.Descriptor instead.

func (DebugOptions_StepMarkerLocation) Number 

func (x DebugOptions_StepMarkerLocation) Number() protoreflect.EnumNumber

func (DebugOptions_StepMarkerLocation) String 

func (x DebugOptions_StepMarkerLocation) String() string

func (DebugOptions_StepMarkerLocation) Type 

func (DebugOptions_StepMarkerLocation) Type() protoreflect.EnumType

type DebugOptions_WhileLoopUnrolling 

type DebugOptions_WhileLoopUnrolling int32
const (
	DebugOptions_WHILE_LOOP_UNROLLING_NO_UNROLL DebugOptions_WhileLoopUnrolling = 0
	// Has the same effect as setting
	// `xla_gpu_enable_while_loop_double_buffering`.
	DebugOptions_WHILE_LOOP_UNROLLING_DOUBLE_BUFFER DebugOptions_WhileLoopUnrolling = 1
	// Enables full loop unrolling using the same strategy as `DOUBLE_BUFFER`.
	DebugOptions_WHILE_LOOP_UNROLLING_FULL_UNROLL DebugOptions_WhileLoopUnrolling = 2
)

func (DebugOptions_WhileLoopUnrolling) Descriptor 

func (DebugOptions_WhileLoopUnrolling) Descriptor() protoreflect.EnumDescriptor

func (DebugOptions_WhileLoopUnrolling) Enum 

func (x DebugOptions_WhileLoopUnrolling) Enum() *DebugOptions_WhileLoopUnrolling

func (DebugOptions_WhileLoopUnrolling) EnumDescriptor deprecated 

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

Deprecated: Use DebugOptions_WhileLoopUnrolling.Descriptor instead.

func (DebugOptions_WhileLoopUnrolling) Number 

func (x DebugOptions_WhileLoopUnrolling) Number() protoreflect.EnumNumber

func (DebugOptions_WhileLoopUnrolling) String 

func (x DebugOptions_WhileLoopUnrolling) String() string

func (DebugOptions_WhileLoopUnrolling) Type 

func (DebugOptions_WhileLoopUnrolling) Type() protoreflect.EnumType

type DeconstructTupleRequest 

type DeconstructTupleRequest struct {
	TupleHandle *GlobalDataHandle `protobuf:"bytes,2,opt,name=tuple_handle,json=tupleHandle,proto3" json:"tuple_handle,omitempty"`
	// contains filtered or unexported fields
}

func (*DeconstructTupleRequest) Descriptor deprecated 

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

Deprecated: Use DeconstructTupleRequest.ProtoReflect.Descriptor instead.

func (*DeconstructTupleRequest) GetTupleHandle 

func (x *DeconstructTupleRequest) GetTupleHandle() *GlobalDataHandle

func (*DeconstructTupleRequest) ProtoMessage 

func (*DeconstructTupleRequest) ProtoMessage()

func (*DeconstructTupleRequest) ProtoReflect 

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

func (*DeconstructTupleRequest) Reset 

func (x *DeconstructTupleRequest) Reset()

func (*DeconstructTupleRequest) String 

func (x *DeconstructTupleRequest) String() string

type DeconstructTupleResponse 

type DeconstructTupleResponse struct {
	ElementHandles []*GlobalDataHandle `protobuf:"bytes,1,rep,name=element_handles,json=elementHandles,proto3" json:"element_handles,omitempty"`
	// contains filtered or unexported fields
}

func (*DeconstructTupleResponse) Descriptor deprecated 

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

Deprecated: Use DeconstructTupleResponse.ProtoReflect.Descriptor instead.

func (*DeconstructTupleResponse) GetElementHandles 

func (x *DeconstructTupleResponse) GetElementHandles() []*GlobalDataHandle

func (*DeconstructTupleResponse) ProtoMessage 

func (*DeconstructTupleResponse) ProtoMessage()

func (*DeconstructTupleResponse) ProtoReflect 

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

func (*DeconstructTupleResponse) Reset 

func (x *DeconstructTupleResponse) Reset()

func (*DeconstructTupleResponse) String 

func (x *DeconstructTupleResponse) 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 DnnVersionInfoProto 

type DnnVersionInfoProto struct {
	Major int32 `protobuf:"varint,1,opt,name=major,proto3" json:"major,omitempty"`
	Minor int32 `protobuf:"varint,2,opt,name=minor,proto3" json:"minor,omitempty"`
	Patch int32 `protobuf:"varint,3,opt,name=patch,proto3" json:"patch,omitempty"`
	// contains filtered or unexported fields
}

func (*DnnVersionInfoProto) Descriptor deprecated 

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

Deprecated: Use DnnVersionInfoProto.ProtoReflect.Descriptor instead.

func (*DnnVersionInfoProto) GetMajor 

func (x *DnnVersionInfoProto) GetMajor() int32

func (*DnnVersionInfoProto) GetMinor 

func (x *DnnVersionInfoProto) GetMinor() int32

func (*DnnVersionInfoProto) GetPatch 

func (x *DnnVersionInfoProto) GetPatch() int32

func (*DnnVersionInfoProto) ProtoMessage 

func (*DnnVersionInfoProto) ProtoMessage()

func (*DnnVersionInfoProto) ProtoReflect 

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

func (*DnnVersionInfoProto) Reset 

func (x *DnnVersionInfoProto) Reset()

func (*DnnVersionInfoProto) String 

func (x *DnnVersionInfoProto) String() string

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 ExecutableAndOptionsProto 

type ExecutableAndOptionsProto struct {
	SerializedExecutable []byte               `protobuf:"bytes,1,opt,name=serialized_executable,json=serializedExecutable,proto3" json:"serialized_executable,omitempty"`
	CompileOptions       *CompileOptionsProto `protobuf:"bytes,2,opt,name=compile_options,json=compileOptions,proto3" json:"compile_options,omitempty"`
	// contains filtered or unexported fields
}

Helper for serializing opaque executables alongside CompileOptions.

func (*ExecutableAndOptionsProto) Descriptor deprecated 

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

Deprecated: Use ExecutableAndOptionsProto.ProtoReflect.Descriptor instead.

func (*ExecutableAndOptionsProto) GetCompileOptions 

func (x *ExecutableAndOptionsProto) GetCompileOptions() *CompileOptionsProto

func (*ExecutableAndOptionsProto) GetSerializedExecutable 

func (x *ExecutableAndOptionsProto) GetSerializedExecutable() []byte

func (*ExecutableAndOptionsProto) ProtoMessage 

func (*ExecutableAndOptionsProto) ProtoMessage()

func (*ExecutableAndOptionsProto) ProtoReflect 

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

func (*ExecutableAndOptionsProto) Reset 

func (x *ExecutableAndOptionsProto) Reset()

func (*ExecutableAndOptionsProto) String 

func (x *ExecutableAndOptionsProto) String() string

type ExecutableBuildOptionsProto 

type ExecutableBuildOptionsProto struct {

	// If set, this is the device to build the computation for. Valid
	// device_ordinal values are: 0 to # of devices - 1. These values are
	// identical to the device ordinal values used by StreamExecutor. The built
	// executable will be executable on any device equivalent to the specified
	// device as determined by Backend::devices_equivalent(). A value of -1
	// indicates this option has not been set.
	DeviceOrdinal int64 `protobuf:"varint,1,opt,name=device_ordinal,json=deviceOrdinal,proto3" json:"device_ordinal,omitempty"`
	// If set, this specifies the layout of the result of the computation. If not
	// set, the service will chose the layout of the result. A Shape is used to
	// store the layout to accommodate tuple result shapes. A value of nullptr
	// indicates the option has not been set.
	ResultLayout *ShapeProto `protobuf:"bytes,2,opt,name=result_layout,json=resultLayout,proto3" json:"result_layout,omitempty"`
	// Expose access to the XLA compilation environments, which will be passed to
	// the compilation process.
	CompEnvs *CompilationEnvironmentsProto `protobuf:"bytes,13,opt,name=comp_envs,json=compEnvs,proto3" json:"comp_envs,omitempty"`
	// Expose access to the XLA debug options which will be passed to the
	// compilation process.
	DebugOptions *DebugOptions `protobuf:"bytes,3,opt,name=debug_options,json=debugOptions,proto3" json:"debug_options,omitempty"`
	// The number of replicas of this computation that are to be executed.
	// Defaults to 1.
	NumReplicas int64 `protobuf:"varint,4,opt,name=num_replicas,json=numReplicas,proto3" json:"num_replicas,omitempty"`
	// The number of partitions in this computation. Defaults to 1.
	NumPartitions int64 `protobuf:"varint,5,opt,name=num_partitions,json=numPartitions,proto3" json:"num_partitions,omitempty"`
	// Indicates whether to use SPMD (true) or MPMD (false) partitioning when
	// num_partitions > 1 and XLA is requested to partition the input program.
	UseSpmdPartitioning bool `protobuf:"varint,6,opt,name=use_spmd_partitioning,json=useSpmdPartitioning,proto3" json:"use_spmd_partitioning,omitempty"`
	// Whether to automatically generate XLA shardings for SPMD partitioner.
	UseAutoSpmdPartitioning bool `` /* 135-byte string literal not displayed */
	// Whether HLOs should be deduplicated.
	DeduplicateHlo bool `protobuf:"varint,8,opt,name=deduplicate_hlo,json=deduplicateHlo,proto3" json:"deduplicate_hlo,omitempty"`
	// If set, this specifies a static device assignment for the computation.
	// Otherwise, the computation will be compiled generically and can be run with
	// any device assignment compatible with the computation's replica and
	// partition counts.
	DeviceAssignment *DeviceAssignmentProto `protobuf:"bytes,9,opt,name=device_assignment,json=deviceAssignment,proto3" json:"device_assignment,omitempty"`
	// Whether input and output buffers are aliased if the associated parameter is
	// passed-through XLA modules without being changed.
	AliasPassthroughParams bool `` /* 131-byte string literal not displayed */
	// By default, XLA builds an executable by invoking standard compilation, i.e.
	// running Compiler::Compile, or both Compiler::RunHloPasses and
	// Compiler::RunBackend. When run_backend_only is set to true, XLA builds an
	// executable by invoking only RunBackend and skip invoking RunHloPasses,
	// which can be used to compile post-optimizations HLO modules.
	RunBackendOnly bool `protobuf:"varint,11,opt,name=run_backend_only,json=runBackendOnly,proto3" json:"run_backend_only,omitempty"`
	// Allows sharding propagation to propagate to the parameters. This changes
	// the input shape of the computation (which is undesirable), but it can be
	// used to allow to run partial compilation to determine what would be the
	// input sharding of a computation if XLA would be allowed to propagate the
	// sharding which can be used by higher level framework as a way to query
	// intermediate sharding of operations when multiple computation would be
	// chained and merged together.
	// This is a vector of bool, because the user can control which parameters can
	// have the sharding substituted. If only one boolean value is passed in the
	// vector that is interpreted as the value to be applied for every parameter.
	AllowSpmdShardingPropagationToParameters []bool `` /* 198-byte string literal not displayed */
	// Allows sharding propagation to propagate to the outputs. This changes the
	// output shape of the computation (which is undesirable), but it can be used
	// to allow to run partial compilation to determine what would be the output
	// sharding of a computation if XLA would be allowed to propagate the sharding
	// which can be used by higher level framework as a way to query intermediate
	// sharding of operations when multiple computation would be chained and
	// merged together.
	// This is a vector of bool, because the user can control (if the output of
	// the computation is a tuple) which elements of the tuple can have the
	// sharding substituted and which don't. If only one boolean value is passed
	// in the vector that's interpreted as the value to be applied for every
	// single element of the output tuple. One value per element of the tuple
	// means that each value is attached to one of the output elements.
	AllowSpmdShardingPropagationToOutput []bool `` /* 186-byte string literal not displayed */
	// Opaque profile data for any feedback directed optimizations.
	FdoProfile       []byte `protobuf:"bytes,14,opt,name=fdo_profile,json=fdoProfile,proto3" json:"fdo_profile,omitempty"`
	DeviceMemorySize int64  `protobuf:"varint,15,opt,name=device_memory_size,json=deviceMemorySize,proto3" json:"device_memory_size,omitempty"`
	// Mesh shape in auto sharding options.
	AutoSpmdPartitioningMeshShape []int64 `` /* 163-byte string literal not displayed */
	// Mesh ids in auto sharding options.
	AutoSpmdPartitioningMeshIds []int64 `` /* 157-byte string literal not displayed */
	// contains filtered or unexported fields
}

A serialization of xla::ExecutableBuildOptions. Next id: 19.

func (*ExecutableBuildOptionsProto) Descriptor deprecated 

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

Deprecated: Use ExecutableBuildOptionsProto.ProtoReflect.Descriptor instead.

func (*ExecutableBuildOptionsProto) GetAliasPassthroughParams 

func (x *ExecutableBuildOptionsProto) GetAliasPassthroughParams() bool

func (*ExecutableBuildOptionsProto) GetAllowSpmdShardingPropagationToOutput 

func (x *ExecutableBuildOptionsProto) GetAllowSpmdShardingPropagationToOutput() []bool

func (*ExecutableBuildOptionsProto) GetAllowSpmdShardingPropagationToParameters 

func (x *ExecutableBuildOptionsProto) GetAllowSpmdShardingPropagationToParameters() []bool

func (*ExecutableBuildOptionsProto) GetAutoSpmdPartitioningMeshIds 

func (x *ExecutableBuildOptionsProto) GetAutoSpmdPartitioningMeshIds() []int64

func (*ExecutableBuildOptionsProto) GetAutoSpmdPartitioningMeshShape 

func (x *ExecutableBuildOptionsProto) GetAutoSpmdPartitioningMeshShape() []int64

func (*ExecutableBuildOptionsProto) GetCompEnvs 

func (x *ExecutableBuildOptionsProto) GetCompEnvs() *CompilationEnvironmentsProto

func (*ExecutableBuildOptionsProto) GetDebugOptions 

func (x *ExecutableBuildOptionsProto) GetDebugOptions() *DebugOptions

func (*ExecutableBuildOptionsProto) GetDeduplicateHlo 

func (x *ExecutableBuildOptionsProto) GetDeduplicateHlo() bool

func (*ExecutableBuildOptionsProto) GetDeviceAssignment 

func (x *ExecutableBuildOptionsProto) GetDeviceAssignment() *DeviceAssignmentProto

func (*ExecutableBuildOptionsProto) GetDeviceMemorySize 

func (x *ExecutableBuildOptionsProto) GetDeviceMemorySize() int64

func (*ExecutableBuildOptionsProto) GetDeviceOrdinal 

func (x *ExecutableBuildOptionsProto) GetDeviceOrdinal() int64

func (*ExecutableBuildOptionsProto) GetFdoProfile 

func (x *ExecutableBuildOptionsProto) GetFdoProfile() []byte

func (*ExecutableBuildOptionsProto) GetNumPartitions 

func (x *ExecutableBuildOptionsProto) GetNumPartitions() int64

func (*ExecutableBuildOptionsProto) GetNumReplicas 

func (x *ExecutableBuildOptionsProto) GetNumReplicas() int64

func (*ExecutableBuildOptionsProto) GetResultLayout 

func (x *ExecutableBuildOptionsProto) GetResultLayout() *ShapeProto

func (*ExecutableBuildOptionsProto) GetRunBackendOnly 

func (x *ExecutableBuildOptionsProto) GetRunBackendOnly() bool

func (*ExecutableBuildOptionsProto) GetUseAutoSpmdPartitioning 

func (x *ExecutableBuildOptionsProto) GetUseAutoSpmdPartitioning() bool

func (*ExecutableBuildOptionsProto) GetUseSpmdPartitioning 

func (x *ExecutableBuildOptionsProto) GetUseSpmdPartitioning() bool

func (*ExecutableBuildOptionsProto) ProtoMessage 

func (*ExecutableBuildOptionsProto) ProtoMessage()

func (*ExecutableBuildOptionsProto) ProtoReflect 

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

func (*ExecutableBuildOptionsProto) Reset 

func (x *ExecutableBuildOptionsProto) Reset()

func (*ExecutableBuildOptionsProto) String 

func (x *ExecutableBuildOptionsProto) String() string

type ExecuteGraphParallelRequest 

type ExecuteGraphParallelRequest struct {
	Requests []*ExecuteGraphRequest `protobuf:"bytes,1,rep,name=requests,proto3" json:"requests,omitempty"`
	// contains filtered or unexported fields
}

func (*ExecuteGraphParallelRequest) Descriptor deprecated 

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

Deprecated: Use ExecuteGraphParallelRequest.ProtoReflect.Descriptor instead.

func (*ExecuteGraphParallelRequest) GetRequests 

func (x *ExecuteGraphParallelRequest) GetRequests() []*ExecuteGraphRequest

func (*ExecuteGraphParallelRequest) ProtoMessage 

func (*ExecuteGraphParallelRequest) ProtoMessage()

func (*ExecuteGraphParallelRequest) ProtoReflect 

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

func (*ExecuteGraphParallelRequest) Reset 

func (x *ExecuteGraphParallelRequest) Reset()

func (*ExecuteGraphParallelRequest) String 

func (x *ExecuteGraphParallelRequest) String() string

type ExecuteGraphRequest 

type ExecuteGraphRequest struct {
	Computation *HloModuleProto     `protobuf:"bytes,1,opt,name=computation,proto3" json:"computation,omitempty"`
	Arguments   []*GlobalDataHandle `protobuf:"bytes,2,rep,name=arguments,proto3" json:"arguments,omitempty"`
	// Options that affect how XLA compiles and runs code to service this request.
	ExecutionOptions *ExecutionOptions `protobuf:"bytes,3,opt,name=execution_options,json=executionOptions,proto3" json:"execution_options,omitempty"`
	// contains filtered or unexported fields
}

TODO(b/118493728): Remove this and ExecuteGraphParallelRequest and replace the uses with calls to Compile and Execute.

func (*ExecuteGraphRequest) Descriptor deprecated 

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

Deprecated: Use ExecuteGraphRequest.ProtoReflect.Descriptor instead.

func (*ExecuteGraphRequest) GetArguments 

func (x *ExecuteGraphRequest) GetArguments() []*GlobalDataHandle

func (*ExecuteGraphRequest) GetComputation 

func (x *ExecuteGraphRequest) GetComputation() *HloModuleProto

func (*ExecuteGraphRequest) GetExecutionOptions 

func (x *ExecuteGraphRequest) GetExecutionOptions() *ExecutionOptions

func (*ExecuteGraphRequest) ProtoMessage 

func (*ExecuteGraphRequest) ProtoMessage()

func (*ExecuteGraphRequest) ProtoReflect 

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

func (*ExecuteGraphRequest) Reset 

func (x *ExecuteGraphRequest) Reset()

func (*ExecuteGraphRequest) String 

func (x *ExecuteGraphRequest) String() string

type ExecuteParallelResponse 

type ExecuteParallelResponse struct {
	Responses []*ExecuteResponse `protobuf:"bytes,1,rep,name=responses,proto3" json:"responses,omitempty"`
	// contains filtered or unexported fields
}

func (*ExecuteParallelResponse) Descriptor deprecated 

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

Deprecated: Use ExecuteParallelResponse.ProtoReflect.Descriptor instead.

func (*ExecuteParallelResponse) GetResponses 

func (x *ExecuteParallelResponse) GetResponses() []*ExecuteResponse

func (*ExecuteParallelResponse) ProtoMessage 

func (*ExecuteParallelResponse) ProtoMessage()

func (*ExecuteParallelResponse) ProtoReflect 

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

func (*ExecuteParallelResponse) Reset 

func (x *ExecuteParallelResponse) Reset()

func (*ExecuteParallelResponse) String 

func (x *ExecuteParallelResponse) String() string

type ExecuteRequest 

type ExecuteRequest struct {
	Handle *ExecutionHandle `protobuf:"bytes,1,opt,name=handle,proto3" json:"handle,omitempty"`
	// The shape and layout of the arguments must be the same as the those of the
	// executable's parameters.
	Arguments []*GlobalDataHandle `protobuf:"bytes,2,rep,name=arguments,proto3" json:"arguments,omitempty"`
	// contains filtered or unexported fields
}

func (*ExecuteRequest) Descriptor deprecated 

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

Deprecated: Use ExecuteRequest.ProtoReflect.Descriptor instead.

func (*ExecuteRequest) GetArguments 

func (x *ExecuteRequest) GetArguments() []*GlobalDataHandle

func (*ExecuteRequest) GetHandle 

func (x *ExecuteRequest) GetHandle() *ExecutionHandle

func (*ExecuteRequest) ProtoMessage 

func (*ExecuteRequest) ProtoMessage()

func (*ExecuteRequest) ProtoReflect 

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

func (*ExecuteRequest) Reset 

func (x *ExecuteRequest) Reset()

func (*ExecuteRequest) String 

func (x *ExecuteRequest) String() string

type ExecuteResponse 

type ExecuteResponse struct {
	Output  *GlobalDataHandle `protobuf:"bytes,1,opt,name=output,proto3" json:"output,omitempty"`
	Profile *ExecutionProfile `protobuf:"bytes,2,opt,name=profile,proto3" json:"profile,omitempty"`
	// contains filtered or unexported fields
}

func (*ExecuteResponse) Descriptor deprecated 

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

Deprecated: Use ExecuteResponse.ProtoReflect.Descriptor instead.

func (*ExecuteResponse) GetOutput 

func (x *ExecuteResponse) GetOutput() *GlobalDataHandle

func (*ExecuteResponse) GetProfile 

func (x *ExecuteResponse) GetProfile() *ExecutionProfile

func (*ExecuteResponse) ProtoMessage 

func (*ExecuteResponse) ProtoMessage()

func (*ExecuteResponse) ProtoReflect 

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

func (*ExecuteResponse) Reset 

func (x *ExecuteResponse) Reset()

func (*ExecuteResponse) String 

func (x *ExecuteResponse) 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 ExecutionOptions 

type ExecutionOptions struct {

	// This optional field's layout is used as a hint when storing the output of
	// this computation.  Subsequent transfers of this output array to the client
	// may be faster when using this layout.
	//
	// We use a Shape here to accommodate computations that return a tuple.
	ShapeWithOutputLayout *ShapeProto `` /* 128-byte string literal not displayed */
	// Used to seed random-number generators used in this computation.  If this is
	// 0, we generate a seed ourselves.
	//
	// TODO(b/32083678): Changing the seed unnecessarily forces a recompilation.
	Seed         uint64        `protobuf:"varint,3,opt,name=seed,proto3" json:"seed,omitempty"`
	DebugOptions *DebugOptions `protobuf:"bytes,4,opt,name=debug_options,json=debugOptions,proto3" json:"debug_options,omitempty"`
	// This optional field specifies a particular set of devices to run the
	// computation on. The computation will be partitioned across these devices.
	// If not provided, the default device will be chosen.
	DeviceHandles []*DeviceHandle `protobuf:"bytes,5,rep,name=device_handles,json=deviceHandles,proto3" json:"device_handles,omitempty"`
	// Number of replicas of the computation to run. If zero, uses the default
	// number of replicas for the XLA service.
	NumReplicas int32 `protobuf:"varint,6,opt,name=num_replicas,json=numReplicas,proto3" json:"num_replicas,omitempty"`
	// This optional field specifies the device assignment if known at compile
	// time.
	DeviceAssignment *DeviceAssignmentProto `protobuf:"bytes,7,opt,name=device_assignment,json=deviceAssignment,proto3" json:"device_assignment,omitempty"`
	// Alias input and output buffers for parameters that are passed-through XLA
	// modules without being changed.
	AliasPassthroughParams bool `` /* 130-byte string literal not displayed */
	// Number of partitions of the computation to run (model parallelism).
	// If zero, uses the default number of partitions for the XLA service.
	NumPartitions int32 `protobuf:"varint,9,opt,name=num_partitions,json=numPartitions,proto3" json:"num_partitions,omitempty"`
	// Used to identify a set of programs that should be launch together.
	LaunchId int32 `protobuf:"varint,10,opt,name=launch_id,json=launchId,proto3" json:"launch_id,omitempty"`
	// Indicates whether to use SPMD (true) or MPMD (false) partitioning when
	// num_partitions > 1 and XLA is requested to partition the input program.
	UseSpmdPartitioning bool `protobuf:"varint,11,opt,name=use_spmd_partitioning,json=useSpmdPartitioning,proto3" json:"use_spmd_partitioning,omitempty"`
	// Whether to automatically generate XLA shardings for SPMD partitioner.
	UseAutoSpmdPartitioning bool `` /* 136-byte string literal not displayed */
	// Device mesh shape used to create the sharding search space when
	// use_auto_spmd_partitioning=true.
	AutoSpmdPartitioningMeshShape []int64 `` /* 163-byte string literal not displayed */
	// Device mesh ids compatible with the above mesh_shape used when
	// use_auto_spmd_partitioning=true.
	AutoSpmdPartitioningMeshIds []int64 `` /* 157-byte string literal not displayed */
	// If set, deduplicate hlo into function calls to reduce binary size. Only
	// works on TPU.
	DeduplicateHlo bool `protobuf:"varint,12,opt,name=deduplicate_hlo,json=deduplicateHlo,proto3" json:"deduplicate_hlo,omitempty"`
	// Allows sharding propagation to propagate to the parameters. This changes
	// the input shape of the computation (which is undesirable), but it can be
	// used to allow to run partial compilation to determine what would be the
	// input sharding of a computation if XLA would be allowed to propagate the
	// sharding which can be used by higher level framework as a way to query
	// intermediate sharding of operations when multiple computation would be
	// chained and merged together.
	// This is a vector of bool, because the user can control which parameters can
	// have the sharding substituted. If only one boolean value is passed in the
	// vector that is interpreted as the value to be applied for every parameter.
	AllowSpmdShardingPropagationToParameters []bool `` /* 198-byte string literal not displayed */
	// Allows sharding propagation to propagate to the outputs. This changes the
	// output shape of the computation (which is undesirable), but it can be used
	// to allow to run partial compilation to determine what would be the output
	// sharding of a computation if XLA would be allowed to propagate the sharding
	// which can be used by higher level framework as a way to query intermediate
	// sharding of operations when multiple computation would be chained and
	// merged together.
	// This is a vector of bool, because the user can control (if the output of
	// the computation is a tuple) which elements of the tuple can have the
	// sharding substituted and which don't. If only one boolean value is passed
	// in the vector that's interpreted as the value to be applied for every
	// single element of the output tuple. One value per element of the tuple
	// means that each value is attached to one of the output elements.
	AllowSpmdShardingPropagationToOutput []bool `` /* 186-byte string literal not displayed */
	// Whether to broadcast args across all replicas. One entry per arg.
	ParamRequiresBroadcastViaCollectives []bool `` /* 184-byte string literal not displayed */
	// If enabled, the compiler may generate sharding and unsharding programs as
	// separate HLO modules, and modify the main program's input and output to
	// be sharded.
	AllowSeparateShardingPrograms bool `` /* 154-byte string literal not displayed */
	// The list of input/output pairs in the main program that could be sharded.
	ShardableValueUpdatePairs []*ShardableValueUpdatePairProto `` /* 141-byte string literal not displayed */
	// Profiling data for feedback directed optimizations. Note that this is not
	// the only way to feed FDO data into the compiler and individual backends
	// may choose to get FDO data by other means.
	FdoProfile []byte `protobuf:"bytes,21,opt,name=fdo_profile,json=fdoProfile,proto3" json:"fdo_profile,omitempty"`
	// Amount of device memory available for the executable to use.
	DeviceMemorySize int64 `protobuf:"varint,22,opt,name=device_memory_size,json=deviceMemorySize,proto3" json:"device_memory_size,omitempty"`
	// contains filtered or unexported fields
}

These settings control how XLA compiles and/or runs code. Not all settings will have an effect on every platform.

When adding new fields, keep in mind that boolean fields default to false. Next id: 24.

func (*ExecutionOptions) Descriptor deprecated 

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

Deprecated: Use ExecutionOptions.ProtoReflect.Descriptor instead.

func (*ExecutionOptions) GetAliasPassthroughParams 

func (x *ExecutionOptions) GetAliasPassthroughParams() bool

func (*ExecutionOptions) GetAllowSeparateShardingPrograms 

func (x *ExecutionOptions) GetAllowSeparateShardingPrograms() bool

func (*ExecutionOptions) GetAllowSpmdShardingPropagationToOutput 

func (x *ExecutionOptions) GetAllowSpmdShardingPropagationToOutput() []bool

func (*ExecutionOptions) GetAllowSpmdShardingPropagationToParameters 

func (x *ExecutionOptions) GetAllowSpmdShardingPropagationToParameters() []bool

func (*ExecutionOptions) GetAutoSpmdPartitioningMeshIds 

func (x *ExecutionOptions) GetAutoSpmdPartitioningMeshIds() []int64

func (*ExecutionOptions) GetAutoSpmdPartitioningMeshShape 

func (x *ExecutionOptions) GetAutoSpmdPartitioningMeshShape() []int64

func (*ExecutionOptions) GetDebugOptions 

func (x *ExecutionOptions) GetDebugOptions() *DebugOptions

func (*ExecutionOptions) GetDeduplicateHlo 

func (x *ExecutionOptions) GetDeduplicateHlo() bool

func (*ExecutionOptions) GetDeviceAssignment 

func (x *ExecutionOptions) GetDeviceAssignment() *DeviceAssignmentProto

func (*ExecutionOptions) GetDeviceHandles 

func (x *ExecutionOptions) GetDeviceHandles() []*DeviceHandle

func (*ExecutionOptions) GetDeviceMemorySize 

func (x *ExecutionOptions) GetDeviceMemorySize() int64

func (*ExecutionOptions) GetFdoProfile 

func (x *ExecutionOptions) GetFdoProfile() []byte

func (*ExecutionOptions) GetLaunchId 

func (x *ExecutionOptions) GetLaunchId() int32

func (*ExecutionOptions) GetNumPartitions 

func (x *ExecutionOptions) GetNumPartitions() int32

func (*ExecutionOptions) GetNumReplicas 

func (x *ExecutionOptions) GetNumReplicas() int32

func (*ExecutionOptions) GetParamRequiresBroadcastViaCollectives 

func (x *ExecutionOptions) GetParamRequiresBroadcastViaCollectives() []bool

func (*ExecutionOptions) GetSeed 

func (x *ExecutionOptions) GetSeed() uint64

func (*ExecutionOptions) GetShapeWithOutputLayout 

func (x *ExecutionOptions) GetShapeWithOutputLayout() *ShapeProto

func (*ExecutionOptions) GetShardableValueUpdatePairs 

func (x *ExecutionOptions) GetShardableValueUpdatePairs() []*ShardableValueUpdatePairProto

func (*ExecutionOptions) GetUseAutoSpmdPartitioning 

func (x *ExecutionOptions) GetUseAutoSpmdPartitioning() bool

func (*ExecutionOptions) GetUseSpmdPartitioning 

func (x *ExecutionOptions) GetUseSpmdPartitioning() bool

func (*ExecutionOptions) ProtoMessage 

func (*ExecutionOptions) ProtoMessage()

func (*ExecutionOptions) ProtoReflect 

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

func (*ExecutionOptions) Reset 

func (x *ExecutionOptions) Reset()

func (*ExecutionOptions) String 

func (x *ExecutionOptions) 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 FMHAMaskKind 

type FMHAMaskKind int32

FusedMHAMaskKind kind 

const (
	FMHAMaskKind_NO_MASK        FMHAMaskKind = 0
	FMHAMaskKind_PADDING        FMHAMaskKind = 1
	FMHAMaskKind_CAUSAL         FMHAMaskKind = 2
	FMHAMaskKind_PADDING_CAUSAL FMHAMaskKind = 3
	FMHAMaskKind_ALIBI          FMHAMaskKind = 4
)

func (FMHAMaskKind) Descriptor 

func (FMHAMaskKind) Descriptor() protoreflect.EnumDescriptor

func (FMHAMaskKind) Enum 

func (x FMHAMaskKind) Enum() *FMHAMaskKind

func (FMHAMaskKind) EnumDescriptor deprecated 

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

Deprecated: Use FMHAMaskKind.Descriptor instead.

func (FMHAMaskKind) Number 

func (x FMHAMaskKind) Number() protoreflect.EnumNumber

func (FMHAMaskKind) String 

func (x FMHAMaskKind) String() string

func (FMHAMaskKind) Type 

func (FMHAMaskKind) Type() protoreflect.EnumType

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 FilterLayout 

type FilterLayout int32

Describes how a convolution filter is laid out in the memory. 

const (
	// Naming convention:
	// Y <-> row or height
	// X <-> column or width
	// Output <-> output feature, or N
	// Input <-> input feature, or N
	// TODO(timshen): turn them into cuDNN names, e.g. kNCHW.
	FilterLayout_kOutputInputYX   FilterLayout = 0 // cuDNN's NCHW layout
	FilterLayout_kOutputYXInput   FilterLayout = 1 // cuDNN's NHWC layout
	FilterLayout_kOutputInputYX4  FilterLayout = 2 // cuDNN's NCHW_VECT_C layout with 4-elem vectors
	FilterLayout_kOutputInputYX32 FilterLayout = 5 // cuDNN's NCHW_VECT_C layout with 32-elem vectors
	// cuDNN-specific filter reordering (using `cudnnReorderFilterAndBias`)
	// When the filter is reordered, so is the bias (if present).
	FilterLayout_kOutputInputYX32_CudnnReordered FilterLayout = 6
	FilterLayout_kInputYXOutput                  FilterLayout = 3
	FilterLayout_kYXInputOutput                  FilterLayout = 4
)

func (FilterLayout) Descriptor 

func (FilterLayout) Descriptor() protoreflect.EnumDescriptor

func (FilterLayout) Enum 

func (x FilterLayout) Enum() *FilterLayout

func (FilterLayout) EnumDescriptor deprecated 

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

Deprecated: Use FilterLayout.Descriptor instead.

func (FilterLayout) Number 

func (x FilterLayout) Number() protoreflect.EnumNumber

func (FilterLayout) String 

func (x FilterLayout) String() string

func (FilterLayout) Type 

func (FilterLayout) 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 FusedMHAKind 

type FusedMHAKind int32

FusedMHAKind kind 

const (
	FusedMHAKind_BMM1_OUTPUT_UNKNOWN    FusedMHAKind = 0
	FusedMHAKind_BMM1_OUTPUT_INPUT_TYPE FusedMHAKind = 1
	FusedMHAKind_BMM1_OUTPUT_FLOAT      FusedMHAKind = 2
)

func (FusedMHAKind) Descriptor 

func (FusedMHAKind) Descriptor() protoreflect.EnumDescriptor

func (FusedMHAKind) Enum 

func (x FusedMHAKind) Enum() *FusedMHAKind

func (FusedMHAKind) EnumDescriptor deprecated 

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

Deprecated: Use FusedMHAKind.Descriptor instead.

func (FusedMHAKind) Number 

func (x FusedMHAKind) Number() protoreflect.EnumNumber

func (FusedMHAKind) String 

func (x FusedMHAKind) String() string

func (FusedMHAKind) Type 

func (FusedMHAKind) Type() protoreflect.EnumType

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"`
	// 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) GetStartIndexMap 

func (x *GatherDimensionNumbers) GetStartIndexMap() []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 GetDeviceHandlesRequest 

type GetDeviceHandlesRequest struct {
	DeviceCount int64 `protobuf:"varint,1,opt,name=device_count,json=deviceCount,proto3" json:"device_count,omitempty"`
	// contains filtered or unexported fields
}

func (*GetDeviceHandlesRequest) Descriptor deprecated 

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

Deprecated: Use GetDeviceHandlesRequest.ProtoReflect.Descriptor instead.

func (*GetDeviceHandlesRequest) GetDeviceCount 

func (x *GetDeviceHandlesRequest) GetDeviceCount() int64

func (*GetDeviceHandlesRequest) ProtoMessage 

func (*GetDeviceHandlesRequest) ProtoMessage()

func (*GetDeviceHandlesRequest) ProtoReflect 

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

func (*GetDeviceHandlesRequest) Reset 

func (x *GetDeviceHandlesRequest) Reset()

func (*GetDeviceHandlesRequest) String 

func (x *GetDeviceHandlesRequest) String() string

type GetDeviceHandlesResponse 

type GetDeviceHandlesResponse struct {
	DeviceHandles []*DeviceHandle `protobuf:"bytes,1,rep,name=device_handles,json=deviceHandles,proto3" json:"device_handles,omitempty"`
	// contains filtered or unexported fields
}

func (*GetDeviceHandlesResponse) Descriptor deprecated 

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

Deprecated: Use GetDeviceHandlesResponse.ProtoReflect.Descriptor instead.

func (*GetDeviceHandlesResponse) GetDeviceHandles 

func (x *GetDeviceHandlesResponse) GetDeviceHandles() []*DeviceHandle

func (*GetDeviceHandlesResponse) ProtoMessage 

func (*GetDeviceHandlesResponse) ProtoMessage()

func (*GetDeviceHandlesResponse) ProtoReflect 

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

func (*GetDeviceHandlesResponse) Reset 

func (x *GetDeviceHandlesResponse) Reset()

func (*GetDeviceHandlesResponse) String 

func (x *GetDeviceHandlesResponse) String() string

type GetShapeRequest 

type GetShapeRequest struct {
	Data *GlobalDataHandle `protobuf:"bytes,1,opt,name=data,proto3" json:"data,omitempty"`
	// contains filtered or unexported fields
}

func (*GetShapeRequest) Descriptor deprecated 

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

Deprecated: Use GetShapeRequest.ProtoReflect.Descriptor instead.

func (*GetShapeRequest) GetData 

func (x *GetShapeRequest) GetData() *GlobalDataHandle

func (*GetShapeRequest) ProtoMessage 

func (*GetShapeRequest) ProtoMessage()

func (*GetShapeRequest) ProtoReflect 

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

func (*GetShapeRequest) Reset 

func (x *GetShapeRequest) Reset()

func (*GetShapeRequest) String 

func (x *GetShapeRequest) String() string

type GetShapeResponse 

type GetShapeResponse struct {
	Shape *ShapeProto `protobuf:"bytes,1,opt,name=shape,proto3" json:"shape,omitempty"`
	// contains filtered or unexported fields
}

func (*GetShapeResponse) Descriptor deprecated 

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

Deprecated: Use GetShapeResponse.ProtoReflect.Descriptor instead.

func (*GetShapeResponse) GetShape 

func (x *GetShapeResponse) GetShape() *ShapeProto

func (*GetShapeResponse) ProtoMessage 

func (*GetShapeResponse) ProtoMessage()

func (*GetShapeResponse) ProtoReflect 

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

func (*GetShapeResponse) Reset 

func (x *GetShapeResponse) Reset()

func (*GetShapeResponse) String 

func (x *GetShapeResponse) 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 GpuCompilationEnvironment 

type GpuCompilationEnvironment struct {

	// Temporary dummy flag is added to test the flow.
	// To be removed when we add flags here.
	DummyFlag int64 `protobuf:"varint,1,opt,name=dummy_flag,json=dummyFlag,proto3" json:"dummy_flag,omitempty"`
	// contains filtered or unexported fields
}

Contains flags which affects the GPU compilation result. These flags are part of Debug Options as of now, and will be migrated to this proto.

func (*GpuCompilationEnvironment) Descriptor deprecated 

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

Deprecated: Use GpuCompilationEnvironment.ProtoReflect.Descriptor instead.

func (*GpuCompilationEnvironment) GetDummyFlag 

func (x *GpuCompilationEnvironment) GetDummyFlag() int64

func (*GpuCompilationEnvironment) ProtoMessage 

func (*GpuCompilationEnvironment) ProtoMessage()

func (*GpuCompilationEnvironment) ProtoReflect 

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

func (*GpuCompilationEnvironment) Reset 

func (x *GpuCompilationEnvironment) Reset()

func (*GpuCompilationEnvironment) String 

func (x *GpuCompilationEnvironment) String() string

type GpuDeviceInfoProto 

type GpuDeviceInfoProto struct {
	ThreadsPerBlockLimit      int32   `` /* 126-byte string literal not displayed */
	ThreadsPerWarp            int32   `protobuf:"varint,2,opt,name=threads_per_warp,json=threadsPerWarp,proto3" json:"threads_per_warp,omitempty"`
	SharedMemoryPerBlock      int32   `` /* 126-byte string literal not displayed */
	SharedMemoryPerCore       int32   `protobuf:"varint,4,opt,name=shared_memory_per_core,json=sharedMemoryPerCore,proto3" json:"shared_memory_per_core,omitempty"`
	ThreadsPerCoreLimit       int32   `protobuf:"varint,5,opt,name=threads_per_core_limit,json=threadsPerCoreLimit,proto3" json:"threads_per_core_limit,omitempty"`
	CoreCount                 int32   `protobuf:"varint,6,opt,name=core_count,json=coreCount,proto3" json:"core_count,omitempty"`
	FpusPerCore               int64   `protobuf:"varint,7,opt,name=fpus_per_core,json=fpusPerCore,proto3" json:"fpus_per_core,omitempty"`
	BlockDimLimitX            int32   `protobuf:"varint,8,opt,name=block_dim_limit_x,json=blockDimLimitX,proto3" json:"block_dim_limit_x,omitempty"`
	BlockDimLimitY            int32   `protobuf:"varint,9,opt,name=block_dim_limit_y,json=blockDimLimitY,proto3" json:"block_dim_limit_y,omitempty"`
	BlockDimLimitZ            int32   `protobuf:"varint,10,opt,name=block_dim_limit_z,json=blockDimLimitZ,proto3" json:"block_dim_limit_z,omitempty"`
	MemoryBandwidth           int64   `protobuf:"varint,11,opt,name=memory_bandwidth,json=memoryBandwidth,proto3" json:"memory_bandwidth,omitempty"`
	L2CacheSize               int64   `protobuf:"varint,12,opt,name=l2_cache_size,json=l2CacheSize,proto3" json:"l2_cache_size,omitempty"`
	ClockRateGhz              float32 `protobuf:"fixed32,13,opt,name=clock_rate_ghz,json=clockRateGhz,proto3" json:"clock_rate_ghz,omitempty"`
	DeviceMemorySize          int64   `protobuf:"varint,14,opt,name=device_memory_size,json=deviceMemorySize,proto3" json:"device_memory_size,omitempty"`
	SharedMemoryPerBlockOptin int32   `` /* 144-byte string literal not displayed */
	// Types that are assignable to ComputeCapability:
	//
	//	*GpuDeviceInfoProto_CudaComputeCapability
	//	*GpuDeviceInfoProto_RocmComputeCapability
	ComputeCapability isGpuDeviceInfoProto_ComputeCapability `protobuf_oneof:"compute_capability"`
	// contains filtered or unexported fields
}

func (*GpuDeviceInfoProto) Descriptor deprecated 

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

Deprecated: Use GpuDeviceInfoProto.ProtoReflect.Descriptor instead.

func (*GpuDeviceInfoProto) GetBlockDimLimitX 

func (x *GpuDeviceInfoProto) GetBlockDimLimitX() int32

func (*GpuDeviceInfoProto) GetBlockDimLimitY 

func (x *GpuDeviceInfoProto) GetBlockDimLimitY() int32

func (*GpuDeviceInfoProto) GetBlockDimLimitZ 

func (x *GpuDeviceInfoProto) GetBlockDimLimitZ() int32

func (*GpuDeviceInfoProto) GetClockRateGhz 

func (x *GpuDeviceInfoProto) GetClockRateGhz() float32

func (*GpuDeviceInfoProto) GetComputeCapability 

func (m *GpuDeviceInfoProto) GetComputeCapability() isGpuDeviceInfoProto_ComputeCapability

func (*GpuDeviceInfoProto) GetCoreCount 

func (x *GpuDeviceInfoProto) GetCoreCount() int32

func (*GpuDeviceInfoProto) GetCudaComputeCapability 

func (x *GpuDeviceInfoProto) GetCudaComputeCapability() *CudaComputeCapabilityProto

func (*GpuDeviceInfoProto) GetDeviceMemorySize 

func (x *GpuDeviceInfoProto) GetDeviceMemorySize() int64

func (*GpuDeviceInfoProto) GetFpusPerCore 

func (x *GpuDeviceInfoProto) GetFpusPerCore() int64

func (*GpuDeviceInfoProto) GetL2CacheSize 

func (x *GpuDeviceInfoProto) GetL2CacheSize() int64

func (*GpuDeviceInfoProto) GetMemoryBandwidth 

func (x *GpuDeviceInfoProto) GetMemoryBandwidth() int64

func (*GpuDeviceInfoProto) GetRocmComputeCapability 

func (x *GpuDeviceInfoProto) GetRocmComputeCapability() *RocmComputeCapabilityProto

func (*GpuDeviceInfoProto) GetSharedMemoryPerBlock 

func (x *GpuDeviceInfoProto) GetSharedMemoryPerBlock() int32

func (*GpuDeviceInfoProto) GetSharedMemoryPerBlockOptin 

func (x *GpuDeviceInfoProto) GetSharedMemoryPerBlockOptin() int32

func (*GpuDeviceInfoProto) GetSharedMemoryPerCore 

func (x *GpuDeviceInfoProto) GetSharedMemoryPerCore() int32

func (*GpuDeviceInfoProto) GetThreadsPerBlockLimit 

func (x *GpuDeviceInfoProto) GetThreadsPerBlockLimit() int32

func (*GpuDeviceInfoProto) GetThreadsPerCoreLimit 

func (x *GpuDeviceInfoProto) GetThreadsPerCoreLimit() int32

func (*GpuDeviceInfoProto) GetThreadsPerWarp 

func (x *GpuDeviceInfoProto) GetThreadsPerWarp() int32

func (*GpuDeviceInfoProto) ProtoMessage 

func (*GpuDeviceInfoProto) ProtoMessage()

func (*GpuDeviceInfoProto) ProtoReflect 

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

func (*GpuDeviceInfoProto) Reset 

func (x *GpuDeviceInfoProto) Reset()

func (*GpuDeviceInfoProto) String 

func (x *GpuDeviceInfoProto) String() string

type GpuDeviceInfoProto_CudaComputeCapability 

type GpuDeviceInfoProto_CudaComputeCapability struct {
	CudaComputeCapability *CudaComputeCapabilityProto `protobuf:"bytes,16,opt,name=cuda_compute_capability,json=cudaComputeCapability,proto3,oneof"`
}

type GpuDeviceInfoProto_RocmComputeCapability 

type GpuDeviceInfoProto_RocmComputeCapability struct {
	RocmComputeCapability *RocmComputeCapabilityProto `protobuf:"bytes,17,opt,name=rocm_compute_capability,json=rocmComputeCapability,proto3,oneof"`
}

type GpuTargetConfigProto 

type GpuTargetConfigProto struct {
	GpuDeviceInfo  *GpuDeviceInfoProto  `protobuf:"bytes,1,opt,name=gpu_device_info,json=gpuDeviceInfo,proto3" json:"gpu_device_info,omitempty"`
	PlatformName   string               `protobuf:"bytes,4,opt,name=platform_name,json=platformName,proto3" json:"platform_name,omitempty"`
	DnnVersionInfo *DnnVersionInfoProto `protobuf:"bytes,5,opt,name=dnn_version_info,json=dnnVersionInfo,proto3" json:"dnn_version_info,omitempty"`
	// TODO(b/248362914): Autotuning results should be separate from
	// GpuTargetConfig because autotuning can be updated regularly separate from
	// the target.
	AutotuneResults      *AutotuneResults `protobuf:"bytes,6,opt,name=autotune_results,json=autotuneResults,proto3" json:"autotune_results,omitempty"`
	DeviceDescriptionStr string           `protobuf:"bytes,7,opt,name=device_description_str,json=deviceDescriptionStr,proto3" json:"device_description_str,omitempty"`
	// contains filtered or unexported fields
}

func (*GpuTargetConfigProto) Descriptor deprecated 

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

Deprecated: Use GpuTargetConfigProto.ProtoReflect.Descriptor instead.

func (*GpuTargetConfigProto) GetAutotuneResults 

func (x *GpuTargetConfigProto) GetAutotuneResults() *AutotuneResults

func (*GpuTargetConfigProto) GetDeviceDescriptionStr 

func (x *GpuTargetConfigProto) GetDeviceDescriptionStr() string

func (*GpuTargetConfigProto) GetDnnVersionInfo 

func (x *GpuTargetConfigProto) GetDnnVersionInfo() *DnnVersionInfoProto

func (*GpuTargetConfigProto) GetGpuDeviceInfo 

func (x *GpuTargetConfigProto) GetGpuDeviceInfo() *GpuDeviceInfoProto

func (*GpuTargetConfigProto) GetPlatformName 

func (x *GpuTargetConfigProto) GetPlatformName() string

func (*GpuTargetConfigProto) ProtoMessage 

func (*GpuTargetConfigProto) ProtoMessage()

func (*GpuTargetConfigProto) ProtoReflect 

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

func (*GpuTargetConfigProto) Reset 

func (x *GpuTargetConfigProto) Reset()

func (*GpuTargetConfigProto) String 

func (x *GpuTargetConfigProto) String() string

type HeapSimulatorTrace 

type HeapSimulatorTrace struct {
	Events                []*HeapSimulatorTrace_Event `protobuf:"bytes,1,rep,name=events,proto3" json:"events,omitempty"`
	WholeModuleSimulation bool                        `` /* 127-byte string literal not displayed */
	BufferAllocationIndex int64                       `` /* 127-byte string literal not displayed */
	// contains filtered or unexported fields
}

A trace of a HeapSimulator run.

func (*HeapSimulatorTrace) Descriptor deprecated 

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

Deprecated: Use HeapSimulatorTrace.ProtoReflect.Descriptor instead.

func (*HeapSimulatorTrace) GetBufferAllocationIndex 

func (x *HeapSimulatorTrace) GetBufferAllocationIndex() int64

func (*HeapSimulatorTrace) GetEvents 

func (x *HeapSimulatorTrace) GetEvents() []*HeapSimulatorTrace_Event

func (*HeapSimulatorTrace) GetWholeModuleSimulation 

func (x *HeapSimulatorTrace) GetWholeModuleSimulation() bool

func (*HeapSimulatorTrace) ProtoMessage 

func (*HeapSimulatorTrace) ProtoMessage()

func (*HeapSimulatorTrace) ProtoReflect 

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

func (*HeapSimulatorTrace) Reset 

func (x *HeapSimulatorTrace) Reset()

func (*HeapSimulatorTrace) String 

func (x *HeapSimulatorTrace) String() string

type HeapSimulatorTrace_Event 

type HeapSimulatorTrace_Event struct {
	Kind HeapSimulatorTrace_Event_Kind `protobuf:"varint,1,opt,name=kind,proto3,enum=xla.HeapSimulatorTrace_Event_Kind" json:"kind,omitempty"`
	// The id of the LogicalBuffer that the event applies to.
	BufferId int64 `protobuf:"varint,2,opt,name=buffer_id,json=bufferId,proto3" json:"buffer_id,omitempty"`
	// The HloInstruction that the simulation was processing that caused this
	// event to occur, identified by its computation and instruction name. E.g.
	// buffers defined by instruction A are allocated when processing A.
	ComputationName string `protobuf:"bytes,3,opt,name=computation_name,json=computationName,proto3" json:"computation_name,omitempty"`
	InstructionName string `protobuf:"bytes,4,opt,name=instruction_name,json=instructionName,proto3" json:"instruction_name,omitempty"`
	// The id of the canonical LogicalBuffer that the buffer shares with. Only
	// set for SHARE_WITH events.
	ShareWithCanonicalId int64 `` /* 126-byte string literal not displayed */
	// contains filtered or unexported fields
}

The trace includes a list of events, where each event describes one action performed by the heap simulator.

func (*HeapSimulatorTrace_Event) Descriptor deprecated 

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

Deprecated: Use HeapSimulatorTrace_Event.ProtoReflect.Descriptor instead.

func (*HeapSimulatorTrace_Event) GetBufferId 

func (x *HeapSimulatorTrace_Event) GetBufferId() int64

func (*HeapSimulatorTrace_Event) GetComputationName 

func (x *HeapSimulatorTrace_Event) GetComputationName() string

func (*HeapSimulatorTrace_Event) GetInstructionName 

func (x *HeapSimulatorTrace_Event) GetInstructionName() string

func (*HeapSimulatorTrace_Event) GetKind 

func (x *HeapSimulatorTrace_Event) GetKind() HeapSimulatorTrace_Event_Kind

func (*HeapSimulatorTrace_Event) GetShareWithCanonicalId 

func (x *HeapSimulatorTrace_Event) GetShareWithCanonicalId() int64

func (*HeapSimulatorTrace_Event) ProtoMessage 

func (*HeapSimulatorTrace_Event) ProtoMessage()

func (*HeapSimulatorTrace_Event) ProtoReflect 

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

func (*HeapSimulatorTrace_Event) Reset 

func (x *HeapSimulatorTrace_Event) Reset()

func (*HeapSimulatorTrace_Event) String 

func (x *HeapSimulatorTrace_Event) String() string

type HeapSimulatorTrace_Event_Kind 

type HeapSimulatorTrace_Event_Kind int32
const (
	HeapSimulatorTrace_Event_ALLOC HeapSimulatorTrace_Event_Kind = 0 // A memory region was allocated for the buffer.
	HeapSimulatorTrace_Event_FREE  HeapSimulatorTrace_Event_Kind = 1 // A memory region was freed for the buffer.
	// A buffer was shared with another (canonical) buffer. This is similar to
	// ALLOC, except that instead of allocating a new region of memory, the
	// memory region of the canonical buffer is directly re-used. Multiple
	// buffers may share with the same canonical buffer. The lifetime of the
	// canonical buffer is extended to the union of all lifetimes.
	HeapSimulatorTrace_Event_SHARE_WITH HeapSimulatorTrace_Event_Kind = 2
)

func (HeapSimulatorTrace_Event_Kind) Descriptor 

func (HeapSimulatorTrace_Event_Kind) Descriptor() protoreflect.EnumDescriptor

func (HeapSimulatorTrace_Event_Kind) Enum 

func (x HeapSimulatorTrace_Event_Kind) Enum() *HeapSimulatorTrace_Event_Kind

func (HeapSimulatorTrace_Event_Kind) EnumDescriptor deprecated 

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

Deprecated: Use HeapSimulatorTrace_Event_Kind.Descriptor instead.

func (HeapSimulatorTrace_Event_Kind) Number 

func (x HeapSimulatorTrace_Event_Kind) Number() protoreflect.EnumNumber

func (HeapSimulatorTrace_Event_Kind) String 

func (x HeapSimulatorTrace_Event_Kind) String() string

func (HeapSimulatorTrace_Event_Kind) Type 

func (HeapSimulatorTrace_Event_Kind) Type() protoreflect.EnumType

type HloBufferDonorProto 

type HloBufferDonorProto struct {
	Entries []*HloBufferDonorProto_BufferDonorEntryProto `protobuf:"bytes,1,rep,name=entries,proto3" json:"entries,omitempty"`
	// contains filtered or unexported fields
}

func (*HloBufferDonorProto) Descriptor deprecated 

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

Deprecated: Use HloBufferDonorProto.ProtoReflect.Descriptor instead.

func (*HloBufferDonorProto) GetEntries 

func (x *HloBufferDonorProto) GetEntries() []*HloBufferDonorProto_BufferDonorEntryProto

func (*HloBufferDonorProto) ProtoMessage 

func (*HloBufferDonorProto) ProtoMessage()

func (*HloBufferDonorProto) ProtoReflect 

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

func (*HloBufferDonorProto) Reset 

func (x *HloBufferDonorProto) Reset()

func (*HloBufferDonorProto) String 

func (x *HloBufferDonorProto) String() string

type HloBufferDonorProto_BufferDonorEntryProto 

type HloBufferDonorProto_BufferDonorEntryProto struct {

	// Number of the parameter in entry computation.
	ParameterNumber int64 `protobuf:"varint,1,opt,name=parameter_number,json=parameterNumber,proto3" json:"parameter_number,omitempty"`
	// ShapeIndex of the parameter instruction.
	ParameterShapeIndex []int64 `` /* 128-byte string literal not displayed */
	// contains filtered or unexported fields
}

The following proto describes an input (described by parameter number and a ShapeIndex of the parameter) that can donate its butter to any output tensor. It is similar to HloInputOutputAliasProto, but without a paired output. For example: 

entry = {
 parameter_number=0,
 parameter_shape_index={1, 2},
}

This entry indicates that the first parameter's {1, 2} element can donate its buffer.

func (*HloBufferDonorProto_BufferDonorEntryProto) Descriptor deprecated 

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

Deprecated: Use HloBufferDonorProto_BufferDonorEntryProto.ProtoReflect.Descriptor instead.

func (*HloBufferDonorProto_BufferDonorEntryProto) GetParameterNumber 

func (x *HloBufferDonorProto_BufferDonorEntryProto) GetParameterNumber() int64

func (*HloBufferDonorProto_BufferDonorEntryProto) GetParameterShapeIndex 

func (x *HloBufferDonorProto_BufferDonorEntryProto) GetParameterShapeIndex() []int64

func (*HloBufferDonorProto_BufferDonorEntryProto) ProtoMessage 

func (*HloBufferDonorProto_BufferDonorEntryProto) ProtoMessage()

func (*HloBufferDonorProto_BufferDonorEntryProto) ProtoReflect 

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

func (*HloBufferDonorProto_BufferDonorEntryProto) Reset 

func (x *HloBufferDonorProto_BufferDonorEntryProto) Reset()

func (*HloBufferDonorProto_BufferDonorEntryProto) String 

func (x *HloBufferDonorProto_BufferDonorEntryProto) String() string

type HloComputationProto 

type HloComputationProto struct {
	Name string `protobuf:"bytes,1,opt,name=name,proto3" json:"name,omitempty"`
	// The array of instructions is always in a valid dependency order, where
	// operands appear before their users.
	Instructions []*HloInstructionProto `protobuf:"bytes,2,rep,name=instructions,proto3" json:"instructions,omitempty"`
	ProgramShape *ProgramShapeProto     `protobuf:"bytes,4,opt,name=program_shape,json=programShape,proto3" json:"program_shape,omitempty"`
	// The id of this computation.
	Id int64 `protobuf:"varint,5,opt,name=id,proto3" json:"id,omitempty"`
	// The id of the root of the computation.
	RootId int64 `protobuf:"varint,6,opt,name=root_id,json=rootId,proto3" json:"root_id,omitempty"`
	// Whether this is a fusion computation. Fusion computations should use this
	// to determine whether they are a fusion in CreateFromProto since the
	// parent fusion_instruction_ may get removed and be nullptr.
	IsFusionComputation bool `protobuf:"varint,7,opt,name=is_fusion_computation,json=isFusionComputation,proto3" json:"is_fusion_computation,omitempty"`
	// The name of execution thread this computation belongs to.
	ExecutionThread string `protobuf:"bytes,8,opt,name=execution_thread,json=executionThread,proto3" json:"execution_thread,omitempty"`
	// contains filtered or unexported fields
}

Serialization of HloComputation.

func (*HloComputationProto) Descriptor deprecated 

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

Deprecated: Use HloComputationProto.ProtoReflect.Descriptor instead.

func (*HloComputationProto) GetExecutionThread 

func (x *HloComputationProto) GetExecutionThread() string

func (*HloComputationProto) GetId 

func (x *HloComputationProto) GetId() int64

func (*HloComputationProto) GetInstructions 

func (x *HloComputationProto) GetInstructions() []*HloInstructionProto

func (*HloComputationProto) GetIsFusionComputation 

func (x *HloComputationProto) GetIsFusionComputation() bool

func (*HloComputationProto) GetName 

func (x *HloComputationProto) GetName() string

func (*HloComputationProto) GetProgramShape 

func (x *HloComputationProto) GetProgramShape() *ProgramShapeProto

func (*HloComputationProto) GetRootId 

func (x *HloComputationProto) GetRootId() int64

func (*HloComputationProto) ProtoMessage 

func (*HloComputationProto) ProtoMessage()

func (*HloComputationProto) ProtoReflect 

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

func (*HloComputationProto) Reset 

func (x *HloComputationProto) Reset()

func (*HloComputationProto) String 

func (x *HloComputationProto) String() string

type HloInputOutputAliasProto 

type HloInputOutputAliasProto struct {
	Entries []*HloInputOutputAliasProto_AliasEntryProto `protobuf:"bytes,1,rep,name=entries,proto3" json:"entries,omitempty"`
	// contains filtered or unexported fields
}

func (*HloInputOutputAliasProto) Descriptor deprecated 

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

Deprecated: Use HloInputOutputAliasProto.ProtoReflect.Descriptor instead.

func (*HloInputOutputAliasProto) GetEntries 

func (x *HloInputOutputAliasProto) GetEntries() []*HloInputOutputAliasProto_AliasEntryProto

func (*HloInputOutputAliasProto) ProtoMessage 

func (*HloInputOutputAliasProto) ProtoMessage()

func (*HloInputOutputAliasProto) ProtoReflect 

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

func (*HloInputOutputAliasProto) Reset 

func (x *HloInputOutputAliasProto) Reset()

func (*HloInputOutputAliasProto) String 

func (x *HloInputOutputAliasProto) String() string

type HloInputOutputAliasProto_AliasEntryProto 

type HloInputOutputAliasProto_AliasEntryProto struct {

	// ShapeIndex of the root hlo.
	OutputShapeIndex []int64 `protobuf:"varint,1,rep,packed,name=output_shape_index,json=outputShapeIndex,proto3" json:"output_shape_index,omitempty"`
	// Number of the parameter in entry computation.
	ParameterNumber int64 `protobuf:"varint,2,opt,name=parameter_number,json=parameterNumber,proto3" json:"parameter_number,omitempty"`
	// ShapeIndex of the parameter instruction.
	ParameterShapeIndex []int64 `` /* 128-byte string literal not displayed */
	// The kind of alias to be setup.
	Kind Kind `protobuf:"varint,4,opt,name=kind,proto3,enum=xla.Kind" json:"kind,omitempty"`
	// contains filtered or unexported fields
}

The following proto describes a pair of aliased an input (described by parameter number and a ShapeIndex of the parameter) and an output (described by a ShapeIndex of the root instruction). For example: 

entry = {
 output_shape_index={1},
 parameter_number=0,
 parameter_shape_index={1, 2},
}

This entry indicates that the first parameter's {1, 2} element is aliased with the {1} element of the root instruction.

func (*HloInputOutputAliasProto_AliasEntryProto) Descriptor deprecated 

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

Deprecated: Use HloInputOutputAliasProto_AliasEntryProto.ProtoReflect.Descriptor instead.

func (*HloInputOutputAliasProto_AliasEntryProto) GetKind 

func (x *HloInputOutputAliasProto_AliasEntryProto) GetKind() Kind

func (*HloInputOutputAliasProto_AliasEntryProto) GetOutputShapeIndex 

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

func (*HloInputOutputAliasProto_AliasEntryProto) GetParameterNumber 

func (x *HloInputOutputAliasProto_AliasEntryProto) GetParameterNumber() int64

func (*HloInputOutputAliasProto_AliasEntryProto) GetParameterShapeIndex 

func (x *HloInputOutputAliasProto_AliasEntryProto) GetParameterShapeIndex() []int64

func (*HloInputOutputAliasProto_AliasEntryProto) ProtoMessage 

func (*HloInputOutputAliasProto_AliasEntryProto) ProtoMessage()

func (*HloInputOutputAliasProto_AliasEntryProto) ProtoReflect 

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

func (*HloInputOutputAliasProto_AliasEntryProto) Reset 

func (x *HloInputOutputAliasProto_AliasEntryProto) Reset()

func (*HloInputOutputAliasProto_AliasEntryProto) String 

func (x *HloInputOutputAliasProto_AliasEntryProto) String() string

type HloInstructionProto 

type HloInstructionProto struct {
	Name     string      `protobuf:"bytes,1,opt,name=name,proto3" json:"name,omitempty"`
	Opcode   string      `protobuf:"bytes,2,opt,name=opcode,proto3" json:"opcode,omitempty"`
	Shape    *ShapeProto `protobuf:"bytes,3,opt,name=shape,proto3" json:"shape,omitempty"`
	Metadata *OpMetadata `protobuf:"bytes,7,opt,name=metadata,proto3" json:"metadata,omitempty"`
	// Literal, only present for kConstant.
	Literal *LiteralProto `protobuf:"bytes,8,opt,name=literal,proto3" json:"literal,omitempty"`
	// Parameter number is only present for kParameter.
	ParameterNumber int64 `protobuf:"varint,9,opt,name=parameter_number,json=parameterNumber,proto3" json:"parameter_number,omitempty"`
	// Fusion state, only present for kFusion.
	FusionKind string `protobuf:"bytes,11,opt,name=fusion_kind,json=fusionKind,proto3" json:"fusion_kind,omitempty"`
	// Index for kGetTupleElement.
	TupleIndex int64 `protobuf:"varint,13,opt,name=tuple_index,json=tupleIndex,proto3" json:"tuple_index,omitempty"`
	// Dimensions present for some operations that require reshaping or
	// broadcasting, including Reshape, Reduce, ReduceWindow, and Reverse.
	Dimensions []int64 `protobuf:"varint,14,rep,packed,name=dimensions,proto3" json:"dimensions,omitempty"`
	// Describes the window in a windowed operation such as convolution.
	Window *Window `protobuf:"bytes,15,opt,name=window,proto3" json:"window,omitempty"`
	// Describes the dimension numbers used for a convolution.
	ConvolutionDimensionNumbers *ConvolutionDimensionNumbers `` /* 145-byte string literal not displayed */
	// The number of feature groups. Used for a convolution. Must be a divisor of
	// the input feature dimension and output feature dimension. If not specified,
	// it will use a default value of 1.
	FeatureGroupCount int64                                  `protobuf:"varint,50,opt,name=feature_group_count,json=featureGroupCount,proto3" json:"feature_group_count,omitempty"`
	BatchGroupCount   int64                                  `protobuf:"varint,58,opt,name=batch_group_count,json=batchGroupCount,proto3" json:"batch_group_count,omitempty"`
	SliceDimensions   []*HloInstructionProto_SliceDimensions `protobuf:"bytes,17,rep,name=slice_dimensions,json=sliceDimensions,proto3" json:"slice_dimensions,omitempty"`
	// The bit sizes for a reduce-precision operation.
	ExponentBits int32 `protobuf:"varint,18,opt,name=exponent_bits,json=exponentBits,proto3" json:"exponent_bits,omitempty"`
	MantissaBits int32 `protobuf:"varint,19,opt,name=mantissa_bits,json=mantissaBits,proto3" json:"mantissa_bits,omitempty"`
	// Describes the [start, start + size) range size for a dynamic slice
	// ('start' is specified dynamically in the second operand of the operation).
	DynamicSliceSizes []int64 `protobuf:"varint,20,rep,packed,name=dynamic_slice_sizes,json=dynamicSliceSizes,proto3" json:"dynamic_slice_sizes,omitempty"`
	// The padding configuration that describes the edge padding and interior
	// padding of this pad instruction. Only set for pad instructions.
	PaddingConfig *PaddingConfig `protobuf:"bytes,21,opt,name=padding_config,json=paddingConfig,proto3" json:"padding_config,omitempty"`
	// Outfeed configuration information, only present for kOutfeed.
	OutfeedConfig []byte `protobuf:"bytes,22,opt,name=outfeed_config,json=outfeedConfig,proto3" json:"outfeed_config,omitempty"`
	// The distribution requested for random number generation.
	// Only present for kRng.
	Distribution RandomDistribution `protobuf:"varint,23,opt,name=distribution,proto3,enum=xla.RandomDistribution" json:"distribution,omitempty"`
	// A small float number added to the variance to avoid divide-by-zero error.
	// Only present for kBatchNormTraining, kBatchNormInference, and
	// kBatchNormGrad.
	Epsilon float32 `protobuf:"fixed32,24,opt,name=epsilon,proto3" json:"epsilon,omitempty"`
	// An integer value representing the index of the feature dimension.
	// Only present for kBatchNormTraining, kBatchNormInference, and
	// kBatchNormGrad.
	FeatureIndex int64 `protobuf:"varint,25,opt,name=feature_index,json=featureIndex,proto3" json:"feature_index,omitempty"`
	// Represents a unique identifier for each Send/Recv instruction pair or
	// optionally for collective instructions (AllReduce, CollectivePermute,
	// AllToAll). Non-positive channel_id is equivalent to no channel id.
	ChannelId int64 `protobuf:"varint,26,opt,name=channel_id,json=channelId,proto3" json:"channel_id,omitempty"`
	// The string representation of the infeed configuration.
	InfeedConfig []byte `protobuf:"bytes,27,opt,name=infeed_config,json=infeedConfig,proto3" json:"infeed_config,omitempty"`
	// Name of a external target (eg, global symbol) to call, only present for
	// kCustomCall.
	CustomCallTarget string `protobuf:"bytes,28,opt,name=custom_call_target,json=customCallTarget,proto3" json:"custom_call_target,omitempty"`
	// Shape of outfeed request.
	OutfeedShape *ShapeProto `protobuf:"bytes,29,opt,name=outfeed_shape,json=outfeedShape,proto3" json:"outfeed_shape,omitempty"`
	// Describes the dimension numbers used for a dot operation
	DotDimensionNumbers *DotDimensionNumbers `protobuf:"bytes,30,opt,name=dot_dimension_numbers,json=dotDimensionNumbers,proto3" json:"dot_dimension_numbers,omitempty"`
	// FFT type (FFT, IFFT, etc).
	FftType FftType `protobuf:"varint,31,opt,name=fft_type,json=fftType,proto3,enum=xla.FftType" json:"fft_type,omitempty"`
	// FFT length.
	FftLength []int64 `protobuf:"varint,32,rep,packed,name=fft_length,json=fftLength,proto3" json:"fft_length,omitempty"`
	// Comparison direction only used for kCompare.
	ComparisonDirection string `protobuf:"bytes,63,opt,name=comparison_direction,json=comparisonDirection,proto3" json:"comparison_direction,omitempty"`
	// Gather dimension numbers.
	GatherDimensionNumbers *GatherDimensionNumbers `` /* 130-byte string literal not displayed */
	GatherSliceSizes       []int64                 `protobuf:"varint,34,rep,packed,name=gather_slice_sizes,json=gatherSliceSizes,proto3" json:"gather_slice_sizes,omitempty"`
	// The id of this instruction.
	Id                    int64       `protobuf:"varint,35,opt,name=id,proto3" json:"id,omitempty"`
	OperandIds            []int64     `protobuf:"varint,36,rep,packed,name=operand_ids,json=operandIds,proto3" json:"operand_ids,omitempty"`
	ControlPredecessorIds []int64     `` /* 135-byte string literal not displayed */
	CalledComputationIds  []int64     `` /* 132-byte string literal not displayed */
	Sharding              *OpSharding `protobuf:"bytes,40,opt,name=sharding,proto3" json:"sharding,omitempty"`
	// Backend configuration for the instruction. Has backend-specific meaning.
	BackendConfig []byte `protobuf:"bytes,43,opt,name=backend_config,json=backendConfig,proto3" json:"backend_config,omitempty"`
	// Cross replica op fields.
	ReplicaGroups []*ReplicaGroup `protobuf:"bytes,49,rep,name=replica_groups,json=replicaGroups,proto3" json:"replica_groups,omitempty"`
	// Deprecated, but keeping it for backward compatibility. Use channel_id.
	// Non-positive all_reduce_id is equivalent to no all_reduce_id.
	//
	// Deprecated: Marked as deprecated in xla/service/hlo.proto.
	AllReduceId int64 `protobuf:"varint,45,opt,name=all_reduce_id,json=allReduceId,proto3" json:"all_reduce_id,omitempty"`
	// If true, interprets ids in ReplicaGroup as global device ids, which is
	// a linearized id of `replica_id * partition_count + partition_id`.
	UseGlobalDeviceIds bool `protobuf:"varint,71,opt,name=use_global_device_ids,json=useGlobalDeviceIds,proto3" json:"use_global_device_ids,omitempty"`
	// Whether this Send/Recv instruction transfers data to/from the host. Only
	// present for Send and Recv instructions and their SendDone and RecvDone
	// partners.
	IsHostTransfer bool `protobuf:"varint,47,opt,name=is_host_transfer,json=isHostTransfer,proto3" json:"is_host_transfer,omitempty"`
	// Whether this Sort instruction should be stable.
	IsStable                bool                     `protobuf:"varint,60,opt,name=is_stable,json=isStable,proto3" json:"is_stable,omitempty"`
	ScatterDimensionNumbers *ScatterDimensionNumbers `` /* 133-byte string literal not displayed */
	// Precision configuration for the instruction. Has backend-specific meaning.
	PrecisionConfig *PrecisionConfig `protobuf:"bytes,51,opt,name=precision_config,json=precisionConfig,proto3" json:"precision_config,omitempty"`
	// Collective permute field.
	SourceTargetPairs []*SourceTarget `protobuf:"bytes,52,rep,name=source_target_pairs,json=sourceTargetPairs,proto3" json:"source_target_pairs,omitempty"`
	// Sharding for kDomain instructions.
	DomainEntrySharding *OpSharding `protobuf:"bytes,54,opt,name=domain_entry_sharding,json=domainEntrySharding,proto3" json:"domain_entry_sharding,omitempty"`
	DomainExitSharding  *OpSharding `protobuf:"bytes,55,opt,name=domain_exit_sharding,json=domainExitSharding,proto3" json:"domain_exit_sharding,omitempty"`
	// For custom call this indicates that the layouts are constrained. If
	// constrain_layout is true then the 'shape' field must contain a layout, and
	// 'operand_shapes_with_layout' must contain a shape with layout for each
	// operand.
	ConstrainLayout         bool          `protobuf:"varint,56,opt,name=constrain_layout,json=constrainLayout,proto3" json:"constrain_layout,omitempty"`
	OperandShapesWithLayout []*ShapeProto `` /* 135-byte string literal not displayed */
	// Options for TriangularSolve
	TriangularSolveOptions *TriangularSolveOptions `` /* 130-byte string literal not displayed */
	// Options for Cholesky
	CholeskyOptions *CholeskyOptions `protobuf:"bytes,62,opt,name=cholesky_options,json=choleskyOptions,proto3" json:"cholesky_options,omitempty"`
	// Describes how parameters behave with regards to replicas.
	ParameterReplication *ParameterReplication `protobuf:"bytes,61,opt,name=parameter_replication,json=parameterReplication,proto3" json:"parameter_replication,omitempty"`
	// Whether the kCustomCall instruction has side-effects, only present for
	// kCustomCall.
	CustomCallHasSideEffect bool `` /* 138-byte string literal not displayed */
	// A list of OutputOperandAliasing pairs that specifies aliasing buffers
	// between output and operands for kCustomCall and kFusion.
	OutputOperandAliasing []*OutputOperandAliasing `` /* 127-byte string literal not displayed */
	// Specifies the desired schedule for the custom-call. The field is only
	// present for custom-call.
	CustomCallSchedule CustomCallSchedule `` /* 147-byte string literal not displayed */
	// The delta value for kRngGetAndUpdateState.
	Delta int64 `protobuf:"varint,66,opt,name=delta,proto3" json:"delta,omitempty"`
	// Specifies if the gather/scatter indices are guaranteed to be sorted by the
	// caller.
	IndicesAreSorted bool `protobuf:"varint,67,opt,name=indices_are_sorted,json=indicesAreSorted,proto3" json:"indices_are_sorted,omitempty"`
	// Frontend attributes to pass to the XLA backend.
	FrontendAttributes *FrontendAttributes `protobuf:"bytes,68,opt,name=frontend_attributes,json=frontendAttributes,proto3" json:"frontend_attributes,omitempty"`
	// Specifies if all elements updated are guaranteed to be unique by
	// the caller.
	UniqueIndices bool `protobuf:"varint,69,opt,name=unique_indices,json=uniqueIndices,proto3" json:"unique_indices,omitempty"`
	// RNG algorithm used by kRngBitGenerator.
	RngAlgorithm RandomAlgorithm `protobuf:"varint,70,opt,name=rng_algorithm,json=rngAlgorithm,proto3,enum=xla.RandomAlgorithm" json:"rng_algorithm,omitempty"`
	// The comparison type used for kCompare.
	ComparisonType string `protobuf:"bytes,72,opt,name=comparison_type,json=comparisonType,proto3" json:"comparison_type,omitempty"`
	// Specifies if this is a cross-program-prefetch, used by kCopyStart.
	// Deprecated and replaced by optional_cross_program_prefetch_index.
	//
	// Deprecated: Marked as deprecated in xla/service/hlo.proto.
	IsCrossProgramPrefetch bool `` /* 133-byte string literal not displayed */
	// Specifies the cross-program-prefetch index used by kCopyStart. Uses oneof
	// to emulate the 'optional' keyword for proto3 versions before v3.15.0
	// released 2021/2/18.
	//
	// Types that are assignable to OptionalCrossProgramPrefetchIndex:
	//
	//	*HloInstructionProto_CrossProgramPrefetchIndex
	OptionalCrossProgramPrefetchIndex isHloInstructionProto_OptionalCrossProgramPrefetchIndex `protobuf_oneof:"optional_cross_program_prefetch_index"`
	// If a convolution is dynamic, a dynamic padding type will be specified.
	PaddingType PaddingType `protobuf:"varint,75,opt,name=padding_type,json=paddingType,proto3,enum=xla.PaddingType" json:"padding_type,omitempty"`
	// The API version used by the custom call function. This field is only
	// present for custom-call.
	// TODO(b/189822916): Remove this field when all clients are migrated to the
	// status-returning API.
	CustomCallApiVersion CustomCallApiVersion `` /* 157-byte string literal not displayed */
	// Represents a unique execution thread name for one or more async groups.
	// Each HLO module may contain a main thread and one or more parallel threads.
	// Empty async_execution_thread is equivalent to main thread.
	AsyncExecutionThread string `protobuf:"bytes,79,opt,name=async_execution_thread,json=asyncExecutionThread,proto3" json:"async_execution_thread,omitempty"`
	// Represents the K value for top-k.
	K int64 `protobuf:"varint,81,opt,name=k,proto3" json:"k,omitempty"`
	// Represents the largest flag for top-k.
	Largest bool `protobuf:"varint,85,opt,name=largest,proto3" json:"largest,omitempty"`
	// Represents the information for tracking propagation of values within HLO
	// graph.
	StatisticsViz *StatisticsViz `protobuf:"bytes,82,opt,name=statistics_viz,json=statisticsViz,proto3" json:"statistics_viz,omitempty"`
	// Sparsity descriptor for dot operation.
	DotSparsity []*SparsityDescriptor `protobuf:"bytes,86,rep,name=dot_sparsity,json=dotSparsity,proto3" json:"dot_sparsity,omitempty"`
	// contains filtered or unexported fields
}

Serialization of HloInstruction. Next ID: 87

func (*HloInstructionProto) Descriptor deprecated 

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

Deprecated: Use HloInstructionProto.ProtoReflect.Descriptor instead.

func (*HloInstructionProto) GetAllReduceId deprecated 

func (x *HloInstructionProto) GetAllReduceId() int64

Deprecated: Marked as deprecated in xla/service/hlo.proto.

func (*HloInstructionProto) GetAsyncExecutionThread 

func (x *HloInstructionProto) GetAsyncExecutionThread() string

func (*HloInstructionProto) GetBackendConfig 

func (x *HloInstructionProto) GetBackendConfig() []byte

func (*HloInstructionProto) GetBatchGroupCount 

func (x *HloInstructionProto) GetBatchGroupCount() int64

func (*HloInstructionProto) GetCalledComputationIds 

func (x *HloInstructionProto) GetCalledComputationIds() []int64

func (*HloInstructionProto) GetChannelId 

func (x *HloInstructionProto) GetChannelId() int64

func (*HloInstructionProto) GetCholeskyOptions 

func (x *HloInstructionProto) GetCholeskyOptions() *CholeskyOptions

func (*HloInstructionProto) GetComparisonDirection 

func (x *HloInstructionProto) GetComparisonDirection() string

func (*HloInstructionProto) GetComparisonType 

func (x *HloInstructionProto) GetComparisonType() string

func (*HloInstructionProto) GetConstrainLayout 

func (x *HloInstructionProto) GetConstrainLayout() bool

func (*HloInstructionProto) GetControlPredecessorIds 

func (x *HloInstructionProto) GetControlPredecessorIds() []int64

func (*HloInstructionProto) GetConvolutionDimensionNumbers 

func (x *HloInstructionProto) GetConvolutionDimensionNumbers() *ConvolutionDimensionNumbers

func (*HloInstructionProto) GetCrossProgramPrefetchIndex 

func (x *HloInstructionProto) GetCrossProgramPrefetchIndex() int32

func (*HloInstructionProto) GetCustomCallApiVersion 

func (x *HloInstructionProto) GetCustomCallApiVersion() CustomCallApiVersion

func (*HloInstructionProto) GetCustomCallHasSideEffect 

func (x *HloInstructionProto) GetCustomCallHasSideEffect() bool

func (*HloInstructionProto) GetCustomCallSchedule 

func (x *HloInstructionProto) GetCustomCallSchedule() CustomCallSchedule

func (*HloInstructionProto) GetCustomCallTarget 

func (x *HloInstructionProto) GetCustomCallTarget() string

func (*HloInstructionProto) GetDelta 

func (x *HloInstructionProto) GetDelta() int64

func (*HloInstructionProto) GetDimensions 

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

func (*HloInstructionProto) GetDistribution 

func (x *HloInstructionProto) GetDistribution() RandomDistribution

func (*HloInstructionProto) GetDomainEntrySharding 

func (x *HloInstructionProto) GetDomainEntrySharding() *OpSharding

func (*HloInstructionProto) GetDomainExitSharding 

func (x *HloInstructionProto) GetDomainExitSharding() *OpSharding

func (*HloInstructionProto) GetDotDimensionNumbers 

func (x *HloInstructionProto) GetDotDimensionNumbers() *DotDimensionNumbers

func (*HloInstructionProto) GetDotSparsity 

func (x *HloInstructionProto) GetDotSparsity() []*SparsityDescriptor

func (*HloInstructionProto) GetDynamicSliceSizes 

func (x *HloInstructionProto) GetDynamicSliceSizes() []int64

func (*HloInstructionProto) GetEpsilon 

func (x *HloInstructionProto) GetEpsilon() float32

func (*HloInstructionProto) GetExponentBits 

func (x *HloInstructionProto) GetExponentBits() int32

func (*HloInstructionProto) GetFeatureGroupCount 

func (x *HloInstructionProto) GetFeatureGroupCount() int64

func (*HloInstructionProto) GetFeatureIndex 

func (x *HloInstructionProto) GetFeatureIndex() int64

func (*HloInstructionProto) GetFftLength 

func (x *HloInstructionProto) GetFftLength() []int64

func (*HloInstructionProto) GetFftType 

func (x *HloInstructionProto) GetFftType() FftType

func (*HloInstructionProto) GetFrontendAttributes 

func (x *HloInstructionProto) GetFrontendAttributes() *FrontendAttributes

func (*HloInstructionProto) GetFusionKind 

func (x *HloInstructionProto) GetFusionKind() string

func (*HloInstructionProto) GetGatherDimensionNumbers 

func (x *HloInstructionProto) GetGatherDimensionNumbers() *GatherDimensionNumbers

func (*HloInstructionProto) GetGatherSliceSizes 

func (x *HloInstructionProto) GetGatherSliceSizes() []int64

func (*HloInstructionProto) GetId 

func (x *HloInstructionProto) GetId() int64

func (*HloInstructionProto) GetIndicesAreSorted 

func (x *HloInstructionProto) GetIndicesAreSorted() bool

func (*HloInstructionProto) GetInfeedConfig 

func (x *HloInstructionProto) GetInfeedConfig() []byte

func (*HloInstructionProto) GetIsCrossProgramPrefetch deprecated 

func (x *HloInstructionProto) GetIsCrossProgramPrefetch() bool

Deprecated: Marked as deprecated in xla/service/hlo.proto.

func (*HloInstructionProto) GetIsHostTransfer 

func (x *HloInstructionProto) GetIsHostTransfer() bool

func (*HloInstructionProto) GetIsStable 

func (x *HloInstructionProto) GetIsStable() bool

func (*HloInstructionProto) GetK 

func (x *HloInstructionProto) GetK() int64

func (*HloInstructionProto) GetLargest 

func (x *HloInstructionProto) GetLargest() bool

func (*HloInstructionProto) GetLiteral 

func (x *HloInstructionProto) GetLiteral() *LiteralProto

func (*HloInstructionProto) GetMantissaBits 

func (x *HloInstructionProto) GetMantissaBits() int32

func (*HloInstructionProto) GetMetadata 

func (x *HloInstructionProto) GetMetadata() *OpMetadata

func (*HloInstructionProto) GetName 

func (x *HloInstructionProto) GetName() string

func (*HloInstructionProto) GetOpcode 

func (x *HloInstructionProto) GetOpcode() string

func (*HloInstructionProto) GetOperandIds 

func (x *HloInstructionProto) GetOperandIds() []int64

func (*HloInstructionProto) GetOperandShapesWithLayout 

func (x *HloInstructionProto) GetOperandShapesWithLayout() []*ShapeProto

func (*HloInstructionProto) GetOptionalCrossProgramPrefetchIndex 

func (m *HloInstructionProto) GetOptionalCrossProgramPrefetchIndex() isHloInstructionProto_OptionalCrossProgramPrefetchIndex

func (*HloInstructionProto) GetOutfeedConfig 

func (x *HloInstructionProto) GetOutfeedConfig() []byte

func (*HloInstructionProto) GetOutfeedShape 

func (x *HloInstructionProto) GetOutfeedShape() *ShapeProto

func (*HloInstructionProto) GetOutputOperandAliasing 

func (x *HloInstructionProto) GetOutputOperandAliasing() []*OutputOperandAliasing

func (*HloInstructionProto) GetPaddingConfig 

func (x *HloInstructionProto) GetPaddingConfig() *PaddingConfig

func (*HloInstructionProto) GetPaddingType 

func (x *HloInstructionProto) GetPaddingType() PaddingType

func (*HloInstructionProto) GetParameterNumber 

func (x *HloInstructionProto) GetParameterNumber() int64

func (*HloInstructionProto) GetParameterReplication 

func (x *HloInstructionProto) GetParameterReplication() *ParameterReplication

func (*HloInstructionProto) GetPrecisionConfig 

func (x *HloInstructionProto) GetPrecisionConfig() *PrecisionConfig

func (*HloInstructionProto) GetReplicaGroups 

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

func (*HloInstructionProto) GetRngAlgorithm 

func (x *HloInstructionProto) GetRngAlgorithm() RandomAlgorithm

func (*HloInstructionProto) GetScatterDimensionNumbers 

func (x *HloInstructionProto) GetScatterDimensionNumbers() *ScatterDimensionNumbers

func (*HloInstructionProto) GetShape 

func (x *HloInstructionProto) GetShape() *ShapeProto

func (*HloInstructionProto) GetSharding 

func (x *HloInstructionProto) GetSharding() *OpSharding

func (*HloInstructionProto) GetSliceDimensions 

func (x *HloInstructionProto) GetSliceDimensions() []*HloInstructionProto_SliceDimensions

func (*HloInstructionProto) GetSourceTargetPairs 

func (x *HloInstructionProto) GetSourceTargetPairs() []*SourceTarget

func (*HloInstructionProto) GetStatisticsViz 

func (x *HloInstructionProto) GetStatisticsViz() *StatisticsViz

func (*HloInstructionProto) GetTriangularSolveOptions 

func (x *HloInstructionProto) GetTriangularSolveOptions() *TriangularSolveOptions

func (*HloInstructionProto) GetTupleIndex 

func (x *HloInstructionProto) GetTupleIndex() int64

func (*HloInstructionProto) GetUniqueIndices 

func (x *HloInstructionProto) GetUniqueIndices() bool

func (*HloInstructionProto) GetUseGlobalDeviceIds 

func (x *HloInstructionProto) GetUseGlobalDeviceIds() bool

func (*HloInstructionProto) GetWindow 

func (x *HloInstructionProto) GetWindow() *Window

func (*HloInstructionProto) ProtoMessage 

func (*HloInstructionProto) ProtoMessage()

func (*HloInstructionProto) ProtoReflect 

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

func (*HloInstructionProto) Reset 

func (x *HloInstructionProto) Reset()

func (*HloInstructionProto) String 

func (x *HloInstructionProto) String() string

type HloInstructionProto_CrossProgramPrefetchIndex 

type HloInstructionProto_CrossProgramPrefetchIndex struct {
	CrossProgramPrefetchIndex int32 `protobuf:"varint,80,opt,name=cross_program_prefetch_index,json=crossProgramPrefetchIndex,proto3,oneof"`
}

type HloInstructionProto_SliceDimensions 

type HloInstructionProto_SliceDimensions struct {
	Start  int64 `protobuf:"varint,1,opt,name=start,proto3" json:"start,omitempty"`
	Limit  int64 `protobuf:"varint,2,opt,name=limit,proto3" json:"limit,omitempty"`
	Stride int64 `protobuf:"varint,3,opt,name=stride,proto3" json:"stride,omitempty"`
	// contains filtered or unexported fields
}

Describes the [begin, end) index range and stride for slices.

func (*HloInstructionProto_SliceDimensions) Descriptor deprecated 

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

Deprecated: Use HloInstructionProto_SliceDimensions.ProtoReflect.Descriptor instead.

func (*HloInstructionProto_SliceDimensions) GetLimit 

func (x *HloInstructionProto_SliceDimensions) GetLimit() int64

func (*HloInstructionProto_SliceDimensions) GetStart 

func (x *HloInstructionProto_SliceDimensions) GetStart() int64

func (*HloInstructionProto_SliceDimensions) GetStride 

func (x *HloInstructionProto_SliceDimensions) GetStride() int64

func (*HloInstructionProto_SliceDimensions) ProtoMessage 

func (*HloInstructionProto_SliceDimensions) ProtoMessage()

func (*HloInstructionProto_SliceDimensions) ProtoReflect 

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

func (*HloInstructionProto_SliceDimensions) Reset 

func (x *HloInstructionProto_SliceDimensions) Reset()

func (*HloInstructionProto_SliceDimensions) String 

func (x *HloInstructionProto_SliceDimensions) String() string

type HloModuleConfigProto 

type HloModuleConfigProto struct {
	EntryComputationLayout                   *ProgramShapeProto                          `` /* 129-byte string literal not displayed */
	Seed                                     uint64                                      `protobuf:"varint,2,opt,name=seed,proto3" json:"seed,omitempty"`
	LaunchId                                 int32                                       `protobuf:"varint,3,opt,name=launch_id,json=launchId,proto3" json:"launch_id,omitempty"`
	ReplicaCount                             int64                                       `protobuf:"varint,4,opt,name=replica_count,json=replicaCount,proto3" json:"replica_count,omitempty"`
	NumPartitions                            int64                                       `protobuf:"varint,5,opt,name=num_partitions,json=numPartitions,proto3" json:"num_partitions,omitempty"`
	ParamRequiresBroadcastViaCollectives     []bool                                      `` /* 183-byte string literal not displayed */
	UseSpmdPartitioning                      bool                                        `protobuf:"varint,7,opt,name=use_spmd_partitioning,json=useSpmdPartitioning,proto3" json:"use_spmd_partitioning,omitempty"`
	UseAutoSpmdPartitioning                  bool                                        `` /* 135-byte string literal not displayed */
	AutoSpmdPartitioningMeshShape            []int64                                     `` /* 162-byte string literal not displayed */
	AutoSpmdPartitioningMeshIds              []int64                                     `` /* 157-byte string literal not displayed */
	DeduplicateHlo                           bool                                        `protobuf:"varint,11,opt,name=deduplicate_hlo,json=deduplicateHlo,proto3" json:"deduplicate_hlo,omitempty"`
	IntraOpParallelismThreads                int64                                       `` /* 142-byte string literal not displayed */
	DeviceType                               string                                      `protobuf:"bytes,13,opt,name=device_type,json=deviceType,proto3" json:"device_type,omitempty"`
	DebugOptions                             *DebugOptions                               `protobuf:"bytes,14,opt,name=debug_options,json=debugOptions,proto3" json:"debug_options,omitempty"`
	StaticDeviceAssignment                   *DeviceAssignmentProto                      `` /* 130-byte string literal not displayed */
	AllowSeparateShardingPrograms            bool                                        `` /* 154-byte string literal not displayed */
	ShardableValueUpdatePairs                []*ShardableValueUpdatePairProto            `` /* 141-byte string literal not displayed */
	AliasPassthroughParams                   bool                                        `` /* 131-byte string literal not displayed */
	ContentAwareComputationSorting           bool                                        `` /* 157-byte string literal not displayed */
	FusionConfigCollection                   HloModuleConfigProto_FusionConfigCollection `` /* 184-byte string literal not displayed */
	FusionConfig                             []*HloModuleConfigProto_BoolList            `protobuf:"bytes,20,rep,name=fusion_config,json=fusionConfig,proto3" json:"fusion_config,omitempty"`
	DotConfig                                map[string]*HloModuleConfigProto_Int64List  `` /* 177-byte string literal not displayed */
	LayoutConfig                             []*HloModuleConfigProto_Int64ListList       `protobuf:"bytes,22,rep,name=layout_config,json=layoutConfig,proto3" json:"layout_config,omitempty"`
	MemorySpaceAssignmentConfig              []uint64                                    `` /* 155-byte string literal not displayed */
	PhaseOrderingConfig                      []*HloModuleConfigProto_BoolList            `protobuf:"bytes,24,rep,name=phase_ordering_config,json=phaseOrderingConfig,proto3" json:"phase_ordering_config,omitempty"`
	PhaseIndex                               int32                                       `protobuf:"varint,25,opt,name=phase_index,json=phaseIndex,proto3" json:"phase_index,omitempty"`
	AllowSpmdShardingPropagationToParameters []bool                                      `` /* 198-byte string literal not displayed */
	AllowSpmdShardingPropagationToOutput     []bool                                      `` /* 186-byte string literal not displayed */
	AnalysisAllowanceMap                     map[string]int64                            `` /* 213-byte string literal not displayed */
	MatrixUnitOperandPrecision               PrecisionConfig_Precision                   `` /* 180-byte string literal not displayed */
	FdoProfile                               []byte                                      `protobuf:"bytes,31,opt,name=fdo_profile,json=fdoProfile,proto3" json:"fdo_profile,omitempty"`
	DeviceMemorySize                         int64                                       `protobuf:"varint,32,opt,name=device_memory_size,json=deviceMemorySize,proto3" json:"device_memory_size,omitempty"`
	// contains filtered or unexported fields
}

Serialization of HloModuleConfig. See the C++ class definition for descriptions of each field. There are no guarantees of backwards or forwards compatibility. Next id: 34.

func (*HloModuleConfigProto) Descriptor deprecated 

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

Deprecated: Use HloModuleConfigProto.ProtoReflect.Descriptor instead.

func (*HloModuleConfigProto) GetAliasPassthroughParams 

func (x *HloModuleConfigProto) GetAliasPassthroughParams() bool

func (*HloModuleConfigProto) GetAllowSeparateShardingPrograms 

func (x *HloModuleConfigProto) GetAllowSeparateShardingPrograms() bool

func (*HloModuleConfigProto) GetAllowSpmdShardingPropagationToOutput 

func (x *HloModuleConfigProto) GetAllowSpmdShardingPropagationToOutput() []bool

func (*HloModuleConfigProto) GetAllowSpmdShardingPropagationToParameters 

func (x *HloModuleConfigProto) GetAllowSpmdShardingPropagationToParameters() []bool

func (*HloModuleConfigProto) GetAnalysisAllowanceMap 

func (x *HloModuleConfigProto) GetAnalysisAllowanceMap() map[string]int64

func (*HloModuleConfigProto) GetAutoSpmdPartitioningMeshIds 

func (x *HloModuleConfigProto) GetAutoSpmdPartitioningMeshIds() []int64

func (*HloModuleConfigProto) GetAutoSpmdPartitioningMeshShape 

func (x *HloModuleConfigProto) GetAutoSpmdPartitioningMeshShape() []int64

func (*HloModuleConfigProto) GetContentAwareComputationSorting 

func (x *HloModuleConfigProto) GetContentAwareComputationSorting() bool

func (*HloModuleConfigProto) GetDebugOptions 

func (x *HloModuleConfigProto) GetDebugOptions() *DebugOptions

func (*HloModuleConfigProto) GetDeduplicateHlo 

func (x *HloModuleConfigProto) GetDeduplicateHlo() bool

func (*HloModuleConfigProto) GetDeviceMemorySize 

func (x *HloModuleConfigProto) GetDeviceMemorySize() int64

func (*HloModuleConfigProto) GetDeviceType 

func (x *HloModuleConfigProto) GetDeviceType() string

func (*HloModuleConfigProto) GetDotConfig 

func (x *HloModuleConfigProto) GetDotConfig() map[string]*HloModuleConfigProto_Int64List

func (*HloModuleConfigProto) GetEntryComputationLayout 

func (x *HloModuleConfigProto) GetEntryComputationLayout() *ProgramShapeProto

func (*HloModuleConfigProto) GetFdoProfile 

func (x *HloModuleConfigProto) GetFdoProfile() []byte

func (*HloModuleConfigProto) GetFusionConfig 

func (x *HloModuleConfigProto) GetFusionConfig() []*HloModuleConfigProto_BoolList

func (*HloModuleConfigProto) GetFusionConfigCollection 

func (x *HloModuleConfigProto) GetFusionConfigCollection() HloModuleConfigProto_FusionConfigCollection

func (*HloModuleConfigProto) GetIntraOpParallelismThreads 

func (x *HloModuleConfigProto) GetIntraOpParallelismThreads() int64

func (*HloModuleConfigProto) GetLaunchId 

func (x *HloModuleConfigProto) GetLaunchId() int32

func (*HloModuleConfigProto) GetLayoutConfig 

func (x *HloModuleConfigProto) GetLayoutConfig() []*HloModuleConfigProto_Int64ListList

func (*HloModuleConfigProto) GetMatrixUnitOperandPrecision 

func (x *HloModuleConfigProto) GetMatrixUnitOperandPrecision() PrecisionConfig_Precision

func (*HloModuleConfigProto) GetMemorySpaceAssignmentConfig 

func (x *HloModuleConfigProto) GetMemorySpaceAssignmentConfig() []uint64

func (*HloModuleConfigProto) GetNumPartitions 

func (x *HloModuleConfigProto) GetNumPartitions() int64

func (*HloModuleConfigProto) GetParamRequiresBroadcastViaCollectives 

func (x *HloModuleConfigProto) GetParamRequiresBroadcastViaCollectives() []bool

func (*HloModuleConfigProto) GetPhaseIndex 

func (x *HloModuleConfigProto) GetPhaseIndex() int32

func (*HloModuleConfigProto) GetPhaseOrderingConfig 

func (x *HloModuleConfigProto) GetPhaseOrderingConfig() []*HloModuleConfigProto_BoolList

func (*HloModuleConfigProto) GetReplicaCount 

func (x *HloModuleConfigProto) GetReplicaCount() int64

func (*HloModuleConfigProto) GetSeed 

func (x *HloModuleConfigProto) GetSeed() uint64

func (*HloModuleConfigProto) GetShardableValueUpdatePairs 

func (x *HloModuleConfigProto) GetShardableValueUpdatePairs() []*ShardableValueUpdatePairProto

func (*HloModuleConfigProto) GetStaticDeviceAssignment 

func (x *HloModuleConfigProto) GetStaticDeviceAssignment() *DeviceAssignmentProto

func (*HloModuleConfigProto) GetUseAutoSpmdPartitioning 

func (x *HloModuleConfigProto) GetUseAutoSpmdPartitioning() bool

func (*HloModuleConfigProto) GetUseSpmdPartitioning 

func (x *HloModuleConfigProto) GetUseSpmdPartitioning() bool

func (*HloModuleConfigProto) ProtoMessage 

func (*HloModuleConfigProto) ProtoMessage()

func (*HloModuleConfigProto) ProtoReflect 

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

func (*HloModuleConfigProto) Reset 

func (x *HloModuleConfigProto) Reset()

func (*HloModuleConfigProto) String 

func (x *HloModuleConfigProto) String() string

type HloModuleConfigProto_BoolList 

type HloModuleConfigProto_BoolList struct {
	Vals []bool `protobuf:"varint,1,rep,packed,name=vals,proto3" json:"vals,omitempty"`
	// contains filtered or unexported fields
}

func (*HloModuleConfigProto_BoolList) Descriptor deprecated 

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

Deprecated: Use HloModuleConfigProto_BoolList.ProtoReflect.Descriptor instead.

func (*HloModuleConfigProto_BoolList) GetVals 

func (x *HloModuleConfigProto_BoolList) GetVals() []bool

func (*HloModuleConfigProto_BoolList) ProtoMessage 

func (*HloModuleConfigProto_BoolList) ProtoMessage()

func (*HloModuleConfigProto_BoolList) ProtoReflect 

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

func (*HloModuleConfigProto_BoolList) Reset 

func (x *HloModuleConfigProto_BoolList) Reset()

func (*HloModuleConfigProto_BoolList) String 

func (x *HloModuleConfigProto_BoolList) String() string

type HloModuleConfigProto_FusionConfigCollection 

type HloModuleConfigProto_FusionConfigCollection int32
const (
	HloModuleConfigProto_OFF      HloModuleConfigProto_FusionConfigCollection = 0 // Do not collect configuration.
	HloModuleConfigProto_PER_EDGE HloModuleConfigProto_FusionConfigCollection = 1 // Collect per-edge configuration.
	HloModuleConfigProto_PER_NODE HloModuleConfigProto_FusionConfigCollection = 2 // Collect per-node configuration.
)

func (HloModuleConfigProto_FusionConfigCollection) Descriptor 

func (HloModuleConfigProto_FusionConfigCollection) Descriptor() protoreflect.EnumDescriptor

func (HloModuleConfigProto_FusionConfigCollection) Enum 

func (x HloModuleConfigProto_FusionConfigCollection) Enum() *HloModuleConfigProto_FusionConfigCollection

func (HloModuleConfigProto_FusionConfigCollection) EnumDescriptor deprecated 

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

Deprecated: Use HloModuleConfigProto_FusionConfigCollection.Descriptor instead.

func (HloModuleConfigProto_FusionConfigCollection) Number 

func (x HloModuleConfigProto_FusionConfigCollection) Number() protoreflect.EnumNumber

func (HloModuleConfigProto_FusionConfigCollection) String 

func (x HloModuleConfigProto_FusionConfigCollection) String() string

func (HloModuleConfigProto_FusionConfigCollection) Type 

func (HloModuleConfigProto_FusionConfigCollection) Type() protoreflect.EnumType

type HloModuleConfigProto_Int64List 

type HloModuleConfigProto_Int64List struct {
	Vals []int64 `protobuf:"varint,1,rep,packed,name=vals,proto3" json:"vals,omitempty"`
	// contains filtered or unexported fields
}

func (*HloModuleConfigProto_Int64List) Descriptor deprecated 

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

Deprecated: Use HloModuleConfigProto_Int64List.ProtoReflect.Descriptor instead.

func (*HloModuleConfigProto_Int64List) GetVals 

func (x *HloModuleConfigProto_Int64List) GetVals() []int64

func (*HloModuleConfigProto_Int64List) ProtoMessage 

func (*HloModuleConfigProto_Int64List) ProtoMessage()

func (*HloModuleConfigProto_Int64List) ProtoReflect 

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

func (*HloModuleConfigProto_Int64List) Reset 

func (x *HloModuleConfigProto_Int64List) Reset()

func (*HloModuleConfigProto_Int64List) String 

func (x *HloModuleConfigProto_Int64List) String() string

type HloModuleConfigProto_Int64ListList 

type HloModuleConfigProto_Int64ListList struct {
	Lists []*HloModuleConfigProto_Int64List `protobuf:"bytes,1,rep,name=lists,proto3" json:"lists,omitempty"`
	// contains filtered or unexported fields
}

func (*HloModuleConfigProto_Int64ListList) Descriptor deprecated 

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

Deprecated: Use HloModuleConfigProto_Int64ListList.ProtoReflect.Descriptor instead.

func (*HloModuleConfigProto_Int64ListList) GetLists 

func (x *HloModuleConfigProto_Int64ListList) GetLists() []*HloModuleConfigProto_Int64List

func (*HloModuleConfigProto_Int64ListList) ProtoMessage 

func (*HloModuleConfigProto_Int64ListList) ProtoMessage()

func (*HloModuleConfigProto_Int64ListList) ProtoReflect 

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

func (*HloModuleConfigProto_Int64ListList) Reset 

func (x *HloModuleConfigProto_Int64ListList) Reset()

func (*HloModuleConfigProto_Int64ListList) String 

func (x *HloModuleConfigProto_Int64ListList) String() string

type HloModuleGroupProto 

type HloModuleGroupProto struct {
	Name       string            `protobuf:"bytes,1,opt,name=name,proto3" json:"name,omitempty"`
	HloModules []*HloModuleProto `protobuf:"bytes,2,rep,name=hlo_modules,json=hloModules,proto3" json:"hlo_modules,omitempty"`
	// contains filtered or unexported fields
}

An abstraction representing a set of HLO module built to run concurrently across different devices.

func (*HloModuleGroupProto) Descriptor deprecated 

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

Deprecated: Use HloModuleGroupProto.ProtoReflect.Descriptor instead.

func (*HloModuleGroupProto) GetHloModules 

func (x *HloModuleGroupProto) GetHloModules() []*HloModuleProto

func (*HloModuleGroupProto) GetName 

func (x *HloModuleGroupProto) GetName() string

func (*HloModuleGroupProto) ProtoMessage 

func (*HloModuleGroupProto) ProtoMessage()

func (*HloModuleGroupProto) ProtoReflect 

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

func (*HloModuleGroupProto) Reset 

func (x *HloModuleGroupProto) Reset()

func (*HloModuleGroupProto) String 

func (x *HloModuleGroupProto) String() string

type HloModuleMetadataProto 

type HloModuleMetadataProto struct {

	// Uniquely identifies an HloModuleMetadata. Equal to the first unique_id
	// of the module (a module may go through multiple unique_ids). If a module
	// is partitioned into multiple modules, those modules will each have a new
	// HloModuleMetadata with a different canonical_module_id.
	CanonicalModuleId int64 `protobuf:"varint,1,opt,name=canonical_module_id,json=canonicalModuleId,proto3" json:"canonical_module_id,omitempty"`
	// Name of the module group that the module is part of.
	ModuleGroupName string `protobuf:"bytes,2,opt,name=module_group_name,json=moduleGroupName,proto3" json:"module_group_name,omitempty"`
	// The canonical module id of the module that this one is partitioned from,
	// if applicable.
	OriginalModuleId int64 `protobuf:"varint,3,opt,name=original_module_id,json=originalModuleId,proto3" json:"original_module_id,omitempty"`
	// The canonical module ids of the modules that this one is partitioned into,
	// if applicable.
	PartitionedModuleIds []int64 `` /* 131-byte string literal not displayed */
	// Metadata for the HLO passes that are run on the module.
	PassMetadata []*HloPassMetadata `protobuf:"bytes,5,rep,name=pass_metadata,json=passMetadata,proto3" json:"pass_metadata,omitempty"`
	// contains filtered or unexported fields
}

Metadata for an HLO module. Dumped after HLO passes and before LLO lowering with filename module_####.metadata.textproto, where #### is canonical_module_id.

func (*HloModuleMetadataProto) Descriptor deprecated 

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

Deprecated: Use HloModuleMetadataProto.ProtoReflect.Descriptor instead.

func (*HloModuleMetadataProto) GetCanonicalModuleId 

func (x *HloModuleMetadataProto) GetCanonicalModuleId() int64

func (*HloModuleMetadataProto) GetModuleGroupName 

func (x *HloModuleMetadataProto) GetModuleGroupName() string

func (*HloModuleMetadataProto) GetOriginalModuleId 

func (x *HloModuleMetadataProto) GetOriginalModuleId() int64

func (*HloModuleMetadataProto) GetPartitionedModuleIds 

func (x *HloModuleMetadataProto) GetPartitionedModuleIds() []int64

func (*HloModuleMetadataProto) GetPassMetadata 

func (x *HloModuleMetadataProto) GetPassMetadata() []*HloPassMetadata

func (*HloModuleMetadataProto) ProtoMessage 

func (*HloModuleMetadataProto) ProtoMessage()

func (*HloModuleMetadataProto) ProtoReflect 

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

func (*HloModuleMetadataProto) Reset 

func (x *HloModuleMetadataProto) Reset()

func (*HloModuleMetadataProto) String 

func (x *HloModuleMetadataProto) String() string

type HloModuleProto 

type HloModuleProto struct {
	Name                 string `protobuf:"bytes,1,opt,name=name,proto3" json:"name,omitempty"`
	EntryComputationName string `protobuf:"bytes,2,opt,name=entry_computation_name,json=entryComputationName,proto3" json:"entry_computation_name,omitempty"`
	EntryComputationId   int64  `protobuf:"varint,6,opt,name=entry_computation_id,json=entryComputationId,proto3" json:"entry_computation_id,omitempty"`
	// The array of computations is always in a valid dependency order, where
	// callees appear before their callers.
	Computations []*HloComputationProto `protobuf:"bytes,3,rep,name=computations,proto3" json:"computations,omitempty"`
	// The host program shape (with layout) of the entry computation.
	HostProgramShape *ProgramShapeProto `protobuf:"bytes,4,opt,name=host_program_shape,json=hostProgramShape,proto3" json:"host_program_shape,omitempty"`
	// The id of this module.
	Id int64 `protobuf:"varint,5,opt,name=id,proto3" json:"id,omitempty"`
	// The schedule for this module.
	Schedule *HloScheduleProto `protobuf:"bytes,7,opt,name=schedule,proto3" json:"schedule,omitempty"`
	// Describes alias information between inputs and outputs.
	InputOutputAlias *HloInputOutputAliasProto `protobuf:"bytes,8,opt,name=input_output_alias,json=inputOutputAlias,proto3" json:"input_output_alias,omitempty"`
	// Describes the information of input buffer donors.
	BufferDonor            *HloBufferDonorProto    `protobuf:"bytes,18,opt,name=buffer_donor,json=bufferDonor,proto3" json:"buffer_donor,omitempty"`
	CrossProgramPrefetches []*CrossProgramPrefetch `` /* 130-byte string literal not displayed */
	// True if the module contains dynamic computation.
	IsDynamic               bool          `protobuf:"varint,11,opt,name=is_dynamic,json=isDynamic,proto3" json:"is_dynamic,omitempty"`
	SpmdOutputSharding      *OpSharding   `protobuf:"bytes,12,opt,name=spmd_output_sharding,json=spmdOutputSharding,proto3" json:"spmd_output_sharding,omitempty"`
	SpmdParametersShardings []*OpSharding `` /* 133-byte string literal not displayed */
	// Uses AutoSharding pass or not.
	UseAutoSpmdPartitioning bool `` /* 136-byte string literal not displayed */
	// Profile information for the HLO module.
	ProfileInfo []*HloModuleProto_ProfileInfo `protobuf:"bytes,13,rep,name=profile_info,json=profileInfo,proto3" json:"profile_info,omitempty"`
	// DeviceAssignment object information.
	DeviceAssignment *DeviceAssignmentProto `protobuf:"bytes,15,opt,name=device_assignment,json=deviceAssignment,proto3" json:"device_assignment,omitempty"`
	// Stack frames index.
	StackFrameIndex *StackFrameIndexProto `protobuf:"bytes,17,opt,name=stack_frame_index,json=stackFrameIndex,proto3" json:"stack_frame_index,omitempty"`
	// Frontend attributes to pass to the XLA backend.
	FrontendAttributes *FrontendAttributes `protobuf:"bytes,19,opt,name=frontend_attributes,json=frontendAttributes,proto3" json:"frontend_attributes,omitempty"`
	// contains filtered or unexported fields
}

Serialization of HloModule.

func (*HloModuleProto) Descriptor deprecated 

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

Deprecated: Use HloModuleProto.ProtoReflect.Descriptor instead.

func (*HloModuleProto) GetBufferDonor 

func (x *HloModuleProto) GetBufferDonor() *HloBufferDonorProto

func (*HloModuleProto) GetComputations 

func (x *HloModuleProto) GetComputations() []*HloComputationProto

func (*HloModuleProto) GetCrossProgramPrefetches 

func (x *HloModuleProto) GetCrossProgramPrefetches() []*CrossProgramPrefetch

func (*HloModuleProto) GetDeviceAssignment 

func (x *HloModuleProto) GetDeviceAssignment() *DeviceAssignmentProto

func (*HloModuleProto) GetEntryComputationId 

func (x *HloModuleProto) GetEntryComputationId() int64

func (*HloModuleProto) GetEntryComputationName 

func (x *HloModuleProto) GetEntryComputationName() string

func (*HloModuleProto) GetFrontendAttributes 

func (x *HloModuleProto) GetFrontendAttributes() *FrontendAttributes

func (*HloModuleProto) GetHostProgramShape 

func (x *HloModuleProto) GetHostProgramShape() *ProgramShapeProto

func (*HloModuleProto) GetId 

func (x *HloModuleProto) GetId() int64

func (*HloModuleProto) GetInputOutputAlias 

func (x *HloModuleProto) GetInputOutputAlias() *HloInputOutputAliasProto

func (*HloModuleProto) GetIsDynamic 

func (x *HloModuleProto) GetIsDynamic() bool

func (*HloModuleProto) GetName 

func (x *HloModuleProto) GetName() string

func (*HloModuleProto) GetProfileInfo 

func (x *HloModuleProto) GetProfileInfo() []*HloModuleProto_ProfileInfo

func (*HloModuleProto) GetSchedule 

func (x *HloModuleProto) GetSchedule() *HloScheduleProto

func (*HloModuleProto) GetSpmdOutputSharding 

func (x *HloModuleProto) GetSpmdOutputSharding() *OpSharding

func (*HloModuleProto) GetSpmdParametersShardings 

func (x *HloModuleProto) GetSpmdParametersShardings() []*OpSharding

func (*HloModuleProto) GetStackFrameIndex 

func (x *HloModuleProto) GetStackFrameIndex() *StackFrameIndexProto

func (*HloModuleProto) GetUseAutoSpmdPartitioning 

func (x *HloModuleProto) GetUseAutoSpmdPartitioning() bool

func (*HloModuleProto) ProtoMessage 

func (*HloModuleProto) ProtoMessage()

func (*HloModuleProto) ProtoReflect 

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

func (*HloModuleProto) Reset 

func (x *HloModuleProto) Reset()

func (*HloModuleProto) String 

func (x *HloModuleProto) String() string

type HloModuleProtoWithConfig 

type HloModuleProtoWithConfig struct {
	HloModule *HloModuleProto       `protobuf:"bytes,1,opt,name=hlo_module,json=hloModule,proto3" json:"hlo_module,omitempty"`
	Config    *HloModuleConfigProto `protobuf:"bytes,2,opt,name=config,proto3" json:"config,omitempty"`
	// contains filtered or unexported fields
}

func (*HloModuleProtoWithConfig) Descriptor deprecated 

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

Deprecated: Use HloModuleProtoWithConfig.ProtoReflect.Descriptor instead.

func (*HloModuleProtoWithConfig) GetConfig 

func (x *HloModuleProtoWithConfig) GetConfig() *HloModuleConfigProto

func (*HloModuleProtoWithConfig) GetHloModule 

func (x *HloModuleProtoWithConfig) GetHloModule() *HloModuleProto

func (*HloModuleProtoWithConfig) ProtoMessage 

func (*HloModuleProtoWithConfig) ProtoMessage()

func (*HloModuleProtoWithConfig) ProtoReflect 

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

func (*HloModuleProtoWithConfig) Reset 

func (x *HloModuleProtoWithConfig) Reset()

func (*HloModuleProtoWithConfig) String 

func (x *HloModuleProtoWithConfig) String() string

type HloModuleProto_ProfileInfo 

type HloModuleProto_ProfileInfo struct {

	// The optimization profiles that this module contains.
	ProfileType HloModuleProto_ProfileType `` /* 131-byte string literal not displayed */
	// Speedup of tuned config compared to default config.
	RelativeSpeedup float64 `protobuf:"fixed64,2,opt,name=relative_speedup,json=relativeSpeedup,proto3" json:"relative_speedup,omitempty"`
	// The source of the optimization profile that this module 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 profile.
	CompilationEvent CompilationEvent `` /* 136-byte string literal not displayed */
	// The fingerprint of the unoptimized module this profile was applied to.
	Fingerprint string `protobuf:"bytes,5,opt,name=fingerprint,proto3" json:"fingerprint,omitempty"`
	// contains filtered or unexported fields
}

Information about the optimization profile that this module contains.

func (*HloModuleProto_ProfileInfo) Descriptor deprecated 

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

Deprecated: Use HloModuleProto_ProfileInfo.ProtoReflect.Descriptor instead.

func (*HloModuleProto_ProfileInfo) GetCompilationEvent 

func (x *HloModuleProto_ProfileInfo) GetCompilationEvent() CompilationEvent

func (*HloModuleProto_ProfileInfo) GetFingerprint 

func (x *HloModuleProto_ProfileInfo) GetFingerprint() string

func (*HloModuleProto_ProfileInfo) GetProfileSource 

func (x *HloModuleProto_ProfileInfo) GetProfileSource() ProfileSource

func (*HloModuleProto_ProfileInfo) GetProfileType 

func (x *HloModuleProto_ProfileInfo) GetProfileType() HloModuleProto_ProfileType

func (*HloModuleProto_ProfileInfo) GetRelativeSpeedup 

func (x *HloModuleProto_ProfileInfo) GetRelativeSpeedup() float64

func (*HloModuleProto_ProfileInfo) ProtoMessage 

func (*HloModuleProto_ProfileInfo) ProtoMessage()

func (*HloModuleProto_ProfileInfo) ProtoReflect 

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

func (*HloModuleProto_ProfileInfo) Reset 

func (x *HloModuleProto_ProfileInfo) Reset()

func (*HloModuleProto_ProfileInfo) String 

func (x *HloModuleProto_ProfileInfo) String() string

type HloModuleProto_ProfileType 

type HloModuleProto_ProfileType int32

The type of optimization profile in use for module-level optimizations. 

const (
	HloModuleProto_INVALID HloModuleProto_ProfileType = 0
	HloModuleProto_FLAG    HloModuleProto_ProfileType = 1
	HloModuleProto_FUSION  HloModuleProto_ProfileType = 2
	HloModuleProto_LAYOUT  HloModuleProto_ProfileType = 3
	HloModuleProto_DOT     HloModuleProto_ProfileType = 4
)

func (HloModuleProto_ProfileType) Descriptor 

func (HloModuleProto_ProfileType) Descriptor() protoreflect.EnumDescriptor

func (HloModuleProto_ProfileType) Enum 

func (x HloModuleProto_ProfileType) Enum() *HloModuleProto_ProfileType

func (HloModuleProto_ProfileType) EnumDescriptor deprecated 

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

Deprecated: Use HloModuleProto_ProfileType.Descriptor instead.

func (HloModuleProto_ProfileType) Number 

func (x HloModuleProto_ProfileType) Number() protoreflect.EnumNumber

func (HloModuleProto_ProfileType) String 

func (x HloModuleProto_ProfileType) String() string

func (HloModuleProto_ProfileType) Type 

func (HloModuleProto_ProfileType) Type() protoreflect.EnumType

type HloPassMetadata 

type HloPassMetadata struct {

	// For a given module, pass_id uniquely identifies a run of an HLO pass on
	// that module. Note that a pass_id may not always refer to the same pass
	// because the order of passes during compilation may change. For finding
	// metadata for a particular pass, pass_name and pipeline_name would be more
	// reliable, although note that they may not be unique.
	PassId       int64  `protobuf:"varint,1,opt,name=pass_id,json=passId,proto3" json:"pass_id,omitempty"`
	PassName     string `protobuf:"bytes,2,opt,name=pass_name,json=passName,proto3" json:"pass_name,omitempty"`
	PipelineName string `protobuf:"bytes,3,opt,name=pipeline_name,json=pipelineName,proto3" json:"pipeline_name,omitempty"`
	// Filenames of the dumps of the module after this pass ran. Module may be
	// dumped in multiple formats, and the order of formats in this field will
	// stay consistent across passes.
	DumpFilenames []string `protobuf:"bytes,4,rep,name=dump_filenames,json=dumpFilenames,proto3" json:"dump_filenames,omitempty"`
	// Return value of pass.Run(). True if this pass changed the module, or, in
	// the case where the module was run through this pass as part of a module
	// group, true if this pass changed any module in the same module group.
	ModuleChanged bool `protobuf:"varint,5,opt,name=module_changed,json=moduleChanged,proto3" json:"module_changed,omitempty"`
	// The unique_id of the module that this pass is run on. May be different from
	// the canonical_module_id of the HloModuleMetadata that this HloPassMetadata
	// is inside.
	ModuleId int64 `protobuf:"varint,6,opt,name=module_id,json=moduleId,proto3" json:"module_id,omitempty"`
	// If the module went through this pass as part of a module group, this is
	// set as the ids of all the modules in the module group. Empty otherwise.
	ModuleGroupModuleIds []int64 `` /* 133-byte string literal not displayed */
	// Timestamp before and after the pass is run. Note they may be equal.
	StartTimestampUsec int64 `protobuf:"varint,8,opt,name=start_timestamp_usec,json=startTimestampUsec,proto3" json:"start_timestamp_usec,omitempty"`
	EndTimestampUsec   int64 `protobuf:"varint,9,opt,name=end_timestamp_usec,json=endTimestampUsec,proto3" json:"end_timestamp_usec,omitempty"`
	// Custom metadata for the pass.
	CustomMetadata *anypb.Any `protobuf:"bytes,10,opt,name=custom_metadata,json=customMetadata,proto3" json:"custom_metadata,omitempty"`
	// contains filtered or unexported fields
}

Metadata for one run of an HLO pass on a module. Provides more information when processing debug dumps of HloProtos about the order of HLO passes and various other stats like duration. `pass_id` may also be used to identify a particular run of a pass in debug info that propagates through stages of compilation.

func (*HloPassMetadata) Descriptor deprecated 

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

Deprecated: Use HloPassMetadata.ProtoReflect.Descriptor instead.

func (*HloPassMetadata) GetCustomMetadata 

func (x *HloPassMetadata) GetCustomMetadata() *anypb.Any

func (*HloPassMetadata) GetDumpFilenames 

func (x *HloPassMetadata) GetDumpFilenames() []string

func (*HloPassMetadata) GetEndTimestampUsec 

func (x *HloPassMetadata) GetEndTimestampUsec() int64

func (*HloPassMetadata) GetModuleChanged 

func (x *HloPassMetadata) GetModuleChanged() bool

func (*HloPassMetadata) GetModuleGroupModuleIds 

func (x *HloPassMetadata) GetModuleGroupModuleIds() []int64

func (*HloPassMetadata) GetModuleId 

func (x *HloPassMetadata) GetModuleId() int64

func (*HloPassMetadata) GetPassId 

func (x *HloPassMetadata) GetPassId() int64

func (*HloPassMetadata) GetPassName 

func (x *HloPassMetadata) GetPassName() string

func (*HloPassMetadata) GetPipelineName 

func (x *HloPassMetadata) GetPipelineName() string

func (*HloPassMetadata) GetStartTimestampUsec 

func (x *HloPassMetadata) GetStartTimestampUsec() int64

func (*HloPassMetadata) ProtoMessage 

func (*HloPassMetadata) ProtoMessage()

func (*HloPassMetadata) ProtoReflect 

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

func (*HloPassMetadata) Reset 

func (x *HloPassMetadata) Reset()

func (*HloPassMetadata) String 

func (x *HloPassMetadata) String() string

type HloProto 

type HloProto struct {
	HloModule        *HloModuleProto        `protobuf:"bytes,1,opt,name=hlo_module,json=hloModule,proto3" json:"hlo_module,omitempty"`
	BufferAssignment *BufferAssignmentProto `protobuf:"bytes,3,opt,name=buffer_assignment,json=bufferAssignment,proto3" json:"buffer_assignment,omitempty"`
	// contains filtered or unexported fields
}

Grouping message that contains all of the information above.

func (*HloProto) Descriptor deprecated 

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

Deprecated: Use HloProto.ProtoReflect.Descriptor instead.

func (*HloProto) GetBufferAssignment 

func (x *HloProto) GetBufferAssignment() *BufferAssignmentProto

func (*HloProto) GetHloModule 

func (x *HloProto) GetHloModule() *HloModuleProto

func (*HloProto) ProtoMessage 

func (*HloProto) ProtoMessage()

func (*HloProto) ProtoReflect 

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

func (*HloProto) Reset 

func (x *HloProto) Reset()

func (*HloProto) String 

func (x *HloProto) String() string

type HloScheduleProto 

type HloScheduleProto struct {

	// Map from computation id to sequence.
	Sequences map[int64]*HloScheduleProto_InstructionSequence `` /* 160-byte string literal not displayed */
	// contains filtered or unexported fields
}

Serialization of an HLO schedule. An HLO schedule contains a total order of instructions for each non-fusion computation in the module.

func (*HloScheduleProto) Descriptor deprecated 

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

Deprecated: Use HloScheduleProto.ProtoReflect.Descriptor instead.

func (*HloScheduleProto) GetSequences 

func (x *HloScheduleProto) GetSequences() map[int64]*HloScheduleProto_InstructionSequence

func (*HloScheduleProto) ProtoMessage 

func (*HloScheduleProto) ProtoMessage()

func (*HloScheduleProto) ProtoReflect 

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

func (*HloScheduleProto) Reset 

func (x *HloScheduleProto) Reset()

func (*HloScheduleProto) String 

func (x *HloScheduleProto) String() string

type HloScheduleProto_InstructionSequence 

type HloScheduleProto_InstructionSequence struct {
	InstructionIds []int64 `protobuf:"varint,1,rep,packed,name=instruction_ids,json=instructionIds,proto3" json:"instruction_ids,omitempty"`
	// contains filtered or unexported fields
}

func (*HloScheduleProto_InstructionSequence) Descriptor deprecated 

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

Deprecated: Use HloScheduleProto_InstructionSequence.ProtoReflect.Descriptor instead.

func (*HloScheduleProto_InstructionSequence) GetInstructionIds 

func (x *HloScheduleProto_InstructionSequence) GetInstructionIds() []int64

func (*HloScheduleProto_InstructionSequence) ProtoMessage 

func (*HloScheduleProto_InstructionSequence) ProtoMessage()

func (*HloScheduleProto_InstructionSequence) ProtoReflect 

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

func (*HloScheduleProto_InstructionSequence) Reset 

func (x *HloScheduleProto_InstructionSequence) Reset()

func (*HloScheduleProto_InstructionSequence) String 

func (x *HloScheduleProto_InstructionSequence) String() string

type HloSnapshot 

type HloSnapshot struct {

	// The hlo graph.
	Hlo *HloProto `protobuf:"bytes,1,opt,name=hlo,proto3" json:"hlo,omitempty"`
	// The arguments passed to the graph.
	Arguments []*LiteralProto `protobuf:"bytes,2,rep,name=arguments,proto3" json:"arguments,omitempty"`
	// The result of the graph.
	Result *LiteralProto `protobuf:"bytes,3,opt,name=result,proto3" json:"result,omitempty"`
	// The name of the platform used to run the graph.
	ExecutionPlatform string `protobuf:"bytes,4,opt,name=execution_platform,json=executionPlatform,proto3" json:"execution_platform,omitempty"`
	// contains filtered or unexported fields
}

Encapsulates HloProto together with the arguments, result, and execution_platform. This message is used for purposes such as analysis/replay/file-storage.

func (*HloSnapshot) Descriptor deprecated 

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

Deprecated: Use HloSnapshot.ProtoReflect.Descriptor instead.

func (*HloSnapshot) GetArguments 

func (x *HloSnapshot) GetArguments() []*LiteralProto

func (*HloSnapshot) GetExecutionPlatform 

func (x *HloSnapshot) GetExecutionPlatform() string

func (*HloSnapshot) GetHlo 

func (x *HloSnapshot) GetHlo() *HloProto

func (*HloSnapshot) GetResult 

func (x *HloSnapshot) GetResult() *LiteralProto

func (*HloSnapshot) ProtoMessage 

func (*HloSnapshot) ProtoMessage()

func (*HloSnapshot) ProtoReflect 

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

func (*HloSnapshot) Reset 

func (x *HloSnapshot) Reset()

func (*HloSnapshot) String 

func (x *HloSnapshot) String() string

type Kind 

type Kind int32
const (
	// Define a UNDEFINED_ALIAS equal to zero to get around the default-0 proto3
	// behavior and missing has_*() APIs.
	Kind_UNDEFINED_ALIAS Kind = 0
	// The buffers may or may not alias at runtime.
	Kind_MAY_ALIAS Kind = 1
	// The buffers must alias at runtime.
	Kind_MUST_ALIAS Kind = 2
)

func (Kind) Descriptor 

func (Kind) Descriptor() protoreflect.EnumDescriptor

func (Kind) Enum 

func (x Kind) Enum() *Kind

func (Kind) EnumDescriptor deprecated 

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

Deprecated: Use Kind.Descriptor instead.

func (Kind) Number 

func (x Kind) Number() protoreflect.EnumNumber

func (Kind) String 

func (x Kind) String() string

func (Kind) Type 

func (Kind) Type() protoreflect.EnumType

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"`
	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"`
	SparseIndices  []int64 `protobuf:"varint,14,rep,packed,name=sparse_indices,json=sparseIndices,proto3" json:"sparse_indices,omitempty"` // Next = 28
	// 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) GetF8E4M3B11Fnuzs 

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

func (*LiteralProto) GetF8E4M3Fns 

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

func (*LiteralProto) GetF8E4M3Fnuzs 

func (x *LiteralProto) GetF8E4M3Fnuzs() []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 LoadDataRequest 

type LoadDataRequest struct {

	// Describes the path of the ColumnIO tablet to load.
	ColumnioTabletPath string `protobuf:"bytes,1,opt,name=columnio_tablet_path,json=columnioTabletPath,proto3" json:"columnio_tablet_path,omitempty"`
	// Describes the field to load within the ColumnIO tablet.
	ColumnioField string `protobuf:"bytes,2,opt,name=columnio_field,json=columnioField,proto3" json:"columnio_field,omitempty"`
	// Individual element shape, excluding rows.
	ElementShape *ShapeProto `protobuf:"bytes,3,opt,name=element_shape,json=elementShape,proto3" json:"element_shape,omitempty"`
	// Warning: ColumnIO does not support random-access, so use offset with
	// caution in performance-critical scenarios.
	Offset int64 `protobuf:"varint,4,opt,name=offset,proto3" json:"offset,omitempty"`
	// Maximum number of elements (with shape element_shape) to load.
	Limit int64 `protobuf:"varint,5,opt,name=limit,proto3" json:"limit,omitempty"`
	// If more than one item is requested (via limit > 1), then this request
	// attribute zips together the produced vectors.
	Zip bool `protobuf:"varint,6,opt,name=zip,proto3" json:"zip,omitempty"`
	// contains filtered or unexported fields
}

func (*LoadDataRequest) Descriptor deprecated 

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

Deprecated: Use LoadDataRequest.ProtoReflect.Descriptor instead.

func (*LoadDataRequest) GetColumnioField 

func (x *LoadDataRequest) GetColumnioField() string

func (*LoadDataRequest) GetColumnioTabletPath 

func (x *LoadDataRequest) GetColumnioTabletPath() string

func (*LoadDataRequest) GetElementShape 

func (x *LoadDataRequest) GetElementShape() *ShapeProto

func (*LoadDataRequest) GetLimit 

func (x *LoadDataRequest) GetLimit() int64

func (*LoadDataRequest) GetOffset 

func (x *LoadDataRequest) GetOffset() int64

func (*LoadDataRequest) GetZip 

func (x *LoadDataRequest) GetZip() bool

func (*LoadDataRequest) ProtoMessage 

func (*LoadDataRequest) ProtoMessage()

func (*LoadDataRequest) ProtoReflect 

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

func (*LoadDataRequest) Reset 

func (x *LoadDataRequest) Reset()

func (*LoadDataRequest) String 

func (x *LoadDataRequest) String() string

type LoadDataResponse 

type LoadDataResponse struct {
	Data          *GlobalDataHandle `protobuf:"bytes,1,opt,name=data,proto3" json:"data,omitempty"`
	DataShape     *ShapeProto       `protobuf:"bytes,2,opt,name=data_shape,json=dataShape,proto3" json:"data_shape,omitempty"`
	AvailableRows int64             `protobuf:"varint,3,opt,name=available_rows,json=availableRows,proto3" json:"available_rows,omitempty"`
	RowsLoaded    int64             `protobuf:"varint,4,opt,name=rows_loaded,json=rowsLoaded,proto3" json:"rows_loaded,omitempty"`
	Nanoseconds   int64             `protobuf:"varint,5,opt,name=nanoseconds,proto3" json:"nanoseconds,omitempty"`
	// contains filtered or unexported fields
}

func (*LoadDataResponse) Descriptor deprecated 

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

Deprecated: Use LoadDataResponse.ProtoReflect.Descriptor instead.

func (*LoadDataResponse) GetAvailableRows 

func (x *LoadDataResponse) GetAvailableRows() int64

func (*LoadDataResponse) GetData 

func (x *LoadDataResponse) GetData() *GlobalDataHandle

func (*LoadDataResponse) GetDataShape 

func (x *LoadDataResponse) GetDataShape() *ShapeProto

func (*LoadDataResponse) GetNanoseconds 

func (x *LoadDataResponse) GetNanoseconds() int64

func (*LoadDataResponse) GetRowsLoaded 

func (x *LoadDataResponse) GetRowsLoaded() int64

func (*LoadDataResponse) ProtoMessage 

func (*LoadDataResponse) ProtoMessage()

func (*LoadDataResponse) ProtoReflect 

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

func (*LoadDataResponse) Reset 

func (x *LoadDataResponse) Reset()

func (*LoadDataResponse) String 

func (x *LoadDataResponse) String() string

type LogicalBufferProto 

type LogicalBufferProto struct {
	Id   int64 `protobuf:"varint,1,opt,name=id,proto3" json:"id,omitempty"`
	Size int64 `protobuf:"varint,2,opt,name=size,proto3" json:"size,omitempty"`
	// The location where the buffer is defined.
	DefinedAt *LogicalBufferProto_Location `protobuf:"bytes,3,opt,name=defined_at,json=definedAt,proto3" json:"defined_at,omitempty"`
	Color     int64                        `protobuf:"varint,4,opt,name=color,proto3" json:"color,omitempty"`
	// contains filtered or unexported fields
}

Serialization of LogicalBuffer.

func (*LogicalBufferProto) Descriptor deprecated 

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

Deprecated: Use LogicalBufferProto.ProtoReflect.Descriptor instead.

func (*LogicalBufferProto) GetColor 

func (x *LogicalBufferProto) GetColor() int64

func (*LogicalBufferProto) GetDefinedAt 

func (x *LogicalBufferProto) GetDefinedAt() *LogicalBufferProto_Location

func (*LogicalBufferProto) GetId 

func (x *LogicalBufferProto) GetId() int64

func (*LogicalBufferProto) GetSize 

func (x *LogicalBufferProto) GetSize() int64

func (*LogicalBufferProto) ProtoMessage 

func (*LogicalBufferProto) ProtoMessage()

func (*LogicalBufferProto) ProtoReflect 

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

func (*LogicalBufferProto) Reset 

func (x *LogicalBufferProto) Reset()

func (*LogicalBufferProto) String 

func (x *LogicalBufferProto) String() string

type LogicalBufferProto_Location 

type LogicalBufferProto_Location struct {

	// TODO(b/239098765): Remove instruction_name and computation_name.
	//
	// Deprecated: Marked as deprecated in xla/service/hlo.proto.
	InstructionName string  `protobuf:"bytes,2,opt,name=instruction_name,json=instructionName,proto3" json:"instruction_name,omitempty"`
	InstructionId   int64   `protobuf:"varint,4,opt,name=instruction_id,json=instructionId,proto3" json:"instruction_id,omitempty"`
	ShapeIndex      []int64 `protobuf:"varint,3,rep,packed,name=shape_index,json=shapeIndex,proto3" json:"shape_index,omitempty"`
	// contains filtered or unexported fields
}

Location represents an instruction and its shape index, which uniquely identifies a point where a buffer is needed.

func (*LogicalBufferProto_Location) Descriptor deprecated 

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

Deprecated: Use LogicalBufferProto_Location.ProtoReflect.Descriptor instead.

func (*LogicalBufferProto_Location) GetInstructionId 

func (x *LogicalBufferProto_Location) GetInstructionId() int64

func (*LogicalBufferProto_Location) GetInstructionName deprecated 

func (x *LogicalBufferProto_Location) GetInstructionName() string

Deprecated: Marked as deprecated in xla/service/hlo.proto.

func (*LogicalBufferProto_Location) GetShapeIndex 

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

func (*LogicalBufferProto_Location) ProtoMessage 

func (*LogicalBufferProto_Location) ProtoMessage()

func (*LogicalBufferProto_Location) ProtoReflect 

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

func (*LogicalBufferProto_Location) Reset 

func (x *LogicalBufferProto_Location) Reset()

func (*LogicalBufferProto_Location) String 

func (x *LogicalBufferProto_Location) String() string

type NormKind 

type NormKind int32

NormKind kind 

const (
	NormKind_LAYER_FWD_INFER NormKind = 0
	NormKind_LAYER_FWD_TRAIN NormKind = 1
	NormKind_LAYER_BWD       NormKind = 2
)

func (NormKind) Descriptor 

func (NormKind) Descriptor() protoreflect.EnumDescriptor

func (NormKind) Enum 

func (x NormKind) Enum() *NormKind

func (NormKind) EnumDescriptor deprecated 

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

Deprecated: Use NormKind.Descriptor instead.

func (NormKind) Number 

func (x NormKind) Number() protoreflect.EnumNumber

func (NormKind) String 

func (x NormKind) String() string

func (NormKind) Type 

func (NormKind) Type() protoreflect.EnumType

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"`
	// 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) 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 OptionOverrideProto 

type OptionOverrideProto struct {

	// Types that are assignable to Value:
	//
	//	*OptionOverrideProto_StringField
	//	*OptionOverrideProto_BoolField
	//	*OptionOverrideProto_IntField
	//	*OptionOverrideProto_DoubleField
	Value isOptionOverrideProto_Value `protobuf_oneof:"value"`
	// contains filtered or unexported fields
}

func (*OptionOverrideProto) Descriptor deprecated 

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

Deprecated: Use OptionOverrideProto.ProtoReflect.Descriptor instead.

func (*OptionOverrideProto) GetBoolField 

func (x *OptionOverrideProto) GetBoolField() bool

func (*OptionOverrideProto) GetDoubleField 

func (x *OptionOverrideProto) GetDoubleField() float64

func (*OptionOverrideProto) GetIntField 

func (x *OptionOverrideProto) GetIntField() int64

func (*OptionOverrideProto) GetStringField 

func (x *OptionOverrideProto) GetStringField() string

func (*OptionOverrideProto) GetValue 

func (m *OptionOverrideProto) GetValue() isOptionOverrideProto_Value

func (*OptionOverrideProto) ProtoMessage 

func (*OptionOverrideProto) ProtoMessage()

func (*OptionOverrideProto) ProtoReflect 

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

func (*OptionOverrideProto) Reset 

func (x *OptionOverrideProto) Reset()

func (*OptionOverrideProto) String 

func (x *OptionOverrideProto) String() string

type OptionOverrideProto_BoolField 

type OptionOverrideProto_BoolField struct {
	BoolField bool `protobuf:"varint,2,opt,name=bool_field,json=boolField,proto3,oneof"`
}

type OptionOverrideProto_DoubleField 

type OptionOverrideProto_DoubleField struct {
	DoubleField float64 `protobuf:"fixed64,4,opt,name=double_field,json=doubleField,proto3,oneof"`
}

type OptionOverrideProto_IntField 

type OptionOverrideProto_IntField struct {
	IntField int64 `protobuf:"varint,3,opt,name=int_field,json=intField,proto3,oneof"`
}

type OptionOverrideProto_StringField 

type OptionOverrideProto_StringField struct {
	StringField string `protobuf:"bytes,1,opt,name=string_field,json=stringField,proto3,oneof"`
}

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.
	//
	// 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.
	//
	// 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_F8E4M3FN      PrimitiveType = 20
	PrimitiveType_F8E4M3B11FNUZ PrimitiveType = 23
	// 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 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 ResetDeviceRequest 

type ResetDeviceRequest struct {
	DeviceHandle *DeviceHandle `protobuf:"bytes,1,opt,name=device_handle,json=deviceHandle,proto3" json:"device_handle,omitempty"`
	// contains filtered or unexported fields
}

func (*ResetDeviceRequest) Descriptor deprecated 

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

Deprecated: Use ResetDeviceRequest.ProtoReflect.Descriptor instead.

func (*ResetDeviceRequest) GetDeviceHandle 

func (x *ResetDeviceRequest) GetDeviceHandle() *DeviceHandle

func (*ResetDeviceRequest) ProtoMessage 

func (*ResetDeviceRequest) ProtoMessage()

func (*ResetDeviceRequest) ProtoReflect 

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

func (*ResetDeviceRequest) Reset 

func (x *ResetDeviceRequest) Reset()

func (*ResetDeviceRequest) String 

func (x *ResetDeviceRequest) String() string

type ResetDeviceResponse 

type ResetDeviceResponse struct {
	// contains filtered or unexported fields
}

func (*ResetDeviceResponse) Descriptor deprecated 

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

Deprecated: Use ResetDeviceResponse.ProtoReflect.Descriptor instead.

func (*ResetDeviceResponse) ProtoMessage 

func (*ResetDeviceResponse) ProtoMessage()

func (*ResetDeviceResponse) ProtoReflect 

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

func (*ResetDeviceResponse) Reset 

func (x *ResetDeviceResponse) Reset()

func (*ResetDeviceResponse) String 

func (x *ResetDeviceResponse) String() string

type RocmComputeCapabilityProto 

type RocmComputeCapabilityProto struct {
	GcnArchName string `protobuf:"bytes,1,opt,name=gcn_arch_name,json=gcnArchName,proto3" json:"gcn_arch_name,omitempty"`
	// contains filtered or unexported fields
}

func (*RocmComputeCapabilityProto) Descriptor deprecated 

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

Deprecated: Use RocmComputeCapabilityProto.ProtoReflect.Descriptor instead.

func (*RocmComputeCapabilityProto) GetGcnArchName 

func (x *RocmComputeCapabilityProto) GetGcnArchName() string

func (*RocmComputeCapabilityProto) ProtoMessage 

func (*RocmComputeCapabilityProto) ProtoMessage()

func (*RocmComputeCapabilityProto) ProtoReflect 

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

func (*RocmComputeCapabilityProto) Reset 

func (x *RocmComputeCapabilityProto) Reset()

func (*RocmComputeCapabilityProto) String 

func (x *RocmComputeCapabilityProto) String() string

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"`
	// 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) GetInsertedWindowDims 

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

func (*ScatterDimensionNumbers) GetScatterDimsToOperandDims 

func (x *ScatterDimensionNumbers) GetScatterDimsToOperandDims() []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 ScheduleProto 

type ScheduleProto struct {
	Instructions []*ScheduleProto_Instruction `protobuf:"bytes,1,rep,name=instructions,proto3" json:"instructions,omitempty"`
	// Computation id (matches the id in HloComputationProto).
	ComputationId        int64           `protobuf:"varint,2,opt,name=computation_id,json=computationId,proto3" json:"computation_id,omitempty"`
	HloModule            *HloModuleProto `protobuf:"bytes,3,opt,name=hlo_module,json=hloModule,proto3" json:"hlo_module,omitempty"`
	CyclesPerMicrosecond int64           `protobuf:"varint,4,opt,name=cycles_per_microsecond,json=cyclesPerMicrosecond,proto3" json:"cycles_per_microsecond,omitempty"`
	// contains filtered or unexported fields
}

A trace estimated by the Latency Hiding Scheduler.

func (*ScheduleProto) Descriptor deprecated 

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

Deprecated: Use ScheduleProto.ProtoReflect.Descriptor instead.

func (*ScheduleProto) GetComputationId 

func (x *ScheduleProto) GetComputationId() int64

func (*ScheduleProto) GetCyclesPerMicrosecond 

func (x *ScheduleProto) GetCyclesPerMicrosecond() int64

func (*ScheduleProto) GetHloModule 

func (x *ScheduleProto) GetHloModule() *HloModuleProto

func (*ScheduleProto) GetInstructions 

func (x *ScheduleProto) GetInstructions() []*ScheduleProto_Instruction

func (*ScheduleProto) ProtoMessage 

func (*ScheduleProto) ProtoMessage()

func (*ScheduleProto) ProtoReflect 

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

func (*ScheduleProto) Reset 

func (x *ScheduleProto) Reset()

func (*ScheduleProto) String 

func (x *ScheduleProto) String() string

type ScheduleProto_Instruction 

type ScheduleProto_Instruction struct {

	// Instruction id (matches the id in HloInstructionProto).
	Id int64 `protobuf:"varint,1,opt,name=id,proto3" json:"id,omitempty"`
	// Start and end timestamps in cycles.
	StartTimestampCycles float64 `protobuf:"fixed64,2,opt,name=start_timestamp_cycles,json=startTimestampCycles,proto3" json:"start_timestamp_cycles,omitempty"`
	EndTimestampCycles   float64 `protobuf:"fixed64,3,opt,name=end_timestamp_cycles,json=endTimestampCycles,proto3" json:"end_timestamp_cycles,omitempty"`
	// contains filtered or unexported fields
}

func (*ScheduleProto_Instruction) Descriptor deprecated 

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

Deprecated: Use ScheduleProto_Instruction.ProtoReflect.Descriptor instead.

func (*ScheduleProto_Instruction) GetEndTimestampCycles 

func (x *ScheduleProto_Instruction) GetEndTimestampCycles() float64

func (*ScheduleProto_Instruction) GetId 

func (x *ScheduleProto_Instruction) GetId() int64

func (*ScheduleProto_Instruction) GetStartTimestampCycles 

func (x *ScheduleProto_Instruction) GetStartTimestampCycles() float64

func (*ScheduleProto_Instruction) ProtoMessage 

func (*ScheduleProto_Instruction) ProtoMessage()

func (*ScheduleProto_Instruction) ProtoReflect 

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

func (*ScheduleProto_Instruction) Reset 

func (x *ScheduleProto_Instruction) Reset()

func (*ScheduleProto_Instruction) String 

func (x *ScheduleProto_Instruction) 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 ShardableValueUpdatePairProto 

type ShardableValueUpdatePairProto struct {
	InputParameterNumber int64   `protobuf:"varint,1,opt,name=input_parameter_number,json=inputParameterNumber,proto3" json:"input_parameter_number,omitempty"`
	ParameterShapeIndex  []int64 `` /* 128-byte string literal not displayed */
	OutputShapeIndex     []int64 `protobuf:"varint,3,rep,packed,name=output_shape_index,json=outputShapeIndex,proto3" json:"output_shape_index,omitempty"`
	// contains filtered or unexported fields
}

func (*ShardableValueUpdatePairProto) Descriptor deprecated 

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

Deprecated: Use ShardableValueUpdatePairProto.ProtoReflect.Descriptor instead.

func (*ShardableValueUpdatePairProto) GetInputParameterNumber 

func (x *ShardableValueUpdatePairProto) GetInputParameterNumber() int64

func (*ShardableValueUpdatePairProto) GetOutputShapeIndex 

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

func (*ShardableValueUpdatePairProto) GetParameterShapeIndex 

func (x *ShardableValueUpdatePairProto) GetParameterShapeIndex() []int64

func (*ShardableValueUpdatePairProto) ProtoMessage 

func (*ShardableValueUpdatePairProto) ProtoMessage()

func (*ShardableValueUpdatePairProto) ProtoReflect 

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

func (*ShardableValueUpdatePairProto) Reset 

func (x *ShardableValueUpdatePairProto) Reset()

func (*ShardableValueUpdatePairProto) String 

func (x *ShardableValueUpdatePairProto) 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 StackFrameIndexProto 

type StackFrameIndexProto struct {

	// Flat index array of file names.
	FileNames []string `protobuf:"bytes,1,rep,name=file_names,json=fileNames,proto3" json:"file_names,omitempty"`
	// Flat index array of function names.
	FunctionNames []string `protobuf:"bytes,2,rep,name=function_names,json=functionNames,proto3" json:"function_names,omitempty"`
	// Flat index array of file locations.
	FileLocations []*StackFrameIndexProto_FileLocation `protobuf:"bytes,3,rep,name=file_locations,json=fileLocations,proto3" json:"file_locations,omitempty"`
	// Flat index array of frames.
	StackFrames []*StackFrameIndexProto_StackFrame `protobuf:"bytes,4,rep,name=stack_frames,json=stackFrames,proto3" json:"stack_frames,omitempty"`
	// contains filtered or unexported fields
}

Serialization of stack frames index representations. Stack frames index presented in four flat arrays: 1. File names array. 2. Function names array. 3. File location array. 4. Frame array. All reference ids in sub-protos are 1-based positions of the entity in the flat array. Ids are 1-based to keep 0 value as representation of non-set property.

func (*StackFrameIndexProto) Descriptor deprecated 

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

Deprecated: Use StackFrameIndexProto.ProtoReflect.Descriptor instead.

func (*StackFrameIndexProto) GetFileLocations 

func (x *StackFrameIndexProto) GetFileLocations() []*StackFrameIndexProto_FileLocation

func (*StackFrameIndexProto) GetFileNames 

func (x *StackFrameIndexProto) GetFileNames() []string

func (*StackFrameIndexProto) GetFunctionNames 

func (x *StackFrameIndexProto) GetFunctionNames() []string

func (*StackFrameIndexProto) GetStackFrames 

func (x *StackFrameIndexProto) GetStackFrames() []*StackFrameIndexProto_StackFrame

func (*StackFrameIndexProto) ProtoMessage 

func (*StackFrameIndexProto) ProtoMessage()

func (*StackFrameIndexProto) ProtoReflect 

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

func (*StackFrameIndexProto) Reset 

func (x *StackFrameIndexProto) Reset()

func (*StackFrameIndexProto) String 

func (x *StackFrameIndexProto) String() string

type StackFrameIndexProto_FileLocation 

type StackFrameIndexProto_FileLocation struct {

	// 1-based position of file name.
	FileNameId int32 `protobuf:"varint,1,opt,name=file_name_id,json=fileNameId,proto3" json:"file_name_id,omitempty"`
	// 1-based position of function name.
	FunctionNameId int32 `protobuf:"varint,2,opt,name=function_name_id,json=functionNameId,proto3" json:"function_name_id,omitempty"`
	// Line number.
	Line int32 `protobuf:"varint,3,opt,name=line,proto3" json:"line,omitempty"`
	// Column number.
	Column int32 `protobuf:"varint,4,opt,name=column,proto3" json:"column,omitempty"`
	// contains filtered or unexported fields
}

Serialization of file position.

func (*StackFrameIndexProto_FileLocation) Descriptor deprecated 

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

Deprecated: Use StackFrameIndexProto_FileLocation.ProtoReflect.Descriptor instead.

func (*StackFrameIndexProto_FileLocation) GetColumn 

func (x *StackFrameIndexProto_FileLocation) GetColumn() int32

func (*StackFrameIndexProto_FileLocation) GetFileNameId 

func (x *StackFrameIndexProto_FileLocation) GetFileNameId() int32

func (*StackFrameIndexProto_FileLocation) GetFunctionNameId 

func (x *StackFrameIndexProto_FileLocation) GetFunctionNameId() int32

func (*StackFrameIndexProto_FileLocation) GetLine 

func (x *StackFrameIndexProto_FileLocation) GetLine() int32

func (*StackFrameIndexProto_FileLocation) ProtoMessage 

func (*StackFrameIndexProto_FileLocation) ProtoMessage()

func (*StackFrameIndexProto_FileLocation) ProtoReflect 

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

func (*StackFrameIndexProto_FileLocation) Reset 

func (x *StackFrameIndexProto_FileLocation) Reset()

func (*StackFrameIndexProto_FileLocation) String 

func (x *StackFrameIndexProto_FileLocation) String() string

type StackFrameIndexProto_StackFrame 

type StackFrameIndexProto_StackFrame struct {

	// 1-based position of file location.
	FileLocationId int32 `protobuf:"varint,1,opt,name=file_location_id,json=fileLocationId,proto3" json:"file_location_id,omitempty"`
	// 1-based position of the parent frame.
	ParentFrameId int32 `protobuf:"varint,2,opt,name=parent_frame_id,json=parentFrameId,proto3" json:"parent_frame_id,omitempty"`
	// contains filtered or unexported fields
}

Serialization of frame.

func (*StackFrameIndexProto_StackFrame) Descriptor deprecated 

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

Deprecated: Use StackFrameIndexProto_StackFrame.ProtoReflect.Descriptor instead.

func (*StackFrameIndexProto_StackFrame) GetFileLocationId 

func (x *StackFrameIndexProto_StackFrame) GetFileLocationId() int32

func (*StackFrameIndexProto_StackFrame) GetParentFrameId 

func (x *StackFrameIndexProto_StackFrame) GetParentFrameId() int32

func (*StackFrameIndexProto_StackFrame) ProtoMessage 

func (*StackFrameIndexProto_StackFrame) ProtoMessage()

func (*StackFrameIndexProto_StackFrame) ProtoReflect 

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

func (*StackFrameIndexProto_StackFrame) Reset 

func (x *StackFrameIndexProto_StackFrame) Reset()

func (*StackFrameIndexProto_StackFrame) String 

func (x *StackFrameIndexProto_StackFrame) 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 TensorDescriptorProto 

type TensorDescriptorProto struct {
	Dimensions []int64  `protobuf:"varint,1,rep,packed,name=dimensions,proto3" json:"dimensions,omitempty"`
	DataType   DataType `protobuf:"varint,2,opt,name=data_type,json=dataType,proto3,enum=stream_executor.dnn.DataType" json:"data_type,omitempty"`
	// Types that are assignable to LayoutOneof:
	//
	//	*TensorDescriptorProto_DataLayout
	//	*TensorDescriptorProto_FilterLayout
	LayoutOneof isTensorDescriptorProto_LayoutOneof `protobuf_oneof:"layout_oneof"`
	// contains filtered or unexported fields
}

Generic tensor representation.

func (*TensorDescriptorProto) Descriptor deprecated 

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

Deprecated: Use TensorDescriptorProto.ProtoReflect.Descriptor instead.

func (*TensorDescriptorProto) GetDataLayout 

func (x *TensorDescriptorProto) GetDataLayout() DataLayout

func (*TensorDescriptorProto) GetDataType 

func (x *TensorDescriptorProto) GetDataType() DataType

func (*TensorDescriptorProto) GetDimensions 

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

func (*TensorDescriptorProto) GetFilterLayout 

func (x *TensorDescriptorProto) GetFilterLayout() FilterLayout

func (*TensorDescriptorProto) GetLayoutOneof 

func (m *TensorDescriptorProto) GetLayoutOneof() isTensorDescriptorProto_LayoutOneof

func (*TensorDescriptorProto) ProtoMessage 

func (*TensorDescriptorProto) ProtoMessage()

func (*TensorDescriptorProto) ProtoReflect 

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

func (*TensorDescriptorProto) Reset 

func (x *TensorDescriptorProto) Reset()

func (*TensorDescriptorProto) String 

func (x *TensorDescriptorProto) String() string

type TensorDescriptorProto_DataLayout 

type TensorDescriptorProto_DataLayout struct {
	DataLayout DataLayout `protobuf:"varint,3,opt,name=data_layout,json=dataLayout,proto3,enum=stream_executor.dnn.DataLayout,oneof"`
}

type TensorDescriptorProto_FilterLayout 

type TensorDescriptorProto_FilterLayout struct {
	FilterLayout FilterLayout `protobuf:"varint,4,opt,name=filter_layout,json=filterLayout,proto3,enum=stream_executor.dnn.FilterLayout,oneof"`
}

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 TransferFromOutfeedRequest 

type TransferFromOutfeedRequest struct {

	// This optional field directs the service to return the literal in this
	// layout. A shape is used to hold the layout to accommodate tuples.
	ShapeWithLayout *ShapeProto   `protobuf:"bytes,1,opt,name=shape_with_layout,json=shapeWithLayout,proto3" json:"shape_with_layout,omitempty"`
	ReplicaId       int64         `protobuf:"varint,2,opt,name=replica_id,json=replicaId,proto3" json:"replica_id,omitempty"`
	DeviceHandle    *DeviceHandle `protobuf:"bytes,3,opt,name=device_handle,json=deviceHandle,proto3" json:"device_handle,omitempty"`
	// contains filtered or unexported fields
}

func (*TransferFromOutfeedRequest) Descriptor deprecated 

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

Deprecated: Use TransferFromOutfeedRequest.ProtoReflect.Descriptor instead.

func (*TransferFromOutfeedRequest) GetDeviceHandle 

func (x *TransferFromOutfeedRequest) GetDeviceHandle() *DeviceHandle

func (*TransferFromOutfeedRequest) GetReplicaId 

func (x *TransferFromOutfeedRequest) GetReplicaId() int64

func (*TransferFromOutfeedRequest) GetShapeWithLayout 

func (x *TransferFromOutfeedRequest) GetShapeWithLayout() *ShapeProto

func (*TransferFromOutfeedRequest) ProtoMessage 

func (*TransferFromOutfeedRequest) ProtoMessage()

func (*TransferFromOutfeedRequest) ProtoReflect 

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

func (*TransferFromOutfeedRequest) Reset 

func (x *TransferFromOutfeedRequest) Reset()

func (*TransferFromOutfeedRequest) String 

func (x *TransferFromOutfeedRequest) String() string

type TransferFromOutfeedResponse 

type TransferFromOutfeedResponse struct {
	Literal *LiteralProto `protobuf:"bytes,1,opt,name=literal,proto3" json:"literal,omitempty"`
	// contains filtered or unexported fields
}

func (*TransferFromOutfeedResponse) Descriptor deprecated 

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

Deprecated: Use TransferFromOutfeedResponse.ProtoReflect.Descriptor instead.

func (*TransferFromOutfeedResponse) GetLiteral 

func (x *TransferFromOutfeedResponse) GetLiteral() *LiteralProto

func (*TransferFromOutfeedResponse) ProtoMessage 

func (*TransferFromOutfeedResponse) ProtoMessage()

func (*TransferFromOutfeedResponse) ProtoReflect 

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

func (*TransferFromOutfeedResponse) Reset 

func (x *TransferFromOutfeedResponse) Reset()

func (*TransferFromOutfeedResponse) String 

func (x *TransferFromOutfeedResponse) String() string

type TransferToClientRequest 

type TransferToClientRequest struct {
	Data *GlobalDataHandle `protobuf:"bytes,1,opt,name=data,proto3" json:"data,omitempty"`
	// This optional field directs the service to return the literal in this
	// layout. A shape is used to hold the layout to accommodate tuples.
	ShapeWithLayout *ShapeProto `protobuf:"bytes,2,opt,name=shape_with_layout,json=shapeWithLayout,proto3" json:"shape_with_layout,omitempty"`
	// contains filtered or unexported fields
}

func (*TransferToClientRequest) Descriptor deprecated 

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

Deprecated: Use TransferToClientRequest.ProtoReflect.Descriptor instead.

func (*TransferToClientRequest) GetData 

func (x *TransferToClientRequest) GetData() *GlobalDataHandle

func (*TransferToClientRequest) GetShapeWithLayout 

func (x *TransferToClientRequest) GetShapeWithLayout() *ShapeProto

func (*TransferToClientRequest) ProtoMessage 

func (*TransferToClientRequest) ProtoMessage()

func (*TransferToClientRequest) ProtoReflect 

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

func (*TransferToClientRequest) Reset 

func (x *TransferToClientRequest) Reset()

func (*TransferToClientRequest) String 

func (x *TransferToClientRequest) String() string

type TransferToClientResponse 

type TransferToClientResponse struct {
	Literal *LiteralProto `protobuf:"bytes,1,opt,name=literal,proto3" json:"literal,omitempty"`
	// contains filtered or unexported fields
}

func (*TransferToClientResponse) Descriptor deprecated 

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

Deprecated: Use TransferToClientResponse.ProtoReflect.Descriptor instead.

func (*TransferToClientResponse) GetLiteral 

func (x *TransferToClientResponse) GetLiteral() *LiteralProto

func (*TransferToClientResponse) ProtoMessage 

func (*TransferToClientResponse) ProtoMessage()

func (*TransferToClientResponse) ProtoReflect 

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

func (*TransferToClientResponse) Reset 

func (x *TransferToClientResponse) Reset()

func (*TransferToClientResponse) String 

func (x *TransferToClientResponse) String() string

type TransferToInfeedRequest 

type TransferToInfeedRequest struct {
	Literal      *LiteralProto `protobuf:"bytes,1,opt,name=literal,proto3" json:"literal,omitempty"`
	ReplicaId    int64         `protobuf:"varint,2,opt,name=replica_id,json=replicaId,proto3" json:"replica_id,omitempty"`
	DeviceHandle *DeviceHandle `protobuf:"bytes,3,opt,name=device_handle,json=deviceHandle,proto3" json:"device_handle,omitempty"`
	// contains filtered or unexported fields
}

func (*TransferToInfeedRequest) Descriptor deprecated 

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

Deprecated: Use TransferToInfeedRequest.ProtoReflect.Descriptor instead.

func (*TransferToInfeedRequest) GetDeviceHandle 

func (x *TransferToInfeedRequest) GetDeviceHandle() *DeviceHandle

func (*TransferToInfeedRequest) GetLiteral 

func (x *TransferToInfeedRequest) GetLiteral() *LiteralProto

func (*TransferToInfeedRequest) GetReplicaId 

func (x *TransferToInfeedRequest) GetReplicaId() int64

func (*TransferToInfeedRequest) ProtoMessage 

func (*TransferToInfeedRequest) ProtoMessage()

func (*TransferToInfeedRequest) ProtoReflect 

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

func (*TransferToInfeedRequest) Reset 

func (x *TransferToInfeedRequest) Reset()

func (*TransferToInfeedRequest) String 

func (x *TransferToInfeedRequest) String() string

type TransferToInfeedResponse 

type TransferToInfeedResponse struct {
	// contains filtered or unexported fields
}

func (*TransferToInfeedResponse) Descriptor deprecated 

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

Deprecated: Use TransferToInfeedResponse.ProtoReflect.Descriptor instead.

func (*TransferToInfeedResponse) ProtoMessage 

func (*TransferToInfeedResponse) ProtoMessage()

func (*TransferToInfeedResponse) ProtoReflect 

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

func (*TransferToInfeedResponse) Reset 

func (x *TransferToInfeedResponse) Reset()

func (*TransferToInfeedResponse) String 

func (x *TransferToInfeedResponse) String() string

type TransferToServerRequest 

type TransferToServerRequest struct {
	Literal      *LiteralProto `protobuf:"bytes,1,opt,name=literal,proto3" json:"literal,omitempty"`
	DeviceHandle *DeviceHandle `protobuf:"bytes,2,opt,name=device_handle,json=deviceHandle,proto3" json:"device_handle,omitempty"`
	// contains filtered or unexported fields
}

func (*TransferToServerRequest) Descriptor deprecated 

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

Deprecated: Use TransferToServerRequest.ProtoReflect.Descriptor instead.

func (*TransferToServerRequest) GetDeviceHandle 

func (x *TransferToServerRequest) GetDeviceHandle() *DeviceHandle

func (*TransferToServerRequest) GetLiteral 

func (x *TransferToServerRequest) GetLiteral() *LiteralProto

func (*TransferToServerRequest) ProtoMessage 

func (*TransferToServerRequest) ProtoMessage()

func (*TransferToServerRequest) ProtoReflect 

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

func (*TransferToServerRequest) Reset 

func (x *TransferToServerRequest) Reset()

func (*TransferToServerRequest) String 

func (x *TransferToServerRequest) String() string

type TransferToServerResponse 

type TransferToServerResponse struct {
	Data *GlobalDataHandle `protobuf:"bytes,1,opt,name=data,proto3" json:"data,omitempty"`
	// contains filtered or unexported fields
}

func (*TransferToServerResponse) Descriptor deprecated 

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

Deprecated: Use TransferToServerResponse.ProtoReflect.Descriptor instead.

func (*TransferToServerResponse) GetData 

func (x *TransferToServerResponse) GetData() *GlobalDataHandle

func (*TransferToServerResponse) ProtoMessage 

func (*TransferToServerResponse) ProtoMessage()

func (*TransferToServerResponse) ProtoReflect 

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

func (*TransferToServerResponse) Reset 

func (x *TransferToServerResponse) Reset()

func (*TransferToServerResponse) String 

func (x *TransferToServerResponse) 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 UnpackRequest 

type UnpackRequest struct {
	Data *GlobalDataHandle `protobuf:"bytes,1,opt,name=data,proto3" json:"data,omitempty"`
	// contains filtered or unexported fields
}

func (*UnpackRequest) Descriptor deprecated 

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

Deprecated: Use UnpackRequest.ProtoReflect.Descriptor instead.

func (*UnpackRequest) GetData 

func (x *UnpackRequest) GetData() *GlobalDataHandle

func (*UnpackRequest) ProtoMessage 

func (*UnpackRequest) ProtoMessage()

func (*UnpackRequest) ProtoReflect 

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

func (*UnpackRequest) Reset 

func (x *UnpackRequest) Reset()

func (*UnpackRequest) String 

func (x *UnpackRequest) String() string

type UnpackResponse 

type UnpackResponse struct {
	TiedData []*GlobalDataHandle `protobuf:"bytes,1,rep,name=tied_data,json=tiedData,proto3" json:"tied_data,omitempty"`
	// contains filtered or unexported fields
}

func (*UnpackResponse) Descriptor deprecated 

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

Deprecated: Use UnpackResponse.ProtoReflect.Descriptor instead.

func (*UnpackResponse) GetTiedData 

func (x *UnpackResponse) GetTiedData() []*GlobalDataHandle

func (*UnpackResponse) ProtoMessage 

func (*UnpackResponse) ProtoMessage()

func (*UnpackResponse) ProtoReflect 

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

func (*UnpackResponse) Reset 

func (x *UnpackResponse) Reset()

func (*UnpackResponse) String 

func (x *UnpackResponse) String() string

type UnregisterRequest 

type UnregisterRequest struct {
	Data []*GlobalDataHandle `protobuf:"bytes,1,rep,name=data,proto3" json:"data,omitempty"`
	// contains filtered or unexported fields
}

func (*UnregisterRequest) Descriptor deprecated 

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

Deprecated: Use UnregisterRequest.ProtoReflect.Descriptor instead.

func (*UnregisterRequest) GetData 

func (x *UnregisterRequest) GetData() []*GlobalDataHandle

func (*UnregisterRequest) ProtoMessage 

func (*UnregisterRequest) ProtoMessage()

func (*UnregisterRequest) ProtoReflect 

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

func (*UnregisterRequest) Reset 

func (x *UnregisterRequest) Reset()

func (*UnregisterRequest) String 

func (x *UnregisterRequest) String() string

type UnregisterResponse 

type UnregisterResponse struct {
	// contains filtered or unexported fields
}

func (*UnregisterResponse) Descriptor deprecated 

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

Deprecated: Use UnregisterResponse.ProtoReflect.Descriptor instead.

func (*UnregisterResponse) ProtoMessage 

func (*UnregisterResponse) ProtoMessage()

func (*UnregisterResponse) ProtoReflect 

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

func (*UnregisterResponse) Reset 

func (x *UnregisterResponse) Reset()

func (*UnregisterResponse) String 

func (x *UnregisterResponse) String() string

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.