protos

package

Go to main page

Versions in this module

v0

v0.6.0

Feb 1, 2025

v0.5.1

Jan 29, 2025

v0.5.0

Dec 19, 2024

v0.4.9

Nov 25, 2024

v0.4.8

Nov 19, 2024

v0.4.7

Nov 17, 2024

v0.4.6

Nov 12, 2024

v0.4.5

Nov 6, 2024

v0.4.4

Oct 24, 2024

v0.4.3

Oct 23, 2024

v0.4.2

Oct 3, 2024

v0.4.1

Sep 28, 2024

v0.4.0

Sep 23, 2024

v0.3.2

Aug 28, 2024

v0.3.1

Aug 27, 2024

v0.3.0

Aug 23, 2024

v0.2.4

Aug 20, 2024

v0.2.3

Aug 16, 2024

v0.2.2

Aug 16, 2024

v0.2.1

Aug 11, 2024

Changes in this version

+ type CollectiveDeviceListProto struct

+ IotaReplicaGroupList *IotaReplicaGroupListProto

+ ReplicaGroups []*ReplicaGroup

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

+ func (*CollectiveDeviceListProto) ProtoMessage()

+ func (x *CollectiveDeviceListProto) GetIotaReplicaGroupList() *IotaReplicaGroupListProto

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

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

+ func (x *CollectiveDeviceListProto) Reset()

+ func (x *CollectiveDeviceListProto) String() string

type DebugOptions

+ LegacyCommandBufferCustomCallTargets []string

+ XlaCmdBufferTraceCacheSize int64

+ XlaGpuAutotuneGemmRtol float32

+ XlaGpuPerFusionAutotuneCacheDir string

+ XlaGpuRunPostLayoutCollectivePipeliner bool

+ XlaGpuTempBufferUseSeparateColor bool

+ XlaGpuUnsafePipelinedLoopAnnotator bool

+ XlaSyntaxSugarAsyncOps bool

+ XlaUseShardy bool

+ func (x *DebugOptions) GetLegacyCommandBufferCustomCallTargets() []string

+ func (x *DebugOptions) GetXlaCmdBufferTraceCacheSize() int64

+ func (x *DebugOptions) GetXlaGpuAutotuneGemmRtol() float32

+ func (x *DebugOptions) GetXlaGpuPerFusionAutotuneCacheDir() string

+ func (x *DebugOptions) GetXlaGpuRunPostLayoutCollectivePipeliner() bool

+ func (x *DebugOptions) GetXlaGpuTempBufferUseSeparateColor() bool

+ func (x *DebugOptions) GetXlaGpuUnsafePipelinedLoopAnnotator() bool

+ func (x *DebugOptions) GetXlaSyntaxSugarAsyncOps() bool

+ func (x *DebugOptions) GetXlaUseShardy() bool

type GpuDeviceInfoProto

+ RegistersPerBlockLimit int64

+ RegistersPerCoreLimit int64

+ func (x *GpuDeviceInfoProto) GetRegistersPerBlockLimit() int64

+ func (x *GpuDeviceInfoProto) GetRegistersPerCoreLimit() int64

type HloInstructionProto

+ CollectiveDeviceList *CollectiveDeviceListProto

+ func (x *HloInstructionProto) GetCollectiveDeviceList() *CollectiveDeviceListProto

+ type IotaReplicaGroupListProto struct

+ IotaReshapeDims []int64

+ IotaTransposePerm []int32

+ NumDevicesPerGroup int64

+ NumReplicaGroups int64

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

+ func (*IotaReplicaGroupListProto) ProtoMessage()

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

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

+ func (x *IotaReplicaGroupListProto) GetNumDevicesPerGroup() int64

+ func (x *IotaReplicaGroupListProto) GetNumReplicaGroups() int64

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

+ func (x *IotaReplicaGroupListProto) Reset()

+ func (x *IotaReplicaGroupListProto) String() string

v0.2.0

Jul 18, 2024

v0.1.2

Jul 5, 2024

v0.1.1

Jul 4, 2024

v0.0.1

Jul 3, 2024

Changes in this version

+ var ActivationMode_name = map[int32]string

+ var ActivationMode_value = map[string]int32

+ var AlgorithmProto_MathType_name = map[int32]string

+ var AlgorithmProto_MathType_value = map[string]int32

+ var AutotuneResult_FailureKind_name = map[int32]string

+ var AutotuneResult_FailureKind_value = map[string]int32

+ var ChannelHandle_ChannelType_name = map[int32]string

+ var ChannelHandle_ChannelType_value = map[string]int32

+ var CompilationEvent_name = map[int32]string

+ var CompilationEvent_value = map[string]int32

+ var ConvolutionKind_name = map[int32]string

+ var ConvolutionKind_value = map[string]int32

+ var ConvolutionMode_name = map[int32]string

+ var ConvolutionMode_value = map[string]int32

+ var CustomCallApiVersion_name = map[int32]string

+ var CustomCallApiVersion_value = map[string]int32

+ var CustomCallSchedule_name = map[int32]string

+ var CustomCallSchedule_value = map[string]int32

+ var DataLayout_name = map[int32]string

+ var DataLayout_value = map[string]int32

+ var DataType_name = map[int32]string

+ var DataType_value = map[string]int32

+ var DebugOptions_CollectiveOpType_name = map[int32]string

+ var DebugOptions_CollectiveOpType_value = map[string]int32

+ var DebugOptions_CommandBufferCmdType_name = map[int32]string

+ var DebugOptions_CommandBufferCmdType_value = map[string]int32

+ var DebugOptions_PartitioningAlgorithm_name = map[int32]string

+ var DebugOptions_PartitioningAlgorithm_value = map[string]int32

+ var DebugOptions_ShapeChecks_name = map[int32]string

+ var DebugOptions_ShapeChecks_value = map[string]int32

+ var DebugOptions_StepMarkerLocation_name = map[int32]string

+ var DebugOptions_StepMarkerLocation_value = map[string]int32

+ var DebugOptions_WhileLoopUnrolling_name = map[int32]string

+ var DebugOptions_WhileLoopUnrolling_value = map[string]int32

+ var DimLevelType_name = map[int32]string

+ var DimLevelType_value = map[string]int32

+ var FMHAMaskKind_name = map[int32]string

+ var FMHAMaskKind_value = map[string]int32

+ var FftType_name = map[int32]string

+ var FftType_value = map[string]int32

+ var File_tsl_protobuf_dnn_proto protoreflect.FileDescriptor

+ var File_xla_autotune_results_proto protoreflect.FileDescriptor

+ var File_xla_autotuning_proto protoreflect.FileDescriptor

+ var File_xla_pjrt_compile_options_proto protoreflect.FileDescriptor

+ var File_xla_service_hlo_proto protoreflect.FileDescriptor

+ var File_xla_stream_executor_device_description_proto protoreflect.FileDescriptor

+ var File_xla_xla_data_proto protoreflect.FileDescriptor

+ var File_xla_xla_proto protoreflect.FileDescriptor

+ var FilterLayout_name = map[int32]string

+ var FilterLayout_value = map[string]int32

+ var FusedMHAKind_name = map[int32]string

+ var FusedMHAKind_value = map[string]int32

+ var HeapSimulatorTrace_Event_Kind_name = map[int32]string

+ var HeapSimulatorTrace_Event_Kind_value = map[string]int32

+ var HloModuleConfigProto_FusionConfigCollection_name = map[int32]string

+ var HloModuleConfigProto_FusionConfigCollection_value = map[string]int32

+ var HloModuleProto_ProfileType_name = map[int32]string

+ var HloModuleProto_ProfileType_value = map[string]int32

+ var Kind_name = map[int32]string

+ var Kind_value = map[string]int32

+ var NormKind_name = map[int32]string

+ var NormKind_value = map[string]int32

+ var OpSharding_ShardGroupType_name = map[int32]string

+ var OpSharding_ShardGroupType_value = map[string]int32

+ var OpSharding_Type_name = map[int32]string

+ var OpSharding_Type_value = map[string]int32

+ var PaddingType_name = map[int32]string

+ var PaddingType_value = map[string]int32

+ var PrecisionConfig_Algorithm_name = map[int32]string

+ var PrecisionConfig_Algorithm_value = map[string]int32

+ var PrecisionConfig_Precision_name = map[int32]string

+ var PrecisionConfig_Precision_value = map[string]int32

+ var PrimitiveType_name = map[int32]string

+ var PrimitiveType_value = map[string]int32

+ var ProfileSource_name = map[int32]string

+ var ProfileSource_value = map[string]int32

+ var ProfileType_name = map[int32]string

+ var ProfileType_value = map[string]int32

+ var RandomAlgorithm_name = map[int32]string

+ var RandomAlgorithm_value = map[string]int32

+ var RandomDistribution_name = map[int32]string

+ var RandomDistribution_value = map[string]int32

+ var SparsityType_name = map[int32]string

+ var SparsityType_value = map[string]int32

+ var TriangularSolveOptions_Transpose_name = map[int32]string

+ var TriangularSolveOptions_Transpose_value = map[string]int32

+ type ActivationMode int32

+ const ActivationMode_kBandPass

+ const ActivationMode_kElu

+ const ActivationMode_kGeluExact

+ const ActivationMode_kLeakyRelu

+ const ActivationMode_kNone

+ const ActivationMode_kRelu

+ const ActivationMode_kRelu6

+ const ActivationMode_kReluX

+ const ActivationMode_kSigmoid

+ const ActivationMode_kTanh

+ func (ActivationMode) Descriptor() protoreflect.EnumDescriptor

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

+ func (ActivationMode) Type() protoreflect.EnumType

+ func (x ActivationMode) Enum() *ActivationMode

+ func (x ActivationMode) Number() protoreflect.EnumNumber

+ func (x ActivationMode) String() string

+ type AlgorithmConfigProto struct

+ OptionalAlgorithm isAlgorithmConfigProto_OptionalAlgorithm

+ OptionalAlgorithmNoScratch isAlgorithmConfigProto_OptionalAlgorithmNoScratch

+ OptionalScratchSize isAlgorithmConfigProto_OptionalScratchSize

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

+ func (*AlgorithmConfigProto) ProtoMessage()

+ func (m *AlgorithmConfigProto) GetOptionalAlgorithm() isAlgorithmConfigProto_OptionalAlgorithm

+ func (m *AlgorithmConfigProto) GetOptionalAlgorithmNoScratch() isAlgorithmConfigProto_OptionalAlgorithmNoScratch

+ func (m *AlgorithmConfigProto) GetOptionalScratchSize() isAlgorithmConfigProto_OptionalScratchSize

+ func (x *AlgorithmConfigProto) GetAlgorithm() *AlgorithmProto

+ func (x *AlgorithmConfigProto) GetAlgorithmNoScratch() *AlgorithmProto

+ func (x *AlgorithmConfigProto) GetScratchSize() int64

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

+ func (x *AlgorithmConfigProto) Reset()

+ func (x *AlgorithmConfigProto) String() string

+ type AlgorithmConfigProto_Algorithm struct

+ Algorithm *AlgorithmProto

+ type AlgorithmConfigProto_AlgorithmNoScratch struct

+ AlgorithmNoScratch *AlgorithmProto

+ type AlgorithmConfigProto_ScratchSize struct

+ ScratchSize int64

+ type AlgorithmProto struct

+ AlgoId int64

+ IsCudnnFrontend bool

+ MathType AlgorithmProto_MathType

+ TuningKnobs map[int64]int64

+ WorkspaceSize *wrapperspb.UInt64Value

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

+ func (*AlgorithmProto) ProtoMessage()

+ func (x *AlgorithmProto) GetAlgoId() int64

+ func (x *AlgorithmProto) GetIsCudnnFrontend() bool

+ func (x *AlgorithmProto) GetMathType() AlgorithmProto_MathType

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

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

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

+ func (x *AlgorithmProto) Reset()

+ func (x *AlgorithmProto) String() string

+ type AlgorithmProto_MathType int32

+ const AlgorithmProto_DEFAULT_MATH

+ const AlgorithmProto_TENSOR_OP_MATH

+ func (AlgorithmProto_MathType) Descriptor() protoreflect.EnumDescriptor

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

+ func (AlgorithmProto_MathType) Type() protoreflect.EnumType

+ func (x AlgorithmProto_MathType) Enum() *AlgorithmProto_MathType

+ func (x AlgorithmProto_MathType) Number() protoreflect.EnumNumber

+ func (x AlgorithmProto_MathType) String() string

+ type AutotuneResult struct

+ Failure *AutotuneResult_FailureResult

+ Key isAutotuneResult_Key

+ RunTime *durationpb.Duration

+ ScratchBytes int64

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

+ func (*AutotuneResult) ProtoMessage()

+ func (m *AutotuneResult) GetKey() isAutotuneResult_Key

+ func (x *AutotuneResult) GetAlgorithm() *AlgorithmProto

+ func (x *AutotuneResult) GetConv() *AutotuneResult_ConvKey

+ func (x *AutotuneResult) GetCudaConvPlan() *AutotuneResult_CudaConvPlanKey

+ func (x *AutotuneResult) GetFailure() *AutotuneResult_FailureResult

+ func (x *AutotuneResult) GetGemm() *AutotuneResult_GemmKey

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

+ func (x *AutotuneResult) GetScratchBytes() int64

+ func (x *AutotuneResult) GetTriton() *AutotuneResult_TritonGemmKey

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

+ func (x *AutotuneResult) Reset()

+ func (x *AutotuneResult) String() string

+ type AutotuneResult_Algorithm struct

+ Algorithm *AlgorithmProto

+ type AutotuneResult_Conv struct

+ Conv *AutotuneResult_ConvKey

+ type AutotuneResult_ConvKey struct

+ Algorithm int64

+ TensorOpsEnabled bool

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

+ func (*AutotuneResult_ConvKey) ProtoMessage()

+ func (x *AutotuneResult_ConvKey) GetAlgorithm() int64

+ func (x *AutotuneResult_ConvKey) GetTensorOpsEnabled() bool

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

+ func (x *AutotuneResult_ConvKey) Reset()

+ func (x *AutotuneResult_ConvKey) String() string

+ type AutotuneResult_CudaConvPlan struct

+ CudaConvPlan *AutotuneResult_CudaConvPlanKey

+ type AutotuneResult_CudaConvPlanKey struct

+ ExecPlanId string

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

+ func (*AutotuneResult_CudaConvPlanKey) ProtoMessage()

+ func (x *AutotuneResult_CudaConvPlanKey) GetExecPlanId() string

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

+ func (x *AutotuneResult_CudaConvPlanKey) Reset()

+ func (x *AutotuneResult_CudaConvPlanKey) String() string

+ type AutotuneResult_FailureKind int32

+ const AutotuneResult_DISQUALIFIED

+ const AutotuneResult_REDZONE_MODIFIED

+ const AutotuneResult_UNKNOWN

+ const AutotuneResult_WRONG_RESULT

+ func (AutotuneResult_FailureKind) Descriptor() protoreflect.EnumDescriptor

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

+ func (AutotuneResult_FailureKind) Type() protoreflect.EnumType

+ func (x AutotuneResult_FailureKind) Enum() *AutotuneResult_FailureKind

+ func (x AutotuneResult_FailureKind) Number() protoreflect.EnumNumber

+ func (x AutotuneResult_FailureKind) String() string

+ type AutotuneResult_FailureResult struct

+ BufferAddress int64

+ Key isAutotuneResult_FailureResult_Key

+ Kind AutotuneResult_FailureKind

+ Msg string

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

+ func (*AutotuneResult_FailureResult) ProtoMessage()

+ func (m *AutotuneResult_FailureResult) GetKey() isAutotuneResult_FailureResult_Key

+ func (x *AutotuneResult_FailureResult) GetBufferAddress() int64

+ func (x *AutotuneResult_FailureResult) GetKind() AutotuneResult_FailureKind

+ func (x *AutotuneResult_FailureResult) GetMsg() string

+ func (x *AutotuneResult_FailureResult) GetReferenceAlgorithm() *AlgorithmProto

+ func (x *AutotuneResult_FailureResult) GetReferenceConv() *AutotuneResult_ConvKey

+ func (x *AutotuneResult_FailureResult) GetReferenceCudaConvPlan() *AutotuneResult_CudaConvPlanKey

+ func (x *AutotuneResult_FailureResult) GetReferenceGemm() *AutotuneResult_GemmKey

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

+ func (x *AutotuneResult_FailureResult) Reset()

+ func (x *AutotuneResult_FailureResult) String() string

+ type AutotuneResult_FailureResult_ReferenceAlgorithm struct

+ ReferenceAlgorithm *AlgorithmProto

+ type AutotuneResult_FailureResult_ReferenceConv struct

+ ReferenceConv *AutotuneResult_ConvKey

+ type AutotuneResult_FailureResult_ReferenceCudaConvPlan struct

+ ReferenceCudaConvPlan *AutotuneResult_CudaConvPlanKey

+ type AutotuneResult_FailureResult_ReferenceGemm struct

+ ReferenceGemm *AutotuneResult_GemmKey

+ type AutotuneResult_Gemm struct

+ Gemm *AutotuneResult_GemmKey

+ type AutotuneResult_GemmKey struct

+ Algorithm int64

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

+ func (*AutotuneResult_GemmKey) ProtoMessage()

+ func (x *AutotuneResult_GemmKey) GetAlgorithm() int64

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

+ func (x *AutotuneResult_GemmKey) Reset()

+ func (x *AutotuneResult_GemmKey) String() string

+ type AutotuneResult_Triton struct

+ Triton *AutotuneResult_TritonGemmKey

+ type AutotuneResult_TritonGemmKey struct

+ BlockK int64

+ BlockM int64

+ BlockN int64

+ NumCtas int64

+ NumStages int64

+ NumWarps int64

+ SplitK int64

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

+ func (*AutotuneResult_TritonGemmKey) ProtoMessage()

+ func (x *AutotuneResult_TritonGemmKey) GetBlockK() int64

+ func (x *AutotuneResult_TritonGemmKey) GetBlockM() int64

+ func (x *AutotuneResult_TritonGemmKey) GetBlockN() int64

+ func (x *AutotuneResult_TritonGemmKey) GetNumCtas() int64

+ func (x *AutotuneResult_TritonGemmKey) GetNumStages() int64

+ func (x *AutotuneResult_TritonGemmKey) GetNumWarps() int64

+ func (x *AutotuneResult_TritonGemmKey) GetSplitK() int64

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

+ func (x *AutotuneResult_TritonGemmKey) Reset()

+ func (x *AutotuneResult_TritonGemmKey) String() string

+ type AutotuneResults struct

+ Results []*AutotuneResults_Entry

+ Version int32

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

+ func (*AutotuneResults) ProtoMessage()

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

+ func (x *AutotuneResults) GetVersion() int32

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

+ func (x *AutotuneResults) Reset()

+ func (x *AutotuneResults) String() string

+ type AutotuneResults_Entry struct

+ Device string

+ Hlo string

+ Result *AutotuneResult

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

+ func (*AutotuneResults_Entry) ProtoMessage()

+ func (x *AutotuneResults_Entry) GetDevice() string

+ func (x *AutotuneResults_Entry) GetHlo() string

+ func (x *AutotuneResults_Entry) GetResult() *AutotuneResult

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

+ func (x *AutotuneResults_Entry) Reset()

+ func (x *AutotuneResults_Entry) String() string

+ type AutotuningLog struct

+ BlasVersion string

+ ComputeCapability *ComputeCapability

+ CudnnVersion *CudnnVersion

+ DevicePciBusId string

+ FusionCount int64

+ FusionName string

+ Instr *anypb.Any

+ Results []*AutotuneResult

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

+ func (*AutotuningLog) ProtoMessage()

+ func (x *AutotuningLog) GetBlasVersion() string

+ func (x *AutotuningLog) GetComputeCapability() *ComputeCapability

+ func (x *AutotuningLog) GetCudnnVersion() *CudnnVersion

+ func (x *AutotuningLog) GetDevicePciBusId() string

+ func (x *AutotuningLog) GetFusionCount() int64

+ func (x *AutotuningLog) GetFusionName() string

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

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

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

+ func (x *AutotuningLog) Reset()

+ func (x *AutotuningLog) String() string

+ type AutotuningLogs struct

+ Logs []*AutotuningLog

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

+ func (*AutotuningLogs) ProtoMessage()

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

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

+ func (x *AutotuningLogs) Reset()

+ func (x *AutotuningLogs) String() string

+ type BufferAllocationProto struct

+ Assigned []*BufferAllocationProto_Assigned

+ Color int64

+ Index int64

+ IsConstant bool

+ IsEntryComputationParameter bool

+ IsThreadLocal bool

+ IsTuple bool

+ MaybeLiveOut bool

+ ParameterNumber int64

+ ParameterShapeIndex []int64

+ Size int64

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

+ func (*BufferAllocationProto) ProtoMessage()

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

+ func (x *BufferAllocationProto) GetColor() int64

+ func (x *BufferAllocationProto) GetIndex() int64

+ func (x *BufferAllocationProto) GetIsConstant() bool

+ func (x *BufferAllocationProto) GetIsEntryComputationParameter() bool

+ func (x *BufferAllocationProto) GetIsThreadLocal() bool

+ func (x *BufferAllocationProto) GetIsTuple() bool

+ func (x *BufferAllocationProto) GetMaybeLiveOut() bool

+ func (x *BufferAllocationProto) GetParameterNumber() int64

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

+ func (x *BufferAllocationProto) GetSize() int64

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

+ func (x *BufferAllocationProto) Reset()

+ func (x *BufferAllocationProto) String() string

+ type BufferAllocationProto_Assigned struct

+ LogicalBufferId int64

+ Offset int64

+ Size int64

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

+ func (*BufferAllocationProto_Assigned) ProtoMessage()

+ func (x *BufferAllocationProto_Assigned) GetLogicalBufferId() int64

+ func (x *BufferAllocationProto_Assigned) GetOffset() int64

+ func (x *BufferAllocationProto_Assigned) GetSize() int64

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

+ func (x *BufferAllocationProto_Assigned) Reset()

+ func (x *BufferAllocationProto_Assigned) String() string

+ type BufferAssignmentProto struct

+ BufferAliases []*BufferAssignmentProto_BufferAlias

+ BufferAllocations []*BufferAllocationProto

+ HeapSimulatorTraces []*HeapSimulatorTrace

+ LogicalBuffers []*LogicalBufferProto

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

+ func (*BufferAssignmentProto) ProtoMessage()

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

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

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

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

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

+ func (x *BufferAssignmentProto) Reset()

+ func (x *BufferAssignmentProto) String() string

+ type BufferAssignmentProto_BufferAlias struct

+ Location *LogicalBufferProto_Location

+ SourceBufferId int64

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

+ func (*BufferAssignmentProto_BufferAlias) ProtoMessage()

+ func (x *BufferAssignmentProto_BufferAlias) GetLocation() *LogicalBufferProto_Location

+ func (x *BufferAssignmentProto_BufferAlias) GetSourceBufferId() int64

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

+ func (x *BufferAssignmentProto_BufferAlias) Reset()

+ func (x *BufferAssignmentProto_BufferAlias) String() string

+ type ChannelHandle struct

+ Handle int64

+ Type ChannelHandle_ChannelType

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

+ func (*ChannelHandle) ProtoMessage()

+ func (x *ChannelHandle) GetHandle() int64

+ func (x *ChannelHandle) GetType() ChannelHandle_ChannelType

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

+ func (x *ChannelHandle) Reset()

+ func (x *ChannelHandle) String() string

+ type ChannelHandle_ChannelType int32

+ const ChannelHandle_CHANNEL_TYPE_INVALID

+ const ChannelHandle_DEVICE_TO_DEVICE

+ const ChannelHandle_DEVICE_TO_HOST

+ const ChannelHandle_HOST_TO_DEVICE

+ func (ChannelHandle_ChannelType) Descriptor() protoreflect.EnumDescriptor

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

+ func (ChannelHandle_ChannelType) Type() protoreflect.EnumType

+ func (x ChannelHandle_ChannelType) Enum() *ChannelHandle_ChannelType

+ func (x ChannelHandle_ChannelType) Number() protoreflect.EnumNumber

+ func (x ChannelHandle_ChannelType) String() string

+ type CholeskyOptions struct

+ Lower bool

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

+ func (*CholeskyOptions) ProtoMessage()

+ func (x *CholeskyOptions) GetLower() bool

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

+ func (x *CholeskyOptions) Reset()

+ func (x *CholeskyOptions) String() string

+ type CompilationEnvironmentsProto struct

+ Environments []*anypb.Any

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

+ func (*CompilationEnvironmentsProto) ProtoMessage()

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

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

+ func (x *CompilationEnvironmentsProto) Reset()

+ func (x *CompilationEnvironmentsProto) String() string

+ type CompilationEvent int32

+ const CompilationEvent_COMPILATION_EVENT_FIRST_COMPILATION

+ const CompilationEvent_COMPILATION_EVENT_RECOMPILATION

+ const CompilationEvent_COMPILATION_EVENT_UNKNOWN_EVENT

+ func (CompilationEvent) Descriptor() protoreflect.EnumDescriptor

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

+ func (CompilationEvent) Type() protoreflect.EnumType

+ func (x CompilationEvent) Enum() *CompilationEvent

+ func (x CompilationEvent) Number() protoreflect.EnumNumber

+ func (x CompilationEvent) String() string

+ type CompileOptionsProto struct

+ ArgumentLayouts []*ShapeProto

+ CompilePortableExecutable bool

+ EnvOptionOverrides map[string]*OptionOverrideProto

+ ExecutableBuildOptions *ExecutableBuildOptionsProto

+ ParameterIsTupledArguments bool

+ ProfileVersion int64

+ SerializedMultiSliceConfig []byte

+ TargetConfig *GpuTargetConfigProto

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

+ func (*CompileOptionsProto) ProtoMessage()

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

+ func (x *CompileOptionsProto) GetCompilePortableExecutable() bool

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

+ func (x *CompileOptionsProto) GetExecutableBuildOptions() *ExecutableBuildOptionsProto

+ func (x *CompileOptionsProto) GetParameterIsTupledArguments() bool

+ func (x *CompileOptionsProto) GetProfileVersion() int64

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

+ func (x *CompileOptionsProto) GetTargetConfig() *GpuTargetConfigProto

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

+ func (x *CompileOptionsProto) Reset()

+ func (x *CompileOptionsProto) String() string

+ type CompileRequest struct

+ Computation *HloModuleProto

+ ExecutionOptions *ExecutionOptions

+ InputShapeWithLayout []*ShapeProto

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

+ func (*CompileRequest) ProtoMessage()

+ func (x *CompileRequest) GetComputation() *HloModuleProto

+ func (x *CompileRequest) GetExecutionOptions() *ExecutionOptions

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

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

+ func (x *CompileRequest) Reset()

+ func (x *CompileRequest) String() string

+ type CompileResponse struct

+ Handle *ExecutionHandle

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

+ func (*CompileResponse) ProtoMessage()

+ func (x *CompileResponse) GetHandle() *ExecutionHandle

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

+ func (x *CompileResponse) Reset()

+ func (x *CompileResponse) String() string

+ type ComputationStats struct

+ FlopCount float64

+ TranscendentalCount float64

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

+ func (*ComputationStats) ProtoMessage()

+ func (x *ComputationStats) GetFlopCount() float64

+ func (x *ComputationStats) GetTranscendentalCount() float64

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

+ func (x *ComputationStats) Reset()

+ func (x *ComputationStats) String() string

+ type ComputeCapability struct

+ Major int32

+ Minor int32

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

+ func (*ComputeCapability) ProtoMessage()

+ func (x *ComputeCapability) GetMajor() int32

+ func (x *ComputeCapability) GetMinor() int32

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

+ func (x *ComputeCapability) Reset()

+ func (x *ComputeCapability) String() string

+ type ComputeConstantGraphRequest struct

+ Computation *HloModuleProto

+ OutputLayout *LayoutProto

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

+ func (*ComputeConstantGraphRequest) ProtoMessage()

+ func (x *ComputeConstantGraphRequest) GetComputation() *HloModuleProto

+ func (x *ComputeConstantGraphRequest) GetOutputLayout() *LayoutProto

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

+ func (x *ComputeConstantGraphRequest) Reset()

+ func (x *ComputeConstantGraphRequest) String() string

+ type ComputeConstantResponse struct

+ Literal *LiteralProto

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

+ func (*ComputeConstantResponse) ProtoMessage()

+ func (x *ComputeConstantResponse) GetLiteral() *LiteralProto

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

+ func (x *ComputeConstantResponse) Reset()

+ func (x *ComputeConstantResponse) String() string

+ type ConvolutionDescriptorProto struct

+ ComputeMode DataType

+ ConvolutionMode ConvolutionMode

+ Dilations []int64

+ GroupCount int32

+ Name string

+ Paddings []int64

+ Strides []int64

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

+ func (*ConvolutionDescriptorProto) ProtoMessage()

+ func (x *ConvolutionDescriptorProto) GetComputeMode() DataType

+ func (x *ConvolutionDescriptorProto) GetConvolutionMode() ConvolutionMode

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

+ func (x *ConvolutionDescriptorProto) GetGroupCount() int32

+ func (x *ConvolutionDescriptorProto) GetName() string

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

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

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

+ func (x *ConvolutionDescriptorProto) Reset()

+ func (x *ConvolutionDescriptorProto) String() string

+ type ConvolutionDimensionNumbers struct

+ InputBatchDimension int64

+ InputFeatureDimension int64

+ InputSpatialDimensions []int64

+ KernelInputFeatureDimension int64

+ KernelOutputFeatureDimension int64

+ KernelSpatialDimensions []int64

+ OutputBatchDimension int64

+ OutputFeatureDimension int64

+ OutputSpatialDimensions []int64

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

+ func (*ConvolutionDimensionNumbers) ProtoMessage()

+ func (x *ConvolutionDimensionNumbers) GetInputBatchDimension() int64

+ func (x *ConvolutionDimensionNumbers) GetInputFeatureDimension() int64

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

+ func (x *ConvolutionDimensionNumbers) GetKernelInputFeatureDimension() int64

+ func (x *ConvolutionDimensionNumbers) GetKernelOutputFeatureDimension() int64

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

+ func (x *ConvolutionDimensionNumbers) GetOutputBatchDimension() int64

+ func (x *ConvolutionDimensionNumbers) GetOutputFeatureDimension() int64

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

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

+ func (x *ConvolutionDimensionNumbers) Reset()

+ func (x *ConvolutionDimensionNumbers) String() string

+ type ConvolutionKind int32

+ const ConvolutionKind_BACKWARD_DATA

+ const ConvolutionKind_BACKWARD_FILTER

+ const ConvolutionKind_FORWARD

+ const ConvolutionKind_FORWARD_BIAS_ACTIVATION

+ const ConvolutionKind_FORWARD_GRAPH

+ const ConvolutionKind_INVALID

+ func (ConvolutionKind) Descriptor() protoreflect.EnumDescriptor

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

+ func (ConvolutionKind) Type() protoreflect.EnumType

+ func (x ConvolutionKind) Enum() *ConvolutionKind

+ func (x ConvolutionKind) Number() protoreflect.EnumNumber

+ func (x ConvolutionKind) String() string

+ type ConvolutionMode int32

+ const ConvolutionMode_CONVOLUTION

+ const ConvolutionMode_CROSS_CORRELATION

+ func (ConvolutionMode) Descriptor() protoreflect.EnumDescriptor

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

+ func (ConvolutionMode) Type() protoreflect.EnumType

+ func (x ConvolutionMode) Enum() *ConvolutionMode

+ func (x ConvolutionMode) Number() protoreflect.EnumNumber

+ func (x ConvolutionMode) String() string

+ type CreateChannelHandleRequest struct

+ ChannelType ChannelHandle_ChannelType

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

+ func (*CreateChannelHandleRequest) ProtoMessage()

+ func (x *CreateChannelHandleRequest) GetChannelType() ChannelHandle_ChannelType

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

+ func (x *CreateChannelHandleRequest) Reset()

+ func (x *CreateChannelHandleRequest) String() string

+ type CreateChannelHandleResponse struct

+ Channel *ChannelHandle

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

+ func (*CreateChannelHandleResponse) ProtoMessage()

+ func (x *CreateChannelHandleResponse) GetChannel() *ChannelHandle

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

+ func (x *CreateChannelHandleResponse) Reset()

+ func (x *CreateChannelHandleResponse) String() string

+ type CrossProgramPrefetch struct

+ Index []int64

+ Offset int64

+ Parameter int64

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

+ func (*CrossProgramPrefetch) ProtoMessage()

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

+ func (x *CrossProgramPrefetch) GetOffset() int64

+ func (x *CrossProgramPrefetch) GetParameter() int64

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

+ func (x *CrossProgramPrefetch) Reset()

+ func (x *CrossProgramPrefetch) String() string

+ type CudaComputeCapabilityProto struct

+ Major int32

+ Minor int32

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

+ func (*CudaComputeCapabilityProto) ProtoMessage()

+ func (x *CudaComputeCapabilityProto) GetMajor() int32

+ func (x *CudaComputeCapabilityProto) GetMinor() int32

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

+ func (x *CudaComputeCapabilityProto) Reset()

+ func (x *CudaComputeCapabilityProto) String() string

+ type CudnnVersion struct

+ Major int32

+ Minor int32

+ Patch int32

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

+ func (*CudnnVersion) ProtoMessage()

+ func (x *CudnnVersion) GetMajor() int32

+ func (x *CudnnVersion) GetMinor() int32

+ func (x *CudnnVersion) GetPatch() int32

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

+ func (x *CudnnVersion) Reset()

+ func (x *CudnnVersion) String() string

+ type CustomCallApiVersion int32

+ const CustomCallApiVersion_API_VERSION_ORIGINAL

+ const CustomCallApiVersion_API_VERSION_STATUS_RETURNING

+ const CustomCallApiVersion_API_VERSION_STATUS_RETURNING_UNIFIED

+ const CustomCallApiVersion_API_VERSION_TYPED_FFI

+ const CustomCallApiVersion_API_VERSION_UNSPECIFIED

+ func (CustomCallApiVersion) Descriptor() protoreflect.EnumDescriptor

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

+ func (CustomCallApiVersion) Type() protoreflect.EnumType

+ func (x CustomCallApiVersion) Enum() *CustomCallApiVersion

+ func (x CustomCallApiVersion) Number() protoreflect.EnumNumber

+ func (x CustomCallApiVersion) String() string

+ type CustomCallSchedule int32

+ const CustomCallSchedule_SCHEDULE_EARLIEST

+ const CustomCallSchedule_SCHEDULE_LATEST

+ const CustomCallSchedule_SCHEDULE_NONE

+ func (CustomCallSchedule) Descriptor() protoreflect.EnumDescriptor

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

+ func (CustomCallSchedule) Type() protoreflect.EnumType

+ func (x CustomCallSchedule) Enum() *CustomCallSchedule

+ func (x CustomCallSchedule) Number() protoreflect.EnumNumber

+ func (x CustomCallSchedule) String() string

+ type DataLayout int32

+ const DataLayout_kBatchDepthYX

+ const DataLayout_kBatchDepthYX32

+ const DataLayout_kBatchDepthYX4

+ const DataLayout_kBatchYXDepth

+ const DataLayout_kYXBatchDepth

+ const DataLayout_kYXDepthBatch

+ func (DataLayout) Descriptor() protoreflect.EnumDescriptor

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

+ func (DataLayout) Type() protoreflect.EnumType

+ func (x DataLayout) Enum() *DataLayout

+ func (x DataLayout) Number() protoreflect.EnumNumber

+ func (x DataLayout) String() string

+ type DataType int32

+ const DataType_kBF16

+ const DataType_kComplexDouble

+ const DataType_kComplexFloat

+ const DataType_kDouble

+ const DataType_kF8E4M3FN

+ const DataType_kF8E4M3FNUZ

+ const DataType_kF8E5M2

+ const DataType_kF8E5M2FNUZ

+ const DataType_kFloat

+ const DataType_kHalf

+ const DataType_kInt32

+ const DataType_kInt64

+ const DataType_kInt8

+ func (DataType) Descriptor() protoreflect.EnumDescriptor

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

+ func (DataType) Type() protoreflect.EnumType

+ func (x DataType) Enum() *DataType

+ func (x DataType) Number() protoreflect.EnumNumber

+ func (x DataType) String() string

+ type DebugOptions struct

+ XlaAllowExcessPrecision bool

+ XlaAllowScalarIndexDynamicOps bool

+ XlaBackendExtraOptions map[string]string

+ XlaBackendOptimizationLevel int32

+ XlaCpuEnableConcurrencyOptimizedScheduler bool

+ XlaCpuEnableCustomMatmulTiling bool

+ XlaCpuEnableExperimentalDeallocation bool

+ XlaCpuEnableFastMath bool

+ XlaCpuEnableFastMinMax bool

+ XlaCpuEnableMlirFusionOutlining bool

+ XlaCpuEnableMlirTilingAndFusion bool

+ XlaCpuEnableXprofTraceme bool

+ XlaCpuFastMathHonorDivision bool

+ XlaCpuFastMathHonorFunctions bool

+ XlaCpuFastMathHonorInfs bool

+ XlaCpuFastMathHonorNans bool

+ XlaCpuMatmulTilingKDim int64

+ XlaCpuMatmulTilingMDim int64

+ XlaCpuMatmulTilingNDim int64

+ XlaCpuMultiThreadEigen bool

+ XlaCpuPreferVectorWidth int32

+ XlaCpuStrictDotConvMath bool

+ XlaCpuUseAcl bool

+ XlaCpuUseMklDnn bool

+ XlaCpuUseThunkRuntime bool

+ XlaDebugBufferAssignmentShowMax int64

+ XlaDetailedLogging bool

+ XlaDisableAllHloPasses bool

+ XlaDisableHloPasses []string

+ XlaDumpCompressProtos bool

+ XlaDumpDisableMetadata bool

+ XlaDumpEnableMlirPrettyForm bool

+ XlaDumpFusionVisualization bool

+ XlaDumpHloAsDot bool

+ XlaDumpHloAsHtml bool

+ XlaDumpHloAsLongText bool

+ XlaDumpHloAsProto bool

+ XlaDumpHloAsText bool

+ XlaDumpHloAsUrl bool

+ XlaDumpHloModuleRe string

+ XlaDumpHloPassRe string

+ XlaDumpHloPipelineRe string

+ XlaDumpHloSnapshots bool

+ XlaDumpIncludeTimestamp bool

+ XlaDumpLargeConstants bool

+ XlaDumpLatencyHidingSchedule bool

+ XlaDumpMaxHloModules int32

+ XlaDumpModuleMetadata bool

+ XlaDumpTo string

+ XlaEliminateHloImplicitBroadcast bool

+ XlaEmbedIrInExecutable bool

+ XlaEnableDumping bool

+ XlaEnableHloPassesOnly []string

+ XlaForceHostPlatformDeviceCount int32

+ XlaGpuAlgorithmDenylistPath string

+ XlaGpuAllGatherCombineThresholdBytes int64

+ XlaGpuAllReduceBlueconnectNumDevicesPerHost int32

+ XlaGpuAllReduceCombineThresholdBytes int64

+ XlaGpuAsmExtraFlags string

+ XlaGpuAutoSpmdPartitioningMemoryBudgetGb int32

+ XlaGpuAutoSpmdPartitioningMemoryBudgetRatio float32

+ XlaGpuAutotuneLevel int32

+ XlaGpuAutotuneMaxSolutions int64

+ XlaGpuCollectCostModelStats bool

+ XlaGpuCollectiveInflationFactor int32

+ XlaGpuCollectivePermuteDecomposerThreshold int64

+ XlaGpuCopyInsertionUseRegionAnalysis bool

+ XlaGpuCrashOnVerificationFailures bool

+ XlaGpuCublasFallback bool

+ XlaGpuCudaDataDir string

+ XlaGpuCudnnGemmFusionLevel int32

+ XlaGpuDeterministicOps bool

+ XlaGpuDisableAsyncCollectives []DebugOptions_CollectiveOpType

+ XlaGpuDisableGpuasmOptimizations bool

+ XlaGpuDumpAutotuneLogsTo string

+ XlaGpuDumpAutotuneResultsTo string

+ XlaGpuDumpAutotunedGemmFusions bool

+ XlaGpuDumpLlvmir bool

+ XlaGpuEnableAddressComputationFusion bool

+ XlaGpuEnableAllGatherCombineByDim bool

+ XlaGpuEnableAllReduceSplitter bool

+ XlaGpuEnableAnalyticalLatencyEstimator bool

+ XlaGpuEnableApproxCostlyCollectives bool

+ XlaGpuEnableBf16_3WayGemm bool

+ XlaGpuEnableBf16_6WayGemm bool

+ XlaGpuEnableCommandBuffer []DebugOptions_CommandBufferCmdType

+ XlaGpuEnableCubRadixSort bool

+ XlaGpuEnableCublaslt bool

+ XlaGpuEnableCudnnFmha bool

+ XlaGpuEnableCudnnFrontend bool

+ XlaGpuEnableCudnnInt8X32ConvolutionReordering bool

+ XlaGpuEnableCudnnLayerNorm bool

+ XlaGpuEnableCustomFusions bool

+ XlaGpuEnableCustomFusionsRe string

+ XlaGpuEnableDotStrengthReduction bool

+ XlaGpuEnableFastMinMax bool

+ XlaGpuEnableHighestPriorityAsyncStream bool

+ XlaGpuEnableHostMemoryOffloading bool

+ XlaGpuEnableLatencyHidingScheduler bool

+ XlaGpuEnableLibnvptxcompiler bool

+ XlaGpuEnableLlvmModuleCompilationParallelism bool

+ XlaGpuEnableNcclCliqueOptimization bool

+ XlaGpuEnableNcclCommSplitting bool

+ XlaGpuEnableNcclPerStreamComms bool

+ XlaGpuEnableNcclUserBuffers bool

+ XlaGpuEnablePipelinedAllGather bool

+ XlaGpuEnablePipelinedAllReduce bool

+ XlaGpuEnablePipelinedCollectives bool

+ XlaGpuEnablePipelinedP2P bool

+ XlaGpuEnablePipelinedReduceScatter bool

+ XlaGpuEnablePriorityFusion bool

+ XlaGpuEnableReassociationForConvertedAr bool

+ XlaGpuEnableReduceScatterCombineByDim bool

+ XlaGpuEnableReductionEpilogueFusion bool

+ XlaGpuEnableSharedConstants bool

+ XlaGpuEnableSplitKAutotuning bool

+ XlaGpuEnableTritonGemm bool

+ XlaGpuEnableTritonHopper bool

+ XlaGpuEnableTritonSoftmaxFusion bool

+ XlaGpuEnableTritonSoftmaxPriorityFusion bool

+ XlaGpuEnableWhileLoopDoubleBuffering bool

+ XlaGpuEnableWhileLoopReduceScatterCodeMotion bool

+ XlaGpuEnableWhileLoopUnrolling DebugOptions_WhileLoopUnrolling

+ XlaGpuEnsureMinorDotContractionDims bool

+ XlaGpuExcludeNondeterministicOps bool

+ XlaGpuExhaustiveTilingSearch bool

+ XlaGpuFilterKernelsSpillingRegistersOnAutotuning bool

+ XlaGpuForceCompilationParallelism int32

+ XlaGpuForceConvNchw bool

+ XlaGpuForceConvNhwc bool

+ XlaGpuFtz bool

+ XlaGpuFusedAttentionUseCudnnRng bool

+ XlaGpuGemmRewriteSizeThreshold int64

+ XlaGpuGraphEnableConcurrentRegion bool

+ XlaGpuGraphMinGraphSize int32

+ XlaGpuKernelCacheFile string

+ XlaGpuLhsEnableGpuAsyncTracker bool

+ XlaGpuLlvmIrFile []string

+ XlaGpuLlvmVerificationLevel int32

+ XlaGpuLoadAutotuneResultsFrom string

+ XlaGpuMaxMlirKernels int64

+ XlaGpuMemoryLimitSlopFactor int32

+ XlaGpuMlirEmitterLevel int64

+ XlaGpuMockCustomCalls bool

+ XlaGpuMultiStreamedWindowedEinsum bool

+ XlaGpuNcclCollectiveMaxNchannels int64

+ XlaGpuNcclP2PMaxNchannels int64

+ XlaGpuNcclTerminateOnError bool

+ XlaGpuNcclTerminationTimeoutSeconds int64

+ XlaGpuOverrideGemmAutotuner string

+ XlaGpuPgleProfileFileOrDirectoryPath string

+ XlaGpuPtxFile []string

+ XlaGpuReduceScatterCombineThresholdBytes int64

+ XlaGpuRedzonePaddingBytes int64

+ XlaGpuRedzoneScratchMaxMegabytes int64

+ XlaGpuRequireCompleteAotAutotuneResults bool

+ XlaGpuShapeChecks DebugOptions_ShapeChecks

+ XlaGpuShardAutotuning bool

+ XlaGpuSkipMlirKernels int64

+ XlaGpuStrictConvAlgorithmPicker bool

+ XlaGpuTargetConfigFilename string

+ XlaGpuThresholdForWindowedEinsumMib int64

+ XlaGpuTritonFusionLevel int32

+ XlaGpuTritonGemmAny bool

+ XlaGpuTritonGemmDisableReducedPrecisionReduction bool

+ XlaGpuUnsafeFallbackToDriverOnPtxasNotFound bool

+ XlaGpuUseMemcpyLocalP2P bool

+ XlaGpuUseRuntimeFusion bool

+ XlaGpuVerifyTritonFusionNumerics bool

+ XlaHloEvaluatorUseFastPath bool

+ XlaHloGraphAddresses bool

+ XlaHloGraphShardingColor bool

+ XlaHloProfile bool

+ XlaLlvmDisableExpensivePasses bool

+ XlaLlvmEnableAliasScopeMetadata bool

+ XlaLlvmEnableInvariantLoadMetadata bool

+ XlaLlvmEnableNoaliasMetadata bool

+ XlaLlvmForceInlineBeforeSplit bool

+ XlaMultiheapSizeConstraintPerHeap int32

+ XlaPartitioningAlgorithm DebugOptions_PartitioningAlgorithm

+ XlaReduceWindowRewriteBaseLength int64

+ XlaStepMarkerLocation DebugOptions_StepMarkerLocation

+ XlaTestAllInputLayouts bool

+ XlaTestAllOutputLayouts bool

+ XlaTpuDetectInf bool

+ XlaTpuDetectNan bool

+ XlaUseShardonnay bool

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

+ func (*DebugOptions) ProtoMessage()

+ func (x *DebugOptions) GetXlaAllowExcessPrecision() bool

+ func (x *DebugOptions) GetXlaAllowScalarIndexDynamicOps() bool

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

+ func (x *DebugOptions) GetXlaBackendOptimizationLevel() int32

+ func (x *DebugOptions) GetXlaCpuEnableConcurrencyOptimizedScheduler() bool

+ func (x *DebugOptions) GetXlaCpuEnableCustomMatmulTiling() bool

+ func (x *DebugOptions) GetXlaCpuEnableExperimentalDeallocation() bool

+ func (x *DebugOptions) GetXlaCpuEnableFastMath() bool

+ func (x *DebugOptions) GetXlaCpuEnableFastMinMax() bool

+ func (x *DebugOptions) GetXlaCpuEnableMlirFusionOutlining() bool

+ func (x *DebugOptions) GetXlaCpuEnableMlirTilingAndFusion() bool

+ func (x *DebugOptions) GetXlaCpuEnableXprofTraceme() bool

+ func (x *DebugOptions) GetXlaCpuFastMathHonorDivision() bool

+ func (x *DebugOptions) GetXlaCpuFastMathHonorFunctions() bool

+ func (x *DebugOptions) GetXlaCpuFastMathHonorInfs() bool

+ func (x *DebugOptions) GetXlaCpuFastMathHonorNans() bool

+ func (x *DebugOptions) GetXlaCpuMatmulTilingKDim() int64

+ func (x *DebugOptions) GetXlaCpuMatmulTilingMDim() int64

+ func (x *DebugOptions) GetXlaCpuMatmulTilingNDim() int64

+ func (x *DebugOptions) GetXlaCpuMultiThreadEigen() bool

+ func (x *DebugOptions) GetXlaCpuPreferVectorWidth() int32

+ func (x *DebugOptions) GetXlaCpuStrictDotConvMath() bool

+ func (x *DebugOptions) GetXlaCpuUseAcl() bool

+ func (x *DebugOptions) GetXlaCpuUseMklDnn() bool

+ func (x *DebugOptions) GetXlaCpuUseThunkRuntime() bool

+ func (x *DebugOptions) GetXlaDebugBufferAssignmentShowMax() int64

+ func (x *DebugOptions) GetXlaDetailedLogging() bool

+ func (x *DebugOptions) GetXlaDisableAllHloPasses() bool

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

+ func (x *DebugOptions) GetXlaDumpCompressProtos() bool

+ func (x *DebugOptions) GetXlaDumpDisableMetadata() bool

+ func (x *DebugOptions) GetXlaDumpEnableMlirPrettyForm() bool

+ func (x *DebugOptions) GetXlaDumpFusionVisualization() bool

+ func (x *DebugOptions) GetXlaDumpHloAsDot() bool

+ func (x *DebugOptions) GetXlaDumpHloAsHtml() bool

+ func (x *DebugOptions) GetXlaDumpHloAsLongText() bool

+ func (x *DebugOptions) GetXlaDumpHloAsProto() bool

+ func (x *DebugOptions) GetXlaDumpHloAsText() bool

+ func (x *DebugOptions) GetXlaDumpHloAsUrl() bool

+ func (x *DebugOptions) GetXlaDumpHloModuleRe() string

+ func (x *DebugOptions) GetXlaDumpHloPassRe() string

+ func (x *DebugOptions) GetXlaDumpHloPipelineRe() string

+ func (x *DebugOptions) GetXlaDumpHloSnapshots() bool

+ func (x *DebugOptions) GetXlaDumpIncludeTimestamp() bool

+ func (x *DebugOptions) GetXlaDumpLargeConstants() bool

+ func (x *DebugOptions) GetXlaDumpLatencyHidingSchedule() bool

+ func (x *DebugOptions) GetXlaDumpMaxHloModules() int32

+ func (x *DebugOptions) GetXlaDumpModuleMetadata() bool

+ func (x *DebugOptions) GetXlaDumpTo() string

+ func (x *DebugOptions) GetXlaEliminateHloImplicitBroadcast() bool

+ func (x *DebugOptions) GetXlaEmbedIrInExecutable() bool

+ func (x *DebugOptions) GetXlaEnableDumping() bool

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

+ func (x *DebugOptions) GetXlaForceHostPlatformDeviceCount() int32

+ func (x *DebugOptions) GetXlaGpuAlgorithmDenylistPath() string

+ func (x *DebugOptions) GetXlaGpuAllGatherCombineThresholdBytes() int64

+ func (x *DebugOptions) GetXlaGpuAllReduceBlueconnectNumDevicesPerHost() int32

+ func (x *DebugOptions) GetXlaGpuAllReduceCombineThresholdBytes() int64

+ func (x *DebugOptions) GetXlaGpuAsmExtraFlags() string

+ func (x *DebugOptions) GetXlaGpuAutoSpmdPartitioningMemoryBudgetGb() int32

+ func (x *DebugOptions) GetXlaGpuAutoSpmdPartitioningMemoryBudgetRatio() float32

+ func (x *DebugOptions) GetXlaGpuAutotuneLevel() int32

+ func (x *DebugOptions) GetXlaGpuAutotuneMaxSolutions() int64

+ func (x *DebugOptions) GetXlaGpuCollectCostModelStats() bool

+ func (x *DebugOptions) GetXlaGpuCollectiveInflationFactor() int32

+ func (x *DebugOptions) GetXlaGpuCollectivePermuteDecomposerThreshold() int64

+ func (x *DebugOptions) GetXlaGpuCopyInsertionUseRegionAnalysis() bool

+ func (x *DebugOptions) GetXlaGpuCrashOnVerificationFailures() bool

+ func (x *DebugOptions) GetXlaGpuCublasFallback() bool

+ func (x *DebugOptions) GetXlaGpuCudaDataDir() string

+ func (x *DebugOptions) GetXlaGpuCudnnGemmFusionLevel() int32

+ func (x *DebugOptions) GetXlaGpuDeterministicOps() bool

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

+ func (x *DebugOptions) GetXlaGpuDisableGpuasmOptimizations() bool

+ func (x *DebugOptions) GetXlaGpuDumpAutotuneLogsTo() string

+ func (x *DebugOptions) GetXlaGpuDumpAutotuneResultsTo() string

+ func (x *DebugOptions) GetXlaGpuDumpAutotunedGemmFusions() bool

+ func (x *DebugOptions) GetXlaGpuDumpLlvmir() bool

+ func (x *DebugOptions) GetXlaGpuEnableAddressComputationFusion() bool

+ func (x *DebugOptions) GetXlaGpuEnableAllGatherCombineByDim() bool

+ func (x *DebugOptions) GetXlaGpuEnableAllReduceSplitter() bool

+ func (x *DebugOptions) GetXlaGpuEnableAnalyticalLatencyEstimator() bool

+ func (x *DebugOptions) GetXlaGpuEnableApproxCostlyCollectives() bool

+ func (x *DebugOptions) GetXlaGpuEnableBf16_3WayGemm() bool

+ func (x *DebugOptions) GetXlaGpuEnableBf16_6WayGemm() bool

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

+ func (x *DebugOptions) GetXlaGpuEnableCubRadixSort() bool

+ func (x *DebugOptions) GetXlaGpuEnableCublaslt() bool

+ func (x *DebugOptions) GetXlaGpuEnableCudnnFmha() bool

+ func (x *DebugOptions) GetXlaGpuEnableCudnnFrontend() bool

+ func (x *DebugOptions) GetXlaGpuEnableCudnnInt8X32ConvolutionReordering() bool

+ func (x *DebugOptions) GetXlaGpuEnableCudnnLayerNorm() bool

+ func (x *DebugOptions) GetXlaGpuEnableCustomFusions() bool

+ func (x *DebugOptions) GetXlaGpuEnableCustomFusionsRe() string

+ func (x *DebugOptions) GetXlaGpuEnableDotStrengthReduction() bool

+ func (x *DebugOptions) GetXlaGpuEnableFastMinMax() bool

+ func (x *DebugOptions) GetXlaGpuEnableHighestPriorityAsyncStream() bool

+ func (x *DebugOptions) GetXlaGpuEnableHostMemoryOffloading() bool

+ func (x *DebugOptions) GetXlaGpuEnableLatencyHidingScheduler() bool

+ func (x *DebugOptions) GetXlaGpuEnableLibnvptxcompiler() bool

+ func (x *DebugOptions) GetXlaGpuEnableLlvmModuleCompilationParallelism() bool

+ func (x *DebugOptions) GetXlaGpuEnableNcclCliqueOptimization() bool

+ func (x *DebugOptions) GetXlaGpuEnableNcclCommSplitting() bool

+ func (x *DebugOptions) GetXlaGpuEnableNcclPerStreamComms() bool

+ func (x *DebugOptions) GetXlaGpuEnableNcclUserBuffers() bool

+ func (x *DebugOptions) GetXlaGpuEnablePipelinedAllGather() bool

+ func (x *DebugOptions) GetXlaGpuEnablePipelinedAllReduce() bool

+ func (x *DebugOptions) GetXlaGpuEnablePipelinedCollectives() bool

+ func (x *DebugOptions) GetXlaGpuEnablePipelinedP2P() bool

+ func (x *DebugOptions) GetXlaGpuEnablePipelinedReduceScatter() bool

+ func (x *DebugOptions) GetXlaGpuEnablePriorityFusion() bool

+ func (x *DebugOptions) GetXlaGpuEnableReassociationForConvertedAr() bool

+ func (x *DebugOptions) GetXlaGpuEnableReduceScatterCombineByDim() bool

+ func (x *DebugOptions) GetXlaGpuEnableReductionEpilogueFusion() bool

+ func (x *DebugOptions) GetXlaGpuEnableSharedConstants() bool

+ func (x *DebugOptions) GetXlaGpuEnableSplitKAutotuning() bool

+ func (x *DebugOptions) GetXlaGpuEnableTritonGemm() bool

+ func (x *DebugOptions) GetXlaGpuEnableTritonHopper() bool

+ func (x *DebugOptions) GetXlaGpuEnableTritonSoftmaxFusion() bool

+ func (x *DebugOptions) GetXlaGpuEnableTritonSoftmaxPriorityFusion() bool

+ func (x *DebugOptions) GetXlaGpuEnableWhileLoopDoubleBuffering() bool

+ func (x *DebugOptions) GetXlaGpuEnableWhileLoopReduceScatterCodeMotion() bool

+ func (x *DebugOptions) GetXlaGpuEnableWhileLoopUnrolling() DebugOptions_WhileLoopUnrolling

+ func (x *DebugOptions) GetXlaGpuEnsureMinorDotContractionDims() bool

+ func (x *DebugOptions) GetXlaGpuExcludeNondeterministicOps() bool

+ func (x *DebugOptions) GetXlaGpuExhaustiveTilingSearch() bool

+ func (x *DebugOptions) GetXlaGpuFilterKernelsSpillingRegistersOnAutotuning() bool

+ func (x *DebugOptions) GetXlaGpuForceCompilationParallelism() int32

+ func (x *DebugOptions) GetXlaGpuForceConvNchw() bool

+ func (x *DebugOptions) GetXlaGpuForceConvNhwc() bool

+ func (x *DebugOptions) GetXlaGpuFtz() bool

+ func (x *DebugOptions) GetXlaGpuFusedAttentionUseCudnnRng() bool

+ func (x *DebugOptions) GetXlaGpuGemmRewriteSizeThreshold() int64

+ func (x *DebugOptions) GetXlaGpuGraphEnableConcurrentRegion() bool

+ func (x *DebugOptions) GetXlaGpuGraphMinGraphSize() int32

+ func (x *DebugOptions) GetXlaGpuKernelCacheFile() string

+ func (x *DebugOptions) GetXlaGpuLhsEnableGpuAsyncTracker() bool

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

+ func (x *DebugOptions) GetXlaGpuLlvmVerificationLevel() int32

+ func (x *DebugOptions) GetXlaGpuLoadAutotuneResultsFrom() string

+ func (x *DebugOptions) GetXlaGpuMaxMlirKernels() int64

+ func (x *DebugOptions) GetXlaGpuMemoryLimitSlopFactor() int32

+ func (x *DebugOptions) GetXlaGpuMlirEmitterLevel() int64

+ func (x *DebugOptions) GetXlaGpuMockCustomCalls() bool

+ func (x *DebugOptions) GetXlaGpuMultiStreamedWindowedEinsum() bool

+ func (x *DebugOptions) GetXlaGpuNcclCollectiveMaxNchannels() int64

+ func (x *DebugOptions) GetXlaGpuNcclP2PMaxNchannels() int64

+ func (x *DebugOptions) GetXlaGpuNcclTerminateOnError() bool

+ func (x *DebugOptions) GetXlaGpuNcclTerminationTimeoutSeconds() int64

+ func (x *DebugOptions) GetXlaGpuOverrideGemmAutotuner() string

+ func (x *DebugOptions) GetXlaGpuPgleProfileFileOrDirectoryPath() string

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

+ func (x *DebugOptions) GetXlaGpuReduceScatterCombineThresholdBytes() int64

+ func (x *DebugOptions) GetXlaGpuRedzonePaddingBytes() int64

+ func (x *DebugOptions) GetXlaGpuRedzoneScratchMaxMegabytes() int64

+ func (x *DebugOptions) GetXlaGpuRequireCompleteAotAutotuneResults() bool

+ func (x *DebugOptions) GetXlaGpuShapeChecks() DebugOptions_ShapeChecks

+ func (x *DebugOptions) GetXlaGpuShardAutotuning() bool

+ func (x *DebugOptions) GetXlaGpuSkipMlirKernels() int64

+ func (x *DebugOptions) GetXlaGpuStrictConvAlgorithmPicker() bool

+ func (x *DebugOptions) GetXlaGpuTargetConfigFilename() string

+ func (x *DebugOptions) GetXlaGpuThresholdForWindowedEinsumMib() int64

+ func (x *DebugOptions) GetXlaGpuTritonFusionLevel() int32

+ func (x *DebugOptions) GetXlaGpuTritonGemmAny() bool

+ func (x *DebugOptions) GetXlaGpuTritonGemmDisableReducedPrecisionReduction() bool

+ func (x *DebugOptions) GetXlaGpuUnsafeFallbackToDriverOnPtxasNotFound() bool

+ func (x *DebugOptions) GetXlaGpuUseMemcpyLocalP2P() bool

+ func (x *DebugOptions) GetXlaGpuUseRuntimeFusion() bool

+ func (x *DebugOptions) GetXlaGpuVerifyTritonFusionNumerics() bool

+ func (x *DebugOptions) GetXlaHloEvaluatorUseFastPath() bool

+ func (x *DebugOptions) GetXlaHloGraphAddresses() bool

+ func (x *DebugOptions) GetXlaHloGraphShardingColor() bool

+ func (x *DebugOptions) GetXlaHloProfile() bool

+ func (x *DebugOptions) GetXlaLlvmDisableExpensivePasses() bool

+ func (x *DebugOptions) GetXlaLlvmEnableAliasScopeMetadata() bool

+ func (x *DebugOptions) GetXlaLlvmEnableInvariantLoadMetadata() bool

+ func (x *DebugOptions) GetXlaLlvmEnableNoaliasMetadata() bool

+ func (x *DebugOptions) GetXlaLlvmForceInlineBeforeSplit() bool

+ func (x *DebugOptions) GetXlaMultiheapSizeConstraintPerHeap() int32

+ func (x *DebugOptions) GetXlaPartitioningAlgorithm() DebugOptions_PartitioningAlgorithm

+ func (x *DebugOptions) GetXlaReduceWindowRewriteBaseLength() int64

+ func (x *DebugOptions) GetXlaStepMarkerLocation() DebugOptions_StepMarkerLocation

+ func (x *DebugOptions) GetXlaTestAllInputLayouts() bool

+ func (x *DebugOptions) GetXlaTestAllOutputLayouts() bool

+ func (x *DebugOptions) GetXlaTpuDetectInf() bool

+ func (x *DebugOptions) GetXlaTpuDetectNan() bool

+ func (x *DebugOptions) GetXlaUseShardonnay() bool

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

+ func (x *DebugOptions) Reset()

+ func (x *DebugOptions) String() string

+ type DebugOptions_CollectiveOpType int32

+ const DebugOptions_ALLGATHER

+ const DebugOptions_ALLREDUCE

+ const DebugOptions_ALLTOALL

+ const DebugOptions_COLLECTIVEBROADCAST

+ const DebugOptions_COLLECTIVEPERMUTE

+ const DebugOptions_NOOP

+ const DebugOptions_REDUCESCATTER

+ func (DebugOptions_CollectiveOpType) Descriptor() protoreflect.EnumDescriptor

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

+ func (DebugOptions_CollectiveOpType) Type() protoreflect.EnumType

+ func (x DebugOptions_CollectiveOpType) Enum() *DebugOptions_CollectiveOpType

+ func (x DebugOptions_CollectiveOpType) Number() protoreflect.EnumNumber

+ func (x DebugOptions_CollectiveOpType) String() string

+ type DebugOptions_CommandBufferCmdType int32

+ const DebugOptions_COLLECTIVES

+ const DebugOptions_CONDITIONALS

+ const DebugOptions_CUBLAS

+ const DebugOptions_CUBLASLT

+ const DebugOptions_CUDNN

+ const DebugOptions_CUSTOM_CALL

+ const DebugOptions_FUSION

+ const DebugOptions_INVALID

+ func (DebugOptions_CommandBufferCmdType) Descriptor() protoreflect.EnumDescriptor

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

+ func (DebugOptions_CommandBufferCmdType) Type() protoreflect.EnumType

+ func (x DebugOptions_CommandBufferCmdType) Enum() *DebugOptions_CommandBufferCmdType

+ func (x DebugOptions_CommandBufferCmdType) Number() protoreflect.EnumNumber

+ func (x DebugOptions_CommandBufferCmdType) String() string

+ type DebugOptions_PartitioningAlgorithm int32

+ const DebugOptions_PARTITIONING_ALGORITHM_EXP0

+ const DebugOptions_PARTITIONING_ALGORITHM_EXP1

+ const DebugOptions_PARTITIONING_ALGORITHM_EXP2

+ const DebugOptions_PARTITIONING_ALGORITHM_NOOP

+ func (DebugOptions_PartitioningAlgorithm) Descriptor() protoreflect.EnumDescriptor

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

+ func (DebugOptions_PartitioningAlgorithm) Type() protoreflect.EnumType

+ func (x DebugOptions_PartitioningAlgorithm) Enum() *DebugOptions_PartitioningAlgorithm

+ func (x DebugOptions_PartitioningAlgorithm) Number() protoreflect.EnumNumber

+ func (x DebugOptions_PartitioningAlgorithm) String() string

+ type DebugOptions_ShapeChecks int32

+ const DebugOptions_COMPILE_TIME

+ const DebugOptions_IGNORE

+ const DebugOptions_RUNTIME

+ func (DebugOptions_ShapeChecks) Descriptor() protoreflect.EnumDescriptor

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

+ func (DebugOptions_ShapeChecks) Type() protoreflect.EnumType

+ func (x DebugOptions_ShapeChecks) Enum() *DebugOptions_ShapeChecks

+ func (x DebugOptions_ShapeChecks) Number() protoreflect.EnumNumber

+ func (x DebugOptions_ShapeChecks) String() string

+ type DebugOptions_StepMarkerLocation int32

+ const DebugOptions_STEP_MARK_AT_ENTRY

+ const DebugOptions_STEP_MARK_AT_SECOND_LEVEL_WHILE_LOOP

+ const DebugOptions_STEP_MARK_AT_TOP_LEVEL_WHILE_LOOP

+ const DebugOptions_STEP_MARK_NONE

+ func (DebugOptions_StepMarkerLocation) Descriptor() protoreflect.EnumDescriptor

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

+ func (DebugOptions_StepMarkerLocation) Type() protoreflect.EnumType

+ func (x DebugOptions_StepMarkerLocation) Enum() *DebugOptions_StepMarkerLocation

+ func (x DebugOptions_StepMarkerLocation) Number() protoreflect.EnumNumber

+ func (x DebugOptions_StepMarkerLocation) String() string

+ type DebugOptions_WhileLoopUnrolling int32

+ const DebugOptions_WHILE_LOOP_UNROLLING_DOUBLE_BUFFER

+ const DebugOptions_WHILE_LOOP_UNROLLING_FULL_UNROLL

+ const DebugOptions_WHILE_LOOP_UNROLLING_NO_UNROLL

+ func (DebugOptions_WhileLoopUnrolling) Descriptor() protoreflect.EnumDescriptor

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

+ func (DebugOptions_WhileLoopUnrolling) Type() protoreflect.EnumType

+ func (x DebugOptions_WhileLoopUnrolling) Enum() *DebugOptions_WhileLoopUnrolling

+ func (x DebugOptions_WhileLoopUnrolling) Number() protoreflect.EnumNumber

+ func (x DebugOptions_WhileLoopUnrolling) String() string

+ type DeconstructTupleRequest struct

+ TupleHandle *GlobalDataHandle

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

+ func (*DeconstructTupleRequest) ProtoMessage()

+ func (x *DeconstructTupleRequest) GetTupleHandle() *GlobalDataHandle

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

+ func (x *DeconstructTupleRequest) Reset()

+ func (x *DeconstructTupleRequest) String() string

+ type DeconstructTupleResponse struct

+ ElementHandles []*GlobalDataHandle

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

+ func (*DeconstructTupleResponse) ProtoMessage()

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

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

+ func (x *DeconstructTupleResponse) Reset()

+ func (x *DeconstructTupleResponse) String() string

+ type DeviceAssignmentProto struct

+ ComputationCount int32

+ ComputationDevices []*DeviceAssignmentProto_ComputationDevice

+ ReplicaCount int32

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

+ func (*DeviceAssignmentProto) ProtoMessage()

+ func (x *DeviceAssignmentProto) GetComputationCount() int32

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

+ func (x *DeviceAssignmentProto) GetReplicaCount() int32

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

+ func (x *DeviceAssignmentProto) Reset()

+ func (x *DeviceAssignmentProto) String() string

+ type DeviceAssignmentProto_ComputationDevice struct

+ ReplicaDeviceIds []int64

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

+ func (*DeviceAssignmentProto_ComputationDevice) ProtoMessage()

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

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

+ func (x *DeviceAssignmentProto_ComputationDevice) Reset()

+ func (x *DeviceAssignmentProto_ComputationDevice) String() string

+ type DeviceHandle struct

+ DeviceCount int64

+ Handle int64

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

+ func (*DeviceHandle) ProtoMessage()

+ func (x *DeviceHandle) GetDeviceCount() int64

+ func (x *DeviceHandle) GetHandle() int64

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

+ func (x *DeviceHandle) Reset()

+ func (x *DeviceHandle) String() string

+ type DimLevelType int32

+ const DimLevelType_DIM_COMPRESSED

+ const DimLevelType_DIM_DENSE

+ const DimLevelType_DIM_LOOSE_COMPRESSED

+ const DimLevelType_DIM_SINGLETON

+ func (DimLevelType) Descriptor() protoreflect.EnumDescriptor

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

+ func (DimLevelType) Type() protoreflect.EnumType

+ func (x DimLevelType) Enum() *DimLevelType

+ func (x DimLevelType) Number() protoreflect.EnumNumber

+ func (x DimLevelType) String() string

+ type DnnVersionInfoProto struct

+ Major int32

+ Minor int32

+ Patch int32

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

+ func (*DnnVersionInfoProto) ProtoMessage()

+ func (x *DnnVersionInfoProto) GetMajor() int32

+ func (x *DnnVersionInfoProto) GetMinor() int32

+ func (x *DnnVersionInfoProto) GetPatch() int32

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

+ func (x *DnnVersionInfoProto) Reset()

+ func (x *DnnVersionInfoProto) String() string

+ type DotDimensionNumbers struct

+ LhsBatchDimensions []int64

+ LhsContractingDimensions []int64

+ RhsBatchDimensions []int64

+ RhsContractingDimensions []int64

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

+ func (*DotDimensionNumbers) ProtoMessage()

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

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

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

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

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

+ func (x *DotDimensionNumbers) Reset()

+ func (x *DotDimensionNumbers) String() string

+ type ExecutableAndOptionsProto struct

+ CompileOptions *CompileOptionsProto

+ SerializedExecutable []byte

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

+ func (*ExecutableAndOptionsProto) ProtoMessage()

+ func (x *ExecutableAndOptionsProto) GetCompileOptions() *CompileOptionsProto

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

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

+ func (x *ExecutableAndOptionsProto) Reset()

+ func (x *ExecutableAndOptionsProto) String() string

+ type ExecutableBuildOptionsProto struct

+ AliasPassthroughParams bool

+ AllowSpmdShardingPropagationToOutput []bool

+ AllowSpmdShardingPropagationToParameters []bool

+ AutoSpmdPartitioningMeshIds []int64

+ AutoSpmdPartitioningMeshShape []int64

+ CompEnvs *CompilationEnvironmentsProto

+ DebugOptions *DebugOptions

+ DeduplicateHlo bool

+ DeviceAssignment *DeviceAssignmentProto

+ DeviceMemorySize int64

+ DeviceOrdinal int64

+ FdoProfile []byte

+ NumPartitions int64

+ NumReplicas int64

+ ResultLayout *ShapeProto

+ RunBackendOnly bool

+ UseAutoSpmdPartitioning bool

+ UseSpmdPartitioning bool

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

+ func (*ExecutableBuildOptionsProto) ProtoMessage()

+ func (x *ExecutableBuildOptionsProto) GetAliasPassthroughParams() bool

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

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

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

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

+ func (x *ExecutableBuildOptionsProto) GetCompEnvs() *CompilationEnvironmentsProto

+ func (x *ExecutableBuildOptionsProto) GetDebugOptions() *DebugOptions

+ func (x *ExecutableBuildOptionsProto) GetDeduplicateHlo() bool

+ func (x *ExecutableBuildOptionsProto) GetDeviceAssignment() *DeviceAssignmentProto

+ func (x *ExecutableBuildOptionsProto) GetDeviceMemorySize() int64

+ func (x *ExecutableBuildOptionsProto) GetDeviceOrdinal() int64

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

+ func (x *ExecutableBuildOptionsProto) GetNumPartitions() int64

+ func (x *ExecutableBuildOptionsProto) GetNumReplicas() int64

+ func (x *ExecutableBuildOptionsProto) GetResultLayout() *ShapeProto

+ func (x *ExecutableBuildOptionsProto) GetRunBackendOnly() bool

+ func (x *ExecutableBuildOptionsProto) GetUseAutoSpmdPartitioning() bool

+ func (x *ExecutableBuildOptionsProto) GetUseSpmdPartitioning() bool

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

+ func (x *ExecutableBuildOptionsProto) Reset()

+ func (x *ExecutableBuildOptionsProto) String() string

+ type ExecuteGraphParallelRequest struct

+ Requests []*ExecuteGraphRequest

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

+ func (*ExecuteGraphParallelRequest) ProtoMessage()

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

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

+ func (x *ExecuteGraphParallelRequest) Reset()

+ func (x *ExecuteGraphParallelRequest) String() string

+ type ExecuteGraphRequest struct

+ Arguments []*GlobalDataHandle

+ Computation *HloModuleProto

+ ExecutionOptions *ExecutionOptions

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

+ func (*ExecuteGraphRequest) ProtoMessage()

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

+ func (x *ExecuteGraphRequest) GetComputation() *HloModuleProto

+ func (x *ExecuteGraphRequest) GetExecutionOptions() *ExecutionOptions

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

+ func (x *ExecuteGraphRequest) Reset()

+ func (x *ExecuteGraphRequest) String() string

+ type ExecuteParallelResponse struct

+ Responses []*ExecuteResponse

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

+ func (*ExecuteParallelResponse) ProtoMessage()

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

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

+ func (x *ExecuteParallelResponse) Reset()

+ func (x *ExecuteParallelResponse) String() string

+ type ExecuteRequest struct

+ Arguments []*GlobalDataHandle

+ Handle *ExecutionHandle

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

+ func (*ExecuteRequest) ProtoMessage()

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

+ func (x *ExecuteRequest) GetHandle() *ExecutionHandle

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

+ func (x *ExecuteRequest) Reset()

+ func (x *ExecuteRequest) String() string

+ type ExecuteResponse struct

+ Output *GlobalDataHandle

+ Profile *ExecutionProfile

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

+ func (*ExecuteResponse) ProtoMessage()

+ func (x *ExecuteResponse) GetOutput() *GlobalDataHandle

+ func (x *ExecuteResponse) GetProfile() *ExecutionProfile

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

+ func (x *ExecuteResponse) Reset()

+ func (x *ExecuteResponse) String() string

+ type ExecutionHandle struct

+ Handle int64

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

+ func (*ExecutionHandle) ProtoMessage()

+ func (x *ExecutionHandle) GetHandle() int64

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

+ func (x *ExecutionHandle) Reset()

+ func (x *ExecutionHandle) String() string

+ type ExecutionOptions struct

+ AliasPassthroughParams bool

+ AllowSeparateShardingPrograms bool

+ AllowSpmdShardingPropagationToOutput []bool

+ AllowSpmdShardingPropagationToParameters []bool

+ AutoSpmdPartitioningMeshIds []int64

+ AutoSpmdPartitioningMeshShape []int64

+ DebugOptions *DebugOptions

+ DeduplicateHlo bool

+ DeviceAssignment *DeviceAssignmentProto

+ DeviceHandles []*DeviceHandle

+ DeviceMemorySize int64

+ FdoProfile []byte

+ LaunchId int32

+ NumPartitions int32

+ NumReplicas int32

+ ParamRequiresBroadcastViaCollectives []bool

+ Seed uint64

+ ShapeWithOutputLayout *ShapeProto

+ ShardableValueUpdatePairs []*ShardableValueUpdatePairProto

+ UseAutoSpmdPartitioning bool

+ UseSpmdPartitioning bool

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

+ func (*ExecutionOptions) ProtoMessage()

+ func (x *ExecutionOptions) GetAliasPassthroughParams() bool

+ func (x *ExecutionOptions) GetAllowSeparateShardingPrograms() bool

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

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

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

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

+ func (x *ExecutionOptions) GetDebugOptions() *DebugOptions

+ func (x *ExecutionOptions) GetDeduplicateHlo() bool

+ func (x *ExecutionOptions) GetDeviceAssignment() *DeviceAssignmentProto

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

+ func (x *ExecutionOptions) GetDeviceMemorySize() int64

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

+ func (x *ExecutionOptions) GetLaunchId() int32

+ func (x *ExecutionOptions) GetNumPartitions() int32

+ func (x *ExecutionOptions) GetNumReplicas() int32

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

+ func (x *ExecutionOptions) GetSeed() uint64

+ func (x *ExecutionOptions) GetShapeWithOutputLayout() *ShapeProto

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

+ func (x *ExecutionOptions) GetUseAutoSpmdPartitioning() bool

+ func (x *ExecutionOptions) GetUseSpmdPartitioning() bool

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

+ func (x *ExecutionOptions) Reset()

+ func (x *ExecutionOptions) String() string

+ type ExecutionProfile struct

+ CompilationCacheHit bool

+ CompileTimeMs int64

+ ComputeAndTransferTimeNs int64

+ ComputeCycleCount int64

+ ComputeTimeNs int64

+ ExecutableSizeInBytes int64

+ ProfileCacheHit bool

+ WarmupRunExecuted bool

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

+ func (*ExecutionProfile) ProtoMessage()

+ func (x *ExecutionProfile) GetCompilationCacheHit() bool

+ func (x *ExecutionProfile) GetCompileTimeMs() int64

+ func (x *ExecutionProfile) GetComputeAndTransferTimeNs() int64

+ func (x *ExecutionProfile) GetComputeCycleCount() int64

+ func (x *ExecutionProfile) GetComputeTimeNs() int64

+ func (x *ExecutionProfile) GetExecutableSizeInBytes() int64

+ func (x *ExecutionProfile) GetProfileCacheHit() bool

+ func (x *ExecutionProfile) GetWarmupRunExecuted() bool

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

+ func (x *ExecutionProfile) Reset()

+ func (x *ExecutionProfile) String() string

+ type FMHAMaskKind int32

+ const FMHAMaskKind_ALIBI

+ const FMHAMaskKind_CAUSAL

+ const FMHAMaskKind_NO_MASK

+ const FMHAMaskKind_PADDING

+ const FMHAMaskKind_PADDING_CAUSAL

+ func (FMHAMaskKind) Descriptor() protoreflect.EnumDescriptor

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

+ func (FMHAMaskKind) Type() protoreflect.EnumType

+ func (x FMHAMaskKind) Enum() *FMHAMaskKind

+ func (x FMHAMaskKind) Number() protoreflect.EnumNumber

+ func (x FMHAMaskKind) String() string

+ type FftType int32

+ const FftType_FFT

+ const FftType_IFFT

+ const FftType_IRFFT

+ const FftType_RFFT

+ func (FftType) Descriptor() protoreflect.EnumDescriptor

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

+ func (FftType) Type() protoreflect.EnumType

+ func (x FftType) Enum() *FftType

+ func (x FftType) Number() protoreflect.EnumNumber

+ func (x FftType) String() string

+ type FilterLayout int32

+ const FilterLayout_kInputYXOutput

+ const FilterLayout_kOutputInputYX

+ const FilterLayout_kOutputInputYX32

+ const FilterLayout_kOutputInputYX32_CudnnReordered

+ const FilterLayout_kOutputInputYX4

+ const FilterLayout_kOutputYXInput

+ const FilterLayout_kYXInputOutput

+ func (FilterLayout) Descriptor() protoreflect.EnumDescriptor

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

+ func (FilterLayout) Type() protoreflect.EnumType

+ func (x FilterLayout) Enum() *FilterLayout

+ func (x FilterLayout) Number() protoreflect.EnumNumber

+ func (x FilterLayout) String() string

+ type FrontendAttributes struct

+ Map map[string]string

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

+ func (*FrontendAttributes) ProtoMessage()

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

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

+ func (x *FrontendAttributes) Reset()

+ func (x *FrontendAttributes) String() string

+ type FusedMHAKind int32

+ const FusedMHAKind_BMM1_OUTPUT_FLOAT

+ const FusedMHAKind_BMM1_OUTPUT_INPUT_TYPE

+ const FusedMHAKind_BMM1_OUTPUT_UNKNOWN

+ func (FusedMHAKind) Descriptor() protoreflect.EnumDescriptor

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

+ func (FusedMHAKind) Type() protoreflect.EnumType

+ func (x FusedMHAKind) Enum() *FusedMHAKind

+ func (x FusedMHAKind) Number() protoreflect.EnumNumber

+ func (x FusedMHAKind) String() string

+ type GatherDimensionNumbers struct

+ CollapsedSliceDims []int64

+ IndexVectorDim int64

+ OffsetDims []int64

+ StartIndexMap []int64

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

+ func (*GatherDimensionNumbers) ProtoMessage()

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

+ func (x *GatherDimensionNumbers) GetIndexVectorDim() int64

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

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

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

+ func (x *GatherDimensionNumbers) Reset()

+ func (x *GatherDimensionNumbers) String() string

+ type GetDeviceHandlesRequest struct

+ DeviceCount int64

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

+ func (*GetDeviceHandlesRequest) ProtoMessage()

+ func (x *GetDeviceHandlesRequest) GetDeviceCount() int64

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

+ func (x *GetDeviceHandlesRequest) Reset()

+ func (x *GetDeviceHandlesRequest) String() string

+ type GetDeviceHandlesResponse struct

+ DeviceHandles []*DeviceHandle

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

+ func (*GetDeviceHandlesResponse) ProtoMessage()

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

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

+ func (x *GetDeviceHandlesResponse) Reset()

+ func (x *GetDeviceHandlesResponse) String() string

+ type GetShapeRequest struct

+ Data *GlobalDataHandle

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

+ func (*GetShapeRequest) ProtoMessage()

+ func (x *GetShapeRequest) GetData() *GlobalDataHandle

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

+ func (x *GetShapeRequest) Reset()

+ func (x *GetShapeRequest) String() string

+ type GetShapeResponse struct

+ Shape *ShapeProto

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

+ func (*GetShapeResponse) ProtoMessage()

+ func (x *GetShapeResponse) GetShape() *ShapeProto

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

+ func (x *GetShapeResponse) Reset()

+ func (x *GetShapeResponse) String() string

+ type GlobalDataHandle struct

+ Handle int64

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

+ func (*GlobalDataHandle) ProtoMessage()

+ func (x *GlobalDataHandle) GetHandle() int64

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

+ func (x *GlobalDataHandle) Reset()

+ func (x *GlobalDataHandle) String() string

+ type GpuCompilationEnvironment struct

+ DummyFlag int64

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

+ func (*GpuCompilationEnvironment) ProtoMessage()

+ func (x *GpuCompilationEnvironment) GetDummyFlag() int64

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

+ func (x *GpuCompilationEnvironment) Reset()

+ func (x *GpuCompilationEnvironment) String() string

+ type GpuDeviceInfoProto struct

+ BlockDimLimitX int32

+ BlockDimLimitY int32

+ BlockDimLimitZ int32

+ ClockRateGhz float32

+ ComputeCapability isGpuDeviceInfoProto_ComputeCapability

+ CoreCount int32

+ DeviceMemorySize int64

+ FpusPerCore int64

+ L2CacheSize int64

+ MemoryBandwidth int64

+ SharedMemoryPerBlock int32

+ SharedMemoryPerBlockOptin int32

+ SharedMemoryPerCore int32

+ ThreadsPerBlockLimit int32

+ ThreadsPerCoreLimit int32

+ ThreadsPerWarp int32

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

+ func (*GpuDeviceInfoProto) ProtoMessage()

+ func (m *GpuDeviceInfoProto) GetComputeCapability() isGpuDeviceInfoProto_ComputeCapability

+ func (x *GpuDeviceInfoProto) GetBlockDimLimitX() int32

+ func (x *GpuDeviceInfoProto) GetBlockDimLimitY() int32

+ func (x *GpuDeviceInfoProto) GetBlockDimLimitZ() int32

+ func (x *GpuDeviceInfoProto) GetClockRateGhz() float32

+ func (x *GpuDeviceInfoProto) GetCoreCount() int32

+ func (x *GpuDeviceInfoProto) GetCudaComputeCapability() *CudaComputeCapabilityProto

+ func (x *GpuDeviceInfoProto) GetDeviceMemorySize() int64

+ func (x *GpuDeviceInfoProto) GetFpusPerCore() int64

+ func (x *GpuDeviceInfoProto) GetL2CacheSize() int64

+ func (x *GpuDeviceInfoProto) GetMemoryBandwidth() int64

+ func (x *GpuDeviceInfoProto) GetRocmComputeCapability() *RocmComputeCapabilityProto

+ func (x *GpuDeviceInfoProto) GetSharedMemoryPerBlock() int32

+ func (x *GpuDeviceInfoProto) GetSharedMemoryPerBlockOptin() int32

+ func (x *GpuDeviceInfoProto) GetSharedMemoryPerCore() int32

+ func (x *GpuDeviceInfoProto) GetThreadsPerBlockLimit() int32

+ func (x *GpuDeviceInfoProto) GetThreadsPerCoreLimit() int32

+ func (x *GpuDeviceInfoProto) GetThreadsPerWarp() int32

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

+ func (x *GpuDeviceInfoProto) Reset()

+ func (x *GpuDeviceInfoProto) String() string

+ type GpuDeviceInfoProto_CudaComputeCapability struct

+ CudaComputeCapability *CudaComputeCapabilityProto

+ type GpuDeviceInfoProto_RocmComputeCapability struct

+ RocmComputeCapability *RocmComputeCapabilityProto

+ type GpuTargetConfigProto struct

+ AutotuneResults *AutotuneResults

+ DeviceDescriptionStr string

+ DnnVersionInfo *DnnVersionInfoProto

+ GpuDeviceInfo *GpuDeviceInfoProto

+ PlatformName string

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

+ func (*GpuTargetConfigProto) ProtoMessage()

+ func (x *GpuTargetConfigProto) GetAutotuneResults() *AutotuneResults

+ func (x *GpuTargetConfigProto) GetDeviceDescriptionStr() string

+ func (x *GpuTargetConfigProto) GetDnnVersionInfo() *DnnVersionInfoProto

+ func (x *GpuTargetConfigProto) GetGpuDeviceInfo() *GpuDeviceInfoProto

+ func (x *GpuTargetConfigProto) GetPlatformName() string

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

+ func (x *GpuTargetConfigProto) Reset()

+ func (x *GpuTargetConfigProto) String() string

+ type HeapSimulatorTrace struct

+ BufferAllocationIndex int64

+ Events []*HeapSimulatorTrace_Event

+ WholeModuleSimulation bool

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

+ func (*HeapSimulatorTrace) ProtoMessage()

+ func (x *HeapSimulatorTrace) GetBufferAllocationIndex() int64

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

+ func (x *HeapSimulatorTrace) GetWholeModuleSimulation() bool

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

+ func (x *HeapSimulatorTrace) Reset()

+ func (x *HeapSimulatorTrace) String() string

+ type HeapSimulatorTrace_Event struct

+ BufferId int64

+ ComputationName string

+ InstructionName string

+ Kind HeapSimulatorTrace_Event_Kind

+ ShareWithCanonicalId int64

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

+ func (*HeapSimulatorTrace_Event) ProtoMessage()

+ func (x *HeapSimulatorTrace_Event) GetBufferId() int64

+ func (x *HeapSimulatorTrace_Event) GetComputationName() string

+ func (x *HeapSimulatorTrace_Event) GetInstructionName() string

+ func (x *HeapSimulatorTrace_Event) GetKind() HeapSimulatorTrace_Event_Kind

+ func (x *HeapSimulatorTrace_Event) GetShareWithCanonicalId() int64

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

+ func (x *HeapSimulatorTrace_Event) Reset()

+ func (x *HeapSimulatorTrace_Event) String() string

+ type HeapSimulatorTrace_Event_Kind int32

+ const HeapSimulatorTrace_Event_ALLOC

+ const HeapSimulatorTrace_Event_FREE

+ const HeapSimulatorTrace_Event_SHARE_WITH

+ func (HeapSimulatorTrace_Event_Kind) Descriptor() protoreflect.EnumDescriptor

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

+ func (HeapSimulatorTrace_Event_Kind) Type() protoreflect.EnumType

+ func (x HeapSimulatorTrace_Event_Kind) Enum() *HeapSimulatorTrace_Event_Kind

+ func (x HeapSimulatorTrace_Event_Kind) Number() protoreflect.EnumNumber

+ func (x HeapSimulatorTrace_Event_Kind) String() string

+ type HloBufferDonorProto struct

+ Entries []*HloBufferDonorProto_BufferDonorEntryProto

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

+ func (*HloBufferDonorProto) ProtoMessage()

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

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

+ func (x *HloBufferDonorProto) Reset()

+ func (x *HloBufferDonorProto) String() string

+ type HloBufferDonorProto_BufferDonorEntryProto struct

+ ParameterNumber int64

+ ParameterShapeIndex []int64

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

+ func (*HloBufferDonorProto_BufferDonorEntryProto) ProtoMessage()

+ func (x *HloBufferDonorProto_BufferDonorEntryProto) GetParameterNumber() int64

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

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

+ func (x *HloBufferDonorProto_BufferDonorEntryProto) Reset()

+ func (x *HloBufferDonorProto_BufferDonorEntryProto) String() string

+ type HloComputationProto struct

+ ExecutionThread string

+ Id int64

+ Instructions []*HloInstructionProto

+ IsFusionComputation bool

+ Name string

+ ProgramShape *ProgramShapeProto

+ RootId int64

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

+ func (*HloComputationProto) ProtoMessage()

+ func (x *HloComputationProto) GetExecutionThread() string

+ func (x *HloComputationProto) GetId() int64

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

+ func (x *HloComputationProto) GetIsFusionComputation() bool

+ func (x *HloComputationProto) GetName() string

+ func (x *HloComputationProto) GetProgramShape() *ProgramShapeProto

+ func (x *HloComputationProto) GetRootId() int64

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

+ func (x *HloComputationProto) Reset()

+ func (x *HloComputationProto) String() string

+ type HloInputOutputAliasProto struct

+ Entries []*HloInputOutputAliasProto_AliasEntryProto

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

+ func (*HloInputOutputAliasProto) ProtoMessage()

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

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

+ func (x *HloInputOutputAliasProto) Reset()

+ func (x *HloInputOutputAliasProto) String() string

+ type HloInputOutputAliasProto_AliasEntryProto struct

+ Kind Kind

+ OutputShapeIndex []int64

+ ParameterNumber int64

+ ParameterShapeIndex []int64

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

+ func (*HloInputOutputAliasProto_AliasEntryProto) ProtoMessage()

+ func (x *HloInputOutputAliasProto_AliasEntryProto) GetKind() Kind

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

+ func (x *HloInputOutputAliasProto_AliasEntryProto) GetParameterNumber() int64

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

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

+ func (x *HloInputOutputAliasProto_AliasEntryProto) Reset()

+ func (x *HloInputOutputAliasProto_AliasEntryProto) String() string

+ type HloInstructionProto struct

+ AllReduceId int64

+ AsyncExecutionThread string

+ BackendConfig []byte

+ BatchGroupCount int64

+ CalledComputationIds []int64

+ ChannelId int64

+ CholeskyOptions *CholeskyOptions

+ ComparisonDirection string

+ ComparisonType string

+ ConstrainLayout bool

+ ControlPredecessorIds []int64

+ ConvolutionDimensionNumbers *ConvolutionDimensionNumbers

+ CustomCallApiVersion CustomCallApiVersion

+ CustomCallHasSideEffect bool

+ CustomCallSchedule CustomCallSchedule

+ CustomCallTarget string

+ Delta int64

+ Dimensions []int64

+ Distribution RandomDistribution

+ DomainEntrySharding *OpSharding

+ DomainExitSharding *OpSharding

+ DotDimensionNumbers *DotDimensionNumbers

+ DotSparsity []*SparsityDescriptor

+ DynamicSliceSizes []int64

+ Epsilon float32

+ ExponentBits int32

+ FeatureGroupCount int64

+ FeatureIndex int64

+ FftLength []int64

+ FftType FftType

+ FrontendAttributes *FrontendAttributes

+ FusionKind string

+ GatherDimensionNumbers *GatherDimensionNumbers

+ GatherSliceSizes []int64

+ Id int64

+ IndicesAreSorted bool

+ InfeedConfig []byte

+ IsCrossProgramPrefetch bool

+ IsHostTransfer bool

+ IsStable bool

+ K int64

+ Largest bool

+ Literal *LiteralProto

+ MantissaBits int32

+ Metadata *OpMetadata

+ Name string

+ Opcode string

+ OperandIds []int64

+ OperandShapesWithLayout []*ShapeProto

+ OptionalCrossProgramPrefetchIndex isHloInstructionProto_OptionalCrossProgramPrefetchIndex

+ OutfeedConfig []byte

+ OutfeedShape *ShapeProto

+ OutputOperandAliasing []*OutputOperandAliasing

+ PaddingConfig *PaddingConfig

+ PaddingType PaddingType

+ ParameterNumber int64

+ ParameterReplication *ParameterReplication

+ PrecisionConfig *PrecisionConfig

+ ReplicaGroups []*ReplicaGroup

+ RngAlgorithm RandomAlgorithm

+ ScatterDimensionNumbers *ScatterDimensionNumbers

+ Shape *ShapeProto

+ Sharding *OpSharding

+ SliceDimensions []*HloInstructionProto_SliceDimensions

+ SourceTargetPairs []*SourceTarget

+ StatisticsViz *StatisticsViz

+ TriangularSolveOptions *TriangularSolveOptions

+ TupleIndex int64

+ UniqueIndices bool

+ UseGlobalDeviceIds bool

+ Window *Window

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

+ func (*HloInstructionProto) ProtoMessage()

+ func (m *HloInstructionProto) GetOptionalCrossProgramPrefetchIndex() isHloInstructionProto_OptionalCrossProgramPrefetchIndex

+ func (x *HloInstructionProto) GetAllReduceId() int64

+ func (x *HloInstructionProto) GetAsyncExecutionThread() string

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

+ func (x *HloInstructionProto) GetBatchGroupCount() int64

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

+ func (x *HloInstructionProto) GetChannelId() int64

+ func (x *HloInstructionProto) GetCholeskyOptions() *CholeskyOptions

+ func (x *HloInstructionProto) GetComparisonDirection() string

+ func (x *HloInstructionProto) GetComparisonType() string

+ func (x *HloInstructionProto) GetConstrainLayout() bool

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

+ func (x *HloInstructionProto) GetConvolutionDimensionNumbers() *ConvolutionDimensionNumbers

+ func (x *HloInstructionProto) GetCrossProgramPrefetchIndex() int32

+ func (x *HloInstructionProto) GetCustomCallApiVersion() CustomCallApiVersion

+ func (x *HloInstructionProto) GetCustomCallHasSideEffect() bool

+ func (x *HloInstructionProto) GetCustomCallSchedule() CustomCallSchedule

+ func (x *HloInstructionProto) GetCustomCallTarget() string

+ func (x *HloInstructionProto) GetDelta() int64

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

+ func (x *HloInstructionProto) GetDistribution() RandomDistribution

+ func (x *HloInstructionProto) GetDomainEntrySharding() *OpSharding

+ func (x *HloInstructionProto) GetDomainExitSharding() *OpSharding

+ func (x *HloInstructionProto) GetDotDimensionNumbers() *DotDimensionNumbers

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

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

+ func (x *HloInstructionProto) GetEpsilon() float32

+ func (x *HloInstructionProto) GetExponentBits() int32

+ func (x *HloInstructionProto) GetFeatureGroupCount() int64

+ func (x *HloInstructionProto) GetFeatureIndex() int64

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

+ func (x *HloInstructionProto) GetFftType() FftType

+ func (x *HloInstructionProto) GetFrontendAttributes() *FrontendAttributes

+ func (x *HloInstructionProto) GetFusionKind() string

+ func (x *HloInstructionProto) GetGatherDimensionNumbers() *GatherDimensionNumbers

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

+ func (x *HloInstructionProto) GetId() int64

+ func (x *HloInstructionProto) GetIndicesAreSorted() bool

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

+ func (x *HloInstructionProto) GetIsCrossProgramPrefetch() bool

+ func (x *HloInstructionProto) GetIsHostTransfer() bool

+ func (x *HloInstructionProto) GetIsStable() bool

+ func (x *HloInstructionProto) GetK() int64

+ func (x *HloInstructionProto) GetLargest() bool

+ func (x *HloInstructionProto) GetLiteral() *LiteralProto

+ func (x *HloInstructionProto) GetMantissaBits() int32

+ func (x *HloInstructionProto) GetMetadata() *OpMetadata

+ func (x *HloInstructionProto) GetName() string

+ func (x *HloInstructionProto) GetOpcode() string

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

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

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

+ func (x *HloInstructionProto) GetOutfeedShape() *ShapeProto

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

+ func (x *HloInstructionProto) GetPaddingConfig() *PaddingConfig

+ func (x *HloInstructionProto) GetPaddingType() PaddingType

+ func (x *HloInstructionProto) GetParameterNumber() int64

+ func (x *HloInstructionProto) GetParameterReplication() *ParameterReplication

+ func (x *HloInstructionProto) GetPrecisionConfig() *PrecisionConfig

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

+ func (x *HloInstructionProto) GetRngAlgorithm() RandomAlgorithm

+ func (x *HloInstructionProto) GetScatterDimensionNumbers() *ScatterDimensionNumbers

+ func (x *HloInstructionProto) GetShape() *ShapeProto

+ func (x *HloInstructionProto) GetSharding() *OpSharding

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

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

+ func (x *HloInstructionProto) GetStatisticsViz() *StatisticsViz

+ func (x *HloInstructionProto) GetTriangularSolveOptions() *TriangularSolveOptions

+ func (x *HloInstructionProto) GetTupleIndex() int64

+ func (x *HloInstructionProto) GetUniqueIndices() bool

+ func (x *HloInstructionProto) GetUseGlobalDeviceIds() bool

+ func (x *HloInstructionProto) GetWindow() *Window

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

+ func (x *HloInstructionProto) Reset()

+ func (x *HloInstructionProto) String() string

+ type HloInstructionProto_CrossProgramPrefetchIndex struct

+ CrossProgramPrefetchIndex int32

+ type HloInstructionProto_SliceDimensions struct

+ Limit int64

+ Start int64

+ Stride int64

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

+ func (*HloInstructionProto_SliceDimensions) ProtoMessage()

+ func (x *HloInstructionProto_SliceDimensions) GetLimit() int64

+ func (x *HloInstructionProto_SliceDimensions) GetStart() int64

+ func (x *HloInstructionProto_SliceDimensions) GetStride() int64

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

+ func (x *HloInstructionProto_SliceDimensions) Reset()

+ func (x *HloInstructionProto_SliceDimensions) String() string

+ type HloModuleConfigProto struct

+ AliasPassthroughParams bool

+ AllowSeparateShardingPrograms bool

+ AllowSpmdShardingPropagationToOutput []bool

+ AllowSpmdShardingPropagationToParameters []bool

+ AnalysisAllowanceMap map[string]int64

+ AutoSpmdPartitioningMeshIds []int64

+ AutoSpmdPartitioningMeshShape []int64

+ ContentAwareComputationSorting bool

+ DebugOptions *DebugOptions

+ DeduplicateHlo bool

+ DeviceMemorySize int64

+ DeviceType string

+ DotConfig map[string]*HloModuleConfigProto_Int64List

+ EntryComputationLayout *ProgramShapeProto

+ FdoProfile []byte

+ FusionConfig []*HloModuleConfigProto_BoolList

+ FusionConfigCollection HloModuleConfigProto_FusionConfigCollection

+ IntraOpParallelismThreads int64

+ LaunchId int32

+ LayoutConfig []*HloModuleConfigProto_Int64ListList

+ MatrixUnitOperandPrecision PrecisionConfig_Precision

+ MemorySpaceAssignmentConfig []uint64

+ NumPartitions int64

+ ParamRequiresBroadcastViaCollectives []bool

+ PhaseIndex int32

+ PhaseOrderingConfig []*HloModuleConfigProto_BoolList

+ ReplicaCount int64

+ Seed uint64

+ ShardableValueUpdatePairs []*ShardableValueUpdatePairProto

+ StaticDeviceAssignment *DeviceAssignmentProto

+ UseAutoSpmdPartitioning bool

+ UseSpmdPartitioning bool

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

+ func (*HloModuleConfigProto) ProtoMessage()

+ func (x *HloModuleConfigProto) GetAliasPassthroughParams() bool

+ func (x *HloModuleConfigProto) GetAllowSeparateShardingPrograms() bool

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

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

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

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

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

+ func (x *HloModuleConfigProto) GetContentAwareComputationSorting() bool

+ func (x *HloModuleConfigProto) GetDebugOptions() *DebugOptions

+ func (x *HloModuleConfigProto) GetDeduplicateHlo() bool

+ func (x *HloModuleConfigProto) GetDeviceMemorySize() int64

+ func (x *HloModuleConfigProto) GetDeviceType() string

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

+ func (x *HloModuleConfigProto) GetEntryComputationLayout() *ProgramShapeProto

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

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

+ func (x *HloModuleConfigProto) GetFusionConfigCollection() HloModuleConfigProto_FusionConfigCollection

+ func (x *HloModuleConfigProto) GetIntraOpParallelismThreads() int64

+ func (x *HloModuleConfigProto) GetLaunchId() int32

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

+ func (x *HloModuleConfigProto) GetMatrixUnitOperandPrecision() PrecisionConfig_Precision

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

+ func (x *HloModuleConfigProto) GetNumPartitions() int64

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

+ func (x *HloModuleConfigProto) GetPhaseIndex() int32

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

+ func (x *HloModuleConfigProto) GetReplicaCount() int64

+ func (x *HloModuleConfigProto) GetSeed() uint64

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

+ func (x *HloModuleConfigProto) GetStaticDeviceAssignment() *DeviceAssignmentProto

+ func (x *HloModuleConfigProto) GetUseAutoSpmdPartitioning() bool

+ func (x *HloModuleConfigProto) GetUseSpmdPartitioning() bool

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

+ func (x *HloModuleConfigProto) Reset()

+ func (x *HloModuleConfigProto) String() string

+ type HloModuleConfigProto_BoolList struct

+ Vals []bool

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

+ func (*HloModuleConfigProto_BoolList) ProtoMessage()

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

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

+ func (x *HloModuleConfigProto_BoolList) Reset()

+ func (x *HloModuleConfigProto_BoolList) String() string

+ type HloModuleConfigProto_FusionConfigCollection int32

+ const HloModuleConfigProto_OFF

+ const HloModuleConfigProto_PER_EDGE

+ const HloModuleConfigProto_PER_NODE

+ func (HloModuleConfigProto_FusionConfigCollection) Descriptor() protoreflect.EnumDescriptor

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

+ func (HloModuleConfigProto_FusionConfigCollection) Type() protoreflect.EnumType

+ func (x HloModuleConfigProto_FusionConfigCollection) Enum() *HloModuleConfigProto_FusionConfigCollection

+ func (x HloModuleConfigProto_FusionConfigCollection) Number() protoreflect.EnumNumber

+ func (x HloModuleConfigProto_FusionConfigCollection) String() string

+ type HloModuleConfigProto_Int64List struct

+ Vals []int64

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

+ func (*HloModuleConfigProto_Int64List) ProtoMessage()

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

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

+ func (x *HloModuleConfigProto_Int64List) Reset()

+ func (x *HloModuleConfigProto_Int64List) String() string

+ type HloModuleConfigProto_Int64ListList struct

+ Lists []*HloModuleConfigProto_Int64List

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

+ func (*HloModuleConfigProto_Int64ListList) ProtoMessage()

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

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

+ func (x *HloModuleConfigProto_Int64ListList) Reset()

+ func (x *HloModuleConfigProto_Int64ListList) String() string

+ type HloModuleGroupProto struct

+ HloModules []*HloModuleProto

+ Name string

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

+ func (*HloModuleGroupProto) ProtoMessage()

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

+ func (x *HloModuleGroupProto) GetName() string

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

+ func (x *HloModuleGroupProto) Reset()

+ func (x *HloModuleGroupProto) String() string

+ type HloModuleMetadataProto struct

+ CanonicalModuleId int64

+ ModuleGroupName string

+ OriginalModuleId int64

+ PartitionedModuleIds []int64

+ PassMetadata []*HloPassMetadata

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

+ func (*HloModuleMetadataProto) ProtoMessage()

+ func (x *HloModuleMetadataProto) GetCanonicalModuleId() int64

+ func (x *HloModuleMetadataProto) GetModuleGroupName() string

+ func (x *HloModuleMetadataProto) GetOriginalModuleId() int64

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

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

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

+ func (x *HloModuleMetadataProto) Reset()

+ func (x *HloModuleMetadataProto) String() string

+ type HloModuleProto struct

+ BufferDonor *HloBufferDonorProto

+ Computations []*HloComputationProto

+ CrossProgramPrefetches []*CrossProgramPrefetch

+ DeviceAssignment *DeviceAssignmentProto

+ EntryComputationId int64

+ EntryComputationName string

+ FrontendAttributes *FrontendAttributes

+ HostProgramShape *ProgramShapeProto

+ Id int64

+ InputOutputAlias *HloInputOutputAliasProto

+ IsDynamic bool

+ Name string

+ ProfileInfo []*HloModuleProto_ProfileInfo

+ Schedule *HloScheduleProto

+ SpmdOutputSharding *OpSharding

+ SpmdParametersShardings []*OpSharding

+ StackFrameIndex *StackFrameIndexProto

+ UseAutoSpmdPartitioning bool

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

+ func (*HloModuleProto) ProtoMessage()

+ func (x *HloModuleProto) GetBufferDonor() *HloBufferDonorProto

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

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

+ func (x *HloModuleProto) GetDeviceAssignment() *DeviceAssignmentProto

+ func (x *HloModuleProto) GetEntryComputationId() int64

+ func (x *HloModuleProto) GetEntryComputationName() string

+ func (x *HloModuleProto) GetFrontendAttributes() *FrontendAttributes

+ func (x *HloModuleProto) GetHostProgramShape() *ProgramShapeProto

+ func (x *HloModuleProto) GetId() int64

+ func (x *HloModuleProto) GetInputOutputAlias() *HloInputOutputAliasProto

+ func (x *HloModuleProto) GetIsDynamic() bool

+ func (x *HloModuleProto) GetName() string

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

+ func (x *HloModuleProto) GetSchedule() *HloScheduleProto

+ func (x *HloModuleProto) GetSpmdOutputSharding() *OpSharding

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

+ func (x *HloModuleProto) GetStackFrameIndex() *StackFrameIndexProto

+ func (x *HloModuleProto) GetUseAutoSpmdPartitioning() bool

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

+ func (x *HloModuleProto) Reset()

+ func (x *HloModuleProto) String() string

+ type HloModuleProtoWithConfig struct

+ Config *HloModuleConfigProto

+ HloModule *HloModuleProto

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

+ func (*HloModuleProtoWithConfig) ProtoMessage()

+ func (x *HloModuleProtoWithConfig) GetConfig() *HloModuleConfigProto

+ func (x *HloModuleProtoWithConfig) GetHloModule() *HloModuleProto

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

+ func (x *HloModuleProtoWithConfig) Reset()

+ func (x *HloModuleProtoWithConfig) String() string

+ type HloModuleProto_ProfileInfo struct

+ CompilationEvent CompilationEvent

+ Fingerprint string

+ ProfileSource ProfileSource

+ ProfileType HloModuleProto_ProfileType

+ RelativeSpeedup float64

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

+ func (*HloModuleProto_ProfileInfo) ProtoMessage()

+ func (x *HloModuleProto_ProfileInfo) GetCompilationEvent() CompilationEvent

+ func (x *HloModuleProto_ProfileInfo) GetFingerprint() string

+ func (x *HloModuleProto_ProfileInfo) GetProfileSource() ProfileSource

+ func (x *HloModuleProto_ProfileInfo) GetProfileType() HloModuleProto_ProfileType

+ func (x *HloModuleProto_ProfileInfo) GetRelativeSpeedup() float64

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

+ func (x *HloModuleProto_ProfileInfo) Reset()

+ func (x *HloModuleProto_ProfileInfo) String() string

+ type HloModuleProto_ProfileType int32

+ const HloModuleProto_DOT

+ const HloModuleProto_FLAG

+ const HloModuleProto_FUSION

+ const HloModuleProto_INVALID

+ const HloModuleProto_LAYOUT

+ func (HloModuleProto_ProfileType) Descriptor() protoreflect.EnumDescriptor

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

+ func (HloModuleProto_ProfileType) Type() protoreflect.EnumType

+ func (x HloModuleProto_ProfileType) Enum() *HloModuleProto_ProfileType

+ func (x HloModuleProto_ProfileType) Number() protoreflect.EnumNumber

+ func (x HloModuleProto_ProfileType) String() string

+ type HloPassMetadata struct

+ CustomMetadata *anypb.Any

+ DumpFilenames []string

+ EndTimestampUsec int64

+ ModuleChanged bool

+ ModuleGroupModuleIds []int64

+ ModuleId int64

+ PassId int64

+ PassName string

+ PipelineName string

+ StartTimestampUsec int64

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

+ func (*HloPassMetadata) ProtoMessage()

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

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

+ func (x *HloPassMetadata) GetEndTimestampUsec() int64

+ func (x *HloPassMetadata) GetModuleChanged() bool

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

+ func (x *HloPassMetadata) GetModuleId() int64

+ func (x *HloPassMetadata) GetPassId() int64

+ func (x *HloPassMetadata) GetPassName() string

+ func (x *HloPassMetadata) GetPipelineName() string

+ func (x *HloPassMetadata) GetStartTimestampUsec() int64

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

+ func (x *HloPassMetadata) Reset()

+ func (x *HloPassMetadata) String() string

+ type HloProto struct

+ BufferAssignment *BufferAssignmentProto

+ HloModule *HloModuleProto

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

+ func (*HloProto) ProtoMessage()

+ func (x *HloProto) GetBufferAssignment() *BufferAssignmentProto

+ func (x *HloProto) GetHloModule() *HloModuleProto

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

+ func (x *HloProto) Reset()

+ func (x *HloProto) String() string

+ type HloScheduleProto struct

+ Sequences map[int64]*HloScheduleProto_InstructionSequence

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

+ func (*HloScheduleProto) ProtoMessage()

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

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

+ func (x *HloScheduleProto) Reset()

+ func (x *HloScheduleProto) String() string

+ type HloScheduleProto_InstructionSequence struct

+ InstructionIds []int64

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

+ func (*HloScheduleProto_InstructionSequence) ProtoMessage()

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

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

+ func (x *HloScheduleProto_InstructionSequence) Reset()

+ func (x *HloScheduleProto_InstructionSequence) String() string

+ type HloSnapshot struct

+ Arguments []*LiteralProto

+ ExecutionPlatform string

+ Hlo *HloProto

+ Result *LiteralProto

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

+ func (*HloSnapshot) ProtoMessage()

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

+ func (x *HloSnapshot) GetExecutionPlatform() string

+ func (x *HloSnapshot) GetHlo() *HloProto

+ func (x *HloSnapshot) GetResult() *LiteralProto

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

+ func (x *HloSnapshot) Reset()

+ func (x *HloSnapshot) String() string

+ type Kind int32

+ const Kind_MAY_ALIAS

+ const Kind_MUST_ALIAS

+ const Kind_UNDEFINED_ALIAS

+ func (Kind) Descriptor() protoreflect.EnumDescriptor

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

+ func (Kind) Type() protoreflect.EnumType

+ func (x Kind) Enum() *Kind

+ func (x Kind) Number() protoreflect.EnumNumber

+ func (x Kind) String() string

+ type LayoutProto struct

+ DimLevelTypes []DimLevelType

+ DimOrdered []bool

+ DimUnique []bool

+ DynamicShapeMetadataPrefixBytes int64

+ ElementSizeInBits int64

+ IndexPrimitiveType PrimitiveType

+ MemorySpace int64

+ MinorToMajor []int64

+ PhysicalShape *ShapeProto

+ PointerPrimitiveType PrimitiveType

+ SplitConfigs []*SplitConfigProto

+ TailPaddingAlignmentInElements int64

+ Tiles []*TileProto

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

+ func (*LayoutProto) ProtoMessage()

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

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

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

+ func (x *LayoutProto) GetDynamicShapeMetadataPrefixBytes() int64

+ func (x *LayoutProto) GetElementSizeInBits() int64

+ func (x *LayoutProto) GetIndexPrimitiveType() PrimitiveType

+ func (x *LayoutProto) GetMemorySpace() int64

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

+ func (x *LayoutProto) GetPhysicalShape() *ShapeProto

+ func (x *LayoutProto) GetPointerPrimitiveType() PrimitiveType

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

+ func (x *LayoutProto) GetTailPaddingAlignmentInElements() int64

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

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

+ func (x *LayoutProto) Reset()

+ func (x *LayoutProto) String() string

+ type LiteralProto struct

+ Bf16S []byte

+ C128S []float64

+ C64S []float32

+ F16S []byte

+ F32S []float32

+ F64S []float64

+ F8E4M3B11Fnuzs []byte

+ F8E4M3Fns []byte

+ F8E4M3Fnuzs []byte

+ F8E5M2Fnuzs []byte

+ F8E5M2S []byte

+ Preds []bool

+ S16S []byte

+ S2S []byte

+ S32S []int32

+ S4S []byte

+ S64S []int64

+ S8S []byte

+ Shape *ShapeProto

+ SparseIndices []int64

+ TupleLiterals []*LiteralProto

+ U16S []byte

+ U2S []byte

+ U32S []uint32

+ U4S []byte

+ U64S []uint64

+ U8S []byte

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

+ func (*LiteralProto) ProtoMessage()

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

+ func (x *LiteralProto) GetShape() *ShapeProto

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

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

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

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

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

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

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

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

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

+ func (x *LiteralProto) Reset()

+ func (x *LiteralProto) String() string

+ type LoadDataRequest struct

+ ColumnioField string

+ ColumnioTabletPath string

+ ElementShape *ShapeProto

+ Limit int64

+ Offset int64

+ Zip bool

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

+ func (*LoadDataRequest) ProtoMessage()

+ func (x *LoadDataRequest) GetColumnioField() string

+ func (x *LoadDataRequest) GetColumnioTabletPath() string

+ func (x *LoadDataRequest) GetElementShape() *ShapeProto

+ func (x *LoadDataRequest) GetLimit() int64

+ func (x *LoadDataRequest) GetOffset() int64

+ func (x *LoadDataRequest) GetZip() bool

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

+ func (x *LoadDataRequest) Reset()

+ func (x *LoadDataRequest) String() string

+ type LoadDataResponse struct

+ AvailableRows int64

+ Data *GlobalDataHandle

+ DataShape *ShapeProto

+ Nanoseconds int64

+ RowsLoaded int64

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

+ func (*LoadDataResponse) ProtoMessage()

+ func (x *LoadDataResponse) GetAvailableRows() int64

+ func (x *LoadDataResponse) GetData() *GlobalDataHandle

+ func (x *LoadDataResponse) GetDataShape() *ShapeProto

+ func (x *LoadDataResponse) GetNanoseconds() int64

+ func (x *LoadDataResponse) GetRowsLoaded() int64

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

+ func (x *LoadDataResponse) Reset()

+ func (x *LoadDataResponse) String() string

+ type LogicalBufferProto struct

+ Color int64

+ DefinedAt *LogicalBufferProto_Location

+ Id int64

+ Size int64

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

+ func (*LogicalBufferProto) ProtoMessage()

+ func (x *LogicalBufferProto) GetColor() int64

+ func (x *LogicalBufferProto) GetDefinedAt() *LogicalBufferProto_Location

+ func (x *LogicalBufferProto) GetId() int64

+ func (x *LogicalBufferProto) GetSize() int64

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

+ func (x *LogicalBufferProto) Reset()

+ func (x *LogicalBufferProto) String() string

+ type LogicalBufferProto_Location struct

+ InstructionId int64

+ InstructionName string

+ ShapeIndex []int64

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

+ func (*LogicalBufferProto_Location) ProtoMessage()

+ func (x *LogicalBufferProto_Location) GetInstructionId() int64

+ func (x *LogicalBufferProto_Location) GetInstructionName() string

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

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

+ func (x *LogicalBufferProto_Location) Reset()

+ func (x *LogicalBufferProto_Location) String() string

+ type NormKind int32

+ const NormKind_LAYER_BWD

+ const NormKind_LAYER_FWD_INFER

+ const NormKind_LAYER_FWD_TRAIN

+ func (NormKind) Descriptor() protoreflect.EnumDescriptor

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

+ func (NormKind) Type() protoreflect.EnumType

+ func (x NormKind) Enum() *NormKind

+ func (x NormKind) Number() protoreflect.EnumNumber

+ func (x NormKind) String() string

+ type OpMetadata struct

+ DeduplicatedName string

+ OpName string

+ OpType string

+ PreserveLayout bool

+ ProfileInfo *OpMetadata_ProfileInfo

+ ProfileType []ProfileType

+ SizeOfGeneratedCodeInBytes int64

+ SizeOfMemoryWorkingSetInBytes int64

+ SourceFile string

+ SourceLine int32

+ StackFrameId int32

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

+ func (*OpMetadata) ProtoMessage()

+ func (x *OpMetadata) GetDeduplicatedName() string

+ func (x *OpMetadata) GetOpName() string

+ func (x *OpMetadata) GetOpType() string

+ func (x *OpMetadata) GetPreserveLayout() bool

+ func (x *OpMetadata) GetProfileInfo() *OpMetadata_ProfileInfo

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

+ func (x *OpMetadata) GetSizeOfGeneratedCodeInBytes() int64

+ func (x *OpMetadata) GetSizeOfMemoryWorkingSetInBytes() int64

+ func (x *OpMetadata) GetSourceFile() string

+ func (x *OpMetadata) GetSourceLine() int32

+ func (x *OpMetadata) GetStackFrameId() int32

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

+ func (x *OpMetadata) Reset()

+ func (x *OpMetadata) String() string

+ type OpMetadata_ProfileInfo struct

+ CompilationEvent CompilationEvent

+ ProfileSource ProfileSource

+ ProfileType []ProfileType

+ RelativeSpeedup float64

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

+ func (*OpMetadata_ProfileInfo) ProtoMessage()

+ func (x *OpMetadata_ProfileInfo) GetCompilationEvent() CompilationEvent

+ func (x *OpMetadata_ProfileInfo) GetProfileSource() ProfileSource

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

+ func (x *OpMetadata_ProfileInfo) GetRelativeSpeedup() float64

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

+ func (x *OpMetadata_ProfileInfo) Reset()

+ func (x *OpMetadata_ProfileInfo) String() string

+ type OpSharding struct

+ IotaReshapeDims []int64

+ IotaTransposePerm []int32

+ IsShardGroup bool

+ LastTileDims []OpSharding_Type

+ Metadata []*OpMetadata

+ ReplicateOnLastTileDim bool

+ ShardGroupId int64

+ ShardGroupType OpSharding_ShardGroupType

+ TileAssignmentDevices []int64

+ TileAssignmentDimensions []int64

+ TileShape *ShapeProto

+ TupleShardings []*OpSharding

+ Type OpSharding_Type

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

+ func (*OpSharding) ProtoMessage()

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

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

+ func (x *OpSharding) GetIsShardGroup() bool

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

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

+ func (x *OpSharding) GetReplicateOnLastTileDim() bool

+ func (x *OpSharding) GetShardGroupId() int64

+ func (x *OpSharding) GetShardGroupType() OpSharding_ShardGroupType

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

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

+ func (x *OpSharding) GetTileShape() *ShapeProto

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

+ func (x *OpSharding) GetType() OpSharding_Type

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

+ func (x *OpSharding) Reset()

+ func (x *OpSharding) String() string

+ type OpSharding_ShardGroupType int32

+ const OpSharding_AS

+ const OpSharding_LIKE

+ func (OpSharding_ShardGroupType) Descriptor() protoreflect.EnumDescriptor

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

+ func (OpSharding_ShardGroupType) Type() protoreflect.EnumType

+ func (x OpSharding_ShardGroupType) Enum() *OpSharding_ShardGroupType

+ func (x OpSharding_ShardGroupType) Number() protoreflect.EnumNumber

+ func (x OpSharding_ShardGroupType) String() string

+ type OpSharding_Type int32

+ const OpSharding_MANUAL

+ const OpSharding_MAXIMAL

+ const OpSharding_OTHER

+ const OpSharding_REPLICATED

+ const OpSharding_TUPLE

+ const OpSharding_UNKNOWN

+ func (OpSharding_Type) Descriptor() protoreflect.EnumDescriptor

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

+ func (OpSharding_Type) Type() protoreflect.EnumType

+ func (x OpSharding_Type) Enum() *OpSharding_Type

+ func (x OpSharding_Type) Number() protoreflect.EnumNumber

+ func (x OpSharding_Type) String() string

+ type OptionOverrideProto struct

+ Value isOptionOverrideProto_Value

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

+ func (*OptionOverrideProto) ProtoMessage()

+ func (m *OptionOverrideProto) GetValue() isOptionOverrideProto_Value

+ func (x *OptionOverrideProto) GetBoolField() bool

+ func (x *OptionOverrideProto) GetDoubleField() float64

+ func (x *OptionOverrideProto) GetIntField() int64

+ func (x *OptionOverrideProto) GetStringField() string

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

+ func (x *OptionOverrideProto) Reset()

+ func (x *OptionOverrideProto) String() string

+ type OptionOverrideProto_BoolField struct

+ BoolField bool

+ type OptionOverrideProto_DoubleField struct

+ DoubleField float64

+ type OptionOverrideProto_IntField struct

+ IntField int64

+ type OptionOverrideProto_StringField struct

+ StringField string

+ type OutputOperandAliasing struct

+ OperandIndex int64

+ OperandShapeIndex []int64

+ OutputShapeIndex []int64

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

+ func (*OutputOperandAliasing) ProtoMessage()

+ func (x *OutputOperandAliasing) GetOperandIndex() int64

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

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

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

+ func (x *OutputOperandAliasing) Reset()

+ func (x *OutputOperandAliasing) String() string

+ type PaddingConfig struct

+ Dimensions []*PaddingConfig_PaddingConfigDimension

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

+ func (*PaddingConfig) ProtoMessage()

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

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

+ func (x *PaddingConfig) Reset()

+ func (x *PaddingConfig) String() string

+ type PaddingConfig_PaddingConfigDimension struct

+ EdgePaddingHigh int64

+ EdgePaddingLow int64

+ InteriorPadding int64

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

+ func (*PaddingConfig_PaddingConfigDimension) ProtoMessage()

+ func (x *PaddingConfig_PaddingConfigDimension) GetEdgePaddingHigh() int64

+ func (x *PaddingConfig_PaddingConfigDimension) GetEdgePaddingLow() int64

+ func (x *PaddingConfig_PaddingConfigDimension) GetInteriorPadding() int64

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

+ func (x *PaddingConfig_PaddingConfigDimension) Reset()

+ func (x *PaddingConfig_PaddingConfigDimension) String() string

+ type PaddingType int32

+ const PaddingType_PADDING_INVALID

+ const PaddingType_PADDING_SAME

+ const PaddingType_PADDING_VALID

+ func (PaddingType) Descriptor() protoreflect.EnumDescriptor

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

+ func (PaddingType) Type() protoreflect.EnumType

+ func (x PaddingType) Enum() *PaddingType

+ func (x PaddingType) Number() protoreflect.EnumNumber

+ func (x PaddingType) String() string

+ type ParameterReplication struct

+ ReplicatedAtLeafBuffers []bool

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

+ func (*ParameterReplication) ProtoMessage()

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

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

+ func (x *ParameterReplication) Reset()

+ func (x *ParameterReplication) String() string

+ type PrecisionConfig struct

+ Algorithm PrecisionConfig_Algorithm

+ OperandPrecision []PrecisionConfig_Precision

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

+ func (*PrecisionConfig) ProtoMessage()

+ func (x *PrecisionConfig) GetAlgorithm() PrecisionConfig_Algorithm

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

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

+ func (x *PrecisionConfig) Reset()

+ func (x *PrecisionConfig) String() string

+ type PrecisionConfig_Algorithm int32

+ const PrecisionConfig_ALG_DOT_ANY_F8_ANY_F8_F32

+ const PrecisionConfig_ALG_DOT_ANY_F8_ANY_F8_F32_FAST_ACCUM

+ const PrecisionConfig_ALG_DOT_BF16_BF16_BF16

+ const PrecisionConfig_ALG_DOT_BF16_BF16_F32

+ const PrecisionConfig_ALG_DOT_BF16_BF16_F32_X3

+ const PrecisionConfig_ALG_DOT_BF16_BF16_F32_X6

+ const PrecisionConfig_ALG_DOT_F16_F16_F16

+ const PrecisionConfig_ALG_DOT_F16_F16_F32

+ const PrecisionConfig_ALG_DOT_F32_F32_F32

+ const PrecisionConfig_ALG_DOT_F64_F64_F64

+ const PrecisionConfig_ALG_DOT_TF32_TF32_F32

+ const PrecisionConfig_ALG_DOT_TF32_TF32_F32_X3

+ const PrecisionConfig_ALG_UNSET

+ func (PrecisionConfig_Algorithm) Descriptor() protoreflect.EnumDescriptor

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

+ func (PrecisionConfig_Algorithm) Type() protoreflect.EnumType

+ func (x PrecisionConfig_Algorithm) Enum() *PrecisionConfig_Algorithm

+ func (x PrecisionConfig_Algorithm) Number() protoreflect.EnumNumber

+ func (x PrecisionConfig_Algorithm) String() string

+ type PrecisionConfig_Precision int32

+ const PrecisionConfig_DEFAULT

+ const PrecisionConfig_HIGH

+ const PrecisionConfig_HIGHEST

+ const PrecisionConfig_PACKED_NIBBLE

+ func (PrecisionConfig_Precision) Descriptor() protoreflect.EnumDescriptor

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

+ func (PrecisionConfig_Precision) Type() protoreflect.EnumType

+ func (x PrecisionConfig_Precision) Enum() *PrecisionConfig_Precision

+ func (x PrecisionConfig_Precision) Number() protoreflect.EnumNumber

+ func (x PrecisionConfig_Precision) String() string

+ type PrimitiveType int32

+ const PrimitiveType_BF16

+ const PrimitiveType_C128

+ const PrimitiveType_C64

+ const PrimitiveType_F16

+ const PrimitiveType_F32

+ const PrimitiveType_F64

+ const PrimitiveType_F8E4M3B11FNUZ

+ const PrimitiveType_F8E4M3FN

+ const PrimitiveType_F8E4M3FNUZ

+ const PrimitiveType_F8E5M2

+ const PrimitiveType_F8E5M2FNUZ

+ const PrimitiveType_OPAQUE_TYPE

+ const PrimitiveType_PRED

+ const PrimitiveType_PRIMITIVE_TYPE_INVALID

+ const PrimitiveType_S16

+ const PrimitiveType_S2

+ const PrimitiveType_S32

+ const PrimitiveType_S4

+ const PrimitiveType_S64

+ const PrimitiveType_S8

+ const PrimitiveType_TOKEN

+ const PrimitiveType_TUPLE

+ const PrimitiveType_U16

+ const PrimitiveType_U2

+ const PrimitiveType_U32

+ const PrimitiveType_U4

+ const PrimitiveType_U64

+ const PrimitiveType_U8

+ func (PrimitiveType) Descriptor() protoreflect.EnumDescriptor

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

+ func (PrimitiveType) Type() protoreflect.EnumType

+ func (x PrimitiveType) Enum() *PrimitiveType

+ func (x PrimitiveType) Number() protoreflect.EnumNumber

+ func (x PrimitiveType) String() string

+ type ProfileSource int32

+ const ProfileSource_PROFILE_SOURCE_EMBEDDED

+ const ProfileSource_PROFILE_SOURCE_REMOTE

+ const ProfileSource_PROFILE_SOURCE_UNKNOWN_SOURCE

+ func (ProfileSource) Descriptor() protoreflect.EnumDescriptor

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

+ func (ProfileSource) Type() protoreflect.EnumType

+ func (x ProfileSource) Enum() *ProfileSource

+ func (x ProfileSource) Number() protoreflect.EnumNumber

+ func (x ProfileSource) String() string

+ type ProfileType int32

+ const ProfileType_FLAG

+ const ProfileType_INTEGER

+ const ProfileType_INVALID

+ const ProfileType_WINDOW

+ func (ProfileType) Descriptor() protoreflect.EnumDescriptor

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

+ func (ProfileType) Type() protoreflect.EnumType

+ func (x ProfileType) Enum() *ProfileType

+ func (x ProfileType) Number() protoreflect.EnumNumber

+ func (x ProfileType) String() string

+ type ProgramShapeProto struct

+ ParameterNames []string

+ Parameters []*ShapeProto

+ Result *ShapeProto

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

+ func (*ProgramShapeProto) ProtoMessage()

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

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

+ func (x *ProgramShapeProto) GetResult() *ShapeProto

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

+ func (x *ProgramShapeProto) Reset()

+ func (x *ProgramShapeProto) String() string

+ type RandomAlgorithm int32

+ const RandomAlgorithm_RNG_DEFAULT

+ const RandomAlgorithm_RNG_PHILOX

+ const RandomAlgorithm_RNG_THREE_FRY

+ func (RandomAlgorithm) Descriptor() protoreflect.EnumDescriptor

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

+ func (RandomAlgorithm) Type() protoreflect.EnumType

+ func (x RandomAlgorithm) Enum() *RandomAlgorithm

+ func (x RandomAlgorithm) Number() protoreflect.EnumNumber

+ func (x RandomAlgorithm) String() string

+ type RandomDistribution int32

+ const RandomDistribution_RNG_INVALID

+ const RandomDistribution_RNG_NORMAL

+ const RandomDistribution_RNG_UNIFORM

+ func (RandomDistribution) Descriptor() protoreflect.EnumDescriptor

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

+ func (RandomDistribution) Type() protoreflect.EnumType

+ func (x RandomDistribution) Enum() *RandomDistribution

+ func (x RandomDistribution) Number() protoreflect.EnumNumber

+ func (x RandomDistribution) String() string

+ type ReplicaGroup struct

+ ReplicaIds []int64

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

+ func (*ReplicaGroup) ProtoMessage()

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

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

+ func (x *ReplicaGroup) Reset()

+ func (x *ReplicaGroup) String() string

+ type ResetDeviceRequest struct

+ DeviceHandle *DeviceHandle

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

+ func (*ResetDeviceRequest) ProtoMessage()

+ func (x *ResetDeviceRequest) GetDeviceHandle() *DeviceHandle

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

+ func (x *ResetDeviceRequest) Reset()

+ func (x *ResetDeviceRequest) String() string

+ type ResetDeviceResponse struct

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

+ func (*ResetDeviceResponse) ProtoMessage()

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

+ func (x *ResetDeviceResponse) Reset()

+ func (x *ResetDeviceResponse) String() string

+ type RocmComputeCapabilityProto struct

+ GcnArchName string

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

+ func (*RocmComputeCapabilityProto) ProtoMessage()

+ func (x *RocmComputeCapabilityProto) GetGcnArchName() string

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

+ func (x *RocmComputeCapabilityProto) Reset()

+ func (x *RocmComputeCapabilityProto) String() string

+ type ScatterDimensionNumbers struct

+ IndexVectorDim int64

+ InsertedWindowDims []int64

+ ScatterDimsToOperandDims []int64

+ UpdateWindowDims []int64

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

+ func (*ScatterDimensionNumbers) ProtoMessage()

+ func (x *ScatterDimensionNumbers) GetIndexVectorDim() int64

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

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

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

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

+ func (x *ScatterDimensionNumbers) Reset()

+ func (x *ScatterDimensionNumbers) String() string

+ type ScheduleProto struct

+ ComputationId int64

+ CyclesPerMicrosecond int64

+ HloModule *HloModuleProto

+ Instructions []*ScheduleProto_Instruction

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

+ func (*ScheduleProto) ProtoMessage()

+ func (x *ScheduleProto) GetComputationId() int64

+ func (x *ScheduleProto) GetCyclesPerMicrosecond() int64

+ func (x *ScheduleProto) GetHloModule() *HloModuleProto

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

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

+ func (x *ScheduleProto) Reset()

+ func (x *ScheduleProto) String() string

+ type ScheduleProto_Instruction struct

+ EndTimestampCycles float64

+ Id int64

+ StartTimestampCycles float64

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

+ func (*ScheduleProto_Instruction) ProtoMessage()

+ func (x *ScheduleProto_Instruction) GetEndTimestampCycles() float64

+ func (x *ScheduleProto_Instruction) GetId() int64

+ func (x *ScheduleProto_Instruction) GetStartTimestampCycles() float64

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

+ func (x *ScheduleProto_Instruction) Reset()

+ func (x *ScheduleProto_Instruction) String() string

+ type ShapeProto struct

+ Dimensions []int64

+ ElementType PrimitiveType

+ IsDynamicDimension []bool

+ Layout *LayoutProto

+ TupleShapes []*ShapeProto

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

+ func (*ShapeProto) ProtoMessage()

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

+ func (x *ShapeProto) GetElementType() PrimitiveType

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

+ func (x *ShapeProto) GetLayout() *LayoutProto

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

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

+ func (x *ShapeProto) Reset()

+ func (x *ShapeProto) String() string

+ type ShardableValueUpdatePairProto struct

+ InputParameterNumber int64

+ OutputShapeIndex []int64

+ ParameterShapeIndex []int64

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

+ func (*ShardableValueUpdatePairProto) ProtoMessage()

+ func (x *ShardableValueUpdatePairProto) GetInputParameterNumber() int64

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

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

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

+ func (x *ShardableValueUpdatePairProto) Reset()

+ func (x *ShardableValueUpdatePairProto) String() string

+ type SortOptions struct

+ Descending bool

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

+ func (*SortOptions) ProtoMessage()

+ func (x *SortOptions) GetDescending() bool

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

+ func (x *SortOptions) Reset()

+ func (x *SortOptions) String() string

+ type SourceTarget struct

+ Source int64

+ Target int64

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

+ func (*SourceTarget) ProtoMessage()

+ func (x *SourceTarget) GetSource() int64

+ func (x *SourceTarget) GetTarget() int64

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

+ func (x *SourceTarget) Reset()

+ func (x *SourceTarget) String() string

+ type SparsityDescriptor struct

+ Dimension int32

+ Index int32

+ M int32

+ N int32

+ Type SparsityType

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

+ func (*SparsityDescriptor) ProtoMessage()

+ func (x *SparsityDescriptor) GetDimension() int32

+ func (x *SparsityDescriptor) GetIndex() int32

+ func (x *SparsityDescriptor) GetM() int32

+ func (x *SparsityDescriptor) GetN() int32

+ func (x *SparsityDescriptor) GetType() SparsityType

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

+ func (x *SparsityDescriptor) Reset()

+ func (x *SparsityDescriptor) String() string

+ type SparsityType int32

+ const SparsityType_SPARSITY_INVALID

+ const SparsityType_SPARSITY_STRUCTURED_N_M

+ func (SparsityType) Descriptor() protoreflect.EnumDescriptor

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

+ func (SparsityType) Type() protoreflect.EnumType

+ func (x SparsityType) Enum() *SparsityType

+ func (x SparsityType) Number() protoreflect.EnumNumber

+ func (x SparsityType) String() string

+ type SplitConfigProto struct

+ Dimension int64

+ SplitIndices []int64

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

+ func (*SplitConfigProto) ProtoMessage()

+ func (x *SplitConfigProto) GetDimension() int64

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

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

+ func (x *SplitConfigProto) Reset()

+ func (x *SplitConfigProto) String() string

+ type StackFrameIndexProto struct

+ FileLocations []*StackFrameIndexProto_FileLocation

+ FileNames []string

+ FunctionNames []string

+ StackFrames []*StackFrameIndexProto_StackFrame

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

+ func (*StackFrameIndexProto) ProtoMessage()

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

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

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

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

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

+ func (x *StackFrameIndexProto) Reset()

+ func (x *StackFrameIndexProto) String() string

+ type StackFrameIndexProto_FileLocation struct

+ Column int32

+ FileNameId int32

+ FunctionNameId int32

+ Line int32

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

+ func (*StackFrameIndexProto_FileLocation) ProtoMessage()

+ func (x *StackFrameIndexProto_FileLocation) GetColumn() int32

+ func (x *StackFrameIndexProto_FileLocation) GetFileNameId() int32

+ func (x *StackFrameIndexProto_FileLocation) GetFunctionNameId() int32

+ func (x *StackFrameIndexProto_FileLocation) GetLine() int32

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

+ func (x *StackFrameIndexProto_FileLocation) Reset()

+ func (x *StackFrameIndexProto_FileLocation) String() string

+ type StackFrameIndexProto_StackFrame struct

+ FileLocationId int32

+ ParentFrameId int32

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

+ func (*StackFrameIndexProto_StackFrame) ProtoMessage()

+ func (x *StackFrameIndexProto_StackFrame) GetFileLocationId() int32

+ func (x *StackFrameIndexProto_StackFrame) GetParentFrameId() int32

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

+ func (x *StackFrameIndexProto_StackFrame) Reset()

+ func (x *StackFrameIndexProto_StackFrame) String() string

+ type Statistic struct

+ StatName string

+ StatVal float64

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

+ func (*Statistic) ProtoMessage()

+ func (x *Statistic) GetStatName() string

+ func (x *Statistic) GetStatVal() float64

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

+ func (x *Statistic) Reset()

+ func (x *Statistic) String() string

+ type StatisticsViz struct

+ StatIndexToVisualize int64

+ Statistics []*Statistic

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

+ func (*StatisticsViz) ProtoMessage()

+ func (x *StatisticsViz) GetStatIndexToVisualize() int64

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

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

+ func (x *StatisticsViz) Reset()

+ func (x *StatisticsViz) String() string

+ type TensorDescriptorProto struct

+ DataType DataType

+ Dimensions []int64

+ LayoutOneof isTensorDescriptorProto_LayoutOneof

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

+ func (*TensorDescriptorProto) ProtoMessage()

+ func (m *TensorDescriptorProto) GetLayoutOneof() isTensorDescriptorProto_LayoutOneof

+ func (x *TensorDescriptorProto) GetDataLayout() DataLayout

+ func (x *TensorDescriptorProto) GetDataType() DataType

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

+ func (x *TensorDescriptorProto) GetFilterLayout() FilterLayout

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

+ func (x *TensorDescriptorProto) Reset()

+ func (x *TensorDescriptorProto) String() string

+ type TensorDescriptorProto_DataLayout struct

+ DataLayout DataLayout

+ type TensorDescriptorProto_FilterLayout struct

+ FilterLayout FilterLayout

+ type TileProto struct

+ Dimensions []int64

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

+ func (*TileProto) ProtoMessage()

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

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

+ func (x *TileProto) Reset()

+ func (x *TileProto) String() string

+ type TransferFromOutfeedRequest struct

+ DeviceHandle *DeviceHandle

+ ReplicaId int64

+ ShapeWithLayout *ShapeProto

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

+ func (*TransferFromOutfeedRequest) ProtoMessage()

+ func (x *TransferFromOutfeedRequest) GetDeviceHandle() *DeviceHandle

+ func (x *TransferFromOutfeedRequest) GetReplicaId() int64

+ func (x *TransferFromOutfeedRequest) GetShapeWithLayout() *ShapeProto

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

+ func (x *TransferFromOutfeedRequest) Reset()

+ func (x *TransferFromOutfeedRequest) String() string

+ type TransferFromOutfeedResponse struct

+ Literal *LiteralProto

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

+ func (*TransferFromOutfeedResponse) ProtoMessage()

+ func (x *TransferFromOutfeedResponse) GetLiteral() *LiteralProto

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

+ func (x *TransferFromOutfeedResponse) Reset()

+ func (x *TransferFromOutfeedResponse) String() string

+ type TransferToClientRequest struct

+ Data *GlobalDataHandle

+ ShapeWithLayout *ShapeProto

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

+ func (*TransferToClientRequest) ProtoMessage()

+ func (x *TransferToClientRequest) GetData() *GlobalDataHandle

+ func (x *TransferToClientRequest) GetShapeWithLayout() *ShapeProto

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

+ func (x *TransferToClientRequest) Reset()

+ func (x *TransferToClientRequest) String() string

+ type TransferToClientResponse struct

+ Literal *LiteralProto

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

+ func (*TransferToClientResponse) ProtoMessage()

+ func (x *TransferToClientResponse) GetLiteral() *LiteralProto

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

+ func (x *TransferToClientResponse) Reset()

+ func (x *TransferToClientResponse) String() string

+ type TransferToInfeedRequest struct

+ DeviceHandle *DeviceHandle

+ Literal *LiteralProto

+ ReplicaId int64

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

+ func (*TransferToInfeedRequest) ProtoMessage()

+ func (x *TransferToInfeedRequest) GetDeviceHandle() *DeviceHandle

+ func (x *TransferToInfeedRequest) GetLiteral() *LiteralProto

+ func (x *TransferToInfeedRequest) GetReplicaId() int64

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

+ func (x *TransferToInfeedRequest) Reset()

+ func (x *TransferToInfeedRequest) String() string

+ type TransferToInfeedResponse struct

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

+ func (*TransferToInfeedResponse) ProtoMessage()

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

+ func (x *TransferToInfeedResponse) Reset()

+ func (x *TransferToInfeedResponse) String() string

+ type TransferToServerRequest struct

+ DeviceHandle *DeviceHandle

+ Literal *LiteralProto

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

+ func (*TransferToServerRequest) ProtoMessage()

+ func (x *TransferToServerRequest) GetDeviceHandle() *DeviceHandle

+ func (x *TransferToServerRequest) GetLiteral() *LiteralProto

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

+ func (x *TransferToServerRequest) Reset()

+ func (x *TransferToServerRequest) String() string

+ type TransferToServerResponse struct

+ Data *GlobalDataHandle

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

+ func (*TransferToServerResponse) ProtoMessage()

+ func (x *TransferToServerResponse) GetData() *GlobalDataHandle

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

+ func (x *TransferToServerResponse) Reset()

+ func (x *TransferToServerResponse) String() string

+ type TriangularSolveOptions struct

+ LeftSide bool

+ Lower bool

+ TransposeA TriangularSolveOptions_Transpose

+ UnitDiagonal bool

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

+ func (*TriangularSolveOptions) ProtoMessage()

+ func (x *TriangularSolveOptions) GetLeftSide() bool

+ func (x *TriangularSolveOptions) GetLower() bool

+ func (x *TriangularSolveOptions) GetTransposeA() TriangularSolveOptions_Transpose

+ func (x *TriangularSolveOptions) GetUnitDiagonal() bool

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

+ func (x *TriangularSolveOptions) Reset()

+ func (x *TriangularSolveOptions) String() string

+ type TriangularSolveOptions_Transpose int32

+ const TriangularSolveOptions_ADJOINT

+ const TriangularSolveOptions_NO_TRANSPOSE

+ const TriangularSolveOptions_TRANSPOSE

+ const TriangularSolveOptions_TRANSPOSE_INVALID

+ func (TriangularSolveOptions_Transpose) Descriptor() protoreflect.EnumDescriptor

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

+ func (TriangularSolveOptions_Transpose) Type() protoreflect.EnumType

+ func (x TriangularSolveOptions_Transpose) Enum() *TriangularSolveOptions_Transpose

+ func (x TriangularSolveOptions_Transpose) Number() protoreflect.EnumNumber

+ func (x TriangularSolveOptions_Transpose) String() string

+ type UnpackRequest struct

+ Data *GlobalDataHandle

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

+ func (*UnpackRequest) ProtoMessage()

+ func (x *UnpackRequest) GetData() *GlobalDataHandle

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

+ func (x *UnpackRequest) Reset()

+ func (x *UnpackRequest) String() string

+ type UnpackResponse struct

+ TiedData []*GlobalDataHandle

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

+ func (*UnpackResponse) ProtoMessage()

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

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

+ func (x *UnpackResponse) Reset()

+ func (x *UnpackResponse) String() string

+ type UnregisterRequest struct

+ Data []*GlobalDataHandle

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

+ func (*UnregisterRequest) ProtoMessage()

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

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

+ func (x *UnregisterRequest) Reset()

+ func (x *UnregisterRequest) String() string

+ type UnregisterResponse struct

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

+ func (*UnregisterResponse) ProtoMessage()

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

+ func (x *UnregisterResponse) Reset()

+ func (x *UnregisterResponse) String() string

+ type WhileLoopBackendConfig struct

+ KnownTripCount *WhileLoopBackendConfig_KnownTripCount

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

+ func (*WhileLoopBackendConfig) ProtoMessage()

+ func (x *WhileLoopBackendConfig) GetKnownTripCount() *WhileLoopBackendConfig_KnownTripCount

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

+ func (x *WhileLoopBackendConfig) Reset()

+ func (x *WhileLoopBackendConfig) String() string

+ type WhileLoopBackendConfig_KnownTripCount struct

+ N int64

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

+ func (*WhileLoopBackendConfig_KnownTripCount) ProtoMessage()

+ func (x *WhileLoopBackendConfig_KnownTripCount) GetN() int64

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

+ func (x *WhileLoopBackendConfig_KnownTripCount) Reset()

+ func (x *WhileLoopBackendConfig_KnownTripCount) String() string

+ type Window struct

+ Dimensions []*WindowDimension

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

+ func (*Window) ProtoMessage()

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

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

+ func (x *Window) Reset()

+ func (x *Window) String() string

+ type WindowDimension struct

+ BaseDilation int64

+ PaddingHigh int64

+ PaddingLow int64

+ Size int64

+ Stride int64

+ WindowDilation int64

+ WindowReversal bool

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

+ func (*WindowDimension) ProtoMessage()

+ func (x *WindowDimension) GetBaseDilation() int64

+ func (x *WindowDimension) GetPaddingHigh() int64

+ func (x *WindowDimension) GetPaddingLow() int64

+ func (x *WindowDimension) GetSize() int64

+ func (x *WindowDimension) GetStride() int64

+ func (x *WindowDimension) GetWindowDilation() int64

+ func (x *WindowDimension) GetWindowReversal() bool

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

+ func (x *WindowDimension) Reset()

+ func (x *WindowDimension) String() string

?	: This menu
/	: Search site
f or F	: Jump to
y or Y	: Canonical URL