model_go_proto

package
v0.7.0 Latest Latest
Warning

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

 Go to latest Published: Nov 17, 2023 License: Apache-2.0 Imports: 4 Imported by: 0

Documentation

Index

Constants

This section is empty.

Variables

View Source 
var (
	NodeClass_name = map[int32]string{
		0: "UNKNOWN",
		1: "INTERLEAVE_MANY",
		2: "ASYNC_INTERLEAVE_MANY",
		3: "KNOWN_RATIO",
		4: "ASYNC_KNOWN_RATIO",
		5: "UNKNOWN_RATIO",
		6: "ASYNC_UNKNOWN_RATIO",
	}
	NodeClass_value = map[string]int32{
		"UNKNOWN":               0,
		"INTERLEAVE_MANY":       1,
		"ASYNC_INTERLEAVE_MANY": 2,
		"KNOWN_RATIO":           3,
		"ASYNC_KNOWN_RATIO":     4,
		"UNKNOWN_RATIO":         5,
		"ASYNC_UNKNOWN_RATIO":   6,
	}
)

Enum value maps for NodeClass.

View Source 
var (
	AutotuneAlgorithm_name = map[int32]string{
		0: "DEFAULT",
		1: "HILL_CLIMB",
		2: "GRADIENT_DESCENT",
		3: "MAX_PARALLELISM",
		4: "STAGE_BASED",
	}
	AutotuneAlgorithm_value = map[string]int32{
		"DEFAULT":          0,
		"HILL_CLIMB":       1,
		"GRADIENT_DESCENT": 2,
		"MAX_PARALLELISM":  3,
		"STAGE_BASED":      4,
	}
)

Enum value maps for AutotuneAlgorithm.

View Source 
var File_tensorflow_core_framework_model_proto protoreflect.FileDescriptor

Functions

This section is empty.

Types

type AutotuneAlgorithm 

type AutotuneAlgorithm int32

Algorithm used for model autotuning optimization. 

const (
	AutotuneAlgorithm_DEFAULT          AutotuneAlgorithm = 0
	AutotuneAlgorithm_HILL_CLIMB       AutotuneAlgorithm = 1
	AutotuneAlgorithm_GRADIENT_DESCENT AutotuneAlgorithm = 2
	AutotuneAlgorithm_MAX_PARALLELISM  AutotuneAlgorithm = 3
	AutotuneAlgorithm_STAGE_BASED      AutotuneAlgorithm = 4
)

func (AutotuneAlgorithm) Descriptor 

func (AutotuneAlgorithm) Descriptor() protoreflect.EnumDescriptor

func (AutotuneAlgorithm) Enum 

func (x AutotuneAlgorithm) Enum() *AutotuneAlgorithm

func (AutotuneAlgorithm) EnumDescriptor deprecated 

func (AutotuneAlgorithm) EnumDescriptor() ([]byte, []int)

Deprecated: Use AutotuneAlgorithm.Descriptor instead.

func (AutotuneAlgorithm) Number 

func (x AutotuneAlgorithm) Number() protoreflect.EnumNumber

func (AutotuneAlgorithm) String 

func (x AutotuneAlgorithm) String() string

func (AutotuneAlgorithm) Type 

func (AutotuneAlgorithm) Type() protoreflect.EnumType

type ModelProto 

type ModelProto struct {

	// Map of node IDs to nodes of this model.
	Nodes map[int64]*ModelProto_Node `` /* 152-byte string literal not displayed */
	// ID of the output node of this model.
	Output int64 `protobuf:"varint,2,opt,name=output,proto3" json:"output,omitempty"`
	// Counter for node IDs of this model.
	IdCounter          int64                          `protobuf:"varint,3,opt,name=id_counter,json=idCounter,proto3" json:"id_counter,omitempty"`
	OptimizationParams *ModelProto_OptimizationParams `protobuf:"bytes,5,opt,name=optimization_params,json=optimizationParams,proto3" json:"optimization_params,omitempty"`
	GapTimes           []uint64                       `protobuf:"varint,6,rep,packed,name=gap_times,json=gapTimes,proto3" json:"gap_times,omitempty"`
	// contains filtered or unexported fields
}

Protocol buffer representing the data used by the autotuning modeling framework.

func (*ModelProto) Descriptor deprecated 

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

Deprecated: Use ModelProto.ProtoReflect.Descriptor instead.

func (*ModelProto) GetGapTimes added in v0.5.0 

func (x *ModelProto) GetGapTimes() []uint64

func (*ModelProto) GetIdCounter 

func (x *ModelProto) GetIdCounter() int64

func (*ModelProto) GetNodes 

func (x *ModelProto) GetNodes() map[int64]*ModelProto_Node

func (*ModelProto) GetOptimizationParams 

func (x *ModelProto) GetOptimizationParams() *ModelProto_OptimizationParams

func (*ModelProto) GetOutput 

func (x *ModelProto) GetOutput() int64

func (*ModelProto) ProtoMessage 

func (*ModelProto) ProtoMessage()

func (*ModelProto) ProtoReflect 

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

func (*ModelProto) Reset 

func (x *ModelProto) Reset()

func (*ModelProto) String 

func (x *ModelProto) String() string

type ModelProto_Node 

type ModelProto_Node struct {

	// Unique node ID.
	Id int64 `protobuf:"varint,1,opt,name=id,proto3" json:"id,omitempty"`
	// Human-readable name of the node.
	Name string `protobuf:"bytes,2,opt,name=name,proto3" json:"name,omitempty"`
	// An indication whether autotuning is enabled for this node.
	Autotune bool `protobuf:"varint,3,opt,name=autotune,proto3" json:"autotune,omitempty"`
	// The number of bytes stored in this node's buffer.
	BufferedBytes int64 `protobuf:"varint,4,opt,name=buffered_bytes,json=bufferedBytes,proto3" json:"buffered_bytes,omitempty"`
	// The number of elements stored in this node's buffer.
	BufferedElements int64 `protobuf:"varint,5,opt,name=buffered_elements,json=bufferedElements,proto3" json:"buffered_elements,omitempty"`
	// The number of bytes consumed by the node.
	BytesConsumed int64 `protobuf:"varint,6,opt,name=bytes_consumed,json=bytesConsumed,proto3" json:"bytes_consumed,omitempty"`
	// The number of bytes produced by the node.
	BytesProduced int64 `protobuf:"varint,7,opt,name=bytes_produced,json=bytesProduced,proto3" json:"bytes_produced,omitempty"`
	// The number of elements produced by the node.
	NumElements int64 `protobuf:"varint,8,opt,name=num_elements,json=numElements,proto3" json:"num_elements,omitempty"`
	// The aggregate processing time spent in this node in nanoseconds.
	ProcessingTime int64 `protobuf:"varint,9,opt,name=processing_time,json=processingTime,proto3" json:"processing_time,omitempty"`
	// An indication whether this node records metrics about produced and
	// consumed elements.
	RecordMetrics bool `protobuf:"varint,10,opt,name=record_metrics,json=recordMetrics,proto3" json:"record_metrics,omitempty"`
	// Parameters of this node.
	Parameters []*ModelProto_Node_Parameter `protobuf:"bytes,11,rep,name=parameters,proto3" json:"parameters,omitempty"`
	// Statistic of inputs processing time history.
	InputProcessingTimeSum   float64 `` /* 134-byte string literal not displayed */
	InputProcessingTimeCount int64   `` /* 139-byte string literal not displayed */
	// IDs of inputs of this node.
	Inputs []int64 `protobuf:"varint,14,rep,packed,name=inputs,proto3" json:"inputs,omitempty"`
	// Class of this node.
	NodeClass NodeClass `` /* 127-byte string literal not displayed */
	// Ratio of input to output elements. This is only used by KNOWN_RATIO and
	// ASYNC_KNOWN_RATIO nodes.
	Ratio float64 `protobuf:"fixed64,16,opt,name=ratio,proto3" json:"ratio,omitempty"`
	// Ratio identifies how many parallelism calls are introduced by one
	// buffered element. This is only used by ASYNC_KNOWN_RATIO nodes.
	MemoryRatio float64 `protobuf:"fixed64,17,opt,name=memory_ratio,json=memoryRatio,proto3" json:"memory_ratio,omitempty"`
	// contains filtered or unexported fields
}

General representation of a node in the model.

func (*ModelProto_Node) Descriptor deprecated 

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

Deprecated: Use ModelProto_Node.ProtoReflect.Descriptor instead.

func (*ModelProto_Node) GetAutotune 

func (x *ModelProto_Node) GetAutotune() bool

func (*ModelProto_Node) GetBufferedBytes 

func (x *ModelProto_Node) GetBufferedBytes() int64

func (*ModelProto_Node) GetBufferedElements 

func (x *ModelProto_Node) GetBufferedElements() int64

func (*ModelProto_Node) GetBytesConsumed 

func (x *ModelProto_Node) GetBytesConsumed() int64

func (*ModelProto_Node) GetBytesProduced 

func (x *ModelProto_Node) GetBytesProduced() int64

func (*ModelProto_Node) GetId 

func (x *ModelProto_Node) GetId() int64

func (*ModelProto_Node) GetInputProcessingTimeCount 

func (x *ModelProto_Node) GetInputProcessingTimeCount() int64

func (*ModelProto_Node) GetInputProcessingTimeSum 

func (x *ModelProto_Node) GetInputProcessingTimeSum() float64

func (*ModelProto_Node) GetInputs 

func (x *ModelProto_Node) GetInputs() []int64

func (*ModelProto_Node) GetMemoryRatio 

func (x *ModelProto_Node) GetMemoryRatio() float64

func (*ModelProto_Node) GetName 

func (x *ModelProto_Node) GetName() string

func (*ModelProto_Node) GetNodeClass 

func (x *ModelProto_Node) GetNodeClass() NodeClass

func (*ModelProto_Node) GetNumElements 

func (x *ModelProto_Node) GetNumElements() int64

func (*ModelProto_Node) GetParameters 

func (x *ModelProto_Node) GetParameters() []*ModelProto_Node_Parameter

func (*ModelProto_Node) GetProcessingTime 

func (x *ModelProto_Node) GetProcessingTime() int64

func (*ModelProto_Node) GetRatio 

func (x *ModelProto_Node) GetRatio() float64

func (*ModelProto_Node) GetRecordMetrics 

func (x *ModelProto_Node) GetRecordMetrics() bool

func (*ModelProto_Node) ProtoMessage 

func (*ModelProto_Node) ProtoMessage()

func (*ModelProto_Node) ProtoReflect 

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

func (*ModelProto_Node) Reset 

func (x *ModelProto_Node) Reset()

func (*ModelProto_Node) String 

func (x *ModelProto_Node) String() string

type ModelProto_Node_Parameter 

type ModelProto_Node_Parameter struct {

	// Human-readable name of the parameter.
	Name string `protobuf:"bytes,1,opt,name=name,proto3" json:"name,omitempty"`
	// Identifies the model value of the parameter. This can be different from
	// the actual value (e.g. during optimization search).
	Value float64 `protobuf:"fixed64,2,opt,name=value,proto3" json:"value,omitempty"`
	// The actual value of the parameter.
	StateValue float64 `protobuf:"fixed64,3,opt,name=state_value,json=stateValue,proto3" json:"state_value,omitempty"`
	// Minimum value of the parameter.
	Min float64 `protobuf:"fixed64,4,opt,name=min,proto3" json:"min,omitempty"`
	// Maximum value of the parameter.
	Max float64 `protobuf:"fixed64,5,opt,name=max,proto3" json:"max,omitempty"`
	// Identifies whether the parameter should participate in autotuning.
	Tunable bool `protobuf:"varint,6,opt,name=tunable,proto3" json:"tunable,omitempty"`
	// contains filtered or unexported fields
}

Represents a node parameter.

func (*ModelProto_Node_Parameter) Descriptor deprecated 

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

Deprecated: Use ModelProto_Node_Parameter.ProtoReflect.Descriptor instead.

func (*ModelProto_Node_Parameter) GetMax 

func (x *ModelProto_Node_Parameter) GetMax() float64

func (*ModelProto_Node_Parameter) GetMin 

func (x *ModelProto_Node_Parameter) GetMin() float64

func (*ModelProto_Node_Parameter) GetName 

func (x *ModelProto_Node_Parameter) GetName() string

func (*ModelProto_Node_Parameter) GetStateValue 

func (x *ModelProto_Node_Parameter) GetStateValue() float64

func (*ModelProto_Node_Parameter) GetTunable 

func (x *ModelProto_Node_Parameter) GetTunable() bool

func (*ModelProto_Node_Parameter) GetValue 

func (x *ModelProto_Node_Parameter) GetValue() float64

func (*ModelProto_Node_Parameter) ProtoMessage 

func (*ModelProto_Node_Parameter) ProtoMessage()

func (*ModelProto_Node_Parameter) ProtoReflect 

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

func (*ModelProto_Node_Parameter) Reset 

func (x *ModelProto_Node_Parameter) Reset()

func (*ModelProto_Node_Parameter) String 

func (x *ModelProto_Node_Parameter) String() string

type ModelProto_OptimizationParams 

type ModelProto_OptimizationParams struct {

	// Algorithm used for autotuning optimization.
	Algorithm AutotuneAlgorithm `protobuf:"varint,1,opt,name=algorithm,proto3,enum=tensorflow.data.model.AutotuneAlgorithm" json:"algorithm,omitempty"`
	// Number of available logical threads.
	CpuBudget int64 `protobuf:"varint,2,opt,name=cpu_budget,json=cpuBudget,proto3" json:"cpu_budget,omitempty"`
	// Amount of available memory in bytes.
	RamBudget int64 `protobuf:"varint,3,opt,name=ram_budget,json=ramBudget,proto3" json:"ram_budget,omitempty"`
	// Time between two consecutive `GetNext` calls to the iterator represented
	// by the output node.
	ModelInputTime float64 `protobuf:"fixed64,4,opt,name=model_input_time,json=modelInputTime,proto3" json:"model_input_time,omitempty"`
	// contains filtered or unexported fields
}

Contains parameters of the model autotuning optimization.

func (*ModelProto_OptimizationParams) Descriptor deprecated 

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

Deprecated: Use ModelProto_OptimizationParams.ProtoReflect.Descriptor instead.

func (*ModelProto_OptimizationParams) GetAlgorithm 

func (x *ModelProto_OptimizationParams) GetAlgorithm() AutotuneAlgorithm

func (*ModelProto_OptimizationParams) GetCpuBudget 

func (x *ModelProto_OptimizationParams) GetCpuBudget() int64

func (*ModelProto_OptimizationParams) GetModelInputTime 

func (x *ModelProto_OptimizationParams) GetModelInputTime() float64

func (*ModelProto_OptimizationParams) GetRamBudget 

func (x *ModelProto_OptimizationParams) GetRamBudget() int64

func (*ModelProto_OptimizationParams) ProtoMessage 

func (*ModelProto_OptimizationParams) ProtoMessage()

func (*ModelProto_OptimizationParams) ProtoReflect 

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

func (*ModelProto_OptimizationParams) Reset 

func (x *ModelProto_OptimizationParams) Reset()

func (*ModelProto_OptimizationParams) String 

func (x *ModelProto_OptimizationParams) String() string

type NodeClass 

type NodeClass int32

Class of a node in the performance model. 

const (
	NodeClass_UNKNOWN               NodeClass = 0
	NodeClass_INTERLEAVE_MANY       NodeClass = 1
	NodeClass_ASYNC_INTERLEAVE_MANY NodeClass = 2
	NodeClass_KNOWN_RATIO           NodeClass = 3
	NodeClass_ASYNC_KNOWN_RATIO     NodeClass = 4
	NodeClass_UNKNOWN_RATIO         NodeClass = 5
	NodeClass_ASYNC_UNKNOWN_RATIO   NodeClass = 6
)

func (NodeClass) Descriptor 

func (NodeClass) Descriptor() protoreflect.EnumDescriptor

func (NodeClass) Enum 

func (x NodeClass) Enum() *NodeClass

func (NodeClass) EnumDescriptor deprecated 

func (NodeClass) EnumDescriptor() ([]byte, []int)

Deprecated: Use NodeClass.Descriptor instead.

func (NodeClass) Number 

func (x NodeClass) Number() protoreflect.EnumNumber

func (NodeClass) String 

func (x NodeClass) String() string

func (NodeClass) Type 

func (NodeClass) Type() protoreflect.EnumType

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.