costs

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: 6 Imported by: 0

Documentation

Index

Constants

This section is empty.

Variables

View Source
var File_tensorflow_core_grappler_costs_op_performance_data_proto protoreflect.FileDescriptor

Functions

This section is empty.

Types

type LogNormalDistribution

type LogNormalDistribution struct {
	Mu    float64 `protobuf:"fixed64,1,opt,name=mu,proto3" json:"mu,omitempty"`
	Sigma float64 `protobuf:"fixed64,2,opt,name=sigma,proto3" json:"sigma,omitempty"`
	// contains filtered or unexported fields
}

func (*LogNormalDistribution) Descriptor deprecated

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

Deprecated: Use LogNormalDistribution.ProtoReflect.Descriptor instead.

func (*LogNormalDistribution) GetMu

func (x *LogNormalDistribution) GetMu() float64

func (*LogNormalDistribution) GetSigma

func (x *LogNormalDistribution) GetSigma() float64

func (*LogNormalDistribution) ProtoMessage

func (*LogNormalDistribution) ProtoMessage()

func (*LogNormalDistribution) ProtoReflect

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

func (*LogNormalDistribution) Reset

func (x *LogNormalDistribution) Reset()

func (*LogNormalDistribution) String

func (x *LogNormalDistribution) String() string

type NormalDistribution

type NormalDistribution struct {
	Mu    float64 `protobuf:"fixed64,1,opt,name=mu,proto3" json:"mu,omitempty"`
	Sigma float64 `protobuf:"fixed64,2,opt,name=sigma,proto3" json:"sigma,omitempty"`
	// contains filtered or unexported fields
}

func (*NormalDistribution) Descriptor deprecated

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

Deprecated: Use NormalDistribution.ProtoReflect.Descriptor instead.

func (*NormalDistribution) GetMu

func (x *NormalDistribution) GetMu() float64

func (*NormalDistribution) GetSigma

func (x *NormalDistribution) GetSigma() float64

func (*NormalDistribution) ProtoMessage

func (*NormalDistribution) ProtoMessage()

func (*NormalDistribution) ProtoReflect

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

func (*NormalDistribution) Reset

func (x *NormalDistribution) Reset()

func (*NormalDistribution) String

func (x *NormalDistribution) String() string

type OpInfo

type OpInfo struct {

	// The operation name.  There may be custom parameters in attrs.
	Op string `protobuf:"bytes,1,opt,name=op,proto3" json:"op,omitempty"`
	// Custom parameters impacting the behavior of the op.
	Attr   map[string]*framework.AttrValue `` /* 149-byte string literal not displayed */
	Inputs []*OpInfo_TensorProperties      `protobuf:"bytes,3,rep,name=inputs,proto3" json:"inputs,omitempty"`
	// Optional description of the op outputs
	Outputs []*OpInfo_TensorProperties `protobuf:"bytes,5,rep,name=outputs,proto3" json:"outputs,omitempty"`
	// Device on which the operation is run.
	Device *protobuf.DeviceProperties `protobuf:"bytes,4,opt,name=device,proto3" json:"device,omitempty"`
	// Information about the session configs.
	SessionInfo *SessionInfo `protobuf:"bytes,6,opt,name=session_info,json=sessionInfo,proto3" json:"session_info,omitempty"`
	// contains filtered or unexported fields
}

Description of an operation as well as the parameters expected to impact its performance.

func (*OpInfo) Descriptor deprecated

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

Deprecated: Use OpInfo.ProtoReflect.Descriptor instead.

func (*OpInfo) GetAttr

func (x *OpInfo) GetAttr() map[string]*framework.AttrValue

func (*OpInfo) GetDevice

func (x *OpInfo) GetDevice() *protobuf.DeviceProperties

func (*OpInfo) GetInputs

func (x *OpInfo) GetInputs() []*OpInfo_TensorProperties

func (*OpInfo) GetOp

func (x *OpInfo) GetOp() string

func (*OpInfo) GetOutputs

func (x *OpInfo) GetOutputs() []*OpInfo_TensorProperties

func (*OpInfo) GetSessionInfo

func (x *OpInfo) GetSessionInfo() *SessionInfo

func (*OpInfo) ProtoMessage

func (*OpInfo) ProtoMessage()

func (*OpInfo) ProtoReflect

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

func (*OpInfo) Reset

func (x *OpInfo) Reset()

func (*OpInfo) String

func (x *OpInfo) String() string

type OpInfo_TensorProperties

type OpInfo_TensorProperties struct {
	Dtype framework.DataType          `protobuf:"varint,1,opt,name=dtype,proto3,enum=tensorflow.DataType" json:"dtype,omitempty"`
	Shape *framework.TensorShapeProto `protobuf:"bytes,2,opt,name=shape,proto3" json:"shape,omitempty"`
	Value *framework.TensorProto      `protobuf:"bytes,3,opt,name=value,proto3" json:"value,omitempty"`
	// contains filtered or unexported fields
}

Input data types, shapes and values if known.

func (*OpInfo_TensorProperties) Descriptor deprecated

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

Deprecated: Use OpInfo_TensorProperties.ProtoReflect.Descriptor instead.

func (*OpInfo_TensorProperties) GetDtype

func (*OpInfo_TensorProperties) GetShape

func (*OpInfo_TensorProperties) GetValue

func (*OpInfo_TensorProperties) ProtoMessage

func (*OpInfo_TensorProperties) ProtoMessage()

func (*OpInfo_TensorProperties) ProtoReflect

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

func (*OpInfo_TensorProperties) Reset

func (x *OpInfo_TensorProperties) Reset()

func (*OpInfo_TensorProperties) String

func (x *OpInfo_TensorProperties) String() string

type OpPerformance

type OpPerformance struct {

	// The op
	Op *OpInfo `protobuf:"bytes,1,opt,name=op,proto3" json:"op,omitempty"`
	// Information about the session configs.
	//
	// Deprecated: Marked as deprecated in tensorflow/core/grappler/costs/op_performance_data.proto.
	SessionInfo *SessionInfo `protobuf:"bytes,12,opt,name=session_info,json=sessionInfo,proto3" json:"session_info,omitempty"`
	// The node name (optional). Makes it easier to associate the performance data
	// with a specific graph node.
	Node string `protobuf:"bytes,5,opt,name=node,proto3" json:"node,omitempty"`
	// Temporary memory used by this node (in bytes).
	TemporaryMemorySize int64 `protobuf:"varint,2,opt,name=temporary_memory_size,json=temporaryMemorySize,proto3" json:"temporary_memory_size,omitempty"`
	// Time it takes to run the op (in nanoseconds).
	ComputeCost int64 `protobuf:"varint,3,opt,name=compute_cost,json=computeCost,proto3" json:"compute_cost,omitempty"`
	// Analytical compute cost (in nanoseconds).
	ComputeTime int64 `protobuf:"varint,6,opt,name=compute_time,json=computeTime,proto3" json:"compute_time,omitempty"`
	// Analytical memory access cost (in nanoseconds).
	MemoryTime int64 `protobuf:"varint,7,opt,name=memory_time,json=memoryTime,proto3" json:"memory_time,omitempty"`
	// Percentage of theoretical compute performance.
	ComputeEfficiency float64 `protobuf:"fixed64,4,opt,name=compute_efficiency,json=computeEfficiency,proto3" json:"compute_efficiency,omitempty"`
	// Percentage of theoretical memory performance.
	MemoryEfficiency float64 `protobuf:"fixed64,8,opt,name=memory_efficiency,json=memoryEfficiency,proto3" json:"memory_efficiency,omitempty"`
	// Expected execution time, modeled using one of 2 possible distributions.
	//
	// Types that are assignable to ExecutionTime:
	//
	//	*OpPerformance_ExecutionTimeNormal
	//	*OpPerformance_ExecutionTimeLogNormal
	ExecutionTime isOpPerformance_ExecutionTime `protobuf_oneof:"execution_time"`
	OpMemory      *OpPerformance_OpMemory       `protobuf:"bytes,9,opt,name=op_memory,json=opMemory,proto3" json:"op_memory,omitempty"`
	// contains filtered or unexported fields
}

Performance data for tensorflow operations

func (*OpPerformance) Descriptor deprecated

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

Deprecated: Use OpPerformance.ProtoReflect.Descriptor instead.

func (*OpPerformance) GetComputeCost

func (x *OpPerformance) GetComputeCost() int64

func (*OpPerformance) GetComputeEfficiency

func (x *OpPerformance) GetComputeEfficiency() float64

func (*OpPerformance) GetComputeTime

func (x *OpPerformance) GetComputeTime() int64

func (*OpPerformance) GetExecutionTime

func (m *OpPerformance) GetExecutionTime() isOpPerformance_ExecutionTime

func (*OpPerformance) GetExecutionTimeLogNormal

func (x *OpPerformance) GetExecutionTimeLogNormal() *LogNormalDistribution

func (*OpPerformance) GetExecutionTimeNormal

func (x *OpPerformance) GetExecutionTimeNormal() *NormalDistribution

func (*OpPerformance) GetMemoryEfficiency

func (x *OpPerformance) GetMemoryEfficiency() float64

func (*OpPerformance) GetMemoryTime

func (x *OpPerformance) GetMemoryTime() int64

func (*OpPerformance) GetNode

func (x *OpPerformance) GetNode() string

func (*OpPerformance) GetOp

func (x *OpPerformance) GetOp() *OpInfo

func (*OpPerformance) GetOpMemory

func (x *OpPerformance) GetOpMemory() *OpPerformance_OpMemory

func (*OpPerformance) GetSessionInfo deprecated

func (x *OpPerformance) GetSessionInfo() *SessionInfo

Deprecated: Marked as deprecated in tensorflow/core/grappler/costs/op_performance_data.proto.

func (*OpPerformance) GetTemporaryMemorySize

func (x *OpPerformance) GetTemporaryMemorySize() int64

func (*OpPerformance) ProtoMessage

func (*OpPerformance) ProtoMessage()

func (*OpPerformance) ProtoReflect

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

func (*OpPerformance) Reset

func (x *OpPerformance) Reset()

func (*OpPerformance) String

func (x *OpPerformance) String() string

type OpPerformanceList

type OpPerformanceList struct {
	OpPerformance []*OpPerformance `protobuf:"bytes,1,rep,name=op_performance,json=opPerformance,proto3" json:"op_performance,omitempty"`
	// contains filtered or unexported fields
}

A collection of OpPerformance data points.

func (*OpPerformanceList) Descriptor deprecated

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

Deprecated: Use OpPerformanceList.ProtoReflect.Descriptor instead.

func (*OpPerformanceList) GetOpPerformance

func (x *OpPerformanceList) GetOpPerformance() []*OpPerformance

func (*OpPerformanceList) ProtoMessage

func (*OpPerformanceList) ProtoMessage()

func (*OpPerformanceList) ProtoReflect

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

func (*OpPerformanceList) Reset

func (x *OpPerformanceList) Reset()

func (*OpPerformanceList) String

func (x *OpPerformanceList) String() string

type OpPerformance_ExecutionTimeLogNormal

type OpPerformance_ExecutionTimeLogNormal struct {
	ExecutionTimeLogNormal *LogNormalDistribution `protobuf:"bytes,11,opt,name=execution_time_log_normal,json=executionTimeLogNormal,proto3,oneof"`
}

type OpPerformance_ExecutionTimeNormal

type OpPerformance_ExecutionTimeNormal struct {
	ExecutionTimeNormal *NormalDistribution `protobuf:"bytes,10,opt,name=execution_time_normal,json=executionTimeNormal,proto3,oneof"`
}

type OpPerformance_OpMemory

type OpPerformance_OpMemory struct {

	// The output information may have memory usage and output shapes.
	OutputMemory []int64 `protobuf:"varint,1,rep,packed,name=output_memory,json=outputMemory,proto3" json:"output_memory,omitempty"`
	// Temp and persistent memory allocated by this node.
	TempMemory       int64 `protobuf:"varint,2,opt,name=temp_memory,json=tempMemory,proto3" json:"temp_memory,omitempty"`
	PersistentMemory int64 `protobuf:"varint,4,opt,name=persistent_memory,json=persistentMemory,proto3" json:"persistent_memory,omitempty"`
	// Deprecated: Marked as deprecated in tensorflow/core/grappler/costs/op_performance_data.proto.
	DeviceTempMemory int64 `protobuf:"varint,3,opt,name=device_temp_memory,json=deviceTempMemory,proto3" json:"device_temp_memory,omitempty"`
	// Deprecated: Marked as deprecated in tensorflow/core/grappler/costs/op_performance_data.proto.
	DevicePersistentMemory int64 `` /* 130-byte string literal not displayed */
	// contains filtered or unexported fields
}

Memory usage data for a tensorflow operation.

func (*OpPerformance_OpMemory) Descriptor deprecated

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

Deprecated: Use OpPerformance_OpMemory.ProtoReflect.Descriptor instead.

func (*OpPerformance_OpMemory) GetDevicePersistentMemory deprecated

func (x *OpPerformance_OpMemory) GetDevicePersistentMemory() int64

Deprecated: Marked as deprecated in tensorflow/core/grappler/costs/op_performance_data.proto.

func (*OpPerformance_OpMemory) GetDeviceTempMemory deprecated

func (x *OpPerformance_OpMemory) GetDeviceTempMemory() int64

Deprecated: Marked as deprecated in tensorflow/core/grappler/costs/op_performance_data.proto.

func (*OpPerformance_OpMemory) GetOutputMemory

func (x *OpPerformance_OpMemory) GetOutputMemory() []int64

func (*OpPerformance_OpMemory) GetPersistentMemory

func (x *OpPerformance_OpMemory) GetPersistentMemory() int64

func (*OpPerformance_OpMemory) GetTempMemory

func (x *OpPerformance_OpMemory) GetTempMemory() int64

func (*OpPerformance_OpMemory) ProtoMessage

func (*OpPerformance_OpMemory) ProtoMessage()

func (*OpPerformance_OpMemory) ProtoReflect

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

func (*OpPerformance_OpMemory) Reset

func (x *OpPerformance_OpMemory) Reset()

func (*OpPerformance_OpMemory) String

func (x *OpPerformance_OpMemory) String() string

type SessionInfo

type SessionInfo struct {
	IntraOpParallelism int64 `protobuf:"varint,1,opt,name=intra_op_parallelism,json=intraOpParallelism,proto3" json:"intra_op_parallelism,omitempty"`
	// contains filtered or unexported fields
}

Description of the session when an op is run.

func (*SessionInfo) Descriptor deprecated

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

Deprecated: Use SessionInfo.ProtoReflect.Descriptor instead.

func (*SessionInfo) GetIntraOpParallelism

func (x *SessionInfo) GetIntraOpParallelism() int64

func (*SessionInfo) ProtoMessage

func (*SessionInfo) ProtoMessage()

func (*SessionInfo) ProtoReflect

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

func (*SessionInfo) Reset

func (x *SessionInfo) Reset()

func (*SessionInfo) String

func (x *SessionInfo) String() string

Jump to

Keyboard shortcuts

? : This menu
/ : Search site
f or F : Jump to
y or Y : Canonical URL