hlo

package

Go to main page

Versions in this module

v0

v0.6.0

Feb 1, 2025

v0.5.1

Jan 29, 2025

Changes in this version

+ type HloInputs struct

+ Arguments []*xla_data.LiteralProto

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

+ func (*HloInputs) ProtoMessage()

+ func (x *HloInputs) GetArguments() []*xla_data.LiteralProto

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

+ func (x *HloInputs) Reset()

+ func (x *HloInputs) String() string

type HloModuleProto_ProfileType

+ const HloModuleProto_SHARDING

+ type HloUnoptimizedSnapshot struct

+ HloModule *HloModuleProto

+ Partitions []*HloInputs

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

+ func (*HloUnoptimizedSnapshot) ProtoMessage()

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

+ func (x *HloUnoptimizedSnapshot) GetPartitions() []*HloInputs

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

+ func (x *HloUnoptimizedSnapshot) Reset()

+ func (x *HloUnoptimizedSnapshot) String() string

v0.5.0

Dec 19, 2024

v0.4.9

Nov 25, 2024

Changes in this version

type HloInstructionProto

+ ResultAccuracy *xla_data.ResultAccuracy

+ func (x *HloInstructionProto) GetResultAccuracy() *xla_data.ResultAccuracy

v0.4.8

Nov 19, 2024

v0.4.7

Nov 17, 2024

Changes in this version

+ var HloModuleProto_ProfileGenerationStrategy_name = map[int32]string

+ var HloModuleProto_ProfileGenerationStrategy_value = map[string]int32

type HloInstructionProto

+ RaggedDotDimensionNumbers *xla_data.RaggedDotDimensionNumbers

+ func (x *HloInstructionProto) GetRaggedDotDimensionNumbers() *xla_data.RaggedDotDimensionNumbers

+ type HloModuleProto_ProfileGenerationStrategy int32

+ const HloModuleProto_PROFILE_GENERATION_STRATEGY_CFO

+ const HloModuleProto_PROFILE_GENERATION_STRATEGY_FANTA

+ const HloModuleProto_PROFILE_GENERATION_STRATEGY_GA

+ const HloModuleProto_PROFILE_GENERATION_STRATEGY_UNKNOWN

+ func (HloModuleProto_ProfileGenerationStrategy) Descriptor() protoreflect.EnumDescriptor

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

+ func (HloModuleProto_ProfileGenerationStrategy) Type() protoreflect.EnumType

+ func (x HloModuleProto_ProfileGenerationStrategy) Enum() *HloModuleProto_ProfileGenerationStrategy

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

+ func (x HloModuleProto_ProfileGenerationStrategy) String() string

type HloModuleProto_ProfileInfo

+ ProfileGenerationStrategy HloModuleProto_ProfileGenerationStrategy

+ func (x *HloModuleProto_ProfileInfo) GetProfileGenerationStrategy() HloModuleProto_ProfileGenerationStrategy

type HloPassMetadata

+ KvMetrics []*metrics.KeyValueMetric

+ func (x *HloPassMetadata) GetKvMetrics() []*metrics.KeyValueMetric

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

Changes in this version

+ 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 File_xla_service_hlo_proto protoreflect.FileDescriptor

+ var HeapSimulatorTrace_Event_Kind_name = map[int32]string

+ var HeapSimulatorTrace_Event_Kind_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

+ 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 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 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 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 *xla_data.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() *xla_data.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 *xla_data.CholeskyOptions

+ CollectiveDeviceList *xla_data.CollectiveDeviceListProto

+ ComparisonDirection string

+ ComparisonType string

+ ConstrainLayout bool

+ ControlPredecessorIds []int64

+ ConvolutionDimensionNumbers *xla_data.ConvolutionDimensionNumbers

+ CustomCallApiVersion CustomCallApiVersion

+ CustomCallHasSideEffect bool

+ CustomCallSchedule CustomCallSchedule

+ CustomCallTarget string

+ Delta int64

+ Dimensions []int64

+ Distribution xla_data.RandomDistribution

+ DomainEntrySharding *xla_data.OpSharding

+ DomainExitSharding *xla_data.OpSharding

+ DotDimensionNumbers *xla_data.DotDimensionNumbers

+ DotSparsity []*xla_data.SparsityDescriptor

+ DynamicSliceSizes []int64

+ Epsilon float32

+ ExponentBits int32

+ FeatureGroupCount int64

+ FeatureIndex int64

+ FftLength []int64

+ FftType xla_data.FftType

+ FrontendAttributes *xla_data.FrontendAttributes

+ FusionKind string

+ GatherDimensionNumbers *xla_data.GatherDimensionNumbers

+ GatherSliceSizes []int64

+ Id int64

+ IndicesAreSorted bool

+ InfeedConfig []byte

+ IsComposite bool

+ IsCrossProgramPrefetch bool

+ IsHostTransfer bool

+ IsStable bool

+ K int64

+ Largest bool

+ Literal *xla_data.LiteralProto

+ MantissaBits int32

+ Metadata *xla_data.OpMetadata

+ Name string

+ Opcode string

+ OperandIds []int64

+ OperandShapesWithLayout []*xla_data.ShapeProto

+ OptionalCrossProgramPrefetchIndex isHloInstructionProto_OptionalCrossProgramPrefetchIndex

+ OriginalValue *xla_data.OriginalValueProto

+ OutfeedConfig []byte

+ OutfeedShape *xla_data.ShapeProto

+ OutputOperandAliasing []*xla_data.OutputOperandAliasing

+ PaddingConfig *xla_data.PaddingConfig

+ PaddingType xla_data.PaddingType

+ ParameterNumber int64

+ ParameterReplication *xla_data.ParameterReplication

+ PrecisionConfig *xla_data.PrecisionConfig

+ ReplicaGroups []*xla_data.ReplicaGroup

+ RngAlgorithm xla_data.RandomAlgorithm

+ ScatterDimensionNumbers *xla_data.ScatterDimensionNumbers

+ Shape *xla_data.ShapeProto

+ Sharding *xla_data.OpSharding

+ SliceDimensions []*HloInstructionProto_SliceDimensions

+ SourceTargetPairs []*xla_data.SourceTarget

+ StatisticsViz *xla_data.StatisticsViz

+ TriangularSolveOptions *xla_data.TriangularSolveOptions

+ TupleIndex int64

+ UniqueIndices bool

+ UseGlobalDeviceIds bool

+ Window *xla_data.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() *xla_data.CholeskyOptions

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

+ func (x *HloInstructionProto) GetComparisonDirection() string

+ func (x *HloInstructionProto) GetComparisonType() string

+ func (x *HloInstructionProto) GetConstrainLayout() bool

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

+ func (x *HloInstructionProto) GetConvolutionDimensionNumbers() *xla_data.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() xla_data.RandomDistribution

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

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

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

+ func (x *HloInstructionProto) GetDotSparsity() []*xla_data.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() xla_data.FftType

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

+ func (x *HloInstructionProto) GetFusionKind() string

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

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

+ func (x *HloInstructionProto) GetId() int64

+ func (x *HloInstructionProto) GetIndicesAreSorted() bool

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

+ func (x *HloInstructionProto) GetIsComposite() bool

+ 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() *xla_data.LiteralProto

+ func (x *HloInstructionProto) GetMantissaBits() int32

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

+ func (x *HloInstructionProto) GetName() string

+ func (x *HloInstructionProto) GetOpcode() string

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

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

+ func (x *HloInstructionProto) GetOriginalValue() *xla_data.OriginalValueProto

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

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

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

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

+ func (x *HloInstructionProto) GetPaddingType() xla_data.PaddingType

+ func (x *HloInstructionProto) GetParameterNumber() int64

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

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

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

+ func (x *HloInstructionProto) GetRngAlgorithm() xla_data.RandomAlgorithm

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

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

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

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

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

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

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

+ func (x *HloInstructionProto) GetTupleIndex() int64

+ func (x *HloInstructionProto) GetUniqueIndices() bool

+ func (x *HloInstructionProto) GetUseGlobalDeviceIds() bool

+ func (x *HloInstructionProto) GetWindow() *xla_data.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 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 *xla_data.DeviceAssignmentProto

+ EntryComputationId int64

+ EntryComputationName string

+ FrontendAttributes *xla_data.FrontendAttributes

+ HostProgramShape *xla_data.ProgramShapeProto

+ Id int64

+ InputOutputAlias *HloInputOutputAliasProto

+ IsDynamic bool

+ Name string

+ ProfileInfo []*HloModuleProto_ProfileInfo

+ Schedule *HloScheduleProto

+ SpmdOutputSharding *xla_data.OpSharding

+ SpmdParametersShardings []*xla_data.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() *xla_data.DeviceAssignmentProto

+ func (x *HloModuleProto) GetEntryComputationId() int64

+ func (x *HloModuleProto) GetEntryComputationName() string

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

+ func (x *HloModuleProto) GetHostProgramShape() *xla_data.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() *xla_data.OpSharding

+ func (x *HloModuleProto) GetSpmdParametersShardings() []*xla_data.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 HloModuleProto_ProfileInfo struct

+ CompilationEvent xla_data.CompilationEvent

+ Fingerprint string

+ ProfileSource xla_data.ProfileSource

+ ProfileType HloModuleProto_ProfileType

+ RelativeSpeedup float64

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

+ func (*HloModuleProto_ProfileInfo) ProtoMessage()

+ func (x *HloModuleProto_ProfileInfo) GetCompilationEvent() xla_data.CompilationEvent

+ func (x *HloModuleProto_ProfileInfo) GetFingerprint() string

+ func (x *HloModuleProto_ProfileInfo) GetProfileSource() xla_data.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_FLAGNET

+ 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 []*xla_data.LiteralProto

+ ExecutionPlatform string

+ Hlo *HloProto

+ Result *xla_data.LiteralProto

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

+ func (*HloSnapshot) ProtoMessage()

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

+ func (x *HloSnapshot) GetExecutionPlatform() string

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

+ func (x *HloSnapshot) GetResult() *xla_data.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 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 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

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