service

package
v2.11.0 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: 5 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",
	}
	CustomCallApiVersion_value = map[string]int32{
		"API_VERSION_UNSPECIFIED":              0,
		"API_VERSION_ORIGINAL":                 1,
		"API_VERSION_STATUS_RETURNING":         2,
		"API_VERSION_STATUS_RETURNING_UNIFIED": 3,
	}
)

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

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_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 = 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 CustomCallOutputOperandAliasing pairs that specifies aliasing
	// buffers between output and operands for kCustomCall.
	CustomCallOutputOperandAliasing []*data.CustomCallOutputOperandAliasing `` /* 161-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.
	IsCrossProgramPrefetch bool `` /* 133-byte string literal not displayed */
	// 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"`
	// contains filtered or unexported fields
}

Serialization of HloInstruction. Next ID: 80

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) GetCustomCallApiVersion 

func (x *HloInstructionProto) GetCustomCallApiVersion() CustomCallApiVersion

func (*HloInstructionProto) GetCustomCallHasSideEffect 

func (x *HloInstructionProto) GetCustomCallHasSideEffect() bool

func (*HloInstructionProto) GetCustomCallOutputOperandAliasing 

func (x *HloInstructionProto) GetCustomCallOutputOperandAliasing() []*data.CustomCallOutputOperandAliasing

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 

func (x *HloInstructionProto) GetIsCrossProgramPrefetch() bool

func (*HloInstructionProto) GetIsHostTransfer 

func (x *HloInstructionProto) GetIsHostTransfer() bool

func (*HloInstructionProto) GetIsStable 

func (x *HloInstructionProto) GetIsStable() bool

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) GetOutfeedConfig 

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

func (*HloInstructionProto) GetOutfeedShape 

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

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

	// NOTE: module_name isn't necessary, since all LogicalBuffers are
	// associated with a single HloModule.
	// TODO(b/239098765): Remove instruction_name and computation_name.
	//
	// Deprecated: Marked as deprecated in tensorflow/compiler/xla/service/hlo.proto.
	ComputationName string `protobuf:"bytes,1,opt,name=computation_name,json=computationName,proto3" json:"computation_name,omitempty"`
	// 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) GetComputationName deprecated 

func (x *LogicalBufferProto_Location) GetComputationName() string

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

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 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"`
	// XLA-specific attributes of the executable's entry function.
	EntryFuncAttrs *EntryFunctionAttributes `protobuf:"bytes,2,opt,name=entry_func_attrs,json=entryFuncAttrs,proto3" json:"entry_func_attrs,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) GetEntryFuncAttrs added in v2.11.0 

func (x *XlaRuntimeExecutableProto) GetEntryFuncAttrs() *EntryFunctionAttributes

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.