hlo

package
v0.4.5 Latest Latest
Warning

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

 Go to latest Published: Nov 6, 2024 License: Apache-2.0 Imports: 6 Imported by: 0

Documentation

Index

Constants

This section is empty.

Variables

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

Enum value maps for CustomCallSchedule.

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

Enum value maps for CustomCallApiVersion.

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

Enum value maps for Kind.

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

Enum value maps for HloModuleProto_ProfileType.

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

Enum value maps for HeapSimulatorTrace_Event_Kind.

View Source 
var File_xla_service_hlo_proto protoreflect.FileDescriptor

Functions

This section is empty.

Types

type BufferAllocationProto 

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

Serialization of BufferAllocation.

func (*BufferAllocationProto) Descriptor deprecated 

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

Deprecated: Use BufferAllocationProto.ProtoReflect.Descriptor instead.

func (*BufferAllocationProto) GetAssigned 

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

func (*BufferAllocationProto) GetColor 

func (x *BufferAllocationProto) GetColor() int64

func (*BufferAllocationProto) GetIndex 

func (x *BufferAllocationProto) GetIndex() int64

func (*BufferAllocationProto) GetIsConstant 

func (x *BufferAllocationProto) GetIsConstant() bool

func (*BufferAllocationProto) GetIsEntryComputationParameter 

func (x *BufferAllocationProto) GetIsEntryComputationParameter() bool

func (*BufferAllocationProto) GetIsThreadLocal 

func (x *BufferAllocationProto) GetIsThreadLocal() bool

func (*BufferAllocationProto) GetIsTuple 

func (x *BufferAllocationProto) GetIsTuple() bool

func (*BufferAllocationProto) GetMaybeLiveOut 

func (x *BufferAllocationProto) GetMaybeLiveOut() bool

func (*BufferAllocationProto) GetParameterNumber 

func (x *BufferAllocationProto) GetParameterNumber() int64

func (*BufferAllocationProto) GetParameterShapeIndex 

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

func (*BufferAllocationProto) GetSize 

func (x *BufferAllocationProto) GetSize() int64

func (*BufferAllocationProto) ProtoMessage 

func (*BufferAllocationProto) ProtoMessage()

func (*BufferAllocationProto) ProtoReflect 

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

func (*BufferAllocationProto) Reset 

func (x *BufferAllocationProto) Reset()

func (*BufferAllocationProto) String 

func (x *BufferAllocationProto) String() string

type BufferAllocationProto_Assigned 

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

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

func (*BufferAllocationProto_Assigned) Descriptor deprecated 

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

Deprecated: Use BufferAllocationProto_Assigned.ProtoReflect.Descriptor instead.

func (*BufferAllocationProto_Assigned) GetLogicalBufferId 

func (x *BufferAllocationProto_Assigned) GetLogicalBufferId() int64

func (*BufferAllocationProto_Assigned) GetOffset 

func (x *BufferAllocationProto_Assigned) GetOffset() int64

func (*BufferAllocationProto_Assigned) GetSize 

func (x *BufferAllocationProto_Assigned) GetSize() int64

func (*BufferAllocationProto_Assigned) ProtoMessage 

func (*BufferAllocationProto_Assigned) ProtoMessage()

func (*BufferAllocationProto_Assigned) ProtoReflect 

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

func (*BufferAllocationProto_Assigned) Reset 

func (x *BufferAllocationProto_Assigned) Reset()

func (*BufferAllocationProto_Assigned) String 

func (x *BufferAllocationProto_Assigned) String() string

type BufferAssignmentProto 

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

Serialization of BufferAssignment.

func (*BufferAssignmentProto) Descriptor deprecated 

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

Deprecated: Use BufferAssignmentProto.ProtoReflect.Descriptor instead.

func (*BufferAssignmentProto) GetBufferAliases 

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

func (*BufferAssignmentProto) GetBufferAllocations 

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

func (*BufferAssignmentProto) GetHeapSimulatorTraces 

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

func (*BufferAssignmentProto) GetLogicalBuffers 

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

func (*BufferAssignmentProto) ProtoMessage 

func (*BufferAssignmentProto) ProtoMessage()

func (*BufferAssignmentProto) ProtoReflect 

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

func (*BufferAssignmentProto) Reset 

func (x *BufferAssignmentProto) Reset()

func (*BufferAssignmentProto) String 

func (x *BufferAssignmentProto) String() string

type BufferAssignmentProto_BufferAlias 

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

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

func (*BufferAssignmentProto_BufferAlias) Descriptor deprecated 

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

Deprecated: Use BufferAssignmentProto_BufferAlias.ProtoReflect.Descriptor instead.

func (*BufferAssignmentProto_BufferAlias) GetLocation 

func (x *BufferAssignmentProto_BufferAlias) GetLocation() *LogicalBufferProto_Location

func (*BufferAssignmentProto_BufferAlias) GetSourceBufferId 

func (x *BufferAssignmentProto_BufferAlias) GetSourceBufferId() int64

func (*BufferAssignmentProto_BufferAlias) ProtoMessage 

func (*BufferAssignmentProto_BufferAlias) ProtoMessage()

func (*BufferAssignmentProto_BufferAlias) ProtoReflect 

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

func (*BufferAssignmentProto_BufferAlias) Reset 

func (x *BufferAssignmentProto_BufferAlias) Reset()

func (*BufferAssignmentProto_BufferAlias) String 

func (x *BufferAssignmentProto_BufferAlias) String() string

type CrossProgramPrefetch 

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

func (*CrossProgramPrefetch) Descriptor deprecated 

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

Deprecated: Use CrossProgramPrefetch.ProtoReflect.Descriptor instead.

func (*CrossProgramPrefetch) GetIndex 

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

func (*CrossProgramPrefetch) GetOffset 

func (x *CrossProgramPrefetch) GetOffset() int64

func (*CrossProgramPrefetch) GetParameter 

func (x *CrossProgramPrefetch) GetParameter() int64

func (*CrossProgramPrefetch) ProtoMessage 

func (*CrossProgramPrefetch) ProtoMessage()

func (*CrossProgramPrefetch) ProtoReflect 

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

func (*CrossProgramPrefetch) Reset 

func (x *CrossProgramPrefetch) Reset()

func (*CrossProgramPrefetch) String 

func (x *CrossProgramPrefetch) String() string

type CustomCallApiVersion 

type CustomCallApiVersion int32

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

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

func (CustomCallApiVersion) Descriptor 

func (CustomCallApiVersion) Descriptor() protoreflect.EnumDescriptor

func (CustomCallApiVersion) Enum 

func (x CustomCallApiVersion) Enum() *CustomCallApiVersion

func (CustomCallApiVersion) EnumDescriptor deprecated 

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

Deprecated: Use CustomCallApiVersion.Descriptor instead.

func (CustomCallApiVersion) Number 

func (x CustomCallApiVersion) Number() protoreflect.EnumNumber

func (CustomCallApiVersion) String 

func (x CustomCallApiVersion) String() string

func (CustomCallApiVersion) Type 

func (CustomCallApiVersion) Type() protoreflect.EnumType

type CustomCallSchedule 

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

func (CustomCallSchedule) Descriptor 

func (CustomCallSchedule) Descriptor() protoreflect.EnumDescriptor

func (CustomCallSchedule) Enum 

func (x CustomCallSchedule) Enum() *CustomCallSchedule

func (CustomCallSchedule) EnumDescriptor deprecated 

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

Deprecated: Use CustomCallSchedule.Descriptor instead.

func (CustomCallSchedule) Number 

func (x CustomCallSchedule) Number() protoreflect.EnumNumber

func (CustomCallSchedule) String 

func (x CustomCallSchedule) String() string

func (CustomCallSchedule) Type 

func (CustomCallSchedule) Type() protoreflect.EnumType

type HeapSimulatorTrace 

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

A trace of a HeapSimulator run.

func (*HeapSimulatorTrace) Descriptor deprecated 

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

Deprecated: Use HeapSimulatorTrace.ProtoReflect.Descriptor instead.

func (*HeapSimulatorTrace) GetBufferAllocationIndex 

func (x *HeapSimulatorTrace) GetBufferAllocationIndex() int64

func (*HeapSimulatorTrace) GetEvents 

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

func (*HeapSimulatorTrace) GetWholeModuleSimulation 

func (x *HeapSimulatorTrace) GetWholeModuleSimulation() bool

func (*HeapSimulatorTrace) ProtoMessage 

func (*HeapSimulatorTrace) ProtoMessage()

func (*HeapSimulatorTrace) ProtoReflect 

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

func (*HeapSimulatorTrace) Reset 

func (x *HeapSimulatorTrace) Reset()

func (*HeapSimulatorTrace) String 

func (x *HeapSimulatorTrace) String() string

type HeapSimulatorTrace_Event 

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

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

func (*HeapSimulatorTrace_Event) Descriptor deprecated 

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

Deprecated: Use HeapSimulatorTrace_Event.ProtoReflect.Descriptor instead.

func (*HeapSimulatorTrace_Event) GetBufferId 

func (x *HeapSimulatorTrace_Event) GetBufferId() int64

func (*HeapSimulatorTrace_Event) GetComputationName 

func (x *HeapSimulatorTrace_Event) GetComputationName() string

func (*HeapSimulatorTrace_Event) GetInstructionName 

func (x *HeapSimulatorTrace_Event) GetInstructionName() string

func (*HeapSimulatorTrace_Event) GetKind 

func (x *HeapSimulatorTrace_Event) GetKind() HeapSimulatorTrace_Event_Kind

func (*HeapSimulatorTrace_Event) GetShareWithCanonicalId 

func (x *HeapSimulatorTrace_Event) GetShareWithCanonicalId() int64

func (*HeapSimulatorTrace_Event) ProtoMessage 

func (*HeapSimulatorTrace_Event) ProtoMessage()

func (*HeapSimulatorTrace_Event) ProtoReflect 

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

func (*HeapSimulatorTrace_Event) Reset 

func (x *HeapSimulatorTrace_Event) Reset()

func (*HeapSimulatorTrace_Event) String 

func (x *HeapSimulatorTrace_Event) String() string

type HeapSimulatorTrace_Event_Kind 

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

func (HeapSimulatorTrace_Event_Kind) Descriptor 

func (HeapSimulatorTrace_Event_Kind) Descriptor() protoreflect.EnumDescriptor

func (HeapSimulatorTrace_Event_Kind) Enum 

func (x HeapSimulatorTrace_Event_Kind) Enum() *HeapSimulatorTrace_Event_Kind

func (HeapSimulatorTrace_Event_Kind) EnumDescriptor deprecated 

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

Deprecated: Use HeapSimulatorTrace_Event_Kind.Descriptor instead.

func (HeapSimulatorTrace_Event_Kind) Number 

func (x HeapSimulatorTrace_Event_Kind) Number() protoreflect.EnumNumber

func (HeapSimulatorTrace_Event_Kind) String 

func (x HeapSimulatorTrace_Event_Kind) String() string

func (HeapSimulatorTrace_Event_Kind) Type 

func (HeapSimulatorTrace_Event_Kind) Type() protoreflect.EnumType

type HloBufferDonorProto 

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

func (*HloBufferDonorProto) Descriptor deprecated 

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

Deprecated: Use HloBufferDonorProto.ProtoReflect.Descriptor instead.

func (*HloBufferDonorProto) GetEntries 

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

func (*HloBufferDonorProto) ProtoMessage 

func (*HloBufferDonorProto) ProtoMessage()

func (*HloBufferDonorProto) ProtoReflect 

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

func (*HloBufferDonorProto) Reset 

func (x *HloBufferDonorProto) Reset()

func (*HloBufferDonorProto) String 

func (x *HloBufferDonorProto) String() string

type HloBufferDonorProto_BufferDonorEntryProto 

type HloBufferDonorProto_BufferDonorEntryProto struct {

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

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

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

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

func (*HloBufferDonorProto_BufferDonorEntryProto) Descriptor deprecated 

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

Deprecated: Use HloBufferDonorProto_BufferDonorEntryProto.ProtoReflect.Descriptor instead.

func (*HloBufferDonorProto_BufferDonorEntryProto) GetParameterNumber 

func (x *HloBufferDonorProto_BufferDonorEntryProto) GetParameterNumber() int64

func (*HloBufferDonorProto_BufferDonorEntryProto) GetParameterShapeIndex 

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

func (*HloBufferDonorProto_BufferDonorEntryProto) ProtoMessage 

func (*HloBufferDonorProto_BufferDonorEntryProto) ProtoMessage()

func (*HloBufferDonorProto_BufferDonorEntryProto) ProtoReflect 

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

func (*HloBufferDonorProto_BufferDonorEntryProto) Reset 

func (x *HloBufferDonorProto_BufferDonorEntryProto) Reset()

func (*HloBufferDonorProto_BufferDonorEntryProto) String 

func (x *HloBufferDonorProto_BufferDonorEntryProto) String() string

type HloComputationProto 

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

Serialization of HloComputation.

func (*HloComputationProto) Descriptor deprecated 

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

Deprecated: Use HloComputationProto.ProtoReflect.Descriptor instead.

func (*HloComputationProto) GetExecutionThread 

func (x *HloComputationProto) GetExecutionThread() string

func (*HloComputationProto) GetId 

func (x *HloComputationProto) GetId() int64

func (*HloComputationProto) GetInstructions 

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

func (*HloComputationProto) GetIsFusionComputation 

func (x *HloComputationProto) GetIsFusionComputation() bool

func (*HloComputationProto) GetName 

func (x *HloComputationProto) GetName() string

func (*HloComputationProto) GetProgramShape 

func (x *HloComputationProto) GetProgramShape() *xla_data.ProgramShapeProto

func (*HloComputationProto) GetRootId 

func (x *HloComputationProto) GetRootId() int64

func (*HloComputationProto) ProtoMessage 

func (*HloComputationProto) ProtoMessage()

func (*HloComputationProto) ProtoReflect 

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

func (*HloComputationProto) Reset 

func (x *HloComputationProto) Reset()

func (*HloComputationProto) String 

func (x *HloComputationProto) String() string

type HloInputOutputAliasProto 

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

func (*HloInputOutputAliasProto) Descriptor deprecated 

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

Deprecated: Use HloInputOutputAliasProto.ProtoReflect.Descriptor instead.

func (*HloInputOutputAliasProto) GetEntries 

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

func (*HloInputOutputAliasProto) ProtoMessage 

func (*HloInputOutputAliasProto) ProtoMessage()

func (*HloInputOutputAliasProto) ProtoReflect 

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

func (*HloInputOutputAliasProto) Reset 

func (x *HloInputOutputAliasProto) Reset()

func (*HloInputOutputAliasProto) String 

func (x *HloInputOutputAliasProto) String() string

type HloInputOutputAliasProto_AliasEntryProto 

type HloInputOutputAliasProto_AliasEntryProto struct {

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

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

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

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

func (*HloInputOutputAliasProto_AliasEntryProto) Descriptor deprecated 

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

Deprecated: Use HloInputOutputAliasProto_AliasEntryProto.ProtoReflect.Descriptor instead.

func (*HloInputOutputAliasProto_AliasEntryProto) GetKind 

func (x *HloInputOutputAliasProto_AliasEntryProto) GetKind() Kind

func (*HloInputOutputAliasProto_AliasEntryProto) GetOutputShapeIndex 

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

func (*HloInputOutputAliasProto_AliasEntryProto) GetParameterNumber 

func (x *HloInputOutputAliasProto_AliasEntryProto) GetParameterNumber() int64

func (*HloInputOutputAliasProto_AliasEntryProto) GetParameterShapeIndex 

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

func (*HloInputOutputAliasProto_AliasEntryProto) ProtoMessage 

func (*HloInputOutputAliasProto_AliasEntryProto) ProtoMessage()

func (*HloInputOutputAliasProto_AliasEntryProto) ProtoReflect 

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

func (*HloInputOutputAliasProto_AliasEntryProto) Reset 

func (x *HloInputOutputAliasProto_AliasEntryProto) Reset()

func (*HloInputOutputAliasProto_AliasEntryProto) String 

func (x *HloInputOutputAliasProto_AliasEntryProto) String() string

type HloInstructionProto 

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

Serialization of HloInstruction. Next ID: 90

func (*HloInstructionProto) Descriptor deprecated 

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

Deprecated: Use HloInstructionProto.ProtoReflect.Descriptor instead.

func (*HloInstructionProto) GetAllReduceId deprecated 

func (x *HloInstructionProto) GetAllReduceId() int64

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

func (*HloInstructionProto) GetAsyncExecutionThread 

func (x *HloInstructionProto) GetAsyncExecutionThread() string

func (*HloInstructionProto) GetBackendConfig 

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

func (*HloInstructionProto) GetBatchGroupCount 

func (x *HloInstructionProto) GetBatchGroupCount() int64

func (*HloInstructionProto) GetCalledComputationIds 

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

func (*HloInstructionProto) GetChannelId 

func (x *HloInstructionProto) GetChannelId() int64

func (*HloInstructionProto) GetCholeskyOptions 

func (x *HloInstructionProto) GetCholeskyOptions() *xla_data.CholeskyOptions

func (*HloInstructionProto) GetCollectiveDeviceList 

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

func (*HloInstructionProto) GetComparisonDirection 

func (x *HloInstructionProto) GetComparisonDirection() string

func (*HloInstructionProto) GetComparisonType 

func (x *HloInstructionProto) GetComparisonType() string

func (*HloInstructionProto) GetConstrainLayout 

func (x *HloInstructionProto) GetConstrainLayout() bool

func (*HloInstructionProto) GetControlPredecessorIds 

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

func (*HloInstructionProto) GetConvolutionDimensionNumbers 

func (x *HloInstructionProto) GetConvolutionDimensionNumbers() *xla_data.ConvolutionDimensionNumbers

func (*HloInstructionProto) GetCrossProgramPrefetchIndex 

func (x *HloInstructionProto) GetCrossProgramPrefetchIndex() int32

func (*HloInstructionProto) GetCustomCallApiVersion 

func (x *HloInstructionProto) GetCustomCallApiVersion() CustomCallApiVersion

func (*HloInstructionProto) GetCustomCallHasSideEffect 

func (x *HloInstructionProto) GetCustomCallHasSideEffect() bool

func (*HloInstructionProto) GetCustomCallSchedule 

func (x *HloInstructionProto) GetCustomCallSchedule() CustomCallSchedule

func (*HloInstructionProto) GetCustomCallTarget 

func (x *HloInstructionProto) GetCustomCallTarget() string

func (*HloInstructionProto) GetDelta 

func (x *HloInstructionProto) GetDelta() int64

func (*HloInstructionProto) GetDimensions 

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

func (*HloInstructionProto) GetDistribution 

func (x *HloInstructionProto) GetDistribution() xla_data.RandomDistribution

func (*HloInstructionProto) GetDomainEntrySharding 

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

func (*HloInstructionProto) GetDomainExitSharding 

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

func (*HloInstructionProto) GetDotDimensionNumbers 

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

func (*HloInstructionProto) GetDotSparsity 

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

func (*HloInstructionProto) GetDynamicSliceSizes 

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

func (*HloInstructionProto) GetEpsilon 

func (x *HloInstructionProto) GetEpsilon() float32

func (*HloInstructionProto) GetExponentBits 

func (x *HloInstructionProto) GetExponentBits() int32

func (*HloInstructionProto) GetFeatureGroupCount 

func (x *HloInstructionProto) GetFeatureGroupCount() int64

func (*HloInstructionProto) GetFeatureIndex 

func (x *HloInstructionProto) GetFeatureIndex() int64

func (*HloInstructionProto) GetFftLength 

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

func (*HloInstructionProto) GetFftType 

func (x *HloInstructionProto) GetFftType() xla_data.FftType

func (*HloInstructionProto) GetFrontendAttributes 

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

func (*HloInstructionProto) GetFusionKind 

func (x *HloInstructionProto) GetFusionKind() string

func (*HloInstructionProto) GetGatherDimensionNumbers 

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

func (*HloInstructionProto) GetGatherSliceSizes 

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

func (*HloInstructionProto) GetId 

func (x *HloInstructionProto) GetId() int64

func (*HloInstructionProto) GetIndicesAreSorted 

func (x *HloInstructionProto) GetIndicesAreSorted() bool

func (*HloInstructionProto) GetInfeedConfig 

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

func (*HloInstructionProto) GetIsComposite 

func (x *HloInstructionProto) GetIsComposite() bool

func (*HloInstructionProto) GetIsCrossProgramPrefetch deprecated 

func (x *HloInstructionProto) GetIsCrossProgramPrefetch() bool

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

func (*HloInstructionProto) GetIsHostTransfer 

func (x *HloInstructionProto) GetIsHostTransfer() bool

func (*HloInstructionProto) GetIsStable 

func (x *HloInstructionProto) GetIsStable() bool

func (*HloInstructionProto) GetK 

func (x *HloInstructionProto) GetK() int64

func (*HloInstructionProto) GetLargest 

func (x *HloInstructionProto) GetLargest() bool

func (*HloInstructionProto) GetLiteral 

func (x *HloInstructionProto) GetLiteral() *xla_data.LiteralProto

func (*HloInstructionProto) GetMantissaBits 

func (x *HloInstructionProto) GetMantissaBits() int32

func (*HloInstructionProto) GetMetadata 

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

func (*HloInstructionProto) GetName 

func (x *HloInstructionProto) GetName() string

func (*HloInstructionProto) GetOpcode 

func (x *HloInstructionProto) GetOpcode() string

func (*HloInstructionProto) GetOperandIds 

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

func (*HloInstructionProto) GetOperandShapesWithLayout 

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

func (*HloInstructionProto) GetOptionalCrossProgramPrefetchIndex 

func (m *HloInstructionProto) GetOptionalCrossProgramPrefetchIndex() isHloInstructionProto_OptionalCrossProgramPrefetchIndex

func (*HloInstructionProto) GetOriginalValue 

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

func (*HloInstructionProto) GetOutfeedConfig 

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

func (*HloInstructionProto) GetOutfeedShape 

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

func (*HloInstructionProto) GetOutputOperandAliasing 

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

func (*HloInstructionProto) GetPaddingConfig 

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

func (*HloInstructionProto) GetPaddingType 

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

func (*HloInstructionProto) GetParameterNumber 

func (x *HloInstructionProto) GetParameterNumber() int64

func (*HloInstructionProto) GetParameterReplication 

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

func (*HloInstructionProto) GetPrecisionConfig 

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

func (*HloInstructionProto) GetReplicaGroups deprecated 

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

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

func (*HloInstructionProto) GetRngAlgorithm 

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

func (*HloInstructionProto) GetScatterDimensionNumbers 

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

func (*HloInstructionProto) GetShape 

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

func (*HloInstructionProto) GetSharding 

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

func (*HloInstructionProto) GetSliceDimensions 

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

func (*HloInstructionProto) GetSourceTargetPairs 

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

func (*HloInstructionProto) GetStatisticsViz 

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

func (*HloInstructionProto) GetTriangularSolveOptions 

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

func (*HloInstructionProto) GetTupleIndex 

func (x *HloInstructionProto) GetTupleIndex() int64

func (*HloInstructionProto) GetUniqueIndices 

func (x *HloInstructionProto) GetUniqueIndices() bool

func (*HloInstructionProto) GetUseGlobalDeviceIds 

func (x *HloInstructionProto) GetUseGlobalDeviceIds() bool

func (*HloInstructionProto) GetWindow 

func (x *HloInstructionProto) GetWindow() *xla_data.Window

func (*HloInstructionProto) ProtoMessage 

func (*HloInstructionProto) ProtoMessage()

func (*HloInstructionProto) ProtoReflect 

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

func (*HloInstructionProto) Reset 

func (x *HloInstructionProto) Reset()

func (*HloInstructionProto) String 

func (x *HloInstructionProto) String() string

type HloInstructionProto_CrossProgramPrefetchIndex 

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

type HloInstructionProto_SliceDimensions 

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

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

func (*HloInstructionProto_SliceDimensions) Descriptor deprecated 

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

Deprecated: Use HloInstructionProto_SliceDimensions.ProtoReflect.Descriptor instead.

func (*HloInstructionProto_SliceDimensions) GetLimit 

func (x *HloInstructionProto_SliceDimensions) GetLimit() int64

func (*HloInstructionProto_SliceDimensions) GetStart 

func (x *HloInstructionProto_SliceDimensions) GetStart() int64

func (*HloInstructionProto_SliceDimensions) GetStride 

func (x *HloInstructionProto_SliceDimensions) GetStride() int64

func (*HloInstructionProto_SliceDimensions) ProtoMessage 

func (*HloInstructionProto_SliceDimensions) ProtoMessage()

func (*HloInstructionProto_SliceDimensions) ProtoReflect 

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

func (*HloInstructionProto_SliceDimensions) Reset 

func (x *HloInstructionProto_SliceDimensions) Reset()

func (*HloInstructionProto_SliceDimensions) String 

func (x *HloInstructionProto_SliceDimensions) String() string

type HloModuleGroupProto 

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

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

func (*HloModuleGroupProto) Descriptor deprecated 

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

Deprecated: Use HloModuleGroupProto.ProtoReflect.Descriptor instead.

func (*HloModuleGroupProto) GetHloModules 

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

func (*HloModuleGroupProto) GetName 

func (x *HloModuleGroupProto) GetName() string

func (*HloModuleGroupProto) ProtoMessage 

func (*HloModuleGroupProto) ProtoMessage()

func (*HloModuleGroupProto) ProtoReflect 

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

func (*HloModuleGroupProto) Reset 

func (x *HloModuleGroupProto) Reset()

func (*HloModuleGroupProto) String 

func (x *HloModuleGroupProto) String() string

type HloModuleMetadataProto 

type HloModuleMetadataProto struct {

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

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

func (*HloModuleMetadataProto) Descriptor deprecated 

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

Deprecated: Use HloModuleMetadataProto.ProtoReflect.Descriptor instead.

func (*HloModuleMetadataProto) GetCanonicalModuleId 

func (x *HloModuleMetadataProto) GetCanonicalModuleId() int64

func (*HloModuleMetadataProto) GetModuleGroupName 

func (x *HloModuleMetadataProto) GetModuleGroupName() string

func (*HloModuleMetadataProto) GetOriginalModuleId 

func (x *HloModuleMetadataProto) GetOriginalModuleId() int64

func (*HloModuleMetadataProto) GetPartitionedModuleIds 

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

func (*HloModuleMetadataProto) GetPassMetadata 

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

func (*HloModuleMetadataProto) ProtoMessage 

func (*HloModuleMetadataProto) ProtoMessage()

func (*HloModuleMetadataProto) ProtoReflect 

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

func (*HloModuleMetadataProto) Reset 

func (x *HloModuleMetadataProto) Reset()

func (*HloModuleMetadataProto) String 

func (x *HloModuleMetadataProto) String() string

type HloModuleProto 

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

Serialization of HloModule.

func (*HloModuleProto) Descriptor deprecated 

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

Deprecated: Use HloModuleProto.ProtoReflect.Descriptor instead.

func (*HloModuleProto) GetBufferDonor 

func (x *HloModuleProto) GetBufferDonor() *HloBufferDonorProto

func (*HloModuleProto) GetComputations 

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

func (*HloModuleProto) GetCrossProgramPrefetches 

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

func (*HloModuleProto) GetDeviceAssignment 

func (x *HloModuleProto) GetDeviceAssignment() *xla_data.DeviceAssignmentProto

func (*HloModuleProto) GetEntryComputationId 

func (x *HloModuleProto) GetEntryComputationId() int64

func (*HloModuleProto) GetEntryComputationName 

func (x *HloModuleProto) GetEntryComputationName() string

func (*HloModuleProto) GetFrontendAttributes 

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

func (*HloModuleProto) GetHostProgramShape 

func (x *HloModuleProto) GetHostProgramShape() *xla_data.ProgramShapeProto

func (*HloModuleProto) GetId 

func (x *HloModuleProto) GetId() int64

func (*HloModuleProto) GetInputOutputAlias 

func (x *HloModuleProto) GetInputOutputAlias() *HloInputOutputAliasProto

func (*HloModuleProto) GetIsDynamic 

func (x *HloModuleProto) GetIsDynamic() bool

func (*HloModuleProto) GetName 

func (x *HloModuleProto) GetName() string

func (*HloModuleProto) GetProfileInfo 

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

func (*HloModuleProto) GetSchedule 

func (x *HloModuleProto) GetSchedule() *HloScheduleProto

func (*HloModuleProto) GetSpmdOutputSharding 

func (x *HloModuleProto) GetSpmdOutputSharding() *xla_data.OpSharding

func (*HloModuleProto) GetSpmdParametersShardings 

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

func (*HloModuleProto) GetStackFrameIndex 

func (x *HloModuleProto) GetStackFrameIndex() *StackFrameIndexProto

func (*HloModuleProto) GetUseAutoSpmdPartitioning 

func (x *HloModuleProto) GetUseAutoSpmdPartitioning() bool

func (*HloModuleProto) ProtoMessage 

func (*HloModuleProto) ProtoMessage()

func (*HloModuleProto) ProtoReflect 

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

func (*HloModuleProto) Reset 

func (x *HloModuleProto) Reset()

func (*HloModuleProto) String 

func (x *HloModuleProto) String() string

type HloModuleProto_ProfileInfo 

type HloModuleProto_ProfileInfo struct {

	// The optimization profiles that this module contains.
	ProfileType HloModuleProto_ProfileType `` /* 131-byte string literal not displayed */
	// Speedup of tuned config compared to default config.
	RelativeSpeedup float64 `protobuf:"fixed64,2,opt,name=relative_speedup,json=relativeSpeedup,proto3" json:"relative_speedup,omitempty"`
	// The source of the optimization profile that this module contains.
	ProfileSource xla_data.ProfileSource `protobuf:"varint,3,opt,name=profile_source,json=profileSource,proto3,enum=xla.ProfileSource" json:"profile_source,omitempty"`
	// The compilation event that triggered the use of the profile.
	CompilationEvent xla_data.CompilationEvent `` /* 136-byte string literal not displayed */
	// The fingerprint of the unoptimized module this profile was applied to.
	Fingerprint string `protobuf:"bytes,5,opt,name=fingerprint,proto3" json:"fingerprint,omitempty"`
	// contains filtered or unexported fields
}

Information about the optimization profile that this module contains.

func (*HloModuleProto_ProfileInfo) Descriptor deprecated 

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

Deprecated: Use HloModuleProto_ProfileInfo.ProtoReflect.Descriptor instead.

func (*HloModuleProto_ProfileInfo) GetCompilationEvent 

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

func (*HloModuleProto_ProfileInfo) GetFingerprint 

func (x *HloModuleProto_ProfileInfo) GetFingerprint() string

func (*HloModuleProto_ProfileInfo) GetProfileSource 

func (x *HloModuleProto_ProfileInfo) GetProfileSource() xla_data.ProfileSource

func (*HloModuleProto_ProfileInfo) GetProfileType 

func (x *HloModuleProto_ProfileInfo) GetProfileType() HloModuleProto_ProfileType

func (*HloModuleProto_ProfileInfo) GetRelativeSpeedup 

func (x *HloModuleProto_ProfileInfo) GetRelativeSpeedup() float64

func (*HloModuleProto_ProfileInfo) ProtoMessage 

func (*HloModuleProto_ProfileInfo) ProtoMessage()

func (*HloModuleProto_ProfileInfo) ProtoReflect 

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

func (*HloModuleProto_ProfileInfo) Reset 

func (x *HloModuleProto_ProfileInfo) Reset()

func (*HloModuleProto_ProfileInfo) String 

func (x *HloModuleProto_ProfileInfo) String() string

type HloModuleProto_ProfileType 

type HloModuleProto_ProfileType int32

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

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

func (HloModuleProto_ProfileType) Descriptor 

func (HloModuleProto_ProfileType) Descriptor() protoreflect.EnumDescriptor

func (HloModuleProto_ProfileType) Enum 

func (x HloModuleProto_ProfileType) Enum() *HloModuleProto_ProfileType

func (HloModuleProto_ProfileType) EnumDescriptor deprecated 

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

Deprecated: Use HloModuleProto_ProfileType.Descriptor instead.

func (HloModuleProto_ProfileType) Number 

func (x HloModuleProto_ProfileType) Number() protoreflect.EnumNumber

func (HloModuleProto_ProfileType) String 

func (x HloModuleProto_ProfileType) String() string

func (HloModuleProto_ProfileType) Type 

func (HloModuleProto_ProfileType) Type() protoreflect.EnumType

type HloPassMetadata 

type HloPassMetadata struct {

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

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

func (*HloPassMetadata) Descriptor deprecated 

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

Deprecated: Use HloPassMetadata.ProtoReflect.Descriptor instead.

func (*HloPassMetadata) GetCustomMetadata 

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

func (*HloPassMetadata) GetDumpFilenames 

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

func (*HloPassMetadata) GetEndTimestampUsec 

func (x *HloPassMetadata) GetEndTimestampUsec() int64

func (*HloPassMetadata) GetModuleChanged 

func (x *HloPassMetadata) GetModuleChanged() bool

func (*HloPassMetadata) GetModuleGroupModuleIds 

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

func (*HloPassMetadata) GetModuleId 

func (x *HloPassMetadata) GetModuleId() int64

func (*HloPassMetadata) GetPassId 

func (x *HloPassMetadata) GetPassId() int64

func (*HloPassMetadata) GetPassName 

func (x *HloPassMetadata) GetPassName() string

func (*HloPassMetadata) GetPipelineName 

func (x *HloPassMetadata) GetPipelineName() string

func (*HloPassMetadata) GetStartTimestampUsec 

func (x *HloPassMetadata) GetStartTimestampUsec() int64

func (*HloPassMetadata) ProtoMessage 

func (*HloPassMetadata) ProtoMessage()

func (*HloPassMetadata) ProtoReflect 

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

func (*HloPassMetadata) Reset 

func (x *HloPassMetadata) Reset()

func (*HloPassMetadata) String 

func (x *HloPassMetadata) String() string

type HloProto 

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

Grouping message that contains all of the information above.

func (*HloProto) Descriptor deprecated 

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

Deprecated: Use HloProto.ProtoReflect.Descriptor instead.

func (*HloProto) GetBufferAssignment 

func (x *HloProto) GetBufferAssignment() *BufferAssignmentProto

func (*HloProto) GetHloModule 

func (x *HloProto) GetHloModule() *HloModuleProto

func (*HloProto) ProtoMessage 

func (*HloProto) ProtoMessage()

func (*HloProto) ProtoReflect 

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

func (*HloProto) Reset 

func (x *HloProto) Reset()

func (*HloProto) String 

func (x *HloProto) String() string

type HloScheduleProto 

type HloScheduleProto struct {

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

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

func (*HloScheduleProto) Descriptor deprecated 

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

Deprecated: Use HloScheduleProto.ProtoReflect.Descriptor instead.

func (*HloScheduleProto) GetSequences 

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

func (*HloScheduleProto) ProtoMessage 

func (*HloScheduleProto) ProtoMessage()

func (*HloScheduleProto) ProtoReflect 

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

func (*HloScheduleProto) Reset 

func (x *HloScheduleProto) Reset()

func (*HloScheduleProto) String 

func (x *HloScheduleProto) String() string

type HloScheduleProto_InstructionSequence 

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

func (*HloScheduleProto_InstructionSequence) Descriptor deprecated 

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

Deprecated: Use HloScheduleProto_InstructionSequence.ProtoReflect.Descriptor instead.

func (*HloScheduleProto_InstructionSequence) GetInstructionIds 

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

func (*HloScheduleProto_InstructionSequence) ProtoMessage 

func (*HloScheduleProto_InstructionSequence) ProtoMessage()

func (*HloScheduleProto_InstructionSequence) ProtoReflect 

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

func (*HloScheduleProto_InstructionSequence) Reset 

func (x *HloScheduleProto_InstructionSequence) Reset()

func (*HloScheduleProto_InstructionSequence) String 

func (x *HloScheduleProto_InstructionSequence) String() string

type HloSnapshot 

type HloSnapshot struct {

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

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

func (*HloSnapshot) Descriptor deprecated 

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

Deprecated: Use HloSnapshot.ProtoReflect.Descriptor instead.

func (*HloSnapshot) GetArguments 

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

func (*HloSnapshot) GetExecutionPlatform 

func (x *HloSnapshot) GetExecutionPlatform() string

func (*HloSnapshot) GetHlo 

func (x *HloSnapshot) GetHlo() *HloProto

func (*HloSnapshot) GetResult 

func (x *HloSnapshot) GetResult() *xla_data.LiteralProto

func (*HloSnapshot) ProtoMessage 

func (*HloSnapshot) ProtoMessage()

func (*HloSnapshot) ProtoReflect 

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

func (*HloSnapshot) Reset 

func (x *HloSnapshot) Reset()

func (*HloSnapshot) String 

func (x *HloSnapshot) String() string

type Kind 

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

func (Kind) Descriptor 

func (Kind) Descriptor() protoreflect.EnumDescriptor

func (Kind) Enum 

func (x Kind) Enum() *Kind

func (Kind) EnumDescriptor deprecated 

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

Deprecated: Use Kind.Descriptor instead.

func (Kind) Number 

func (x Kind) Number() protoreflect.EnumNumber

func (Kind) String 

func (x Kind) String() string

func (Kind) Type 

func (Kind) Type() protoreflect.EnumType

type LogicalBufferProto 

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

Serialization of LogicalBuffer.

func (*LogicalBufferProto) Descriptor deprecated 

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

Deprecated: Use LogicalBufferProto.ProtoReflect.Descriptor instead.

func (*LogicalBufferProto) GetColor 

func (x *LogicalBufferProto) GetColor() int64

func (*LogicalBufferProto) GetDefinedAt 

func (x *LogicalBufferProto) GetDefinedAt() *LogicalBufferProto_Location

func (*LogicalBufferProto) GetId 

func (x *LogicalBufferProto) GetId() int64

func (*LogicalBufferProto) GetSize 

func (x *LogicalBufferProto) GetSize() int64

func (*LogicalBufferProto) ProtoMessage 

func (*LogicalBufferProto) ProtoMessage()

func (*LogicalBufferProto) ProtoReflect 

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

func (*LogicalBufferProto) Reset 

func (x *LogicalBufferProto) Reset()

func (*LogicalBufferProto) String 

func (x *LogicalBufferProto) String() string

type LogicalBufferProto_Location 

type LogicalBufferProto_Location struct {

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

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

func (*LogicalBufferProto_Location) Descriptor deprecated 

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

Deprecated: Use LogicalBufferProto_Location.ProtoReflect.Descriptor instead.

func (*LogicalBufferProto_Location) GetInstructionId 

func (x *LogicalBufferProto_Location) GetInstructionId() int64

func (*LogicalBufferProto_Location) GetInstructionName deprecated 

func (x *LogicalBufferProto_Location) GetInstructionName() string

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

func (*LogicalBufferProto_Location) GetShapeIndex 

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

func (*LogicalBufferProto_Location) ProtoMessage 

func (*LogicalBufferProto_Location) ProtoMessage()

func (*LogicalBufferProto_Location) ProtoReflect 

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

func (*LogicalBufferProto_Location) Reset 

func (x *LogicalBufferProto_Location) Reset()

func (*LogicalBufferProto_Location) String 

func (x *LogicalBufferProto_Location) String() string

type StackFrameIndexProto 

type StackFrameIndexProto struct {

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

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

func (*StackFrameIndexProto) Descriptor deprecated 

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

Deprecated: Use StackFrameIndexProto.ProtoReflect.Descriptor instead.

func (*StackFrameIndexProto) GetFileLocations 

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

func (*StackFrameIndexProto) GetFileNames 

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

func (*StackFrameIndexProto) GetFunctionNames 

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

func (*StackFrameIndexProto) GetStackFrames 

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

func (*StackFrameIndexProto) ProtoMessage 

func (*StackFrameIndexProto) ProtoMessage()

func (*StackFrameIndexProto) ProtoReflect 

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

func (*StackFrameIndexProto) Reset 

func (x *StackFrameIndexProto) Reset()

func (*StackFrameIndexProto) String 

func (x *StackFrameIndexProto) String() string

type StackFrameIndexProto_FileLocation 

type StackFrameIndexProto_FileLocation struct {

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

Serialization of file position.

func (*StackFrameIndexProto_FileLocation) Descriptor deprecated 

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

Deprecated: Use StackFrameIndexProto_FileLocation.ProtoReflect.Descriptor instead.

func (*StackFrameIndexProto_FileLocation) GetColumn 

func (x *StackFrameIndexProto_FileLocation) GetColumn() int32

func (*StackFrameIndexProto_FileLocation) GetFileNameId 

func (x *StackFrameIndexProto_FileLocation) GetFileNameId() int32

func (*StackFrameIndexProto_FileLocation) GetFunctionNameId 

func (x *StackFrameIndexProto_FileLocation) GetFunctionNameId() int32

func (*StackFrameIndexProto_FileLocation) GetLine 

func (x *StackFrameIndexProto_FileLocation) GetLine() int32

func (*StackFrameIndexProto_FileLocation) ProtoMessage 

func (*StackFrameIndexProto_FileLocation) ProtoMessage()

func (*StackFrameIndexProto_FileLocation) ProtoReflect 

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

func (*StackFrameIndexProto_FileLocation) Reset 

func (x *StackFrameIndexProto_FileLocation) Reset()

func (*StackFrameIndexProto_FileLocation) String 

func (x *StackFrameIndexProto_FileLocation) String() string

type StackFrameIndexProto_StackFrame 

type StackFrameIndexProto_StackFrame struct {

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

Serialization of frame.

func (*StackFrameIndexProto_StackFrame) Descriptor deprecated 

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

Deprecated: Use StackFrameIndexProto_StackFrame.ProtoReflect.Descriptor instead.

func (*StackFrameIndexProto_StackFrame) GetFileLocationId 

func (x *StackFrameIndexProto_StackFrame) GetFileLocationId() int32

func (*StackFrameIndexProto_StackFrame) GetParentFrameId 

func (x *StackFrameIndexProto_StackFrame) GetParentFrameId() int32

func (*StackFrameIndexProto_StackFrame) ProtoMessage 

func (*StackFrameIndexProto_StackFrame) ProtoMessage()

func (*StackFrameIndexProto_StackFrame) ProtoReflect 

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

func (*StackFrameIndexProto_StackFrame) Reset 

func (x *StackFrameIndexProto_StackFrame) Reset()

func (*StackFrameIndexProto_StackFrame) String 

func (x *StackFrameIndexProto_StackFrame) String() string

Source Files

View all Source files

Jump to

Keyboard shortcuts

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