step_stats_go_proto

package
v1.12.2-0...-1b6f133 Latest Latest
Warning

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

 Go to latest Published: Nov 10, 2020 License: Apache-2.0 Imports: 5 Imported by: 0

Documentation

Index

Constants

This section is empty.

Variables

This section is empty.

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"`
	XXX_NoUnkeyedLiteral struct{} `json:"-"`
	XXX_unrecognized     []byte   `json:"-"`
	XXX_sizecache        int32    `json:"-"`
}

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

func (*AllocationRecord) Descriptor 

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

func (*AllocationRecord) GetAllocBytes 

func (m *AllocationRecord) GetAllocBytes() int64

func (*AllocationRecord) GetAllocMicros 

func (m *AllocationRecord) GetAllocMicros() int64

func (*AllocationRecord) ProtoMessage 

func (*AllocationRecord) ProtoMessage()

func (*AllocationRecord) Reset 

func (m *AllocationRecord) Reset()

func (*AllocationRecord) String 

func (m *AllocationRecord) String() string

func (*AllocationRecord) XXX_DiscardUnknown 

func (m *AllocationRecord) XXX_DiscardUnknown()

func (*AllocationRecord) XXX_Marshal 

func (m *AllocationRecord) XXX_Marshal(b []byte, deterministic bool) ([]byte, error)

func (*AllocationRecord) XXX_Merge 

func (m *AllocationRecord) XXX_Merge(src proto.Message)

func (*AllocationRecord) XXX_Size 

func (m *AllocationRecord) XXX_Size() int

func (*AllocationRecord) XXX_Unmarshal 

func (m *AllocationRecord) XXX_Unmarshal(b []byte) error

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"`
	XXX_NoUnkeyedLiteral struct{} `json:"-"`
	XXX_unrecognized     []byte   `json:"-"`
	XXX_sizecache        int32    `json:"-"`
}

func (*AllocatorMemoryUsed) Descriptor 

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

func (*AllocatorMemoryUsed) GetAllocationRecords 

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

func (*AllocatorMemoryUsed) GetAllocatorBytesInUse 

func (m *AllocatorMemoryUsed) GetAllocatorBytesInUse() int64

func (*AllocatorMemoryUsed) GetAllocatorName 

func (m *AllocatorMemoryUsed) GetAllocatorName() string

func (*AllocatorMemoryUsed) GetLiveBytes 

func (m *AllocatorMemoryUsed) GetLiveBytes() int64

func (*AllocatorMemoryUsed) GetPeakBytes 

func (m *AllocatorMemoryUsed) GetPeakBytes() int64

func (*AllocatorMemoryUsed) GetTotalBytes 

func (m *AllocatorMemoryUsed) GetTotalBytes() int64

func (*AllocatorMemoryUsed) ProtoMessage 

func (*AllocatorMemoryUsed) ProtoMessage()

func (*AllocatorMemoryUsed) Reset 

func (m *AllocatorMemoryUsed) Reset()

func (*AllocatorMemoryUsed) String 

func (m *AllocatorMemoryUsed) String() string

func (*AllocatorMemoryUsed) XXX_DiscardUnknown 

func (m *AllocatorMemoryUsed) XXX_DiscardUnknown()

func (*AllocatorMemoryUsed) XXX_Marshal 

func (m *AllocatorMemoryUsed) XXX_Marshal(b []byte, deterministic bool) ([]byte, error)

func (*AllocatorMemoryUsed) XXX_Merge 

func (m *AllocatorMemoryUsed) XXX_Merge(src proto.Message)

func (*AllocatorMemoryUsed) XXX_Size 

func (m *AllocatorMemoryUsed) XXX_Size() int

func (*AllocatorMemoryUsed) XXX_Unmarshal 

func (m *AllocatorMemoryUsed) XXX_Unmarshal(b []byte) error

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 */
	XXX_NoUnkeyedLiteral struct{}          `json:"-"`
	XXX_unrecognized     []byte            `json:"-"`
	XXX_sizecache        int32             `json:"-"`
}

func (*DeviceStepStats) Descriptor 

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

func (*DeviceStepStats) GetDevice 

func (m *DeviceStepStats) GetDevice() string

func (*DeviceStepStats) GetNodeStats 

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

func (*DeviceStepStats) GetThreadNames 

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

func (*DeviceStepStats) ProtoMessage 

func (*DeviceStepStats) ProtoMessage()

func (*DeviceStepStats) Reset 

func (m *DeviceStepStats) Reset()

func (*DeviceStepStats) String 

func (m *DeviceStepStats) String() string

func (*DeviceStepStats) XXX_DiscardUnknown 

func (m *DeviceStepStats) XXX_DiscardUnknown()

func (*DeviceStepStats) XXX_Marshal 

func (m *DeviceStepStats) XXX_Marshal(b []byte, deterministic bool) ([]byte, error)

func (*DeviceStepStats) XXX_Merge 

func (m *DeviceStepStats) XXX_Merge(src proto.Message)

func (*DeviceStepStats) XXX_Size 

func (m *DeviceStepStats) XXX_Size() int

func (*DeviceStepStats) XXX_Unmarshal 

func (m *DeviceStepStats) XXX_Unmarshal(b []byte) error

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 */
	DeviceTempMemorySize     int64   `` // Deprecated: Do not use.
	/* 126-byte string literal not displayed */
	DevicePersistentMemorySize int64 `` // Deprecated: Do not use.
	/* 144-byte string literal not displayed */
	DevicePersistentTensorAllocIds []int64 `` // Deprecated: Do not use.
	/* 165-byte string literal not displayed */
	XXX_NoUnkeyedLiteral struct{} `json:"-"`
	XXX_unrecognized     []byte   `json:"-"`
	XXX_sizecache        int32    `json:"-"`
}

For memory tracking.

func (*MemoryStats) Descriptor 

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

func (*MemoryStats) GetDevicePersistentMemorySize deprecated 

func (m *MemoryStats) GetDevicePersistentMemorySize() int64

Deprecated: Do not use.

func (*MemoryStats) GetDevicePersistentTensorAllocIds deprecated 

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

Deprecated: Do not use.

func (*MemoryStats) GetDeviceTempMemorySize deprecated 

func (m *MemoryStats) GetDeviceTempMemorySize() int64

Deprecated: Do not use.

func (*MemoryStats) GetPersistentMemorySize 

func (m *MemoryStats) GetPersistentMemorySize() int64

func (*MemoryStats) GetPersistentTensorAllocIds 

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

func (*MemoryStats) GetTempMemorySize 

func (m *MemoryStats) GetTempMemorySize() int64

func (*MemoryStats) ProtoMessage 

func (*MemoryStats) ProtoMessage()

func (*MemoryStats) Reset 

func (m *MemoryStats) Reset()

func (*MemoryStats) String 

func (m *MemoryStats) String() string

func (*MemoryStats) XXX_DiscardUnknown 

func (m *MemoryStats) XXX_DiscardUnknown()

func (*MemoryStats) XXX_Marshal 

func (m *MemoryStats) XXX_Marshal(b []byte, deterministic bool) ([]byte, error)

func (*MemoryStats) XXX_Merge 

func (m *MemoryStats) XXX_Merge(src proto.Message)

func (*MemoryStats) XXX_Size 

func (m *MemoryStats) XXX_Size() int

func (*MemoryStats) XXX_Unmarshal 

func (m *MemoryStats) XXX_Unmarshal(b []byte) error

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"`
	XXX_NoUnkeyedLiteral struct{}                                                 `json:"-"`
	XXX_unrecognized     []byte                                                   `json:"-"`
	XXX_sizecache        int32                                                    `json:"-"`
}

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

func (*NodeExecStats) Descriptor 

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

func (*NodeExecStats) GetAllEndRelMicros 

func (m *NodeExecStats) GetAllEndRelMicros() int64

func (*NodeExecStats) GetAllEndRelNanos 

func (m *NodeExecStats) GetAllEndRelNanos() int64

func (*NodeExecStats) GetAllStartMicros 

func (m *NodeExecStats) GetAllStartMicros() int64

func (*NodeExecStats) GetAllStartNanos 

func (m *NodeExecStats) GetAllStartNanos() int64

func (*NodeExecStats) GetMemory 

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

func (*NodeExecStats) GetMemoryStats 

func (m *NodeExecStats) GetMemoryStats() *MemoryStats

func (*NodeExecStats) GetNodeName 

func (m *NodeExecStats) GetNodeName() string

func (*NodeExecStats) GetOpEndRelMicros 

func (m *NodeExecStats) GetOpEndRelMicros() int64

func (*NodeExecStats) GetOpEndRelNanos 

func (m *NodeExecStats) GetOpEndRelNanos() int64

func (*NodeExecStats) GetOpStartRelMicros 

func (m *NodeExecStats) GetOpStartRelMicros() int64

func (*NodeExecStats) GetOpStartRelNanos 

func (m *NodeExecStats) GetOpStartRelNanos() int64

func (*NodeExecStats) GetOutput 

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

func (*NodeExecStats) GetReferencedTensor 

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

func (*NodeExecStats) GetScheduledMicros 

func (m *NodeExecStats) GetScheduledMicros() int64

func (*NodeExecStats) GetScheduledNanos 

func (m *NodeExecStats) GetScheduledNanos() int64

func (*NodeExecStats) GetThreadId 

func (m *NodeExecStats) GetThreadId() uint32

func (*NodeExecStats) GetTimelineLabel 

func (m *NodeExecStats) GetTimelineLabel() string

func (*NodeExecStats) ProtoMessage 

func (*NodeExecStats) ProtoMessage()

func (*NodeExecStats) Reset 

func (m *NodeExecStats) Reset()

func (*NodeExecStats) String 

func (m *NodeExecStats) String() string

func (*NodeExecStats) XXX_DiscardUnknown 

func (m *NodeExecStats) XXX_DiscardUnknown()

func (*NodeExecStats) XXX_Marshal 

func (m *NodeExecStats) XXX_Marshal(b []byte, deterministic bool) ([]byte, error)

func (*NodeExecStats) XXX_Merge 

func (m *NodeExecStats) XXX_Merge(src proto.Message)

func (*NodeExecStats) XXX_Size 

func (m *NodeExecStats) XXX_Size() int

func (*NodeExecStats) XXX_Unmarshal 

func (m *NodeExecStats) XXX_Unmarshal(b []byte) error

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"`
	XXX_NoUnkeyedLiteral struct{}                                       `json:"-"`
	XXX_unrecognized     []byte                                         `json:"-"`
	XXX_sizecache        int32                                          `json:"-"`
}

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

func (*NodeOutput) Descriptor 

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

func (*NodeOutput) GetSlot 

func (m *NodeOutput) GetSlot() int32

func (*NodeOutput) GetTensorDescription 

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

func (*NodeOutput) ProtoMessage 

func (*NodeOutput) ProtoMessage()

func (*NodeOutput) Reset 

func (m *NodeOutput) Reset()

func (*NodeOutput) String 

func (m *NodeOutput) String() string

func (*NodeOutput) XXX_DiscardUnknown 

func (m *NodeOutput) XXX_DiscardUnknown()

func (*NodeOutput) XXX_Marshal 

func (m *NodeOutput) XXX_Marshal(b []byte, deterministic bool) ([]byte, error)

func (*NodeOutput) XXX_Merge 

func (m *NodeOutput) XXX_Merge(src proto.Message)

func (*NodeOutput) XXX_Size 

func (m *NodeOutput) XXX_Size() int

func (*NodeOutput) XXX_Unmarshal 

func (m *NodeOutput) XXX_Unmarshal(b []byte) error

type StepStats 

type StepStats struct {
	DevStats             []*DeviceStepStats `protobuf:"bytes,1,rep,name=dev_stats,json=devStats,proto3" json:"dev_stats,omitempty"`
	XXX_NoUnkeyedLiteral struct{}           `json:"-"`
	XXX_unrecognized     []byte             `json:"-"`
	XXX_sizecache        int32              `json:"-"`
}

func (*StepStats) Descriptor 

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

func (*StepStats) GetDevStats 

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

func (*StepStats) ProtoMessage 

func (*StepStats) ProtoMessage()

func (*StepStats) Reset 

func (m *StepStats) Reset()

func (*StepStats) String 

func (m *StepStats) String() string

func (*StepStats) XXX_DiscardUnknown 

func (m *StepStats) XXX_DiscardUnknown()

func (*StepStats) XXX_Marshal 

func (m *StepStats) XXX_Marshal(b []byte, deterministic bool) ([]byte, error)

func (*StepStats) XXX_Merge 

func (m *StepStats) XXX_Merge(src proto.Message)

func (*StepStats) XXX_Size 

func (m *StepStats) XXX_Size() int

func (*StepStats) XXX_Unmarshal 

func (m *StepStats) XXX_Unmarshal(b []byte) error

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.