public

package
v2.14.1 Latest Latest
Warning

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

 Go to latest Published: Dec 8, 2023 License: BSD-3-Clause Imports: 4 Imported by: 0

Documentation

Index

Constants

This section is empty.

Variables

View Source 
var (
	TracedValue_ElementType_name = map[int32]string{
		0: "UNKNOWN",
		1: "INTEGRAL",
		2: "UNSIGNED",
		3: "FLOAT",
		4: "COMPLEX",
		5: "TUPLE",
	}
	TracedValue_ElementType_value = map[string]int32{
		"UNKNOWN":  0,
		"INTEGRAL": 1,
		"UNSIGNED": 2,
		"FLOAT":    3,
		"COMPLEX":  4,
		"TUPLE":    5,
	}
)

Enum value maps for TracedValue_ElementType.

View Source 
var File_tensorflow_compiler_xla_mlir_tools_mlir_replay_public_compiler_trace_proto protoreflect.FileDescriptor
View Source 
var File_tensorflow_compiler_xla_mlir_tools_mlir_replay_public_execution_trace_proto protoreflect.FileDescriptor

Functions

This section is empty.

Types

type ExecutionTrace 

type ExecutionTrace struct {

	// The IR that was executed. Note: this should always be filled in the generic
	// format.
	Ir *string `protobuf:"bytes,1,opt,name=ir" json:"ir,omitempty"`
	// The trace of the entry function execution.
	Trace *RegionTrace `protobuf:"bytes,2,opt,name=trace" json:"trace,omitempty"`
	// contains filtered or unexported fields
}

func (*ExecutionTrace) Descriptor deprecated 

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

Deprecated: Use ExecutionTrace.ProtoReflect.Descriptor instead.

func (*ExecutionTrace) GetIr 

func (x *ExecutionTrace) GetIr() string

func (*ExecutionTrace) GetTrace 

func (x *ExecutionTrace) GetTrace() *RegionTrace

func (*ExecutionTrace) ProtoMessage 

func (*ExecutionTrace) ProtoMessage()

func (*ExecutionTrace) ProtoReflect 

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

func (*ExecutionTrace) Reset 

func (x *ExecutionTrace) Reset()

func (*ExecutionTrace) String 

func (x *ExecutionTrace) String() string

type InstructionTrace 

type InstructionTrace struct {
	Name    *string        `protobuf:"bytes,1,opt,name=name" json:"name,omitempty"`
	Args    []*TracedValue `protobuf:"bytes,2,rep,name=args" json:"args,omitempty"`
	Results []*TracedValue `protobuf:"bytes,3,rep,name=results" json:"results,omitempty"` // TODO(jreiffers): Model side effects (e.g. memref.store).
	Regions []*RegionTrace `protobuf:"bytes,4,rep,name=regions" json:"regions,omitempty"`
	// contains filtered or unexported fields
}

func (*InstructionTrace) Descriptor deprecated 

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

Deprecated: Use InstructionTrace.ProtoReflect.Descriptor instead.

func (*InstructionTrace) GetArgs 

func (x *InstructionTrace) GetArgs() []*TracedValue

func (*InstructionTrace) GetName 

func (x *InstructionTrace) GetName() string

func (*InstructionTrace) GetRegions 

func (x *InstructionTrace) GetRegions() []*RegionTrace

func (*InstructionTrace) GetResults 

func (x *InstructionTrace) GetResults() []*TracedValue

func (*InstructionTrace) ProtoMessage 

func (*InstructionTrace) ProtoMessage()

func (*InstructionTrace) ProtoReflect 

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

func (*InstructionTrace) Reset 

func (x *InstructionTrace) Reset()

func (*InstructionTrace) String 

func (x *InstructionTrace) String() string

type MlirCompilationTrace 

type MlirCompilationTrace struct {

	// MLIR modules corresponding to each stage of the compilation pipeline.
	Passes []*MlirCompilationTraceEntry `protobuf:"bytes,1,rep,name=passes" json:"passes,omitempty"`
	// contains filtered or unexported fields
}

func (*MlirCompilationTrace) Descriptor deprecated 

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

Deprecated: Use MlirCompilationTrace.ProtoReflect.Descriptor instead.

func (*MlirCompilationTrace) GetPasses 

func (x *MlirCompilationTrace) GetPasses() []*MlirCompilationTraceEntry

func (*MlirCompilationTrace) ProtoMessage 

func (*MlirCompilationTrace) ProtoMessage()

func (*MlirCompilationTrace) ProtoReflect 

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

func (*MlirCompilationTrace) Reset 

func (x *MlirCompilationTrace) Reset()

func (*MlirCompilationTrace) String 

func (x *MlirCompilationTrace) String() string

type MlirCompilationTraceEntry 

type MlirCompilationTraceEntry struct {

	// The name of the pass that was previously executed.
	AfterPass *string `protobuf:"bytes,1,opt,name=after_pass,json=afterPass" json:"after_pass,omitempty"`
	// MLIR module IR of the state after the pass.
	MlirModule *string `protobuf:"bytes,2,opt,name=mlir_module,json=mlirModule" json:"mlir_module,omitempty"`
	// contains filtered or unexported fields
}

func (*MlirCompilationTraceEntry) Descriptor deprecated 

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

Deprecated: Use MlirCompilationTraceEntry.ProtoReflect.Descriptor instead.

func (*MlirCompilationTraceEntry) GetAfterPass 

func (x *MlirCompilationTraceEntry) GetAfterPass() string

func (*MlirCompilationTraceEntry) GetMlirModule 

func (x *MlirCompilationTraceEntry) GetMlirModule() string

func (*MlirCompilationTraceEntry) ProtoMessage 

func (*MlirCompilationTraceEntry) ProtoMessage()

func (*MlirCompilationTraceEntry) ProtoReflect 

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

func (*MlirCompilationTraceEntry) Reset 

func (x *MlirCompilationTraceEntry) Reset()

func (*MlirCompilationTraceEntry) String 

func (x *MlirCompilationTraceEntry) String() string

type RegionTrace 

type RegionTrace struct {

	// The number of the region that is being executed (within the parent op).
	// For example: '1' for an scf.while's `after` region.
	RegionNumber *int32 `protobuf:"varint,1,opt,name=region_number,json=regionNumber" json:"region_number,omitempty"`
	// The arguments that were passed to the region.
	Bbargs []*TracedValue `protobuf:"bytes,2,rep,name=bbargs" json:"bbargs,omitempty"`
	// One instruction per instruction in the region.
	Instructions []*InstructionTrace `protobuf:"bytes,3,rep,name=instructions" json:"instructions,omitempty"`
	Results      []*TracedValue      `protobuf:"bytes,4,rep,name=results" json:"results,omitempty"`
	// contains filtered or unexported fields
}

func (*RegionTrace) Descriptor deprecated 

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

Deprecated: Use RegionTrace.ProtoReflect.Descriptor instead.

func (*RegionTrace) GetBbargs 

func (x *RegionTrace) GetBbargs() []*TracedValue

func (*RegionTrace) GetInstructions 

func (x *RegionTrace) GetInstructions() []*InstructionTrace

func (*RegionTrace) GetRegionNumber 

func (x *RegionTrace) GetRegionNumber() int32

func (*RegionTrace) GetResults 

func (x *RegionTrace) GetResults() []*TracedValue

func (*RegionTrace) ProtoMessage 

func (*RegionTrace) ProtoMessage()

func (*RegionTrace) ProtoReflect 

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

func (*RegionTrace) Reset 

func (x *RegionTrace) Reset()

func (*RegionTrace) String 

func (x *RegionTrace) String() string

type TracedValue 

type TracedValue struct {

	// The shape - includes vector dimensions.
	// TODO(jreiffers): Model vector dimensions separately.
	Shape         []int64                  `protobuf:"varint,1,rep,name=shape" json:"shape,omitempty"`
	IsScalar      *bool                    `protobuf:"varint,2,opt,name=is_scalar,json=isScalar" json:"is_scalar,omitempty"`
	BitWidth      *int32                   `protobuf:"varint,3,opt,name=bit_width,json=bitWidth" json:"bit_width,omitempty"`
	ElementType   *TracedValue_ElementType `` /* 134-byte string literal not displayed */
	Floats        []float32                `protobuf:"fixed32,5,rep,packed,name=floats" json:"floats,omitempty"`
	Doubles       []float64                `protobuf:"fixed64,6,rep,packed,name=doubles" json:"doubles,omitempty"`
	Ints          []int64                  `protobuf:"varint,7,rep,packed,name=ints" json:"ints,omitempty"`
	Uints         []uint64                 `protobuf:"varint,8,rep,packed,name=uints" json:"uints,omitempty"`
	TupleElements []*TracedValue           `protobuf:"bytes,9,rep,name=tuple_elements,json=tupleElements" json:"tuple_elements,omitempty"`
	// contains filtered or unexported fields
}

func (*TracedValue) Descriptor deprecated 

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

Deprecated: Use TracedValue.ProtoReflect.Descriptor instead.

func (*TracedValue) GetBitWidth 

func (x *TracedValue) GetBitWidth() int32

func (*TracedValue) GetDoubles 

func (x *TracedValue) GetDoubles() []float64

func (*TracedValue) GetElementType 

func (x *TracedValue) GetElementType() TracedValue_ElementType

func (*TracedValue) GetFloats 

func (x *TracedValue) GetFloats() []float32

func (*TracedValue) GetInts 

func (x *TracedValue) GetInts() []int64

func (*TracedValue) GetIsScalar 

func (x *TracedValue) GetIsScalar() bool

func (*TracedValue) GetShape 

func (x *TracedValue) GetShape() []int64

func (*TracedValue) GetTupleElements 

func (x *TracedValue) GetTupleElements() []*TracedValue

func (*TracedValue) GetUints 

func (x *TracedValue) GetUints() []uint64

func (*TracedValue) ProtoMessage 

func (*TracedValue) ProtoMessage()

func (*TracedValue) ProtoReflect 

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

func (*TracedValue) Reset 

func (x *TracedValue) Reset()

func (*TracedValue) String 

func (x *TracedValue) String() string

type TracedValue_ElementType 

type TracedValue_ElementType int32
const (
	TracedValue_UNKNOWN  TracedValue_ElementType = 0
	TracedValue_INTEGRAL TracedValue_ElementType = 1
	TracedValue_UNSIGNED TracedValue_ElementType = 2
	TracedValue_FLOAT    TracedValue_ElementType = 3
	TracedValue_COMPLEX  TracedValue_ElementType = 4
	TracedValue_TUPLE    TracedValue_ElementType = 5
)

func (TracedValue_ElementType) Descriptor 

func (TracedValue_ElementType) Descriptor() protoreflect.EnumDescriptor

func (TracedValue_ElementType) Enum 

func (x TracedValue_ElementType) Enum() *TracedValue_ElementType

func (TracedValue_ElementType) EnumDescriptor deprecated 

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

Deprecated: Use TracedValue_ElementType.Descriptor instead.

func (TracedValue_ElementType) Number 

func (x TracedValue_ElementType) Number() protoreflect.EnumNumber

func (TracedValue_ElementType) String 

func (x TracedValue_ElementType) String() string

func (TracedValue_ElementType) Type 

func (TracedValue_ElementType) Type() protoreflect.EnumType

func (*TracedValue_ElementType) UnmarshalJSON deprecated 

func (x *TracedValue_ElementType) UnmarshalJSON(b []byte) error

Deprecated: Do not use.

Source Files

View all Source files

Jump to

Keyboard shortcuts

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