tpu

package
v2.12.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 File_tensorflow_python_tpu_tensor_tracer_proto protoreflect.FileDescriptor

Functions

This section is empty.

Types

type TensorTracerReport 

type TensorTracerReport struct {
	Config *TensorTracerReport_TensorTracerConfig `protobuf:"bytes,1,opt,name=config,proto3" json:"config,omitempty"`
	// Tensorflow graph.
	Graphdef *framework.GraphDef `protobuf:"bytes,2,opt,name=graphdef,proto3" json:"graphdef,omitempty"`
	// A map from tensor name to its TracedTensorDef.
	Tensordef map[string]*TensorTracerReport_TracedTensorDef `` /* 159-byte string literal not displayed */
	// The fingerprint of the TensorTracerReport (fingerprint calculation excludes
	// this field and graphdef).
	Fingerprint string `protobuf:"bytes,4,opt,name=fingerprint,proto3" json:"fingerprint,omitempty"`
	// The function_name passed to the function_callback
	// that produced this TensorTracerReport
	ConcreteFunctionName string `protobuf:"bytes,5,opt,name=concrete_function_name,json=concreteFunctionName,proto3" json:"concrete_function_name,omitempty"`
	// The index of the last stack frame where the stack traces for all output
	// operations in the graph have the same value.
	LastCommonFrameNo int32 `protobuf:"varint,6,opt,name=last_common_frame_no,json=lastCommonFrameNo,proto3" json:"last_common_frame_no,omitempty"`
	// List of names of output tensors of the function being traced.
	Outputs []string `protobuf:"bytes,7,rep,name=outputs,proto3" json:"outputs,omitempty"`
	// Information about the number of tensors traced and skipped.
	TracingStats *TensorTracerReport_TracingStats `protobuf:"bytes,8,opt,name=tracing_stats,json=tracingStats,proto3" json:"tracing_stats,omitempty"`
	// contains filtered or unexported fields
}

Tensor Tracer Report proto gives information about the trace including: - TensorTracerConfig: version, device, num replicas, trace mode. - Graphdef, e.g., list of operations, tensors - TracedTensorDef:

  • Name of the tensor
  • Tracepoint name if provided.
  • Index of the tensor in the compact cache if traced.
  • Explanation for why the tensor is traced or not.

func (*TensorTracerReport) Descriptor deprecated 

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

Deprecated: Use TensorTracerReport.ProtoReflect.Descriptor instead.

func (*TensorTracerReport) GetConcreteFunctionName 

func (x *TensorTracerReport) GetConcreteFunctionName() string

func (*TensorTracerReport) GetConfig 

func (x *TensorTracerReport) GetConfig() *TensorTracerReport_TensorTracerConfig

func (*TensorTracerReport) GetFingerprint 

func (x *TensorTracerReport) GetFingerprint() string

func (*TensorTracerReport) GetGraphdef 

func (x *TensorTracerReport) GetGraphdef() *framework.GraphDef

func (*TensorTracerReport) GetLastCommonFrameNo 

func (x *TensorTracerReport) GetLastCommonFrameNo() int32

func (*TensorTracerReport) GetOutputs 

func (x *TensorTracerReport) GetOutputs() []string

func (*TensorTracerReport) GetTensordef 

func (x *TensorTracerReport) GetTensordef() map[string]*TensorTracerReport_TracedTensorDef

func (*TensorTracerReport) GetTracingStats 

func (x *TensorTracerReport) GetTracingStats() *TensorTracerReport_TracingStats

func (*TensorTracerReport) ProtoMessage 

func (*TensorTracerReport) ProtoMessage()

func (*TensorTracerReport) ProtoReflect 

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

func (*TensorTracerReport) Reset 

func (x *TensorTracerReport) Reset()

func (*TensorTracerReport) String 

func (x *TensorTracerReport) String() string

type TensorTracerReport_TensorTracerConfig 

type TensorTracerReport_TensorTracerConfig struct {

	// Tensor tracer version, e.g. hostcall, outside compilation.
	Version string `protobuf:"bytes,1,opt,name=version,proto3" json:"version,omitempty"`
	// Traced device, CPU, TPU...
	Device string `protobuf:"bytes,2,opt,name=device,proto3" json:"device,omitempty"`
	// Trace mode, norm, summary, full-trace.
	TraceMode string `protobuf:"bytes,3,opt,name=trace_mode,json=traceMode,proto3" json:"trace_mode,omitempty"`
	// Number of cores, e.g. TPU cores, in the system.
	NumCores int32 `protobuf:"varint,4,opt,name=num_cores,json=numCores,proto3" json:"num_cores,omitempty"`
	// Number of hosts, e.g. compute nodes in the system.
	NumHosts int32 `protobuf:"varint,5,opt,name=num_hosts,json=numHosts,proto3" json:"num_hosts,omitempty"`
	// Keep submode as string for backward compatibility.
	Submode string `protobuf:"bytes,6,opt,name=submode,proto3" json:"submode,omitempty"`
	// Keep num cores per host for backward compatibility.
	NumCoresPerHost int32 `protobuf:"varint,7,opt,name=num_cores_per_host,json=numCoresPerHost,proto3" json:"num_cores_per_host,omitempty"`
	// Id of the included cores, if a subset of cores are traced.
	IncludedCores []int32 `protobuf:"varint,8,rep,packed,name=included_cores,json=includedCores,proto3" json:"included_cores,omitempty"`
	// The names of the signatures corresponding to the cache indices.
	Signatures []string `protobuf:"bytes,9,rep,name=signatures,proto3" json:"signatures,omitempty"`
	// contains filtered or unexported fields
}

func (*TensorTracerReport_TensorTracerConfig) Descriptor deprecated 

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

Deprecated: Use TensorTracerReport_TensorTracerConfig.ProtoReflect.Descriptor instead.

func (*TensorTracerReport_TensorTracerConfig) GetDevice 

func (x *TensorTracerReport_TensorTracerConfig) GetDevice() string

func (*TensorTracerReport_TensorTracerConfig) GetIncludedCores 

func (x *TensorTracerReport_TensorTracerConfig) GetIncludedCores() []int32

func (*TensorTracerReport_TensorTracerConfig) GetNumCores 

func (x *TensorTracerReport_TensorTracerConfig) GetNumCores() int32

func (*TensorTracerReport_TensorTracerConfig) GetNumCoresPerHost 

func (x *TensorTracerReport_TensorTracerConfig) GetNumCoresPerHost() int32

func (*TensorTracerReport_TensorTracerConfig) GetNumHosts 

func (x *TensorTracerReport_TensorTracerConfig) GetNumHosts() int32

func (*TensorTracerReport_TensorTracerConfig) GetSignatures 

func (x *TensorTracerReport_TensorTracerConfig) GetSignatures() []string

func (*TensorTracerReport_TensorTracerConfig) GetSubmode 

func (x *TensorTracerReport_TensorTracerConfig) GetSubmode() string

func (*TensorTracerReport_TensorTracerConfig) GetTraceMode 

func (x *TensorTracerReport_TensorTracerConfig) GetTraceMode() string

func (*TensorTracerReport_TensorTracerConfig) GetVersion 

func (x *TensorTracerReport_TensorTracerConfig) GetVersion() string

func (*TensorTracerReport_TensorTracerConfig) ProtoMessage 

func (*TensorTracerReport_TensorTracerConfig) ProtoMessage()

func (*TensorTracerReport_TensorTracerConfig) ProtoReflect 

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

func (*TensorTracerReport_TensorTracerConfig) Reset 

func (x *TensorTracerReport_TensorTracerConfig) Reset()

func (*TensorTracerReport_TensorTracerConfig) String 

func (x *TensorTracerReport_TensorTracerConfig) String() string

type TensorTracerReport_TracedTensorDef 

type TensorTracerReport_TracedTensorDef struct {

	// Name of the tensor as appears in tf graph.
	Name string `protobuf:"bytes,1,opt,name=name,proto3" json:"name,omitempty"`
	// Cache index of the tensor. This may be different than topological index.
	CacheIndex int32 `protobuf:"varint,2,opt,name=cache_index,json=cacheIndex,proto3" json:"cache_index,omitempty"`
	// If trace points are provided, corresponding tracepoint name of the
	// tensor. Trace points are placed on the edges (tensors) in the tensorflow
	// graph, and they force tensor tracer to trace the corresponding tensor.
	// Tracepoints can be added using the programatic interface
	// tensor_tracer.tensor_tracepoint(tensor, trace_point_name) function.
	// This will add a trace point with the given trace_point_name for the given
	// tensor. If a trace_point is provided for the tensor,
	// trace_point name will be used for the rest of the analysis instead of
	// tensor names. One can use trace_point_name's to compare two models with
	// arbitrary tensor names by providing the same trace point name for the
	// tensors that are comparable.
	TracePointName string `protobuf:"bytes,3,opt,name=trace_point_name,json=tracePointName,proto3" json:"trace_point_name,omitempty"`
	// Whether the tensor is traced or not.
	IsTraced bool `protobuf:"varint,4,opt,name=is_traced,json=isTraced,proto3" json:"is_traced,omitempty"`
	// Detailed explanation why the tensor is traced or not.
	Explanation string `protobuf:"bytes,5,opt,name=explanation,proto3" json:"explanation,omitempty"`
	// Detailed stack of operation
	OpStackInfo *TensorTracerReport_TracedTensorDef_Stack `protobuf:"bytes,6,opt,name=op_stack_info,json=opStackInfo,proto3" json:"op_stack_info,omitempty"`
	// contains filtered or unexported fields
}

func (*TensorTracerReport_TracedTensorDef) Descriptor deprecated 

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

Deprecated: Use TensorTracerReport_TracedTensorDef.ProtoReflect.Descriptor instead.

func (*TensorTracerReport_TracedTensorDef) GetCacheIndex 

func (x *TensorTracerReport_TracedTensorDef) GetCacheIndex() int32

func (*TensorTracerReport_TracedTensorDef) GetExplanation 

func (x *TensorTracerReport_TracedTensorDef) GetExplanation() string

func (*TensorTracerReport_TracedTensorDef) GetIsTraced 

func (x *TensorTracerReport_TracedTensorDef) GetIsTraced() bool

func (*TensorTracerReport_TracedTensorDef) GetName 

func (x *TensorTracerReport_TracedTensorDef) GetName() string

func (*TensorTracerReport_TracedTensorDef) GetOpStackInfo 

func (x *TensorTracerReport_TracedTensorDef) GetOpStackInfo() *TensorTracerReport_TracedTensorDef_Stack

func (*TensorTracerReport_TracedTensorDef) GetTracePointName 

func (x *TensorTracerReport_TracedTensorDef) GetTracePointName() string

func (*TensorTracerReport_TracedTensorDef) ProtoMessage 

func (*TensorTracerReport_TracedTensorDef) ProtoMessage()

func (*TensorTracerReport_TracedTensorDef) ProtoReflect 

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

func (*TensorTracerReport_TracedTensorDef) Reset 

func (x *TensorTracerReport_TracedTensorDef) Reset()

func (*TensorTracerReport_TracedTensorDef) String 

func (x *TensorTracerReport_TracedTensorDef) String() string

type TensorTracerReport_TracedTensorDef_Stack 

type TensorTracerReport_TracedTensorDef_Stack struct {

	// Function names from stack
	StackFnNames []string `protobuf:"bytes,1,rep,name=stack_fn_names,json=stackFnNames,proto3" json:"stack_fn_names,omitempty"`
	// Line in stack
	StackLines []string `protobuf:"bytes,2,rep,name=stack_lines,json=stackLines,proto3" json:"stack_lines,omitempty"`
	// Filenames from stack
	StackFilenames []string `protobuf:"bytes,3,rep,name=stack_filenames,json=stackFilenames,proto3" json:"stack_filenames,omitempty"`
	// Line number in file from stack
	StackLinenos []int32 `protobuf:"varint,4,rep,packed,name=stack_linenos,json=stackLinenos,proto3" json:"stack_linenos,omitempty"`
	// contains filtered or unexported fields
}

func (*TensorTracerReport_TracedTensorDef_Stack) Descriptor deprecated 

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

Deprecated: Use TensorTracerReport_TracedTensorDef_Stack.ProtoReflect.Descriptor instead.

func (*TensorTracerReport_TracedTensorDef_Stack) GetStackFilenames 

func (x *TensorTracerReport_TracedTensorDef_Stack) GetStackFilenames() []string

func (*TensorTracerReport_TracedTensorDef_Stack) GetStackFnNames 

func (x *TensorTracerReport_TracedTensorDef_Stack) GetStackFnNames() []string

func (*TensorTracerReport_TracedTensorDef_Stack) GetStackLinenos 

func (x *TensorTracerReport_TracedTensorDef_Stack) GetStackLinenos() []int32

func (*TensorTracerReport_TracedTensorDef_Stack) GetStackLines 

func (x *TensorTracerReport_TracedTensorDef_Stack) GetStackLines() []string

func (*TensorTracerReport_TracedTensorDef_Stack) ProtoMessage 

func (*TensorTracerReport_TracedTensorDef_Stack) ProtoMessage()

func (*TensorTracerReport_TracedTensorDef_Stack) ProtoReflect 

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

func (*TensorTracerReport_TracedTensorDef_Stack) Reset 

func (x *TensorTracerReport_TracedTensorDef_Stack) Reset()

func (*TensorTracerReport_TracedTensorDef_Stack) String 

func (x *TensorTracerReport_TracedTensorDef_Stack) String() string

type TensorTracerReport_TracingStats 

type TensorTracerReport_TracingStats struct {

	// The total number of tensors in the function.
	TotalTensors int32 `protobuf:"varint,1,opt,name=total_tensors,json=totalTensors,proto3" json:"total_tensors,omitempty"`
	// The number of traced tensors in the function.
	TracedTensors int32 `protobuf:"varint,2,opt,name=traced_tensors,json=tracedTensors,proto3" json:"traced_tensors,omitempty"`
	// Counts of traced tensors by op type.
	TracedTensorTypes map[string]int32 `` /* 203-byte string literal not displayed */
	// The number of tensors added by Tensor Tracer.
	AddedTensors int32 `protobuf:"varint,4,opt,name=added_tensors,json=addedTensors,proto3" json:"added_tensors,omitempty"`
	// contains filtered or unexported fields
}

func (*TensorTracerReport_TracingStats) Descriptor deprecated 

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

Deprecated: Use TensorTracerReport_TracingStats.ProtoReflect.Descriptor instead.

func (*TensorTracerReport_TracingStats) GetAddedTensors 

func (x *TensorTracerReport_TracingStats) GetAddedTensors() int32

func (*TensorTracerReport_TracingStats) GetTotalTensors 

func (x *TensorTracerReport_TracingStats) GetTotalTensors() int32

func (*TensorTracerReport_TracingStats) GetTracedTensorTypes 

func (x *TensorTracerReport_TracingStats) GetTracedTensorTypes() map[string]int32

func (*TensorTracerReport_TracingStats) GetTracedTensors 

func (x *TensorTracerReport_TracingStats) GetTracedTensors() int32

func (*TensorTracerReport_TracingStats) ProtoMessage 

func (*TensorTracerReport_TracingStats) ProtoMessage()

func (*TensorTracerReport_TracingStats) ProtoReflect 

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

func (*TensorTracerReport_TracingStats) Reset 

func (x *TensorTracerReport_TracingStats) Reset()

func (*TensorTracerReport_TracingStats) String 

func (x *TensorTracerReport_TracingStats) String() string

Source Files

View all Source files

Jump to

Keyboard shortcuts

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