step_stats_go_proto

package
v0.0.0-...-88323db Latest Latest
Warning

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

 Go to latest Published: Jan 13, 2024 License: Apache-2.0, BSD-2-Clause Imports: 6 Imported by: 0

Documentation

Index

Constants

This section is empty.

Variables

View Source 
var File_tensorflow_core_framework_step_stats_proto protoreflect.FileDescriptor

Functions

This section is empty.

Types

type AllocationRecord 

type AllocationRecord struct {

	// The timestamp of the operation.
	AllocMicros int64 `protobuf:"varint,1,opt,name=alloc_micros,json=allocMicros,proto3" json:"alloc_micros,omitempty"`
	// Number of bytes allocated, or de-allocated if negative.
	AllocBytes int64 `protobuf:"varint,2,opt,name=alloc_bytes,json=allocBytes,proto3" json:"alloc_bytes,omitempty"`
	// contains filtered or unexported fields
}

An allocation/de-allocation operation performed by the allocator.

func (*AllocationRecord) Descriptor deprecated 

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

Deprecated: Use AllocationRecord.ProtoReflect.Descriptor instead.

func (*AllocationRecord) GetAllocBytes 

func (x *AllocationRecord) GetAllocBytes() int64

func (*AllocationRecord) GetAllocMicros 

func (x *AllocationRecord) GetAllocMicros() int64

func (*AllocationRecord) ProtoMessage 

func (*AllocationRecord) ProtoMessage()

func (*AllocationRecord) ProtoReflect 

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

func (*AllocationRecord) Reset 

func (x *AllocationRecord) Reset()

func (*AllocationRecord) String 

func (x *AllocationRecord) String() string

type AllocatorMemoryUsed 

type AllocatorMemoryUsed struct {
	AllocatorName string `protobuf:"bytes,1,opt,name=allocator_name,json=allocatorName,proto3" json:"allocator_name,omitempty"`
	// These are per-node allocator memory stats.
	TotalBytes int64 `protobuf:"varint,2,opt,name=total_bytes,json=totalBytes,proto3" json:"total_bytes,omitempty"`
	PeakBytes  int64 `protobuf:"varint,3,opt,name=peak_bytes,json=peakBytes,proto3" json:"peak_bytes,omitempty"`
	// The bytes that are not deallocated.
	LiveBytes int64 `protobuf:"varint,4,opt,name=live_bytes,json=liveBytes,proto3" json:"live_bytes,omitempty"`
	// The allocation and deallocation timeline.
	AllocationRecords []*AllocationRecord `protobuf:"bytes,6,rep,name=allocation_records,json=allocationRecords,proto3" json:"allocation_records,omitempty"`
	// These are snapshots of the overall allocator memory stats.
	// The number of live bytes currently allocated by the allocator.
	AllocatorBytesInUse int64 `protobuf:"varint,5,opt,name=allocator_bytes_in_use,json=allocatorBytesInUse,proto3" json:"allocator_bytes_in_use,omitempty"`
	// contains filtered or unexported fields
}

func (*AllocatorMemoryUsed) Descriptor deprecated 

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

Deprecated: Use AllocatorMemoryUsed.ProtoReflect.Descriptor instead.

func (*AllocatorMemoryUsed) GetAllocationRecords 

func (x *AllocatorMemoryUsed) GetAllocationRecords() []*AllocationRecord

func (*AllocatorMemoryUsed) GetAllocatorBytesInUse 

func (x *AllocatorMemoryUsed) GetAllocatorBytesInUse() int64

func (*AllocatorMemoryUsed) GetAllocatorName 

func (x *AllocatorMemoryUsed) GetAllocatorName() string

func (*AllocatorMemoryUsed) GetLiveBytes 

func (x *AllocatorMemoryUsed) GetLiveBytes() int64

func (*AllocatorMemoryUsed) GetPeakBytes 

func (x *AllocatorMemoryUsed) GetPeakBytes() int64

func (*AllocatorMemoryUsed) GetTotalBytes 

func (x *AllocatorMemoryUsed) GetTotalBytes() int64

func (*AllocatorMemoryUsed) ProtoMessage 

func (*AllocatorMemoryUsed) ProtoMessage()

func (*AllocatorMemoryUsed) ProtoReflect 

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

func (*AllocatorMemoryUsed) Reset 

func (x *AllocatorMemoryUsed) Reset()

func (*AllocatorMemoryUsed) String 

func (x *AllocatorMemoryUsed) String() string

type DeviceStepStats 

type DeviceStepStats struct {
	Device    string           `protobuf:"bytes,1,opt,name=device,proto3" json:"device,omitempty"`
	NodeStats []*NodeExecStats `protobuf:"bytes,2,rep,name=node_stats,json=nodeStats,proto3" json:"node_stats,omitempty"`
	// Its key is thread id.
	ThreadNames map[uint32]string `` /* 183-byte string literal not displayed */
	// contains filtered or unexported fields
}

func (*DeviceStepStats) Descriptor deprecated 

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

Deprecated: Use DeviceStepStats.ProtoReflect.Descriptor instead.

func (*DeviceStepStats) GetDevice 

func (x *DeviceStepStats) GetDevice() string

func (*DeviceStepStats) GetNodeStats 

func (x *DeviceStepStats) GetNodeStats() []*NodeExecStats

func (*DeviceStepStats) GetThreadNames 

func (x *DeviceStepStats) GetThreadNames() map[uint32]string

func (*DeviceStepStats) ProtoMessage 

func (*DeviceStepStats) ProtoMessage()

func (*DeviceStepStats) ProtoReflect 

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

func (*DeviceStepStats) Reset 

func (x *DeviceStepStats) Reset()

func (*DeviceStepStats) String 

func (x *DeviceStepStats) String() string

type MemoryStats 

type MemoryStats struct {
	TempMemorySize           int64   `protobuf:"varint,1,opt,name=temp_memory_size,json=tempMemorySize,proto3" json:"temp_memory_size,omitempty"`
	PersistentMemorySize     int64   `protobuf:"varint,3,opt,name=persistent_memory_size,json=persistentMemorySize,proto3" json:"persistent_memory_size,omitempty"`
	PersistentTensorAllocIds []int64 `` /* 145-byte string literal not displayed */
	// Deprecated: Do not use.
	DeviceTempMemorySize int64 `` /* 126-byte string literal not displayed */
	// Deprecated: Do not use.
	DevicePersistentMemorySize int64 `` /* 144-byte string literal not displayed */
	// Deprecated: Do not use.
	DevicePersistentTensorAllocIds []int64 `` /* 165-byte string literal not displayed */
	// contains filtered or unexported fields
}

For memory tracking.

func (*MemoryStats) Descriptor deprecated 

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

Deprecated: Use MemoryStats.ProtoReflect.Descriptor instead.

func (*MemoryStats) GetDevicePersistentMemorySize deprecated 

func (x *MemoryStats) GetDevicePersistentMemorySize() int64

Deprecated: Do not use.

func (*MemoryStats) GetDevicePersistentTensorAllocIds deprecated 

func (x *MemoryStats) GetDevicePersistentTensorAllocIds() []int64

Deprecated: Do not use.

func (*MemoryStats) GetDeviceTempMemorySize deprecated 

func (x *MemoryStats) GetDeviceTempMemorySize() int64

Deprecated: Do not use.

func (*MemoryStats) GetPersistentMemorySize 

func (x *MemoryStats) GetPersistentMemorySize() int64

func (*MemoryStats) GetPersistentTensorAllocIds 

func (x *MemoryStats) GetPersistentTensorAllocIds() []int64

func (*MemoryStats) GetTempMemorySize 

func (x *MemoryStats) GetTempMemorySize() int64

func (*MemoryStats) ProtoMessage 

func (*MemoryStats) ProtoMessage()

func (*MemoryStats) ProtoReflect 

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

func (*MemoryStats) Reset 

func (x *MemoryStats) Reset()

func (*MemoryStats) String 

func (x *MemoryStats) String() string

type NodeExecStats 

type NodeExecStats struct {

	// TODO(tucker): Use some more compact form of node identity than
	// the full string name.  Either all processes should agree on a
	// global id (cost_id?) for each node, or we should use a hash of
	// the name.
	NodeName         string                                                   `protobuf:"bytes,1,opt,name=node_name,json=nodeName,proto3" json:"node_name,omitempty"`
	AllStartMicros   int64                                                    `protobuf:"varint,2,opt,name=all_start_micros,json=allStartMicros,proto3" json:"all_start_micros,omitempty"`
	OpStartRelMicros int64                                                    `protobuf:"varint,3,opt,name=op_start_rel_micros,json=opStartRelMicros,proto3" json:"op_start_rel_micros,omitempty"`
	OpEndRelMicros   int64                                                    `protobuf:"varint,4,opt,name=op_end_rel_micros,json=opEndRelMicros,proto3" json:"op_end_rel_micros,omitempty"`
	AllEndRelMicros  int64                                                    `protobuf:"varint,5,opt,name=all_end_rel_micros,json=allEndRelMicros,proto3" json:"all_end_rel_micros,omitempty"`
	Memory           []*AllocatorMemoryUsed                                   `protobuf:"bytes,6,rep,name=memory,proto3" json:"memory,omitempty"`
	Output           []*NodeOutput                                            `protobuf:"bytes,7,rep,name=output,proto3" json:"output,omitempty"`
	TimelineLabel    string                                                   `protobuf:"bytes,8,opt,name=timeline_label,json=timelineLabel,proto3" json:"timeline_label,omitempty"`
	ScheduledMicros  int64                                                    `protobuf:"varint,9,opt,name=scheduled_micros,json=scheduledMicros,proto3" json:"scheduled_micros,omitempty"`
	ThreadId         uint32                                                   `protobuf:"varint,10,opt,name=thread_id,json=threadId,proto3" json:"thread_id,omitempty"`
	ReferencedTensor []*allocation_description_go_proto.AllocationDescription `protobuf:"bytes,11,rep,name=referenced_tensor,json=referencedTensor,proto3" json:"referenced_tensor,omitempty"`
	MemoryStats      *MemoryStats                                             `protobuf:"bytes,12,opt,name=memory_stats,json=memoryStats,proto3" json:"memory_stats,omitempty"`
	AllStartNanos    int64                                                    `protobuf:"varint,13,opt,name=all_start_nanos,json=allStartNanos,proto3" json:"all_start_nanos,omitempty"`
	OpStartRelNanos  int64                                                    `protobuf:"varint,14,opt,name=op_start_rel_nanos,json=opStartRelNanos,proto3" json:"op_start_rel_nanos,omitempty"`
	OpEndRelNanos    int64                                                    `protobuf:"varint,15,opt,name=op_end_rel_nanos,json=opEndRelNanos,proto3" json:"op_end_rel_nanos,omitempty"`
	AllEndRelNanos   int64                                                    `protobuf:"varint,16,opt,name=all_end_rel_nanos,json=allEndRelNanos,proto3" json:"all_end_rel_nanos,omitempty"`
	ScheduledNanos   int64                                                    `protobuf:"varint,17,opt,name=scheduled_nanos,json=scheduledNanos,proto3" json:"scheduled_nanos,omitempty"`
	// contains filtered or unexported fields
}

Time/size stats recorded for a single execution of a graph node.

func (*NodeExecStats) Descriptor deprecated 

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

Deprecated: Use NodeExecStats.ProtoReflect.Descriptor instead.

func (*NodeExecStats) GetAllEndRelMicros 

func (x *NodeExecStats) GetAllEndRelMicros() int64

func (*NodeExecStats) GetAllEndRelNanos 

func (x *NodeExecStats) GetAllEndRelNanos() int64

func (*NodeExecStats) GetAllStartMicros 

func (x *NodeExecStats) GetAllStartMicros() int64

func (*NodeExecStats) GetAllStartNanos 

func (x *NodeExecStats) GetAllStartNanos() int64

func (*NodeExecStats) GetMemory 

func (x *NodeExecStats) GetMemory() []*AllocatorMemoryUsed

func (*NodeExecStats) GetMemoryStats 

func (x *NodeExecStats) GetMemoryStats() *MemoryStats

func (*NodeExecStats) GetNodeName 

func (x *NodeExecStats) GetNodeName() string

func (*NodeExecStats) GetOpEndRelMicros 

func (x *NodeExecStats) GetOpEndRelMicros() int64

func (*NodeExecStats) GetOpEndRelNanos 

func (x *NodeExecStats) GetOpEndRelNanos() int64

func (*NodeExecStats) GetOpStartRelMicros 

func (x *NodeExecStats) GetOpStartRelMicros() int64

func (*NodeExecStats) GetOpStartRelNanos 

func (x *NodeExecStats) GetOpStartRelNanos() int64

func (*NodeExecStats) GetOutput 

func (x *NodeExecStats) GetOutput() []*NodeOutput

func (*NodeExecStats) GetReferencedTensor 

func (x *NodeExecStats) GetReferencedTensor() []*allocation_description_go_proto.AllocationDescription

func (*NodeExecStats) GetScheduledMicros 

func (x *NodeExecStats) GetScheduledMicros() int64

func (*NodeExecStats) GetScheduledNanos 

func (x *NodeExecStats) GetScheduledNanos() int64

func (*NodeExecStats) GetThreadId 

func (x *NodeExecStats) GetThreadId() uint32

func (*NodeExecStats) GetTimelineLabel 

func (x *NodeExecStats) GetTimelineLabel() string

func (*NodeExecStats) ProtoMessage 

func (*NodeExecStats) ProtoMessage()

func (*NodeExecStats) ProtoReflect 

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

func (*NodeExecStats) Reset 

func (x *NodeExecStats) Reset()

func (*NodeExecStats) String 

func (x *NodeExecStats) String() string

type NodeOutput 

type NodeOutput struct {
	Slot              int32                                          `protobuf:"varint,1,opt,name=slot,proto3" json:"slot,omitempty"`
	TensorDescription *tensor_description_go_proto.TensorDescription `protobuf:"bytes,3,opt,name=tensor_description,json=tensorDescription,proto3" json:"tensor_description,omitempty"`
	// contains filtered or unexported fields
}

Output sizes recorded for a single execution of a graph node.

func (*NodeOutput) Descriptor deprecated 

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

Deprecated: Use NodeOutput.ProtoReflect.Descriptor instead.

func (*NodeOutput) GetSlot 

func (x *NodeOutput) GetSlot() int32

func (*NodeOutput) GetTensorDescription 

func (x *NodeOutput) GetTensorDescription() *tensor_description_go_proto.TensorDescription

func (*NodeOutput) ProtoMessage 

func (*NodeOutput) ProtoMessage()

func (*NodeOutput) ProtoReflect 

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

func (*NodeOutput) Reset 

func (x *NodeOutput) Reset()

func (*NodeOutput) String 

func (x *NodeOutput) String() string

type StepStats 

type StepStats struct {
	DevStats []*DeviceStepStats `protobuf:"bytes,1,rep,name=dev_stats,json=devStats,proto3" json:"dev_stats,omitempty"`
	// contains filtered or unexported fields
}

func (*StepStats) Descriptor deprecated 

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

Deprecated: Use StepStats.ProtoReflect.Descriptor instead.

func (*StepStats) GetDevStats 

func (x *StepStats) GetDevStats() []*DeviceStepStats

func (*StepStats) ProtoMessage 

func (*StepStats) ProtoMessage()

func (*StepStats) ProtoReflect 

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

func (*StepStats) Reset 

func (x *StepStats) Reset()

func (*StepStats) String 

func (x *StepStats) 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.