service

package
v2.14.1 Latest Latest
Warning

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

 Go to latest Published: Dec 8, 2023 License: BSD-3-Clause Imports: 7 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",
	}
	HloModuleProto_ProfileType_value = map[string]int32{
		"INVALID": 0,
		"FLAG":    1,
		"FUSION":  2,
		"LAYOUT":  3,
		"DOT":     4,
	}
)

Enum value maps for HloModuleProto_ProfileType.

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

Enum value maps for HeapSimulatorTrace_Event_Kind.

View Source 
var (
	CompilationLogEntry_CompilationStage_name = map[int32]string{
		0: "UNSPECIFIED",
		1: "END_TO_END",
		2: "HLO_PASSES",
		3: "CODE_GENERATION",
		4: "BACKEND_PASSES",
	}
	CompilationLogEntry_CompilationStage_value = map[string]int32{
		"UNSPECIFIED":     0,
		"END_TO_END":      1,
		"HLO_PASSES":      2,
		"CODE_GENERATION": 3,
		"BACKEND_PASSES":  4,
	}
)

Enum value maps for CompilationLogEntry_CompilationStage.

View Source 
var File_tensorflow_compiler_xla_service_hlo_execution_profile_data_proto protoreflect.FileDescriptor
View Source 
var File_tensorflow_compiler_xla_service_hlo_profile_printer_data_proto protoreflect.FileDescriptor
View Source 
var File_tensorflow_compiler_xla_service_hlo_proto protoreflect.FileDescriptor
View Source 
var File_tensorflow_compiler_xla_service_memory_space_assignment_proto protoreflect.FileDescriptor
View Source 
var File_tensorflow_compiler_xla_service_metrics_proto protoreflect.FileDescriptor
View Source 
var File_tensorflow_compiler_xla_service_test_compilation_environment_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 CompilationLogEntry added in v2.12.0 

type CompilationLogEntry struct {

	// Time when the event captured by this log entry occurred.
	Timestamp *timestamppb.Timestamp `protobuf:"bytes,1,opt,name=timestamp,proto3" json:"timestamp,omitempty"`
	// Compilation stage recorded by this log entry.
	Stage CompilationLogEntry_CompilationStage `protobuf:"varint,2,opt,name=stage,proto3,enum=xla.CompilationLogEntry_CompilationStage" json:"stage,omitempty"`
	// Duration of the given compilation stage.
	Duration *durationpb.Duration `protobuf:"bytes,3,opt,name=duration,proto3" json:"duration,omitempty"`
	// Task index from which this log entry was recorded or
	// -1 if the task index could not be fetched.
	TaskIndex int32 `protobuf:"varint,4,opt,name=task_index,json=taskIndex,proto3" json:"task_index,omitempty"`
	// Pass specific metrics.
	PassMetrics []*PassMetrics `protobuf:"bytes,5,rep,name=pass_metrics,json=passMetrics,proto3" json:"pass_metrics,omitempty"`
	// contains filtered or unexported fields
}

Defines XLA compilation metrics.

func (*CompilationLogEntry) Descriptor deprecated added in v2.12.0 

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

Deprecated: Use CompilationLogEntry.ProtoReflect.Descriptor instead.

func (*CompilationLogEntry) GetDuration added in v2.12.0 

func (x *CompilationLogEntry) GetDuration() *durationpb.Duration

func (*CompilationLogEntry) GetPassMetrics added in v2.13.0 

func (x *CompilationLogEntry) GetPassMetrics() []*PassMetrics

func (*CompilationLogEntry) GetStage added in v2.12.0 

func (x *CompilationLogEntry) GetStage() CompilationLogEntry_CompilationStage

func (*CompilationLogEntry) GetTaskIndex added in v2.12.0 

func (x *CompilationLogEntry) GetTaskIndex() int32

func (*CompilationLogEntry) GetTimestamp added in v2.12.0 

func (x *CompilationLogEntry) GetTimestamp() *timestamppb.Timestamp

func (*CompilationLogEntry) ProtoMessage added in v2.12.0 

func (*CompilationLogEntry) ProtoMessage()

func (*CompilationLogEntry) ProtoReflect added in v2.12.0 

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

func (*CompilationLogEntry) Reset added in v2.12.0 

func (x *CompilationLogEntry) Reset()

func (*CompilationLogEntry) String added in v2.12.0 

func (x *CompilationLogEntry) String() string

type CompilationLogEntry_CompilationStage added in v2.12.0 

type CompilationLogEntry_CompilationStage int32

Defines compilation stages for which metrics are collected. 

const (
	CompilationLogEntry_UNSPECIFIED     CompilationLogEntry_CompilationStage = 0
	CompilationLogEntry_END_TO_END      CompilationLogEntry_CompilationStage = 1
	CompilationLogEntry_HLO_PASSES      CompilationLogEntry_CompilationStage = 2
	CompilationLogEntry_CODE_GENERATION CompilationLogEntry_CompilationStage = 3
	CompilationLogEntry_BACKEND_PASSES  CompilationLogEntry_CompilationStage = 4
)

func (CompilationLogEntry_CompilationStage) Descriptor added in v2.12.0 

func (CompilationLogEntry_CompilationStage) Descriptor() protoreflect.EnumDescriptor

func (CompilationLogEntry_CompilationStage) Enum added in v2.12.0 

func (x CompilationLogEntry_CompilationStage) Enum() *CompilationLogEntry_CompilationStage

func (CompilationLogEntry_CompilationStage) EnumDescriptor deprecated added in v2.12.0 

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

Deprecated: Use CompilationLogEntry_CompilationStage.Descriptor instead.

func (CompilationLogEntry_CompilationStage) Number added in v2.12.0 

func (x CompilationLogEntry_CompilationStage) Number() protoreflect.EnumNumber

func (CompilationLogEntry_CompilationStage) String added in v2.12.0 

func (x CompilationLogEntry_CompilationStage) String() string

func (CompilationLogEntry_CompilationStage) Type added in v2.12.0 

func (CompilationLogEntry_CompilationStage) Type() protoreflect.EnumType

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 added in v2.12.0 

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 DynamicParameterBindingProto 

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

func (*DynamicParameterBindingProto) Descriptor deprecated 

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

Deprecated: Use DynamicParameterBindingProto.ProtoReflect.Descriptor instead.

func (*DynamicParameterBindingProto) GetEntries 

func (x *DynamicParameterBindingProto) GetEntries() []*DynamicParameterBindingProto_Binding

func (*DynamicParameterBindingProto) ProtoMessage 

func (*DynamicParameterBindingProto) ProtoMessage()

func (*DynamicParameterBindingProto) ProtoReflect 

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

func (*DynamicParameterBindingProto) Reset 

func (x *DynamicParameterBindingProto) Reset()

func (*DynamicParameterBindingProto) String 

func (x *DynamicParameterBindingProto) String() string

type DynamicParameterBindingProto_Binding 

type DynamicParameterBindingProto_Binding struct {
	DynamicParamNum   int64   `protobuf:"varint,1,opt,name=dynamic_param_num,json=dynamicParamNum,proto3" json:"dynamic_param_num,omitempty"`
	DynamicParamIndex []int64 `protobuf:"varint,2,rep,packed,name=dynamic_param_index,json=dynamicParamIndex,proto3" json:"dynamic_param_index,omitempty"`
	TargetParamNum    int64   `protobuf:"varint,3,opt,name=target_param_num,json=targetParamNum,proto3" json:"target_param_num,omitempty"`
	TargetParamIndex  []int64 `protobuf:"varint,4,rep,packed,name=target_param_index,json=targetParamIndex,proto3" json:"target_param_index,omitempty"`
	TargetParamDimNum int64   `protobuf:"varint,5,opt,name=target_param_dim_num,json=targetParamDimNum,proto3" json:"target_param_dim_num,omitempty"`
	// contains filtered or unexported fields
}

A list of bindings which indicates that the `target_param_dim_num` in the subshape `target_param_index` of parameter `target_param_num` is a dynamic dimension and its real dynamic size is represented by `dynamic_param_index` in parameter `dynamic_param_num`.

As an example, imagine we have a program: 

ENTRY main {
  a = f32[] parameter(0)
  b = f32[10] parameter(1)
  ROOT root = (f32[], f32[10]) tuple(%a, %b)
}

Let's say 'b' (param index 1) is a dynamic shape whose input has an upperbound of 10 and real size is determined at runtime.'a' represents the real size of b's first dimension.

In this case, the fields are set in the following way: dynamic_param_num = 1 dynamic_param_index = {} target_param_num = 0 target_param_index = {} target_param_dim_num = 0

func (*DynamicParameterBindingProto_Binding) Descriptor deprecated 

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

Deprecated: Use DynamicParameterBindingProto_Binding.ProtoReflect.Descriptor instead.

func (*DynamicParameterBindingProto_Binding) GetDynamicParamIndex 

func (x *DynamicParameterBindingProto_Binding) GetDynamicParamIndex() []int64

func (*DynamicParameterBindingProto_Binding) GetDynamicParamNum 

func (x *DynamicParameterBindingProto_Binding) GetDynamicParamNum() int64

func (*DynamicParameterBindingProto_Binding) GetTargetParamDimNum 

func (x *DynamicParameterBindingProto_Binding) GetTargetParamDimNum() int64

func (*DynamicParameterBindingProto_Binding) GetTargetParamIndex 

func (x *DynamicParameterBindingProto_Binding) GetTargetParamIndex() []int64

func (*DynamicParameterBindingProto_Binding) GetTargetParamNum 

func (x *DynamicParameterBindingProto_Binding) GetTargetParamNum() int64

func (*DynamicParameterBindingProto_Binding) ProtoMessage 

func (*DynamicParameterBindingProto_Binding) ProtoMessage()

func (*DynamicParameterBindingProto_Binding) ProtoReflect 

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

func (*DynamicParameterBindingProto_Binding) Reset 

func (x *DynamicParameterBindingProto_Binding) Reset()

func (*DynamicParameterBindingProto_Binding) String 

func (x *DynamicParameterBindingProto_Binding) String() string

type EntryFunctionAttributes 

type EntryFunctionAttributes struct {
	Buffers []*EntryFunctionAttributes_BufferParameterAttributes `protobuf:"bytes,1,rep,name=buffers,proto3" json:"buffers,omitempty"`
	// xla::Shape in string format.
	ResultXlaShape string `protobuf:"bytes,2,opt,name=result_xla_shape,json=resultXlaShape,proto3" json:"result_xla_shape,omitempty"`
	// contains filtered or unexported fields
}

Encodes attributes for an entry function.

func (*EntryFunctionAttributes) Descriptor deprecated 

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

Deprecated: Use EntryFunctionAttributes.ProtoReflect.Descriptor instead.

func (*EntryFunctionAttributes) GetBuffers 

func (x *EntryFunctionAttributes) GetBuffers() []*EntryFunctionAttributes_BufferParameterAttributes

func (*EntryFunctionAttributes) GetResultXlaShape 

func (x *EntryFunctionAttributes) GetResultXlaShape() string

func (*EntryFunctionAttributes) ProtoMessage 

func (*EntryFunctionAttributes) ProtoMessage()

func (*EntryFunctionAttributes) ProtoReflect 

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

func (*EntryFunctionAttributes) Reset 

func (x *EntryFunctionAttributes) Reset()

func (*EntryFunctionAttributes) String 

func (x *EntryFunctionAttributes) String() string

type EntryFunctionAttributes_BufferParameterAttributes 

type EntryFunctionAttributes_BufferParameterAttributes struct {

	// Represents an lmhlo.params function argument attribute.
	LmhloParams int64 `protobuf:"varint,1,opt,name=lmhlo_params,json=lmhloParams,proto3" json:"lmhlo_params,omitempty"`
	// TODO(hanbinyoon): Deprecate when optional fields are available in proto3
	// (Protocol Buffers v3.15.0).
	LmhloParamsPresent bool `protobuf:"varint,6,opt,name=lmhlo_params_present,json=lmhloParamsPresent,proto3" json:"lmhlo_params_present,omitempty"`
	// Represents an lmhlo.param_shape_index function argument attribute.
	LmhloParamShapeIndex *EntryFunctionAttributes_ShapeIndex `protobuf:"bytes,2,opt,name=lmhlo_param_shape_index,json=lmhloParamShapeIndex,proto3" json:"lmhlo_param_shape_index,omitempty"`
	// Represents an lmhlo.constant_name function argument attribute.
	LmhloConstantName string `protobuf:"bytes,3,opt,name=lmhlo_constant_name,json=lmhloConstantName,proto3" json:"lmhlo_constant_name,omitempty"`
	// Represents an lmhlo.must_alias function argument attribute.
	LmhloMustAlias bool `protobuf:"varint,4,opt,name=lmhlo_must_alias,json=lmhloMustAlias,proto3" json:"lmhlo_must_alias,omitempty"`
	// Represents an lmhlo.params function argument attribute.
	LmhloOutputIndex *EntryFunctionAttributes_ShapeIndex `protobuf:"bytes,5,opt,name=lmhlo_output_index,json=lmhloOutputIndex,proto3" json:"lmhlo_output_index,omitempty"`
	// contains filtered or unexported fields
}

Encodes attributes for a single buffer parameter.

func (*EntryFunctionAttributes_BufferParameterAttributes) Descriptor deprecated 

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

Deprecated: Use EntryFunctionAttributes_BufferParameterAttributes.ProtoReflect.Descriptor instead.

func (*EntryFunctionAttributes_BufferParameterAttributes) GetLmhloConstantName 

func (x *EntryFunctionAttributes_BufferParameterAttributes) GetLmhloConstantName() string

func (*EntryFunctionAttributes_BufferParameterAttributes) GetLmhloMustAlias 

func (x *EntryFunctionAttributes_BufferParameterAttributes) GetLmhloMustAlias() bool

func (*EntryFunctionAttributes_BufferParameterAttributes) GetLmhloOutputIndex 

func (x *EntryFunctionAttributes_BufferParameterAttributes) GetLmhloOutputIndex() *EntryFunctionAttributes_ShapeIndex

func (*EntryFunctionAttributes_BufferParameterAttributes) GetLmhloParamShapeIndex 

func (x *EntryFunctionAttributes_BufferParameterAttributes) GetLmhloParamShapeIndex() *EntryFunctionAttributes_ShapeIndex

func (*EntryFunctionAttributes_BufferParameterAttributes) GetLmhloParams 

func (x *EntryFunctionAttributes_BufferParameterAttributes) GetLmhloParams() int64

func (*EntryFunctionAttributes_BufferParameterAttributes) GetLmhloParamsPresent 

func (x *EntryFunctionAttributes_BufferParameterAttributes) GetLmhloParamsPresent() bool

func (*EntryFunctionAttributes_BufferParameterAttributes) ProtoMessage 

func (*EntryFunctionAttributes_BufferParameterAttributes) ProtoMessage()

func (*EntryFunctionAttributes_BufferParameterAttributes) ProtoReflect 

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

func (*EntryFunctionAttributes_BufferParameterAttributes) Reset 

func (x *EntryFunctionAttributes_BufferParameterAttributes) Reset()

func (*EntryFunctionAttributes_BufferParameterAttributes) String 

func (x *EntryFunctionAttributes_BufferParameterAttributes) String() string

type EntryFunctionAttributes_ShapeIndex 

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

Acts as the underlying container for an xla::ShapeIndex.

func (*EntryFunctionAttributes_ShapeIndex) Descriptor deprecated 

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

Deprecated: Use EntryFunctionAttributes_ShapeIndex.ProtoReflect.Descriptor instead.

func (*EntryFunctionAttributes_ShapeIndex) GetIndices 

func (x *EntryFunctionAttributes_ShapeIndex) GetIndices() []int64

func (*EntryFunctionAttributes_ShapeIndex) ProtoMessage 

func (*EntryFunctionAttributes_ShapeIndex) ProtoMessage()

func (*EntryFunctionAttributes_ShapeIndex) ProtoReflect 

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

func (*EntryFunctionAttributes_ShapeIndex) Reset 

func (x *EntryFunctionAttributes_ShapeIndex) Reset()

func (*EntryFunctionAttributes_ShapeIndex) String 

func (x *EntryFunctionAttributes_ShapeIndex) String() string

type HeapSimulatorTrace 

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

A trace of a HeapSimulator run.

func (*HeapSimulatorTrace) Descriptor deprecated 

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

Deprecated: Use HeapSimulatorTrace.ProtoReflect.Descriptor instead.

func (*HeapSimulatorTrace) GetBufferAllocationIndex 

func (x *HeapSimulatorTrace) GetBufferAllocationIndex() int64

func (*HeapSimulatorTrace) GetEvents 

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

func (*HeapSimulatorTrace) GetWholeModuleSimulation 

func (x *HeapSimulatorTrace) GetWholeModuleSimulation() bool

func (*HeapSimulatorTrace) ProtoMessage 

func (*HeapSimulatorTrace) ProtoMessage()

func (*HeapSimulatorTrace) ProtoReflect 

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

func (*HeapSimulatorTrace) Reset 

func (x *HeapSimulatorTrace) Reset()

func (*HeapSimulatorTrace) String 

func (x *HeapSimulatorTrace) String() string

type HeapSimulatorTrace_Event 

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

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

func (*HeapSimulatorTrace_Event) Descriptor deprecated 

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

Deprecated: Use HeapSimulatorTrace_Event.ProtoReflect.Descriptor instead.

func (*HeapSimulatorTrace_Event) GetBufferId 

func (x *HeapSimulatorTrace_Event) GetBufferId() int64

func (*HeapSimulatorTrace_Event) GetComputationName 

func (x *HeapSimulatorTrace_Event) GetComputationName() string

func (*HeapSimulatorTrace_Event) GetInstructionName 

func (x *HeapSimulatorTrace_Event) GetInstructionName() string

func (*HeapSimulatorTrace_Event) GetKind 

func (x *HeapSimulatorTrace_Event) GetKind() HeapSimulatorTrace_Event_Kind

func (*HeapSimulatorTrace_Event) GetShareWithCanonicalId 

func (x *HeapSimulatorTrace_Event) GetShareWithCanonicalId() int64

func (*HeapSimulatorTrace_Event) ProtoMessage 

func (*HeapSimulatorTrace_Event) ProtoMessage()

func (*HeapSimulatorTrace_Event) ProtoReflect 

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

func (*HeapSimulatorTrace_Event) Reset 

func (x *HeapSimulatorTrace_Event) Reset()

func (*HeapSimulatorTrace_Event) String 

func (x *HeapSimulatorTrace_Event) String() string

type HeapSimulatorTrace_Event_Kind 

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

func (HeapSimulatorTrace_Event_Kind) Descriptor 

func (HeapSimulatorTrace_Event_Kind) Descriptor() protoreflect.EnumDescriptor

func (HeapSimulatorTrace_Event_Kind) Enum 

func (x HeapSimulatorTrace_Event_Kind) Enum() *HeapSimulatorTrace_Event_Kind

func (HeapSimulatorTrace_Event_Kind) EnumDescriptor deprecated 

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

Deprecated: Use HeapSimulatorTrace_Event_Kind.Descriptor instead.

func (HeapSimulatorTrace_Event_Kind) Number 

func (x HeapSimulatorTrace_Event_Kind) Number() protoreflect.EnumNumber

func (HeapSimulatorTrace_Event_Kind) String 

func (x HeapSimulatorTrace_Event_Kind) String() string

func (HeapSimulatorTrace_Event_Kind) Type 

func (HeapSimulatorTrace_Event_Kind) Type() protoreflect.EnumType

type 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 *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() *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 HloExecutionProfileData 

type HloExecutionProfileData struct {
	PrinterData     *HloProfilePrinterData `protobuf:"bytes,1,opt,name=printer_data,json=printerData,proto3" json:"printer_data,omitempty"`
	ProfileCounters []int64                `protobuf:"varint,2,rep,packed,name=profile_counters,json=profileCounters,proto3" json:"profile_counters,omitempty"`
	// contains filtered or unexported fields
}

func (*HloExecutionProfileData) Descriptor deprecated 

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

Deprecated: Use HloExecutionProfileData.ProtoReflect.Descriptor instead.

func (*HloExecutionProfileData) GetPrinterData 

func (x *HloExecutionProfileData) GetPrinterData() *HloProfilePrinterData

func (*HloExecutionProfileData) GetProfileCounters 

func (x *HloExecutionProfileData) GetProfileCounters() []int64

func (*HloExecutionProfileData) ProtoMessage 

func (*HloExecutionProfileData) ProtoMessage()

func (*HloExecutionProfileData) ProtoReflect 

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

func (*HloExecutionProfileData) Reset 

func (x *HloExecutionProfileData) Reset()

func (*HloExecutionProfileData) String 

func (x *HloExecutionProfileData) 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 paremter'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    *data.ShapeProto `protobuf:"bytes,3,opt,name=shape,proto3" json:"shape,omitempty"`
	Metadata *data.OpMetadata `protobuf:"bytes,7,opt,name=metadata,proto3" json:"metadata,omitempty"`
	// Literal, only present for kConstant.
	Literal *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 *data.Window `protobuf:"bytes,15,opt,name=window,proto3" json:"window,omitempty"`
	// Describes the dimension numbers used for a convolution.
	ConvolutionDimensionNumbers *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 *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 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 *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 *data.DotDimensionNumbers `protobuf:"bytes,30,opt,name=dot_dimension_numbers,json=dotDimensionNumbers,proto3" json:"dot_dimension_numbers,omitempty"`
	// FFT type (FFT, IFFT, etc).
	FftType 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 *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              *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"`
	// Cross replica op fields.
	ReplicaGroups []*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 tensorflow/compiler/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 *data.ScatterDimensionNumbers `` /* 133-byte string literal not displayed */
	// Precision configuration for the instruction. Has backend-specific meaning.
	PrecisionConfig *data.PrecisionConfig `protobuf:"bytes,51,opt,name=precision_config,json=precisionConfig,proto3" json:"precision_config,omitempty"`
	// Collective permute field.
	SourceTargetPairs []*data.SourceTarget `protobuf:"bytes,52,rep,name=source_target_pairs,json=sourceTargetPairs,proto3" json:"source_target_pairs,omitempty"`
	// Sharding for kDomain instructions.
	DomainEntrySharding *data.OpSharding `protobuf:"bytes,54,opt,name=domain_entry_sharding,json=domainEntrySharding,proto3" json:"domain_entry_sharding,omitempty"`
	DomainExitSharding  *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 []*data.ShapeProto `` /* 135-byte string literal not displayed */
	// Options for TriangularSolve
	TriangularSolveOptions *data.TriangularSolveOptions `` /* 130-byte string literal not displayed */
	// Options for Cholesky
	CholeskyOptions *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 *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 []*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 *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 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 tensorflow/compiler/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 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 identifier for an async group which consists of an
	// async start, async done, and zero or more async update operations.
	// Negative async_group_id is equivalent to no async group id.
	AsyncGroupId int64 `protobuf:"varint,78,opt,name=async_group_id,json=asyncGroupId,proto3" json:"async_group_id,omitempty"`
	// 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 information for tracking propagation of values within HLO
	// graph.
	StatisticsViz *data.StatisticsViz `protobuf:"bytes,82,opt,name=statistics_viz,json=statisticsViz,proto3" json:"statistics_viz,omitempty"`
	// contains filtered or unexported fields
}

Serialization of HloInstruction. Next ID: 83

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 tensorflow/compiler/xla/service/hlo.proto.

func (*HloInstructionProto) GetAsyncExecutionThread 

func (x *HloInstructionProto) GetAsyncExecutionThread() string

func (*HloInstructionProto) GetAsyncGroupId 

func (x *HloInstructionProto) GetAsyncGroupId() int64

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

func (*HloInstructionProto) GetComparisonDirection 

func (x *HloInstructionProto) GetComparisonDirection() string

func (*HloInstructionProto) GetComparisonType 

func (x *HloInstructionProto) GetComparisonType() string

func (*HloInstructionProto) GetConstrainLayout 

func (x *HloInstructionProto) GetConstrainLayout() bool

func (*HloInstructionProto) GetControlPredecessorIds 

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

func (*HloInstructionProto) GetConvolutionDimensionNumbers 

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

func (*HloInstructionProto) GetCrossProgramPrefetchIndex added in v2.12.0 

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() data.RandomDistribution

func (*HloInstructionProto) GetDomainEntrySharding 

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

func (*HloInstructionProto) GetDomainExitSharding 

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

func (*HloInstructionProto) GetDotDimensionNumbers 

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

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() data.FftType

func (*HloInstructionProto) GetFrontendAttributes 

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

func (*HloInstructionProto) GetFusionKind 

func (x *HloInstructionProto) GetFusionKind() string

func (*HloInstructionProto) GetGatherDimensionNumbers 

func (x *HloInstructionProto) GetGatherDimensionNumbers() *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) GetIsCrossProgramPrefetch deprecated 

func (x *HloInstructionProto) GetIsCrossProgramPrefetch() bool

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

func (*HloInstructionProto) GetIsHostTransfer 

func (x *HloInstructionProto) GetIsHostTransfer() bool

func (*HloInstructionProto) GetIsStable 

func (x *HloInstructionProto) GetIsStable() bool

func (*HloInstructionProto) GetK added in v2.14.0 

func (x *HloInstructionProto) GetK() int64

func (*HloInstructionProto) GetLiteral 

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

func (*HloInstructionProto) GetMantissaBits 

func (x *HloInstructionProto) GetMantissaBits() int32

func (*HloInstructionProto) GetMetadata 

func (x *HloInstructionProto) GetMetadata() *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() []*data.ShapeProto

func (*HloInstructionProto) GetOptionalCrossProgramPrefetchIndex added in v2.12.0 

func (m *HloInstructionProto) GetOptionalCrossProgramPrefetchIndex() isHloInstructionProto_OptionalCrossProgramPrefetchIndex

func (*HloInstructionProto) GetOutfeedConfig 

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

func (*HloInstructionProto) GetOutfeedShape 

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

func (*HloInstructionProto) GetOutputOperandAliasing added in v2.12.0 

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

func (*HloInstructionProto) GetPaddingConfig 

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

func (*HloInstructionProto) GetPaddingType 

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

func (*HloInstructionProto) GetParameterNumber 

func (x *HloInstructionProto) GetParameterNumber() int64

func (*HloInstructionProto) GetParameterReplication 

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

func (*HloInstructionProto) GetPrecisionConfig 

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

func (*HloInstructionProto) GetReplicaGroups 

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

func (*HloInstructionProto) GetRngAlgorithm 

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

func (*HloInstructionProto) GetScatterDimensionNumbers 

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

func (*HloInstructionProto) GetShape 

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

func (*HloInstructionProto) GetSharding 

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

func (*HloInstructionProto) GetSliceDimensions 

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

func (*HloInstructionProto) GetSourceTargetPairs 

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

func (*HloInstructionProto) GetStatisticsViz added in v2.14.0 

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

func (*HloInstructionProto) GetTriangularSolveOptions 

func (x *HloInstructionProto) GetTriangularSolveOptions() *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() *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 added in v2.12.0 

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 *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"`
	DynamicParameterBinding *DynamicParameterBindingProto `` /* 132-byte string literal not displayed */
	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      *data.OpSharding   `protobuf:"bytes,12,opt,name=spmd_output_sharding,json=spmdOutputSharding,proto3" json:"spmd_output_sharding,omitempty"`
	SpmdParametersShardings []*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 *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"`
	// 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) GetComputations 

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

func (*HloModuleProto) GetCrossProgramPrefetches 

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

func (*HloModuleProto) GetDeviceAssignment 

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

func (*HloModuleProto) GetDynamicParameterBinding 

func (x *HloModuleProto) GetDynamicParameterBinding() *DynamicParameterBindingProto

func (*HloModuleProto) GetEntryComputationId 

func (x *HloModuleProto) GetEntryComputationId() int64

func (*HloModuleProto) GetEntryComputationName 

func (x *HloModuleProto) GetEntryComputationName() string

func (*HloModuleProto) GetHostProgramShape 

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

func (*HloModuleProto) GetSpmdParametersShardings 

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

func (*HloModuleProto) GetStackFrameIndex added in v2.14.0 

func (x *HloModuleProto) GetStackFrameIndex() *StackFrameIndexProto

func (*HloModuleProto) GetUseAutoSpmdPartitioning added in v2.11.0 

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 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 data.CompilationEvent `` /* 136-byte string literal not displayed */
	// 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() data.CompilationEvent

func (*HloModuleProto_ProfileInfo) GetProfileSource 

func (x *HloModuleProto_ProfileInfo) GetProfileSource() 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
)

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

type HloProfilePrinterData struct {

	// HloComputationInfos for every HloComputation in the HloModule.
	ComputationInfos []*HloProfilePrinterData_HloComputationInfo `protobuf:"bytes,1,rep,name=computation_infos,json=computationInfos,proto3" json:"computation_infos,omitempty"`
	// The size of the profile counters array we will pretty-print.
	ProfileCountersSize int64 `protobuf:"varint,2,opt,name=profile_counters_size,json=profileCountersSize,proto3" json:"profile_counters_size,omitempty"`
	// Maps extra metric name to the index into the profile counters array.
	ExtraMetrics map[string]int64 `` /* 186-byte string literal not displayed */
	// Name of the entry computation.
	EntryComputation string `protobuf:"bytes,4,opt,name=entry_computation,json=entryComputation,proto3" json:"entry_computation,omitempty"`
	// contains filtered or unexported fields
}

Describes how to pretty-print a profile counter array gathered for a specific HloModule.

func (*HloProfilePrinterData) Descriptor deprecated 

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

Deprecated: Use HloProfilePrinterData.ProtoReflect.Descriptor instead.

func (*HloProfilePrinterData) GetComputationInfos 

func (x *HloProfilePrinterData) GetComputationInfos() []*HloProfilePrinterData_HloComputationInfo

func (*HloProfilePrinterData) GetEntryComputation 

func (x *HloProfilePrinterData) GetEntryComputation() string

func (*HloProfilePrinterData) GetExtraMetrics 

func (x *HloProfilePrinterData) GetExtraMetrics() map[string]int64

func (*HloProfilePrinterData) GetProfileCountersSize 

func (x *HloProfilePrinterData) GetProfileCountersSize() int64

func (*HloProfilePrinterData) ProtoMessage 

func (*HloProfilePrinterData) ProtoMessage()

func (*HloProfilePrinterData) ProtoReflect 

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

func (*HloProfilePrinterData) Reset 

func (x *HloProfilePrinterData) Reset()

func (*HloProfilePrinterData) String 

func (x *HloProfilePrinterData) String() string

type HloProfilePrinterData_HloComputationInfo 

type HloProfilePrinterData_HloComputationInfo struct {
	Name string `protobuf:"bytes,1,opt,name=name,proto3" json:"name,omitempty"`
	// The index into the profile counters array for the HloComputation
	// corresponding to this HloComputationInfo.
	ProfileIndex int64 `protobuf:"varint,2,opt,name=profile_index,json=profileIndex,proto3" json:"profile_index,omitempty"`
	// HloInstructionInfos for every HloInstruction in the HloComputation for
	// corresponding to this HloComputattionInfo.
	InstructionInfos []*HloProfilePrinterData_HloInstructionInfo `protobuf:"bytes,3,rep,name=instruction_infos,json=instructionInfos,proto3" json:"instruction_infos,omitempty"`
	// contains filtered or unexported fields
}

Pretty-printer information about an HloComputation.

func (*HloProfilePrinterData_HloComputationInfo) Descriptor deprecated 

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

Deprecated: Use HloProfilePrinterData_HloComputationInfo.ProtoReflect.Descriptor instead.

func (*HloProfilePrinterData_HloComputationInfo) GetInstructionInfos 

func (x *HloProfilePrinterData_HloComputationInfo) GetInstructionInfos() []*HloProfilePrinterData_HloInstructionInfo

func (*HloProfilePrinterData_HloComputationInfo) GetName 

func (x *HloProfilePrinterData_HloComputationInfo) GetName() string

func (*HloProfilePrinterData_HloComputationInfo) GetProfileIndex 

func (x *HloProfilePrinterData_HloComputationInfo) GetProfileIndex() int64

func (*HloProfilePrinterData_HloComputationInfo) ProtoMessage 

func (*HloProfilePrinterData_HloComputationInfo) ProtoMessage()

func (*HloProfilePrinterData_HloComputationInfo) ProtoReflect 

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

func (*HloProfilePrinterData_HloComputationInfo) Reset 

func (x *HloProfilePrinterData_HloComputationInfo) Reset()

func (*HloProfilePrinterData_HloComputationInfo) String 

func (x *HloProfilePrinterData_HloComputationInfo) String() string

type HloProfilePrinterData_HloInstructionInfo 

type HloProfilePrinterData_HloInstructionInfo struct {
	LongName  string `protobuf:"bytes,1,opt,name=long_name,json=longName,proto3" json:"long_name,omitempty"`
	ShortName string `protobuf:"bytes,2,opt,name=short_name,json=shortName,proto3" json:"short_name,omitempty"`
	Category  string `protobuf:"bytes,3,opt,name=category,proto3" json:"category,omitempty"`
	// Metrics computed by HloCostAnalysis.
	FlopCount           float32 `protobuf:"fixed32,4,opt,name=flop_count,json=flopCount,proto3" json:"flop_count,omitempty"`
	TranscendentalCount float32 `protobuf:"fixed32,5,opt,name=transcendental_count,json=transcendentalCount,proto3" json:"transcendental_count,omitempty"`
	BytesAccessed       int64   `protobuf:"varint,9,opt,name=bytes_accessed,json=bytesAccessed,proto3" json:"bytes_accessed,omitempty"`
	OptimalSeconds      float32 `protobuf:"fixed32,7,opt,name=optimal_seconds,json=optimalSeconds,proto3" json:"optimal_seconds,omitempty"`
	// The index into the profile counters array for the HloInstruction
	// corresponding to this HloInstructionInfo.
	ProfileIndex int64 `protobuf:"varint,8,opt,name=profile_index,json=profileIndex,proto3" json:"profile_index,omitempty"`
	// contains filtered or unexported fields
}

Pretty-printer information about an HloInstruction.

func (*HloProfilePrinterData_HloInstructionInfo) Descriptor deprecated 

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

Deprecated: Use HloProfilePrinterData_HloInstructionInfo.ProtoReflect.Descriptor instead.

func (*HloProfilePrinterData_HloInstructionInfo) GetBytesAccessed 

func (x *HloProfilePrinterData_HloInstructionInfo) GetBytesAccessed() int64

func (*HloProfilePrinterData_HloInstructionInfo) GetCategory 

func (x *HloProfilePrinterData_HloInstructionInfo) GetCategory() string

func (*HloProfilePrinterData_HloInstructionInfo) GetFlopCount 

func (x *HloProfilePrinterData_HloInstructionInfo) GetFlopCount() float32

func (*HloProfilePrinterData_HloInstructionInfo) GetLongName 

func (x *HloProfilePrinterData_HloInstructionInfo) GetLongName() string

func (*HloProfilePrinterData_HloInstructionInfo) GetOptimalSeconds 

func (x *HloProfilePrinterData_HloInstructionInfo) GetOptimalSeconds() float32

func (*HloProfilePrinterData_HloInstructionInfo) GetProfileIndex 

func (x *HloProfilePrinterData_HloInstructionInfo) GetProfileIndex() int64

func (*HloProfilePrinterData_HloInstructionInfo) GetShortName 

func (x *HloProfilePrinterData_HloInstructionInfo) GetShortName() string

func (*HloProfilePrinterData_HloInstructionInfo) GetTranscendentalCount 

func (x *HloProfilePrinterData_HloInstructionInfo) GetTranscendentalCount() float32

func (*HloProfilePrinterData_HloInstructionInfo) ProtoMessage 

func (*HloProfilePrinterData_HloInstructionInfo) ProtoMessage()

func (*HloProfilePrinterData_HloInstructionInfo) ProtoReflect 

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

func (*HloProfilePrinterData_HloInstructionInfo) Reset 

func (x *HloProfilePrinterData_HloInstructionInfo) Reset()

func (*HloProfilePrinterData_HloInstructionInfo) String 

func (x *HloProfilePrinterData_HloInstructionInfo) 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 []*data.LiteralProto `protobuf:"bytes,2,rep,name=arguments,proto3" json:"arguments,omitempty"`
	// The result of the graph.
	Result *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() []*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() *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 tensorflow/compiler/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 tensorflow/compiler/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 MemoryBoundLoopOptimizerOptions added in v2.14.0 

type MemoryBoundLoopOptimizerOptions struct {

	// Enable the memory-bound loop optimizations.
	Enabled *bool `protobuf:"varint,1,opt,name=enabled,proto3,oneof" json:"enabled,omitempty"`
	// The desired ratio of overlapped operations that is sufficient to overlap
	// prefetches with. If this value is 1, the algorithm will try to fully
	// overlap the prefetches with other compute, if less than 1, the algorithm
	// may schedule prefetches such that some of the prefetch is not overlapped,
	// so may become critical. For example, if this value is 0.5, we are willing
	// for the prefetch time to take up to 2X of the overlapped computation time.
	DesiredCopyRatio *float32 `protobuf:"fixed32,2,opt,name=desired_copy_ratio,json=desiredCopyRatio,proto3,oneof" json:"desired_copy_ratio,omitempty"`
	// If true, the algorithm allows a fully pipelined prefetch to be scheduled
	// even if the copy resources haven't reached the desired copy ratio. A fully
	// pipelined prefetch starts the same time as its counterpart in the previous
	// iteration finishes.
	AllowUnsatisfiedFullyPipelinedPrefetch *bool `` /* 188-byte string literal not displayed */
	// contains filtered or unexported fields
}

Options for memory-bound loop optimizations in memory space assignment. If enabled, this pass can optimize memory-bound unrolled loops to maximize the bandwidth utilized and minimize the execution time.

func (*MemoryBoundLoopOptimizerOptions) Descriptor deprecated added in v2.14.0 

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

Deprecated: Use MemoryBoundLoopOptimizerOptions.ProtoReflect.Descriptor instead.

func (*MemoryBoundLoopOptimizerOptions) GetAllowUnsatisfiedFullyPipelinedPrefetch added in v2.14.0 

func (x *MemoryBoundLoopOptimizerOptions) GetAllowUnsatisfiedFullyPipelinedPrefetch() bool

func (*MemoryBoundLoopOptimizerOptions) GetDesiredCopyRatio added in v2.14.0 

func (x *MemoryBoundLoopOptimizerOptions) GetDesiredCopyRatio() float32

func (*MemoryBoundLoopOptimizerOptions) GetEnabled added in v2.14.0 

func (x *MemoryBoundLoopOptimizerOptions) GetEnabled() bool

func (*MemoryBoundLoopOptimizerOptions) ProtoMessage added in v2.14.0 

func (*MemoryBoundLoopOptimizerOptions) ProtoMessage()

func (*MemoryBoundLoopOptimizerOptions) ProtoReflect added in v2.14.0 

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

func (*MemoryBoundLoopOptimizerOptions) Reset added in v2.14.0 

func (x *MemoryBoundLoopOptimizerOptions) Reset()

func (*MemoryBoundLoopOptimizerOptions) String added in v2.14.0 

func (x *MemoryBoundLoopOptimizerOptions) String() string

type PassMetrics added in v2.13.0 

type PassMetrics struct {

	// Unique ID of the module on which the pass was run.
	ModuleId int64 `protobuf:"varint,1,opt,name=module_id,json=moduleId,proto3" json:"module_id,omitempty"`
	// The name of the pass.
	PassName string `protobuf:"bytes,2,opt,name=pass_name,json=passName,proto3" json:"pass_name,omitempty"`
	// Duration of the pass.
	PassDuration *durationpb.Duration `protobuf:"bytes,3,opt,name=pass_duration,json=passDuration,proto3" json:"pass_duration,omitempty"`
	// contains filtered or unexported fields
}

Defines pass specific metrics.

func (*PassMetrics) Descriptor deprecated added in v2.13.0 

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

Deprecated: Use PassMetrics.ProtoReflect.Descriptor instead.

func (*PassMetrics) GetModuleId added in v2.13.0 

func (x *PassMetrics) GetModuleId() int64

func (*PassMetrics) GetPassDuration added in v2.13.0 

func (x *PassMetrics) GetPassDuration() *durationpb.Duration

func (*PassMetrics) GetPassName added in v2.13.0 

func (x *PassMetrics) GetPassName() string

func (*PassMetrics) ProtoMessage added in v2.13.0 

func (*PassMetrics) ProtoMessage()

func (*PassMetrics) ProtoReflect added in v2.13.0 

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

func (*PassMetrics) Reset added in v2.13.0 

func (x *PassMetrics) Reset()

func (*PassMetrics) String added in v2.13.0 

func (x *PassMetrics) String() string

type SlicedPrefetchOptions added in v2.13.0 

type SlicedPrefetchOptions struct {

	// The maximum number of slices into which to slice a prefetch.
	MaxSlices uint32 `protobuf:"varint,1,opt,name=max_slices,json=maxSlices,proto3" json:"max_slices,omitempty"`
	// The minimum tensor size in bytes that we will attempt to slice.
	MinBytes uint64 `protobuf:"varint,2,opt,name=min_bytes,json=minBytes,proto3" json:"min_bytes,omitempty"`
	// This option should never be set to true in production. When this is true,
	// we will crash if we propose a slice (other than the final slice) with a
	// size that is not a multiple of the required hardware alignment. Otherwise,
	// we will choose not to slice such situations, which is always safe.
	FailOnNonAlignmentBoundarySliceProposal bool `` /* 189-byte string literal not displayed */
	// contains filtered or unexported fields
}

Memory space assignment options for slicing prefetches into smaller asynchronous copies, reducing prefetch memory allocation pressure.

No prefetch slicing is performed if max_slices == 0.

TODO(b/275905276): Consider adding another option that indicates that we want slices of a certain size, rather than just always creating max_slices.

func (*SlicedPrefetchOptions) Descriptor deprecated added in v2.13.0 

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

Deprecated: Use SlicedPrefetchOptions.ProtoReflect.Descriptor instead.

func (*SlicedPrefetchOptions) GetFailOnNonAlignmentBoundarySliceProposal added in v2.14.0 

func (x *SlicedPrefetchOptions) GetFailOnNonAlignmentBoundarySliceProposal() bool

func (*SlicedPrefetchOptions) GetMaxSlices added in v2.13.0 

func (x *SlicedPrefetchOptions) GetMaxSlices() uint32

func (*SlicedPrefetchOptions) GetMinBytes added in v2.14.0 

func (x *SlicedPrefetchOptions) GetMinBytes() uint64

func (*SlicedPrefetchOptions) ProtoMessage added in v2.13.0 

func (*SlicedPrefetchOptions) ProtoMessage()

func (*SlicedPrefetchOptions) ProtoReflect added in v2.13.0 

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

func (*SlicedPrefetchOptions) Reset added in v2.13.0 

func (x *SlicedPrefetchOptions) Reset()

func (*SlicedPrefetchOptions) String added in v2.13.0 

func (x *SlicedPrefetchOptions) String() string

type StackFrameIndexProto added in v2.14.0 

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 added in v2.14.0 

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

Deprecated: Use StackFrameIndexProto.ProtoReflect.Descriptor instead.

func (*StackFrameIndexProto) GetFileLocations added in v2.14.0 

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

func (*StackFrameIndexProto) GetFileNames added in v2.14.0 

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

func (*StackFrameIndexProto) GetFunctionNames added in v2.14.0 

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

func (*StackFrameIndexProto) GetStackFrames added in v2.14.0 

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

func (*StackFrameIndexProto) ProtoMessage added in v2.14.0 

func (*StackFrameIndexProto) ProtoMessage()

func (*StackFrameIndexProto) ProtoReflect added in v2.14.0 

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

func (*StackFrameIndexProto) Reset added in v2.14.0 

func (x *StackFrameIndexProto) Reset()

func (*StackFrameIndexProto) String added in v2.14.0 

func (x *StackFrameIndexProto) String() string

type StackFrameIndexProto_FileLocation added in v2.14.0 

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 added in v2.14.0 

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

Deprecated: Use StackFrameIndexProto_FileLocation.ProtoReflect.Descriptor instead.

func (*StackFrameIndexProto_FileLocation) GetColumn added in v2.14.0 

func (x *StackFrameIndexProto_FileLocation) GetColumn() int32

func (*StackFrameIndexProto_FileLocation) GetFileNameId added in v2.14.0 

func (x *StackFrameIndexProto_FileLocation) GetFileNameId() int32

func (*StackFrameIndexProto_FileLocation) GetFunctionNameId added in v2.14.0 

func (x *StackFrameIndexProto_FileLocation) GetFunctionNameId() int32

func (*StackFrameIndexProto_FileLocation) GetLine added in v2.14.0 

func (x *StackFrameIndexProto_FileLocation) GetLine() int32

func (*StackFrameIndexProto_FileLocation) ProtoMessage added in v2.14.0 

func (*StackFrameIndexProto_FileLocation) ProtoMessage()

func (*StackFrameIndexProto_FileLocation) ProtoReflect added in v2.14.0 

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

func (*StackFrameIndexProto_FileLocation) Reset added in v2.14.0 

func (x *StackFrameIndexProto_FileLocation) Reset()

func (*StackFrameIndexProto_FileLocation) String added in v2.14.0 

func (x *StackFrameIndexProto_FileLocation) String() string

type StackFrameIndexProto_StackFrame added in v2.14.0 

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 added in v2.14.0 

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

Deprecated: Use StackFrameIndexProto_StackFrame.ProtoReflect.Descriptor instead.

func (*StackFrameIndexProto_StackFrame) GetFileLocationId added in v2.14.0 

func (x *StackFrameIndexProto_StackFrame) GetFileLocationId() int32

func (*StackFrameIndexProto_StackFrame) GetParentFrameId added in v2.14.0 

func (x *StackFrameIndexProto_StackFrame) GetParentFrameId() int32

func (*StackFrameIndexProto_StackFrame) ProtoMessage added in v2.14.0 

func (*StackFrameIndexProto_StackFrame) ProtoMessage()

func (*StackFrameIndexProto_StackFrame) ProtoReflect added in v2.14.0 

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

func (*StackFrameIndexProto_StackFrame) Reset added in v2.14.0 

func (x *StackFrameIndexProto_StackFrame) Reset()

func (*StackFrameIndexProto_StackFrame) String added in v2.14.0 

func (x *StackFrameIndexProto_StackFrame) String() string

type TestCompilationEnvironment1 

type TestCompilationEnvironment1 struct {
	SomeFlag uint32 `protobuf:"varint,1,opt,name=some_flag,json=someFlag,proto3" json:"some_flag,omitempty"`
	// contains filtered or unexported fields
}

func (*TestCompilationEnvironment1) Descriptor deprecated 

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

Deprecated: Use TestCompilationEnvironment1.ProtoReflect.Descriptor instead.

func (*TestCompilationEnvironment1) GetSomeFlag 

func (x *TestCompilationEnvironment1) GetSomeFlag() uint32

func (*TestCompilationEnvironment1) ProtoMessage 

func (*TestCompilationEnvironment1) ProtoMessage()

func (*TestCompilationEnvironment1) ProtoReflect 

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

func (*TestCompilationEnvironment1) Reset 

func (x *TestCompilationEnvironment1) Reset()

func (*TestCompilationEnvironment1) String 

func (x *TestCompilationEnvironment1) String() string

type TestCompilationEnvironment2 

type TestCompilationEnvironment2 struct {
	SomeOtherFlag uint32 `protobuf:"varint,1,opt,name=some_other_flag,json=someOtherFlag,proto3" json:"some_other_flag,omitempty"`
	// contains filtered or unexported fields
}

func (*TestCompilationEnvironment2) Descriptor deprecated 

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

Deprecated: Use TestCompilationEnvironment2.ProtoReflect.Descriptor instead.

func (*TestCompilationEnvironment2) GetSomeOtherFlag 

func (x *TestCompilationEnvironment2) GetSomeOtherFlag() uint32

func (*TestCompilationEnvironment2) ProtoMessage 

func (*TestCompilationEnvironment2) ProtoMessage()

func (*TestCompilationEnvironment2) ProtoReflect 

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

func (*TestCompilationEnvironment2) Reset 

func (x *TestCompilationEnvironment2) Reset()

func (*TestCompilationEnvironment2) String 

func (x *TestCompilationEnvironment2) String() string

type TestCompilationEnvironment3 

type TestCompilationEnvironment3 struct {
	AThirdFlag uint32 `protobuf:"varint,1,opt,name=a_third_flag,json=aThirdFlag,proto3" json:"a_third_flag,omitempty"`
	// contains filtered or unexported fields
}

func (*TestCompilationEnvironment3) Descriptor deprecated 

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

Deprecated: Use TestCompilationEnvironment3.ProtoReflect.Descriptor instead.

func (*TestCompilationEnvironment3) GetAThirdFlag 

func (x *TestCompilationEnvironment3) GetAThirdFlag() uint32

func (*TestCompilationEnvironment3) ProtoMessage 

func (*TestCompilationEnvironment3) ProtoMessage()

func (*TestCompilationEnvironment3) ProtoReflect 

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

func (*TestCompilationEnvironment3) Reset 

func (x *TestCompilationEnvironment3) Reset()

func (*TestCompilationEnvironment3) String 

func (x *TestCompilationEnvironment3) String() string

type XlaRuntimeExecutableProto added in v2.11.0 

type XlaRuntimeExecutableProto struct {
	HloModuleProto *HloModuleProto `protobuf:"bytes,1,opt,name=hlo_module_proto,json=hloModuleProto,proto3" json:"hlo_module_proto,omitempty"`
	// Serialized object file compiled from the XLA module.
	ObjFile []byte `protobuf:"bytes,3,opt,name=obj_file,json=objFile,proto3" json:"obj_file,omitempty"`
	// Serialized MLIR module corresponding to compiled object file.
	MlirModule string `protobuf:"bytes,4,opt,name=mlir_module,json=mlirModule,proto3" json:"mlir_module,omitempty"`
	// contains filtered or unexported fields
}

Encodes the underlying Xla runtime executable compiled from the XLA module.

func (*XlaRuntimeExecutableProto) Descriptor deprecated added in v2.11.0 

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

Deprecated: Use XlaRuntimeExecutableProto.ProtoReflect.Descriptor instead.

func (*XlaRuntimeExecutableProto) GetHloModuleProto added in v2.11.0 

func (x *XlaRuntimeExecutableProto) GetHloModuleProto() *HloModuleProto

func (*XlaRuntimeExecutableProto) GetMlirModule added in v2.11.0 

func (x *XlaRuntimeExecutableProto) GetMlirModule() string

func (*XlaRuntimeExecutableProto) GetObjFile added in v2.11.0 

func (x *XlaRuntimeExecutableProto) GetObjFile() []byte

func (*XlaRuntimeExecutableProto) ProtoMessage added in v2.11.0 

func (*XlaRuntimeExecutableProto) ProtoMessage()

func (*XlaRuntimeExecutableProto) ProtoReflect added in v2.11.0 

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

func (*XlaRuntimeExecutableProto) Reset added in v2.11.0 

func (x *XlaRuntimeExecutableProto) Reset()

func (*XlaRuntimeExecutableProto) String added in v2.11.0 

func (x *XlaRuntimeExecutableProto) String() string

Source Files

View all Source files

Directories

Path Synopsis

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.