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

func (*CostGraphDef) Descriptor

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

func (*CostGraphDef) GetCost

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

func (*CostGraphDef) GetNode

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

func (*CostGraphDef) ProtoMessage

func (*CostGraphDef) ProtoMessage()

func (*CostGraphDef) Reset

func (m *CostGraphDef) Reset()

func (*CostGraphDef) String

func (m *CostGraphDef) String() string

func (*CostGraphDef) XXX_DiscardUnknown

func (m *CostGraphDef) XXX_DiscardUnknown()

func (*CostGraphDef) XXX_Marshal

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

func (*CostGraphDef) XXX_Merge

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

func (*CostGraphDef) XXX_Size

func (m *CostGraphDef) XXX_Size() int

func (*CostGraphDef) XXX_Unmarshal

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

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

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

func (*CostGraphDef_AggregatedCost) Descriptor

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

func (*CostGraphDef_AggregatedCost) GetCost

func (m *CostGraphDef_AggregatedCost) GetCost() float32

func (*CostGraphDef_AggregatedCost) GetDimension

func (m *CostGraphDef_AggregatedCost) GetDimension() string

func (*CostGraphDef_AggregatedCost) ProtoMessage

func (*CostGraphDef_AggregatedCost) ProtoMessage()

func (*CostGraphDef_AggregatedCost) Reset

func (m *CostGraphDef_AggregatedCost) Reset()

func (*CostGraphDef_AggregatedCost) String

func (m *CostGraphDef_AggregatedCost) String() string

func (*CostGraphDef_AggregatedCost) XXX_DiscardUnknown

func (m *CostGraphDef_AggregatedCost) XXX_DiscardUnknown()

func (*CostGraphDef_AggregatedCost) XXX_Marshal

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

func (*CostGraphDef_AggregatedCost) XXX_Merge

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

func (*CostGraphDef_AggregatedCost) XXX_Size

func (m *CostGraphDef_AggregatedCost) XXX_Size() int

func (*CostGraphDef_AggregatedCost) XXX_Unmarshal

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

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

func (*CostGraphDef_Node) Descriptor

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

func (*CostGraphDef_Node) GetComputeCost

func (m *CostGraphDef_Node) GetComputeCost() int64

func (*CostGraphDef_Node) GetComputeTime

func (m *CostGraphDef_Node) GetComputeTime() int64

func (*CostGraphDef_Node) GetControlInput

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

func (*CostGraphDef_Node) GetDevice

func (m *CostGraphDef_Node) GetDevice() string

func (*CostGraphDef_Node) GetDevicePersistentMemorySize deprecated

func (m *CostGraphDef_Node) GetDevicePersistentMemorySize() int64

Deprecated: Do not use.

func (*CostGraphDef_Node) GetDeviceTempMemorySize deprecated

func (m *CostGraphDef_Node) GetDeviceTempMemorySize() int64

Deprecated: Do not use.

func (*CostGraphDef_Node) GetHostTempMemorySize deprecated

func (m *CostGraphDef_Node) GetHostTempMemorySize() int64

Deprecated: Do not use.

func (*CostGraphDef_Node) GetId

func (m *CostGraphDef_Node) GetId() int32

func (*CostGraphDef_Node) GetInaccurate

func (m *CostGraphDef_Node) GetInaccurate() bool

func (*CostGraphDef_Node) GetInputInfo

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

func (*CostGraphDef_Node) GetIsFinal

func (m *CostGraphDef_Node) GetIsFinal() bool

func (*CostGraphDef_Node) GetMemoryTime

func (m *CostGraphDef_Node) GetMemoryTime() int64

func (*CostGraphDef_Node) GetName

func (m *CostGraphDef_Node) GetName() string

func (*CostGraphDef_Node) GetOutputInfo

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

func (*CostGraphDef_Node) GetPersistentMemorySize

func (m *CostGraphDef_Node) GetPersistentMemorySize() int64

func (*CostGraphDef_Node) GetTemporaryMemorySize

func (m *CostGraphDef_Node) GetTemporaryMemorySize() int64

func (*CostGraphDef_Node) ProtoMessage

func (*CostGraphDef_Node) ProtoMessage()

func (*CostGraphDef_Node) Reset

func (m *CostGraphDef_Node) Reset()

func (*CostGraphDef_Node) String

func (m *CostGraphDef_Node) String() string

func (*CostGraphDef_Node) XXX_DiscardUnknown

func (m *CostGraphDef_Node) XXX_DiscardUnknown()

func (*CostGraphDef_Node) XXX_Marshal

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

func (*CostGraphDef_Node) XXX_Merge

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

func (*CostGraphDef_Node) XXX_Size

func (m *CostGraphDef_Node) XXX_Size() int

func (*CostGraphDef_Node) XXX_Unmarshal

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

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

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

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

func (*CostGraphDef_Node_InputInfo) GetPrecedingNode

func (m *CostGraphDef_Node_InputInfo) GetPrecedingNode() int32

func (*CostGraphDef_Node_InputInfo) GetPrecedingPort

func (m *CostGraphDef_Node_InputInfo) GetPrecedingPort() int32

func (*CostGraphDef_Node_InputInfo) ProtoMessage

func (*CostGraphDef_Node_InputInfo) ProtoMessage()

func (*CostGraphDef_Node_InputInfo) Reset

func (m *CostGraphDef_Node_InputInfo) Reset()

func (*CostGraphDef_Node_InputInfo) String

func (m *CostGraphDef_Node_InputInfo) String() string

func (*CostGraphDef_Node_InputInfo) XXX_DiscardUnknown

func (m *CostGraphDef_Node_InputInfo) XXX_DiscardUnknown()

func (*CostGraphDef_Node_InputInfo) XXX_Marshal

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

func (*CostGraphDef_Node_InputInfo) XXX_Merge

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

func (*CostGraphDef_Node_InputInfo) XXX_Size

func (m *CostGraphDef_Node_InputInfo) XXX_Size() int

func (*CostGraphDef_Node_InputInfo) XXX_Unmarshal

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

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

Outputs of this node.

func (*CostGraphDef_Node_OutputInfo) Descriptor

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

func (*CostGraphDef_Node_OutputInfo) GetAliasInputPort

func (m *CostGraphDef_Node_OutputInfo) GetAliasInputPort() int64

func (*CostGraphDef_Node_OutputInfo) GetDtype

func (*CostGraphDef_Node_OutputInfo) GetShape

func (*CostGraphDef_Node_OutputInfo) GetSize

func (m *CostGraphDef_Node_OutputInfo) GetSize() int64

func (*CostGraphDef_Node_OutputInfo) ProtoMessage

func (*CostGraphDef_Node_OutputInfo) ProtoMessage()

func (*CostGraphDef_Node_OutputInfo) Reset

func (m *CostGraphDef_Node_OutputInfo) Reset()

func (*CostGraphDef_Node_OutputInfo) String

func (*CostGraphDef_Node_OutputInfo) XXX_DiscardUnknown

func (m *CostGraphDef_Node_OutputInfo) XXX_DiscardUnknown()

func (*CostGraphDef_Node_OutputInfo) XXX_Marshal

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

func (*CostGraphDef_Node_OutputInfo) XXX_Merge

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

func (*CostGraphDef_Node_OutputInfo) XXX_Size

func (m *CostGraphDef_Node_OutputInfo) XXX_Size() int

func (*CostGraphDef_Node_OutputInfo) XXX_Unmarshal

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

Jump to

Keyboard shortcuts

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