cost_graph_go_proto

package
v2.8.0+incompatible Latest Latest
Warning

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

 Go to latest Published: Jul 18, 2022 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_cost_graph_proto protoreflect.FileDescriptor

Functions

This section is empty.

Types

type CostGraphDef 

type CostGraphDef struct {
	Node []*CostGraphDef_Node           `protobuf:"bytes,1,rep,name=node,proto3" json:"node,omitempty"`
	Cost []*CostGraphDef_AggregatedCost `protobuf:"bytes,2,rep,name=cost,proto3" json:"cost,omitempty"`
	// contains filtered or unexported fields
}

func (*CostGraphDef) Descriptor deprecated 

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

Deprecated: Use CostGraphDef.ProtoReflect.Descriptor instead.

func (*CostGraphDef) GetCost 

func (x *CostGraphDef) GetCost() []*CostGraphDef_AggregatedCost

func (*CostGraphDef) GetNode 

func (x *CostGraphDef) GetNode() []*CostGraphDef_Node

func (*CostGraphDef) ProtoMessage 

func (*CostGraphDef) ProtoMessage()

func (*CostGraphDef) ProtoReflect 

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

func (*CostGraphDef) Reset 

func (x *CostGraphDef) Reset()

func (*CostGraphDef) String 

func (x *CostGraphDef) String() string

type CostGraphDef_AggregatedCost 

type CostGraphDef_AggregatedCost struct {

	// Aggregated cost value.
	Cost float32 `protobuf:"fixed32,1,opt,name=cost,proto3" json:"cost,omitempty"`
	// Aggregated cost dimension (e.g. 'memory', 'compute', 'network').
	Dimension string `protobuf:"bytes,2,opt,name=dimension,proto3" json:"dimension,omitempty"`
	// contains filtered or unexported fields
}

Total cost of this graph, typically used for balancing decisions.

func (*CostGraphDef_AggregatedCost) Descriptor deprecated 

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

Deprecated: Use CostGraphDef_AggregatedCost.ProtoReflect.Descriptor instead.

func (*CostGraphDef_AggregatedCost) GetCost 

func (x *CostGraphDef_AggregatedCost) GetCost() float32

func (*CostGraphDef_AggregatedCost) GetDimension 

func (x *CostGraphDef_AggregatedCost) GetDimension() string

func (*CostGraphDef_AggregatedCost) ProtoMessage 

func (*CostGraphDef_AggregatedCost) ProtoMessage()

func (*CostGraphDef_AggregatedCost) ProtoReflect 

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

func (*CostGraphDef_AggregatedCost) Reset 

func (x *CostGraphDef_AggregatedCost) Reset()

func (*CostGraphDef_AggregatedCost) String 

func (x *CostGraphDef_AggregatedCost) String() string

type CostGraphDef_Node 

type CostGraphDef_Node struct {

	// The name of the node. Names are globally unique.
	Name string `protobuf:"bytes,1,opt,name=name,proto3" json:"name,omitempty"`
	// The device of the node. Can be empty if the node is mapped to the
	// default partition or partitioning hasn't been run yet.
	Device string `protobuf:"bytes,2,opt,name=device,proto3" json:"device,omitempty"`
	// The id of the node. Node ids are only unique inside a partition.
	Id         int32                           `protobuf:"varint,3,opt,name=id,proto3" json:"id,omitempty"`
	InputInfo  []*CostGraphDef_Node_InputInfo  `protobuf:"bytes,4,rep,name=input_info,json=inputInfo,proto3" json:"input_info,omitempty"`
	OutputInfo []*CostGraphDef_Node_OutputInfo `protobuf:"bytes,5,rep,name=output_info,json=outputInfo,proto3" json:"output_info,omitempty"`
	// Temporary memory used by this node.
	TemporaryMemorySize int64 `protobuf:"varint,6,opt,name=temporary_memory_size,json=temporaryMemorySize,proto3" json:"temporary_memory_size,omitempty"`
	// Persistent memory used by this node.
	PersistentMemorySize int64 `protobuf:"varint,12,opt,name=persistent_memory_size,json=persistentMemorySize,proto3" json:"persistent_memory_size,omitempty"`
	// Deprecated: Do not use.
	HostTempMemorySize int64 `protobuf:"varint,10,opt,name=host_temp_memory_size,json=hostTempMemorySize,proto3" json:"host_temp_memory_size,omitempty"`
	// Deprecated: Do not use.
	DeviceTempMemorySize int64 `` /* 127-byte string literal not displayed */
	// Deprecated: Do not use.
	DevicePersistentMemorySize int64 `` /* 145-byte string literal not displayed */
	// Estimate of the computational cost of this node, in microseconds.
	ComputeCost int64 `protobuf:"varint,9,opt,name=compute_cost,json=computeCost,proto3" json:"compute_cost,omitempty"`
	// Analytical estimate of the computational cost of this node, in
	// microseconds.
	ComputeTime int64 `protobuf:"varint,14,opt,name=compute_time,json=computeTime,proto3" json:"compute_time,omitempty"`
	// Analytical estimate of the memory access cost of this node, in
	// microseconds.
	MemoryTime int64 `protobuf:"varint,15,opt,name=memory_time,json=memoryTime,proto3" json:"memory_time,omitempty"`
	// If true, the output is permanent: it can't be discarded, because this
	// node is part of the "final output". Nodes may depend on final nodes.
	IsFinal bool `protobuf:"varint,7,opt,name=is_final,json=isFinal,proto3" json:"is_final,omitempty"`
	// Ids of the control inputs for this node.
	ControlInput []int32 `protobuf:"varint,8,rep,packed,name=control_input,json=controlInput,proto3" json:"control_input,omitempty"`
	// Are the costs inaccurate?
	Inaccurate bool `protobuf:"varint,17,opt,name=inaccurate,proto3" json:"inaccurate,omitempty"`
	// contains filtered or unexported fields
}

func (*CostGraphDef_Node) Descriptor deprecated 

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

Deprecated: Use CostGraphDef_Node.ProtoReflect.Descriptor instead.

func (*CostGraphDef_Node) GetComputeCost 

func (x *CostGraphDef_Node) GetComputeCost() int64

func (*CostGraphDef_Node) GetComputeTime 

func (x *CostGraphDef_Node) GetComputeTime() int64

func (*CostGraphDef_Node) GetControlInput 

func (x *CostGraphDef_Node) GetControlInput() []int32

func (*CostGraphDef_Node) GetDevice 

func (x *CostGraphDef_Node) GetDevice() string

func (*CostGraphDef_Node) GetDevicePersistentMemorySize deprecated 

func (x *CostGraphDef_Node) GetDevicePersistentMemorySize() int64

Deprecated: Do not use.

func (*CostGraphDef_Node) GetDeviceTempMemorySize deprecated 

func (x *CostGraphDef_Node) GetDeviceTempMemorySize() int64

Deprecated: Do not use.

func (*CostGraphDef_Node) GetHostTempMemorySize deprecated 

func (x *CostGraphDef_Node) GetHostTempMemorySize() int64

Deprecated: Do not use.

func (*CostGraphDef_Node) GetId 

func (x *CostGraphDef_Node) GetId() int32

func (*CostGraphDef_Node) GetInaccurate 

func (x *CostGraphDef_Node) GetInaccurate() bool

func (*CostGraphDef_Node) GetInputInfo 

func (x *CostGraphDef_Node) GetInputInfo() []*CostGraphDef_Node_InputInfo

func (*CostGraphDef_Node) GetIsFinal 

func (x *CostGraphDef_Node) GetIsFinal() bool

func (*CostGraphDef_Node) GetMemoryTime 

func (x *CostGraphDef_Node) GetMemoryTime() int64

func (*CostGraphDef_Node) GetName 

func (x *CostGraphDef_Node) GetName() string

func (*CostGraphDef_Node) GetOutputInfo 

func (x *CostGraphDef_Node) GetOutputInfo() []*CostGraphDef_Node_OutputInfo

func (*CostGraphDef_Node) GetPersistentMemorySize 

func (x *CostGraphDef_Node) GetPersistentMemorySize() int64

func (*CostGraphDef_Node) GetTemporaryMemorySize 

func (x *CostGraphDef_Node) GetTemporaryMemorySize() int64

func (*CostGraphDef_Node) ProtoMessage 

func (*CostGraphDef_Node) ProtoMessage()

func (*CostGraphDef_Node) ProtoReflect 

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

func (*CostGraphDef_Node) Reset 

func (x *CostGraphDef_Node) Reset()

func (*CostGraphDef_Node) String 

func (x *CostGraphDef_Node) String() string

type CostGraphDef_Node_InputInfo 

type CostGraphDef_Node_InputInfo struct {
	PrecedingNode int32 `protobuf:"varint,1,opt,name=preceding_node,json=precedingNode,proto3" json:"preceding_node,omitempty"`
	PrecedingPort int32 `protobuf:"varint,2,opt,name=preceding_port,json=precedingPort,proto3" json:"preceding_port,omitempty"`
	// contains filtered or unexported fields
}

Inputs of this node. They must be executed before this node can be executed. An input is a particular output of another node, specified by the node id and the output index.

func (*CostGraphDef_Node_InputInfo) Descriptor deprecated 

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

Deprecated: Use CostGraphDef_Node_InputInfo.ProtoReflect.Descriptor instead.

func (*CostGraphDef_Node_InputInfo) GetPrecedingNode 

func (x *CostGraphDef_Node_InputInfo) GetPrecedingNode() int32

func (*CostGraphDef_Node_InputInfo) GetPrecedingPort 

func (x *CostGraphDef_Node_InputInfo) GetPrecedingPort() int32

func (*CostGraphDef_Node_InputInfo) ProtoMessage 

func (*CostGraphDef_Node_InputInfo) ProtoMessage()

func (*CostGraphDef_Node_InputInfo) ProtoReflect 

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

func (*CostGraphDef_Node_InputInfo) Reset 

func (x *CostGraphDef_Node_InputInfo) Reset()

func (*CostGraphDef_Node_InputInfo) String 

func (x *CostGraphDef_Node_InputInfo) String() string

type CostGraphDef_Node_OutputInfo 

type CostGraphDef_Node_OutputInfo struct {
	Size int64 `protobuf:"varint,1,opt,name=size,proto3" json:"size,omitempty"`
	// If >= 0, the output is an alias of an input. Note that an alias input
	// may itself be an alias. The algorithm will therefore need to follow
	// those pointers.
	AliasInputPort int64                                   `protobuf:"varint,2,opt,name=alias_input_port,json=aliasInputPort,proto3" json:"alias_input_port,omitempty"`
	Shape          *tensor_shape_go_proto.TensorShapeProto `protobuf:"bytes,3,opt,name=shape,proto3" json:"shape,omitempty"`
	Dtype          types_go_proto.DataType                 `protobuf:"varint,4,opt,name=dtype,proto3,enum=tensorflow.DataType" json:"dtype,omitempty"`
	// contains filtered or unexported fields
}

Outputs of this node.

func (*CostGraphDef_Node_OutputInfo) Descriptor deprecated 

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

Deprecated: Use CostGraphDef_Node_OutputInfo.ProtoReflect.Descriptor instead.

func (*CostGraphDef_Node_OutputInfo) GetAliasInputPort 

func (x *CostGraphDef_Node_OutputInfo) GetAliasInputPort() int64

func (*CostGraphDef_Node_OutputInfo) GetDtype 

func (x *CostGraphDef_Node_OutputInfo) GetDtype() types_go_proto.DataType

func (*CostGraphDef_Node_OutputInfo) GetShape 

func (x *CostGraphDef_Node_OutputInfo) GetShape() *tensor_shape_go_proto.TensorShapeProto

func (*CostGraphDef_Node_OutputInfo) GetSize 

func (x *CostGraphDef_Node_OutputInfo) GetSize() int64

func (*CostGraphDef_Node_OutputInfo) ProtoMessage 

func (*CostGraphDef_Node_OutputInfo) ProtoMessage()

func (*CostGraphDef_Node_OutputInfo) ProtoReflect 

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

func (*CostGraphDef_Node_OutputInfo) Reset 

func (x *CostGraphDef_Node_OutputInfo) Reset()

func (*CostGraphDef_Node_OutputInfo) String 

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