framework

package
v0.1.0 Latest Latest
Warning

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

 Go to latest Published: Feb 8, 2022 License: MIT Imports: 4 Imported by: 0

Documentation

Index

Constants

This section is empty.

Variables

View Source 
var (
	FullTypeId_name = map[int32]string{
		0:     "TFT_UNSET",
		1:     "TFT_VAR",
		2:     "TFT_ANY",
		3:     "TFT_PRODUCT",
		100:   "TFT_CALLABLE",
		1000:  "TFT_TENSOR",
		1001:  "TFT_ARRAY",
		1002:  "TFT_OPTIONAL",
		10102: "TFT_DATASET",
		200:   "TFT_BOOL",
		201:   "TFT_UINT8",
		202:   "TFT_UINT16",
		203:   "TFT_UINT32",
		204:   "TFT_UINT64",
		205:   "TFT_INT8",
		206:   "TFT_INT16",
		207:   "TFT_INT32",
		208:   "TFT_INT64",
		209:   "TFT_HALF",
		210:   "TFT_FLOAT",
		211:   "TFT_DOUBLE",
		215:   "TFT_BFLOAT16",
		212:   "TFT_COMPLEX64",
		213:   "TFT_COMPLEX128",
		214:   "TFT_STRING",
	}
	FullTypeId_value = map[string]int32{
		"TFT_UNSET":      0,
		"TFT_VAR":        1,
		"TFT_ANY":        2,
		"TFT_PRODUCT":    3,
		"TFT_CALLABLE":   100,
		"TFT_TENSOR":     1000,
		"TFT_ARRAY":      1001,
		"TFT_OPTIONAL":   1002,
		"TFT_DATASET":    10102,
		"TFT_BOOL":       200,
		"TFT_UINT8":      201,
		"TFT_UINT16":     202,
		"TFT_UINT32":     203,
		"TFT_UINT64":     204,
		"TFT_INT8":       205,
		"TFT_INT16":      206,
		"TFT_INT32":      207,
		"TFT_INT64":      208,
		"TFT_HALF":       209,
		"TFT_FLOAT":      210,
		"TFT_DOUBLE":     211,
		"TFT_BFLOAT16":   215,
		"TFT_COMPLEX64":  212,
		"TFT_COMPLEX128": 213,
		"TFT_STRING":     214,
	}
)

Enum value maps for FullTypeId.

View Source 
var (
	DataType_name = map[int32]string{
		0:   "DT_INVALID",
		1:   "DT_FLOAT",
		2:   "DT_DOUBLE",
		3:   "DT_INT32",
		4:   "DT_UINT8",
		5:   "DT_INT16",
		6:   "DT_INT8",
		7:   "DT_STRING",
		8:   "DT_COMPLEX64",
		9:   "DT_INT64",
		10:  "DT_BOOL",
		11:  "DT_QINT8",
		12:  "DT_QUINT8",
		13:  "DT_QINT32",
		14:  "DT_BFLOAT16",
		15:  "DT_QINT16",
		16:  "DT_QUINT16",
		17:  "DT_UINT16",
		18:  "DT_COMPLEX128",
		19:  "DT_HALF",
		20:  "DT_RESOURCE",
		21:  "DT_VARIANT",
		22:  "DT_UINT32",
		23:  "DT_UINT64",
		101: "DT_FLOAT_REF",
		102: "DT_DOUBLE_REF",
		103: "DT_INT32_REF",
		104: "DT_UINT8_REF",
		105: "DT_INT16_REF",
		106: "DT_INT8_REF",
		107: "DT_STRING_REF",
		108: "DT_COMPLEX64_REF",
		109: "DT_INT64_REF",
		110: "DT_BOOL_REF",
		111: "DT_QINT8_REF",
		112: "DT_QUINT8_REF",
		113: "DT_QINT32_REF",
		114: "DT_BFLOAT16_REF",
		115: "DT_QINT16_REF",
		116: "DT_QUINT16_REF",
		117: "DT_UINT16_REF",
		118: "DT_COMPLEX128_REF",
		119: "DT_HALF_REF",
		120: "DT_RESOURCE_REF",
		121: "DT_VARIANT_REF",
		122: "DT_UINT32_REF",
		123: "DT_UINT64_REF",
	}
	DataType_value = map[string]int32{
		"DT_INVALID":        0,
		"DT_FLOAT":          1,
		"DT_DOUBLE":         2,
		"DT_INT32":          3,
		"DT_UINT8":          4,
		"DT_INT16":          5,
		"DT_INT8":           6,
		"DT_STRING":         7,
		"DT_COMPLEX64":      8,
		"DT_INT64":          9,
		"DT_BOOL":           10,
		"DT_QINT8":          11,
		"DT_QUINT8":         12,
		"DT_QINT32":         13,
		"DT_BFLOAT16":       14,
		"DT_QINT16":         15,
		"DT_QUINT16":        16,
		"DT_UINT16":         17,
		"DT_COMPLEX128":     18,
		"DT_HALF":           19,
		"DT_RESOURCE":       20,
		"DT_VARIANT":        21,
		"DT_UINT32":         22,
		"DT_UINT64":         23,
		"DT_FLOAT_REF":      101,
		"DT_DOUBLE_REF":     102,
		"DT_INT32_REF":      103,
		"DT_UINT8_REF":      104,
		"DT_INT16_REF":      105,
		"DT_INT8_REF":       106,
		"DT_STRING_REF":     107,
		"DT_COMPLEX64_REF":  108,
		"DT_INT64_REF":      109,
		"DT_BOOL_REF":       110,
		"DT_QINT8_REF":      111,
		"DT_QUINT8_REF":     112,
		"DT_QINT32_REF":     113,
		"DT_BFLOAT16_REF":   114,
		"DT_QINT16_REF":     115,
		"DT_QUINT16_REF":    116,
		"DT_UINT16_REF":     117,
		"DT_COMPLEX128_REF": 118,
		"DT_HALF_REF":       119,
		"DT_RESOURCE_REF":   120,
		"DT_VARIANT_REF":    121,
		"DT_UINT32_REF":     122,
		"DT_UINT64_REF":     123,
	}
)

Enum value maps for DataType.

View Source 
var (
	VariableSynchronization_name = map[int32]string{
		0: "VARIABLE_SYNCHRONIZATION_AUTO",
		1: "VARIABLE_SYNCHRONIZATION_NONE",
		2: "VARIABLE_SYNCHRONIZATION_ON_WRITE",
		3: "VARIABLE_SYNCHRONIZATION_ON_READ",
	}
	VariableSynchronization_value = map[string]int32{
		"VARIABLE_SYNCHRONIZATION_AUTO":     0,
		"VARIABLE_SYNCHRONIZATION_NONE":     1,
		"VARIABLE_SYNCHRONIZATION_ON_WRITE": 2,
		"VARIABLE_SYNCHRONIZATION_ON_READ":  3,
	}
)

Enum value maps for VariableSynchronization.

View Source 
var (
	VariableAggregation_name = map[int32]string{
		0: "VARIABLE_AGGREGATION_NONE",
		1: "VARIABLE_AGGREGATION_SUM",
		2: "VARIABLE_AGGREGATION_MEAN",
		3: "VARIABLE_AGGREGATION_ONLY_FIRST_REPLICA",
	}
	VariableAggregation_value = map[string]int32{
		"VARIABLE_AGGREGATION_NONE":               0,
		"VARIABLE_AGGREGATION_SUM":                1,
		"VARIABLE_AGGREGATION_MEAN":               2,
		"VARIABLE_AGGREGATION_ONLY_FIRST_REPLICA": 3,
	}
)

Enum value maps for VariableAggregation.

View Source 
var File_tensorflow_core_framework_allocation_description_proto protoreflect.FileDescriptor
View Source 
var File_tensorflow_core_framework_attr_value_proto protoreflect.FileDescriptor
View Source 
var File_tensorflow_core_framework_cost_graph_proto protoreflect.FileDescriptor
View Source 
var File_tensorflow_core_framework_full_type_proto protoreflect.FileDescriptor
View Source 
var File_tensorflow_core_framework_function_proto protoreflect.FileDescriptor
View Source 
var File_tensorflow_core_framework_graph_proto protoreflect.FileDescriptor
View Source 
var File_tensorflow_core_framework_node_def_proto protoreflect.FileDescriptor
View Source 
var File_tensorflow_core_framework_op_def_proto protoreflect.FileDescriptor
View Source 
var File_tensorflow_core_framework_resource_handle_proto protoreflect.FileDescriptor
View Source 
var File_tensorflow_core_framework_step_stats_proto protoreflect.FileDescriptor
View Source 
var File_tensorflow_core_framework_tensor_description_proto protoreflect.FileDescriptor
View Source 
var File_tensorflow_core_framework_tensor_proto protoreflect.FileDescriptor
View Source 
var File_tensorflow_core_framework_tensor_shape_proto protoreflect.FileDescriptor
View Source 
var File_tensorflow_core_framework_types_proto protoreflect.FileDescriptor
View Source 
var File_tensorflow_core_framework_variable_proto protoreflect.FileDescriptor
View Source 
var File_tensorflow_core_framework_versions_proto protoreflect.FileDescriptor

Functions

This section is empty.

Types

type AllocationDescription 

type AllocationDescription struct {

	// Total number of bytes requested
	RequestedBytes int64 `protobuf:"varint,1,opt,name=requested_bytes,json=requestedBytes,proto3" json:"requested_bytes,omitempty"`
	// Total number of bytes allocated if known
	AllocatedBytes int64 `protobuf:"varint,2,opt,name=allocated_bytes,json=allocatedBytes,proto3" json:"allocated_bytes,omitempty"`
	// Name of the allocator used
	AllocatorName string `protobuf:"bytes,3,opt,name=allocator_name,json=allocatorName,proto3" json:"allocator_name,omitempty"`
	// Identifier of the allocated buffer if known
	AllocationId int64 `protobuf:"varint,4,opt,name=allocation_id,json=allocationId,proto3" json:"allocation_id,omitempty"`
	// Set if this tensor only has one remaining reference
	HasSingleReference bool `protobuf:"varint,5,opt,name=has_single_reference,json=hasSingleReference,proto3" json:"has_single_reference,omitempty"`
	// Address of the allocation.
	Ptr uint64 `protobuf:"varint,6,opt,name=ptr,proto3" json:"ptr,omitempty"`
	// contains filtered or unexported fields
}

func (*AllocationDescription) Descriptor deprecated 

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

Deprecated: Use AllocationDescription.ProtoReflect.Descriptor instead.

func (*AllocationDescription) GetAllocatedBytes 

func (x *AllocationDescription) GetAllocatedBytes() int64

func (*AllocationDescription) GetAllocationId 

func (x *AllocationDescription) GetAllocationId() int64

func (*AllocationDescription) GetAllocatorName 

func (x *AllocationDescription) GetAllocatorName() string

func (*AllocationDescription) GetHasSingleReference 

func (x *AllocationDescription) GetHasSingleReference() bool

func (*AllocationDescription) GetPtr 

func (x *AllocationDescription) GetPtr() uint64

func (*AllocationDescription) GetRequestedBytes 

func (x *AllocationDescription) GetRequestedBytes() int64

func (*AllocationDescription) ProtoMessage 

func (*AllocationDescription) ProtoMessage()

func (*AllocationDescription) ProtoReflect 

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

func (*AllocationDescription) Reset 

func (x *AllocationDescription) Reset()

func (*AllocationDescription) String 

func (x *AllocationDescription) String() string

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"`
	// contains filtered or unexported fields
}

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

func (*AllocationRecord) Descriptor deprecated 

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

Deprecated: Use AllocationRecord.ProtoReflect.Descriptor instead.

func (*AllocationRecord) GetAllocBytes 

func (x *AllocationRecord) GetAllocBytes() int64

func (*AllocationRecord) GetAllocMicros 

func (x *AllocationRecord) GetAllocMicros() int64

func (*AllocationRecord) ProtoMessage 

func (*AllocationRecord) ProtoMessage()

func (*AllocationRecord) ProtoReflect 

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

func (*AllocationRecord) Reset 

func (x *AllocationRecord) Reset()

func (*AllocationRecord) String 

func (x *AllocationRecord) String() string

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"`
	// contains filtered or unexported fields
}

func (*AllocatorMemoryUsed) Descriptor deprecated 

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

Deprecated: Use AllocatorMemoryUsed.ProtoReflect.Descriptor instead.

func (*AllocatorMemoryUsed) GetAllocationRecords 

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

func (*AllocatorMemoryUsed) GetAllocatorBytesInUse 

func (x *AllocatorMemoryUsed) GetAllocatorBytesInUse() int64

func (*AllocatorMemoryUsed) GetAllocatorName 

func (x *AllocatorMemoryUsed) GetAllocatorName() string

func (*AllocatorMemoryUsed) GetLiveBytes 

func (x *AllocatorMemoryUsed) GetLiveBytes() int64

func (*AllocatorMemoryUsed) GetPeakBytes 

func (x *AllocatorMemoryUsed) GetPeakBytes() int64

func (*AllocatorMemoryUsed) GetTotalBytes 

func (x *AllocatorMemoryUsed) GetTotalBytes() int64

func (*AllocatorMemoryUsed) ProtoMessage 

func (*AllocatorMemoryUsed) ProtoMessage()

func (*AllocatorMemoryUsed) ProtoReflect 

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

func (*AllocatorMemoryUsed) Reset 

func (x *AllocatorMemoryUsed) Reset()

func (*AllocatorMemoryUsed) String 

func (x *AllocatorMemoryUsed) String() string

type AttrValue 

type AttrValue struct {

	// Types that are assignable to Value:
	//	*AttrValue_S
	//	*AttrValue_I
	//	*AttrValue_F
	//	*AttrValue_B
	//	*AttrValue_Type
	//	*AttrValue_Shape
	//	*AttrValue_Tensor
	//	*AttrValue_List
	//	*AttrValue_Func
	//	*AttrValue_Placeholder
	Value isAttrValue_Value `protobuf_oneof:"value"`
	// contains filtered or unexported fields
}

Protocol buffer representing the value for an attr used to configure an Op. Comment indicates the corresponding attr type. Only the field matching the attr type may be filled.

func (*AttrValue) Descriptor deprecated 

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

Deprecated: Use AttrValue.ProtoReflect.Descriptor instead.

func (*AttrValue) GetB 

func (x *AttrValue) GetB() bool

func (*AttrValue) GetF 

func (x *AttrValue) GetF() float32

func (*AttrValue) GetFunc 

func (x *AttrValue) GetFunc() *NameAttrList

func (*AttrValue) GetI 

func (x *AttrValue) GetI() int64

func (*AttrValue) GetList 

func (x *AttrValue) GetList() *AttrValue_ListValue

func (*AttrValue) GetPlaceholder 

func (x *AttrValue) GetPlaceholder() string

func (*AttrValue) GetS 

func (x *AttrValue) GetS() []byte

func (*AttrValue) GetShape 

func (x *AttrValue) GetShape() *TensorShapeProto

func (*AttrValue) GetTensor 

func (x *AttrValue) GetTensor() *TensorProto

func (*AttrValue) GetType 

func (x *AttrValue) GetType() DataType

func (*AttrValue) GetValue 

func (m *AttrValue) GetValue() isAttrValue_Value

func (*AttrValue) ProtoMessage 

func (*AttrValue) ProtoMessage()

func (*AttrValue) ProtoReflect 

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

func (*AttrValue) Reset 

func (x *AttrValue) Reset()

func (*AttrValue) String 

func (x *AttrValue) String() string

type AttrValue_B 

type AttrValue_B struct {
	B bool `protobuf:"varint,5,opt,name=b,proto3,oneof"` // "bool"
}

type AttrValue_F 

type AttrValue_F struct {
	F float32 `protobuf:"fixed32,4,opt,name=f,proto3,oneof"` // "float"
}

type AttrValue_Func 

type AttrValue_Func struct {
	// "func" represents a function. func.name is a function's name or
	// a primitive op's name. func.attr.first is the name of an attr
	// defined for that function. func.attr.second is the value for
	// that attr in the instantiation.
	Func *NameAttrList `protobuf:"bytes,10,opt,name=func,proto3,oneof"`
}

type AttrValue_I 

type AttrValue_I struct {
	I int64 `protobuf:"varint,3,opt,name=i,proto3,oneof"` // "int"
}

type AttrValue_List 

type AttrValue_List struct {
	List *AttrValue_ListValue `protobuf:"bytes,1,opt,name=list,proto3,oneof"` // any "list(...)"
}

type AttrValue_ListValue 

type AttrValue_ListValue struct {
	S      [][]byte            `protobuf:"bytes,2,rep,name=s,proto3" json:"s,omitempty"`                                        // "list(string)"
	I      []int64             `protobuf:"varint,3,rep,packed,name=i,proto3" json:"i,omitempty"`                                // "list(int)"
	F      []float32           `protobuf:"fixed32,4,rep,packed,name=f,proto3" json:"f,omitempty"`                               // "list(float)"
	B      []bool              `protobuf:"varint,5,rep,packed,name=b,proto3" json:"b,omitempty"`                                // "list(bool)"
	Type   []DataType          `protobuf:"varint,6,rep,packed,name=type,proto3,enum=tensorflow.DataType" json:"type,omitempty"` // "list(type)"
	Shape  []*TensorShapeProto `protobuf:"bytes,7,rep,name=shape,proto3" json:"shape,omitempty"`                                // "list(shape)"
	Tensor []*TensorProto      `protobuf:"bytes,8,rep,name=tensor,proto3" json:"tensor,omitempty"`                              // "list(tensor)"
	Func   []*NameAttrList     `protobuf:"bytes,9,rep,name=func,proto3" json:"func,omitempty"`                                  // "list(attr)"
	// contains filtered or unexported fields
}

LINT.IfChange

func (*AttrValue_ListValue) Descriptor deprecated 

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

Deprecated: Use AttrValue_ListValue.ProtoReflect.Descriptor instead.

func (*AttrValue_ListValue) GetB 

func (x *AttrValue_ListValue) GetB() []bool

func (*AttrValue_ListValue) GetF 

func (x *AttrValue_ListValue) GetF() []float32

func (*AttrValue_ListValue) GetFunc 

func (x *AttrValue_ListValue) GetFunc() []*NameAttrList

func (*AttrValue_ListValue) GetI 

func (x *AttrValue_ListValue) GetI() []int64

func (*AttrValue_ListValue) GetS 

func (x *AttrValue_ListValue) GetS() [][]byte

func (*AttrValue_ListValue) GetShape 

func (x *AttrValue_ListValue) GetShape() []*TensorShapeProto

func (*AttrValue_ListValue) GetTensor 

func (x *AttrValue_ListValue) GetTensor() []*TensorProto

func (*AttrValue_ListValue) GetType 

func (x *AttrValue_ListValue) GetType() []DataType

func (*AttrValue_ListValue) ProtoMessage 

func (*AttrValue_ListValue) ProtoMessage()

func (*AttrValue_ListValue) ProtoReflect 

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

func (*AttrValue_ListValue) Reset 

func (x *AttrValue_ListValue) Reset()

func (*AttrValue_ListValue) String 

func (x *AttrValue_ListValue) String() string

type AttrValue_Placeholder 

type AttrValue_Placeholder struct {
	// This is a placeholder only used in nodes defined inside a
	// function.  It indicates the attr value will be supplied when
	// the function is instantiated.  For example, let us suppose a
	// node "N" in function "FN". "N" has an attr "A" with value
	// placeholder = "foo". When FN is instantiated with attr "foo"
	// set to "bar", the instantiated node N's attr A will have been
	// given the value "bar".
	Placeholder string `protobuf:"bytes,9,opt,name=placeholder,proto3,oneof"`
}

type AttrValue_S 

type AttrValue_S struct {
	S []byte `protobuf:"bytes,2,opt,name=s,proto3,oneof"` // "string"
}

type AttrValue_Shape 

type AttrValue_Shape struct {
	Shape *TensorShapeProto `protobuf:"bytes,7,opt,name=shape,proto3,oneof"` // "shape"
}

type AttrValue_Tensor 

type AttrValue_Tensor struct {
	Tensor *TensorProto `protobuf:"bytes,8,opt,name=tensor,proto3,oneof"` // "tensor"
}

type AttrValue_Type 

type AttrValue_Type struct {
	Type DataType `protobuf:"varint,6,opt,name=type,proto3,enum=tensorflow.DataType,oneof"` // "type"
}

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          *TensorShapeProto `protobuf:"bytes,3,opt,name=shape,proto3" json:"shape,omitempty"`
	Dtype          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() DataType

func (*CostGraphDef_Node_OutputInfo) GetShape 

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

type DataType 

type DataType int32

(== suppress_warning documentation-presence ==) LINT.IfChange 

const (
	// Not a legal value for DataType.  Used to indicate a DataType field
	// has not been set.
	DataType_DT_INVALID DataType = 0
	// Data types that all computation devices are expected to be
	// capable to support.
	DataType_DT_FLOAT      DataType = 1
	DataType_DT_DOUBLE     DataType = 2
	DataType_DT_INT32      DataType = 3
	DataType_DT_UINT8      DataType = 4
	DataType_DT_INT16      DataType = 5
	DataType_DT_INT8       DataType = 6
	DataType_DT_STRING     DataType = 7
	DataType_DT_COMPLEX64  DataType = 8 // Single-precision complex
	DataType_DT_INT64      DataType = 9
	DataType_DT_BOOL       DataType = 10
	DataType_DT_QINT8      DataType = 11 // Quantized int8
	DataType_DT_QUINT8     DataType = 12 // Quantized uint8
	DataType_DT_QINT32     DataType = 13 // Quantized int32
	DataType_DT_BFLOAT16   DataType = 14 // Float32 truncated to 16 bits.  Only for cast ops.
	DataType_DT_QINT16     DataType = 15 // Quantized int16
	DataType_DT_QUINT16    DataType = 16 // Quantized uint16
	DataType_DT_UINT16     DataType = 17
	DataType_DT_COMPLEX128 DataType = 18 // Double-precision complex
	DataType_DT_HALF       DataType = 19
	DataType_DT_RESOURCE   DataType = 20
	DataType_DT_VARIANT    DataType = 21 // Arbitrary C++ data types
	DataType_DT_UINT32     DataType = 22
	DataType_DT_UINT64     DataType = 23
	// Do not use!  These are only for parameters.  Every enum above
	// should have a corresponding value below (verified by types_test).
	DataType_DT_FLOAT_REF      DataType = 101
	DataType_DT_DOUBLE_REF     DataType = 102
	DataType_DT_INT32_REF      DataType = 103
	DataType_DT_UINT8_REF      DataType = 104
	DataType_DT_INT16_REF      DataType = 105
	DataType_DT_INT8_REF       DataType = 106
	DataType_DT_STRING_REF     DataType = 107
	DataType_DT_COMPLEX64_REF  DataType = 108
	DataType_DT_INT64_REF      DataType = 109
	DataType_DT_BOOL_REF       DataType = 110
	DataType_DT_QINT8_REF      DataType = 111
	DataType_DT_QUINT8_REF     DataType = 112
	DataType_DT_QINT32_REF     DataType = 113
	DataType_DT_BFLOAT16_REF   DataType = 114
	DataType_DT_QINT16_REF     DataType = 115
	DataType_DT_QUINT16_REF    DataType = 116
	DataType_DT_UINT16_REF     DataType = 117
	DataType_DT_COMPLEX128_REF DataType = 118
	DataType_DT_HALF_REF       DataType = 119
	DataType_DT_RESOURCE_REF   DataType = 120
	DataType_DT_VARIANT_REF    DataType = 121
	DataType_DT_UINT32_REF     DataType = 122
	DataType_DT_UINT64_REF     DataType = 123
)

func (DataType) Descriptor 

func (DataType) Descriptor() protoreflect.EnumDescriptor

func (DataType) Enum 

func (x DataType) Enum() *DataType

func (DataType) EnumDescriptor deprecated 

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

Deprecated: Use DataType.Descriptor instead.

func (DataType) Number 

func (x DataType) Number() protoreflect.EnumNumber

func (DataType) String 

func (x DataType) String() string

func (DataType) Type 

func (DataType) Type() protoreflect.EnumType

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 */
	// contains filtered or unexported fields
}

func (*DeviceStepStats) Descriptor deprecated 

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

Deprecated: Use DeviceStepStats.ProtoReflect.Descriptor instead.

func (*DeviceStepStats) GetDevice 

func (x *DeviceStepStats) GetDevice() string

func (*DeviceStepStats) GetNodeStats 

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

func (*DeviceStepStats) GetThreadNames 

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

func (*DeviceStepStats) ProtoMessage 

func (*DeviceStepStats) ProtoMessage()

func (*DeviceStepStats) ProtoReflect 

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

func (*DeviceStepStats) Reset 

func (x *DeviceStepStats) Reset()

func (*DeviceStepStats) String 

func (x *DeviceStepStats) String() string

type FullTypeDef 

type FullTypeDef struct {

	// The principal type represented by this object. This may be a concrete type
	// (Tensor, Dataset) a type variable (used for dependent types) a type
	// symbol (Any, Union). See FullTypeId for details.
	TypeId FullTypeId     `protobuf:"varint,1,opt,name=type_id,json=typeId,proto3,enum=tensorflow.FullTypeId" json:"type_id,omitempty"`
	Args   []*FullTypeDef `protobuf:"bytes,2,rep,name=args,proto3" json:"args,omitempty"`
	// Literal values of this type object, if the the type admits one.
	// For example, a type variable admits a string attribute - its name.
	// Shape-related types may admit int attributes - their static shape values.
	// Fields for more data types to be added as needed.
	//
	// Types that are assignable to Attr:
	//	*FullTypeDef_S
	Attr isFullTypeDef_Attr `protobuf_oneof:"attr"`
	// contains filtered or unexported fields
}

Highly experimental and very likely to change. This encoding uses tags instead of dedicated messages for regularity. In particular the encoding imposes no restrictions on what the parameters of any type should be, which in particular needs to be true for type symbols.

func (*FullTypeDef) Descriptor deprecated 

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

Deprecated: Use FullTypeDef.ProtoReflect.Descriptor instead.

func (*FullTypeDef) GetArgs 

func (x *FullTypeDef) GetArgs() []*FullTypeDef

func (*FullTypeDef) GetAttr 

func (m *FullTypeDef) GetAttr() isFullTypeDef_Attr

func (*FullTypeDef) GetS 

func (x *FullTypeDef) GetS() string

func (*FullTypeDef) GetTypeId 

func (x *FullTypeDef) GetTypeId() FullTypeId

func (*FullTypeDef) ProtoMessage 

func (*FullTypeDef) ProtoMessage()

func (*FullTypeDef) ProtoReflect 

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

func (*FullTypeDef) Reset 

func (x *FullTypeDef) Reset()

func (*FullTypeDef) String 

func (x *FullTypeDef) String() string

type FullTypeDef_S 

type FullTypeDef_S struct {
	S string `protobuf:"bytes,3,opt,name=s,proto3,oneof"`
}

type FullTypeId 

type FullTypeId int32

Experimental. Represents the complete type information of a TensorFlow value. 

const (
	// The default represents an uninitialized values.
	FullTypeId_TFT_UNSET FullTypeId = 0
	// Type variables may serve as placeholder for any other type ID in type
	// templates.
	//
	// Examples:
	//   TFT_DATASET[TFT_VAR["T"]] is a Dataset returning a type indicated by "T".
	//   TFT_TENSOR[TFT_VAR["T"]] is a Tensor of n element type indicated by "T".
	//   TFT_TENSOR[TFT_VAR["T"]], TFT_TENSOR[TFT_VAR["T"]] are two tensors of
	//     identical element types.
	//   TFT_TENSOR[TFT_VAR["P"]], TFT_TENSOR[TFT_VAR["Q"]] are two tensors of
	//     potentially different element types.
	//
	FullTypeId_TFT_VAR FullTypeId = 1
	// Wildcard type. Describes a parameter of unknown type. In TensorFlow, that
	// can mean either a "Top" type (accepts any type), or a dynamically typed
	// object whose type is unknown in context.
	// Important: "unknown" does not necessarily mean undeterminable!
	FullTypeId_TFT_ANY FullTypeId = 2
	// The algebraic product type. This is an algebraic type that may be used just
	// for logical grouping. Not to confused with TFT_TUPLE which describes a
	// concrete object of several elements.
	//
	// Example:
	//   TFT_DATASET[TFT_PRODUCT[TFT_TENSOR[TFT_INT32], TFT_TENSOR[TFT_FLOAT64]]]
	//     is a Dataset producing two tensors, an integer one and a float one.
	//
	FullTypeId_TFT_PRODUCT FullTypeId = 3
	// Callable types describe functions and ops.
	//
	// Parametrization:
	//   TFT_CALLABLE[<arg type>, <return type>]
	//   * <arg_type> is the type of the arguments; TFT_PRODUCT represents
	//   multiple
	//     arguments.
	//   * <return_type> is the return type; TFT_PRODUCT represents multiple
	//     return values (that means that callables returning multiple things
	//     don't necessarily return a single tuple).
	//
	// Example:
	//   TFT_CALLABLE[
	//     TFT_ANY,
	//     TFT_PRODUCT[TFT_TENSOR[TFT_INT32], TFT_TENSOR[TFT_FLOAT64]],
	//   ]
	//     is a callable with unspecified (for now) input arguments, and
	//     two return values of type tensor.
	//
	FullTypeId_TFT_CALLABLE FullTypeId = 100
	// The usual Tensor. This is a parametric type.
	//
	// Parametrization:
	//   TFT_TENSOR[<element type>, <shape type>]
	//   * <element_type> is currently limited to one of the element types
	//     defined below.
	//   * <shape_type> is not yet defined, and may only be TFT_UNKNOWN for now.
	//
	// A TFT_SHAPE type will be defined in the future.
	//
	// Example:
	//   TFT_TENSOR[TFT_INT32, TFT_UNKNOWN]
	//     is a Tensor of int32 element type and unknown shape.
	//
	// TODO(mdan): Define TFT_SHAPE and add more examples.
	FullTypeId_TFT_TENSOR FullTypeId = 1000
	// Array (or tensorflow::TensorList in the variant type registry).
	// Note: this is not to be confused with the deprecated `TensorArray*` ops
	// which are not supported by FullType.
	// This type represents a random-access list whose elements can be
	// described by a single type. Although immutable, Array is expected to
	// support efficient mutation semantics (i.e. element update) in the
	// user-facing API.
	// The element type may be generic or even TFT_ANY for a heterogenous list.
	//
	// Parametrization:
	//   TFT_ARRAY[<element type>]
	//   * <element_type> may be any concrete type.
	//
	// Examples:
	//   TFT_ARRAY[TFT_TENSOR[TFT_INT32]] is a TensorArray holding int32 Tensors
	//     of any shape.
	//   TFT_ARRAY[TFT_TENSOR[TFT_UNKNOWN]] is a TensorArray holding Tensors of
	//     mixed element types.
	//   TFT_ARRAY[TFT_UNKNOWN] is a TensorArray holding any element type.
	//   TFT_ARRAY[] is equivalent to TFT_ARRAY[TFT_UNKNOWN].
	//   TFT_ARRAY[TFT_ARRAY[]] is an array or arrays (of unknown types).
	FullTypeId_TFT_ARRAY FullTypeId = 1001
	// Optional (or tensorflow::OptionalVariant in the variant type registry).
	// This type represents a value that may either hold an element of a single
	// specified type, or nothing at all.
	//
	// Parametrization:
	//   TFT_OPTIONAL[<element type>]
	//   * <element_type> may be any concrete type.
	//
	// Examples:
	//   TFT_OPTIONAL[TFT_TENSOR[TFT_INT32]] is an Optional holding an int32
	//     Tensor of any shape.
	FullTypeId_TFT_OPTIONAL FullTypeId = 1002
	// Datasets created by tf.data ops and APIs. Datasets have generator/iterable
	// semantics, that is, one can construct an iterator from them. Like
	// Array, they are considered to return elements that can be described
	// by a single type. Unlike Array, they do not support random access or
	// mutation, and can potentially produce an infinite number of elements.
	// A datasets can produce logical structures (e.g. multiple elements). This
	// is expressed using TFT_PRODUCT.
	//
	//
	// Parametrization: TFT_ARRAY[<element type>].
	// <element_type> may be a concrete type or a type symbol. It represents the
	//   data type of the elements produced by the dataset.
	//
	// Examples:
	//   TFT_DATSET[TFT_TENSOR[TFT_INT32]] is a Dataset producing single int32
	//     Tensors of unknown shape.
	//   TFT_DATSET[TFT_PRODUCT[TFT_TENSOR[TFT_INT32], TFT_TENSOR[TFT_FLOAT32]] is
	//   a
	//     Dataset producing pairs of Tensors, one integer and one float.
	// Note: The high ID number is to prepare for the eventuality that Datasets
	// will be supported by user types in the future.
	FullTypeId_TFT_DATASET FullTypeId = 10102
	// The bool element type.
	// TODO(mdan): Quantized types, legacy representations (e.g. ref)
	FullTypeId_TFT_BOOL FullTypeId = 200
	// Integer element types.
	FullTypeId_TFT_UINT8  FullTypeId = 201
	FullTypeId_TFT_UINT16 FullTypeId = 202
	FullTypeId_TFT_UINT32 FullTypeId = 203
	FullTypeId_TFT_UINT64 FullTypeId = 204
	FullTypeId_TFT_INT8   FullTypeId = 205
	FullTypeId_TFT_INT16  FullTypeId = 206
	FullTypeId_TFT_INT32  FullTypeId = 207
	FullTypeId_TFT_INT64  FullTypeId = 208
	// Floating-point element types.
	FullTypeId_TFT_HALF     FullTypeId = 209
	FullTypeId_TFT_FLOAT    FullTypeId = 210
	FullTypeId_TFT_DOUBLE   FullTypeId = 211
	FullTypeId_TFT_BFLOAT16 FullTypeId = 215
	// Complex element types.
	// TODO(mdan): Represent as TFT_COMPLEX[TFT_DOUBLE] instead?
	FullTypeId_TFT_COMPLEX64  FullTypeId = 212
	FullTypeId_TFT_COMPLEX128 FullTypeId = 213
	// The string element type.
	FullTypeId_TFT_STRING FullTypeId = 214
)

func (FullTypeId) Descriptor 

func (FullTypeId) Descriptor() protoreflect.EnumDescriptor

func (FullTypeId) Enum 

func (x FullTypeId) Enum() *FullTypeId

func (FullTypeId) EnumDescriptor deprecated 

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

Deprecated: Use FullTypeId.Descriptor instead.

func (FullTypeId) Number 

func (x FullTypeId) Number() protoreflect.EnumNumber

func (FullTypeId) String 

func (x FullTypeId) String() string

func (FullTypeId) Type 

func (FullTypeId) Type() protoreflect.EnumType

type FunctionDef 

type FunctionDef struct {

	// The definition of the function's name, arguments, return values,
	// attrs etc.
	Signature *OpDef `protobuf:"bytes,1,opt,name=signature,proto3" json:"signature,omitempty"`
	// Attributes specific to this function definition.
	Attr    map[string]*AttrValue            `` /* 149-byte string literal not displayed */
	ArgAttr map[uint32]*FunctionDef_ArgAttrs `` /* 171-byte string literal not displayed */
	// Unique IDs for each resource argument, used to track aliasing resources. If
	// Argument A and Argument B alias each other, then
	// resource_arg_unique_ids[A.index] == resource_arg_unique_ids[B.index].
	//
	// If this field is empty, none of the arguments could alias; otherwise, every
	// resource argument should have an entry in this field.
	//
	// When instantiated, the unique IDs will be attached to the _Arg nodes'
	// "_resource_arg_unique_id" attribute.
	ResourceArgUniqueId map[uint32]uint32 `` /* 212-byte string literal not displayed */
	// By convention, "op" in node_def is resolved by consulting with a
	// user-defined library first. If not resolved, "func" is assumed to
	// be a builtin op.
	NodeDef []*NodeDef `protobuf:"bytes,3,rep,name=node_def,json=nodeDef,proto3" json:"node_def,omitempty"`
	// A mapping from the output arg names from `signature` to the
	// outputs from `node_def` that should be returned by the function.
	Ret map[string]string `` /* 147-byte string literal not displayed */
	// A mapping from control output names from `signature` to node names in
	// `node_def` which should be control outputs of this function.
	ControlRet map[string]string `` /* 179-byte string literal not displayed */
	// contains filtered or unexported fields
}

A function can be instantiated when the runtime can bind every attr with a value. When a GraphDef has a call to a function, it must have binding for every attr defined in the signature.

TODO(zhifengc):

  • device spec, etc.

func (*FunctionDef) Descriptor deprecated 

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

Deprecated: Use FunctionDef.ProtoReflect.Descriptor instead.

func (*FunctionDef) GetArgAttr 

func (x *FunctionDef) GetArgAttr() map[uint32]*FunctionDef_ArgAttrs

func (*FunctionDef) GetAttr 

func (x *FunctionDef) GetAttr() map[string]*AttrValue

func (*FunctionDef) GetControlRet 

func (x *FunctionDef) GetControlRet() map[string]string

func (*FunctionDef) GetNodeDef 

func (x *FunctionDef) GetNodeDef() []*NodeDef

func (*FunctionDef) GetResourceArgUniqueId 

func (x *FunctionDef) GetResourceArgUniqueId() map[uint32]uint32

func (*FunctionDef) GetRet 

func (x *FunctionDef) GetRet() map[string]string

func (*FunctionDef) GetSignature 

func (x *FunctionDef) GetSignature() *OpDef

func (*FunctionDef) ProtoMessage 

func (*FunctionDef) ProtoMessage()

func (*FunctionDef) ProtoReflect 

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

func (*FunctionDef) Reset 

func (x *FunctionDef) Reset()

func (*FunctionDef) String 

func (x *FunctionDef) String() string

type FunctionDefLibrary 

type FunctionDefLibrary struct {
	Function            []*FunctionDef        `protobuf:"bytes,1,rep,name=function,proto3" json:"function,omitempty"`
	Gradient            []*GradientDef        `protobuf:"bytes,2,rep,name=gradient,proto3" json:"gradient,omitempty"`
	RegisteredGradients []*RegisteredGradient `protobuf:"bytes,3,rep,name=registered_gradients,json=registeredGradients,proto3" json:"registered_gradients,omitempty"`
	// contains filtered or unexported fields
}

A library is a set of named functions.

func (*FunctionDefLibrary) Descriptor deprecated 

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

Deprecated: Use FunctionDefLibrary.ProtoReflect.Descriptor instead.

func (*FunctionDefLibrary) GetFunction 

func (x *FunctionDefLibrary) GetFunction() []*FunctionDef

func (*FunctionDefLibrary) GetGradient 

func (x *FunctionDefLibrary) GetGradient() []*GradientDef

func (*FunctionDefLibrary) GetRegisteredGradients 

func (x *FunctionDefLibrary) GetRegisteredGradients() []*RegisteredGradient

func (*FunctionDefLibrary) ProtoMessage 

func (*FunctionDefLibrary) ProtoMessage()

func (*FunctionDefLibrary) ProtoReflect 

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

func (*FunctionDefLibrary) Reset 

func (x *FunctionDefLibrary) Reset()

func (*FunctionDefLibrary) String 

func (x *FunctionDefLibrary) String() string

type FunctionDef_ArgAttrs 

type FunctionDef_ArgAttrs struct {
	Attr map[string]*AttrValue `` /* 149-byte string literal not displayed */
	// contains filtered or unexported fields
}

Attributes for function arguments. These attributes are the same set of valid attributes as to _Arg nodes.

func (*FunctionDef_ArgAttrs) Descriptor deprecated 

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

Deprecated: Use FunctionDef_ArgAttrs.ProtoReflect.Descriptor instead.

func (*FunctionDef_ArgAttrs) GetAttr 

func (x *FunctionDef_ArgAttrs) GetAttr() map[string]*AttrValue

func (*FunctionDef_ArgAttrs) ProtoMessage 

func (*FunctionDef_ArgAttrs) ProtoMessage()

func (*FunctionDef_ArgAttrs) ProtoReflect 

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

func (*FunctionDef_ArgAttrs) Reset 

func (x *FunctionDef_ArgAttrs) Reset()

func (*FunctionDef_ArgAttrs) String 

func (x *FunctionDef_ArgAttrs) String() string

type GradientDef 

type GradientDef struct {
	FunctionName string `protobuf:"bytes,1,opt,name=function_name,json=functionName,proto3" json:"function_name,omitempty"` // The function name.
	GradientFunc string `protobuf:"bytes,2,opt,name=gradient_func,json=gradientFunc,proto3" json:"gradient_func,omitempty"` // The gradient function's name.
	// contains filtered or unexported fields
}

GradientDef defines the gradient function of a function defined in a function library.

A gradient function g (specified by gradient_func) for a function f (specified by function_name) must follow the following:

The function 'f' must be a numerical function which takes N inputs and produces M outputs. Its gradient function 'g', which is a function taking N + M inputs and produces N outputs.

I.e. if we have 

(y1, y2, ..., y_M) = f(x1, x2, ..., x_N),

then, g is 

(dL/dx1, dL/dx2, ..., dL/dx_N) = g(x1, x2, ..., x_N,
                                  dL/dy1, dL/dy2, ..., dL/dy_M),

where L is a scalar-value function of (x1, x2, ..., xN) (e.g., the loss function). dL/dx_i is the partial derivative of L with respect to x_i.

func (*GradientDef) Descriptor deprecated 

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

Deprecated: Use GradientDef.ProtoReflect.Descriptor instead.

func (*GradientDef) GetFunctionName 

func (x *GradientDef) GetFunctionName() string

func (*GradientDef) GetGradientFunc 

func (x *GradientDef) GetGradientFunc() string

func (*GradientDef) ProtoMessage 

func (*GradientDef) ProtoMessage()

func (*GradientDef) ProtoReflect 

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

func (*GradientDef) Reset 

func (x *GradientDef) Reset()

func (*GradientDef) String 

func (x *GradientDef) String() string

type GraphDef 

type GraphDef struct {
	Node []*NodeDef `protobuf:"bytes,1,rep,name=node,proto3" json:"node,omitempty"`
	// Compatibility versions of the graph.  See core/public/version.h for version
	// history.  The GraphDef version is distinct from the TensorFlow version, and
	// each release of TensorFlow will support a range of GraphDef versions.
	Versions *VersionDef `protobuf:"bytes,4,opt,name=versions,proto3" json:"versions,omitempty"`
	// Deprecated single version field; use versions above instead.  Since all
	// GraphDef changes before "versions" was introduced were forward
	// compatible, this field is entirely ignored.
	//
	// Deprecated: Do not use.
	Version int32 `protobuf:"varint,3,opt,name=version,proto3" json:"version,omitempty"`
	// "library" provides user-defined functions.
	//
	// Naming:
	//   * library.function.name are in a flat namespace.
	//     NOTE: We may need to change it to be hierarchical to support
	//     different orgs. E.g.,
	//     { "/google/nn", { ... }},
	//     { "/google/vision", { ... }}
	//     { "/org_foo/module_bar", { ... }}
	//     map<string, FunctionDefLib> named_lib;
	//   * If node[i].op is the name of one function in "library",
	//     node[i] is deemed as a function call. Otherwise, node[i].op
	//     must be a primitive operation supported by the runtime.
	//
	//
	// Function call semantics:
	//
	//   * The callee may start execution as soon as some of its inputs
	//     are ready. The caller may want to use Tuple() mechanism to
	//     ensure all inputs are ready in the same time.
	//
	//   * The consumer of return values may start executing as soon as
	//     the return values the consumer depends on are ready.  The
	//     consumer may want to use Tuple() mechanism to ensure the
	//     consumer does not start until all return values of the callee
	//     function are ready.
	Library *FunctionDefLibrary `protobuf:"bytes,2,opt,name=library,proto3" json:"library,omitempty"`
	// contains filtered or unexported fields
}

Represents the graph of operations

func (*GraphDef) Descriptor deprecated 

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

Deprecated: Use GraphDef.ProtoReflect.Descriptor instead.

func (*GraphDef) GetLibrary 

func (x *GraphDef) GetLibrary() *FunctionDefLibrary

func (*GraphDef) GetNode 

func (x *GraphDef) GetNode() []*NodeDef

func (*GraphDef) GetVersion deprecated 

func (x *GraphDef) GetVersion() int32

Deprecated: Do not use.

func (*GraphDef) GetVersions 

func (x *GraphDef) GetVersions() *VersionDef

func (*GraphDef) ProtoMessage 

func (*GraphDef) ProtoMessage()

func (*GraphDef) ProtoReflect 

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

func (*GraphDef) Reset 

func (x *GraphDef) Reset()

func (*GraphDef) String 

func (x *GraphDef) String() string

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 */
	// Deprecated: Do not use.
	DeviceTempMemorySize int64 `` /* 126-byte string literal not displayed */
	// Deprecated: Do not use.
	DevicePersistentMemorySize int64 `` /* 144-byte string literal not displayed */
	// Deprecated: Do not use.
	DevicePersistentTensorAllocIds []int64 `` /* 165-byte string literal not displayed */
	// contains filtered or unexported fields
}

For memory tracking.

func (*MemoryStats) Descriptor deprecated 

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

Deprecated: Use MemoryStats.ProtoReflect.Descriptor instead.

func (*MemoryStats) GetDevicePersistentMemorySize deprecated 

func (x *MemoryStats) GetDevicePersistentMemorySize() int64

Deprecated: Do not use.

func (*MemoryStats) GetDevicePersistentTensorAllocIds deprecated 

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

Deprecated: Do not use.

func (*MemoryStats) GetDeviceTempMemorySize deprecated 

func (x *MemoryStats) GetDeviceTempMemorySize() int64

Deprecated: Do not use.

func (*MemoryStats) GetPersistentMemorySize 

func (x *MemoryStats) GetPersistentMemorySize() int64

func (*MemoryStats) GetPersistentTensorAllocIds 

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

func (*MemoryStats) GetTempMemorySize 

func (x *MemoryStats) GetTempMemorySize() int64

func (*MemoryStats) ProtoMessage 

func (*MemoryStats) ProtoMessage()

func (*MemoryStats) ProtoReflect 

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

func (*MemoryStats) Reset 

func (x *MemoryStats) Reset()

func (*MemoryStats) String 

func (x *MemoryStats) String() string

type NameAttrList 

type NameAttrList struct {
	Name string                `protobuf:"bytes,1,opt,name=name,proto3" json:"name,omitempty"`
	Attr map[string]*AttrValue `` /* 149-byte string literal not displayed */
	// contains filtered or unexported fields
}

A list of attr names and their values. The whole list is attached with a string name. E.g., MatMul[T=float].

func (*NameAttrList) Descriptor deprecated 

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

Deprecated: Use NameAttrList.ProtoReflect.Descriptor instead.

func (*NameAttrList) GetAttr 

func (x *NameAttrList) GetAttr() map[string]*AttrValue

func (*NameAttrList) GetName 

func (x *NameAttrList) GetName() string

func (*NameAttrList) ProtoMessage 

func (*NameAttrList) ProtoMessage()

func (*NameAttrList) ProtoReflect 

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

func (*NameAttrList) Reset 

func (x *NameAttrList) Reset()

func (*NameAttrList) String 

func (x *NameAttrList) String() string

type NodeDef 

type NodeDef struct {

	// The name given to this operator. Used for naming inputs,
	// logging, visualization, etc.  Unique within a single GraphDef.
	// Must match the regexp "[A-Za-z0-9.][A-Za-z0-9_>./]*".
	Name string `protobuf:"bytes,1,opt,name=name,proto3" json:"name,omitempty"`
	// The operation name.  There may be custom parameters in attrs.
	// Op names starting with an underscore are reserved for internal use.
	Op string `protobuf:"bytes,2,opt,name=op,proto3" json:"op,omitempty"`
	// Each input is "node:src_output" with "node" being a string name and
	// "src_output" indicating which output tensor to use from "node". If
	// "src_output" is 0 the ":0" suffix can be omitted.  Regular inputs
	// may optionally be followed by control inputs that have the format
	// "^node".
	Input []string `protobuf:"bytes,3,rep,name=input,proto3" json:"input,omitempty"`
	// A (possibly partial) specification for the device on which this
	// node should be placed.
	// The expected syntax for this string is as follows:
	//
	// DEVICE_SPEC ::= PARTIAL_SPEC
	//
	// PARTIAL_SPEC ::= ("/" CONSTRAINT) *
	// CONSTRAINT ::= ("job:" JOB_NAME)
	//              | ("replica:" [1-9][0-9]*)
	//              | ("task:" [1-9][0-9]*)
	//              | ("device:" [A-Za-z]* ":" ([1-9][0-9]* | "*") )
	//
	// Valid values for this string include:
	// * "/job:worker/replica:0/task:1/device:GPU:3"  (full specification)
	// * "/job:worker/device:GPU:3"                   (partial specification)
	// * ""                                    (no specification)
	//
	// If the constraints do not resolve to a single device (or if this
	// field is empty or not present), the runtime will attempt to
	// choose a device automatically.
	Device string `protobuf:"bytes,4,opt,name=device,proto3" json:"device,omitempty"`
	// Operation-specific graph-construction-time configuration.
	// Note that this should include all attrs defined in the
	// corresponding OpDef, including those with a value matching
	// the default -- this allows the default to change and makes
	// NodeDefs easier to interpret on their own.  However, if
	// an attr with a default is not specified in this list, the
	// default will be used.
	// The "names" (keys) must match the regexp "[a-z][a-z0-9_]+" (and
	// one of the names from the corresponding OpDef's attr field).
	// The values must have a type matching the corresponding OpDef
	// attr's type field.
	// TODO(josh11b): Add some examples here showing best practices.
	Attr map[string]*AttrValue `` /* 149-byte string literal not displayed */
	// This stores debug information associated with the node.
	ExperimentalDebugInfo *NodeDef_ExperimentalDebugInfo `` /* 126-byte string literal not displayed */
	// contains filtered or unexported fields
}

func (*NodeDef) Descriptor deprecated 

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

Deprecated: Use NodeDef.ProtoReflect.Descriptor instead.

func (*NodeDef) GetAttr 

func (x *NodeDef) GetAttr() map[string]*AttrValue

func (*NodeDef) GetDevice 

func (x *NodeDef) GetDevice() string

func (*NodeDef) GetExperimentalDebugInfo 

func (x *NodeDef) GetExperimentalDebugInfo() *NodeDef_ExperimentalDebugInfo

func (*NodeDef) GetInput 

func (x *NodeDef) GetInput() []string

func (*NodeDef) GetName 

func (x *NodeDef) GetName() string

func (*NodeDef) GetOp 

func (x *NodeDef) GetOp() string

func (*NodeDef) ProtoMessage 

func (*NodeDef) ProtoMessage()

func (*NodeDef) ProtoReflect 

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

func (*NodeDef) Reset 

func (x *NodeDef) Reset()

func (*NodeDef) String 

func (x *NodeDef) String() string

type NodeDef_ExperimentalDebugInfo 

type NodeDef_ExperimentalDebugInfo struct {

	// Opaque string inserted into error messages created by the runtime.
	//
	// This is intended to store the list of names of the nodes from the
	// original graph that this node was derived. For example if this node, say
	// C, was result of a fusion of 2 nodes A and B, then 'original_node' would
	// be {A, B}. This information can be used to map errors originating at the
	// current node to some top level source code.
	OriginalNodeNames []string `protobuf:"bytes,1,rep,name=original_node_names,json=originalNodeNames,proto3" json:"original_node_names,omitempty"`
	// This is intended to store the list of names of the functions from the
	// original graph that this node was derived. For example if this node, say
	// C, was result of a fusion of node A in function FA and node B in function
	// FB, then `original_funcs` would be {FA, FB}. If the node is in the top
	// level graph, the `original_func` is empty. This information, with the
	// `original_node_names` can be used to map errors originating at the
	// current ndoe to some top level source code.
	OriginalFuncNames []string `protobuf:"bytes,2,rep,name=original_func_names,json=originalFuncNames,proto3" json:"original_func_names,omitempty"`
	// contains filtered or unexported fields
}

func (*NodeDef_ExperimentalDebugInfo) Descriptor deprecated 

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

Deprecated: Use NodeDef_ExperimentalDebugInfo.ProtoReflect.Descriptor instead.

func (*NodeDef_ExperimentalDebugInfo) GetOriginalFuncNames 

func (x *NodeDef_ExperimentalDebugInfo) GetOriginalFuncNames() []string

func (*NodeDef_ExperimentalDebugInfo) GetOriginalNodeNames 

func (x *NodeDef_ExperimentalDebugInfo) GetOriginalNodeNames() []string

func (*NodeDef_ExperimentalDebugInfo) ProtoMessage 

func (*NodeDef_ExperimentalDebugInfo) ProtoMessage()

func (*NodeDef_ExperimentalDebugInfo) ProtoReflect 

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

func (*NodeDef_ExperimentalDebugInfo) Reset 

func (x *NodeDef_ExperimentalDebugInfo) Reset()

func (*NodeDef_ExperimentalDebugInfo) String 

func (x *NodeDef_ExperimentalDebugInfo) String() string

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 []*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"`
	// contains filtered or unexported fields
}

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

func (*NodeExecStats) Descriptor deprecated 

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

Deprecated: Use NodeExecStats.ProtoReflect.Descriptor instead.

func (*NodeExecStats) GetAllEndRelMicros 

func (x *NodeExecStats) GetAllEndRelMicros() int64

func (*NodeExecStats) GetAllEndRelNanos 

func (x *NodeExecStats) GetAllEndRelNanos() int64

func (*NodeExecStats) GetAllStartMicros 

func (x *NodeExecStats) GetAllStartMicros() int64

func (*NodeExecStats) GetAllStartNanos 

func (x *NodeExecStats) GetAllStartNanos() int64

func (*NodeExecStats) GetMemory 

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

func (*NodeExecStats) GetMemoryStats 

func (x *NodeExecStats) GetMemoryStats() *MemoryStats

func (*NodeExecStats) GetNodeName 

func (x *NodeExecStats) GetNodeName() string

func (*NodeExecStats) GetOpEndRelMicros 

func (x *NodeExecStats) GetOpEndRelMicros() int64

func (*NodeExecStats) GetOpEndRelNanos 

func (x *NodeExecStats) GetOpEndRelNanos() int64

func (*NodeExecStats) GetOpStartRelMicros 

func (x *NodeExecStats) GetOpStartRelMicros() int64

func (*NodeExecStats) GetOpStartRelNanos 

func (x *NodeExecStats) GetOpStartRelNanos() int64

func (*NodeExecStats) GetOutput 

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

func (*NodeExecStats) GetReferencedTensor 

func (x *NodeExecStats) GetReferencedTensor() []*AllocationDescription

func (*NodeExecStats) GetScheduledMicros 

func (x *NodeExecStats) GetScheduledMicros() int64

func (*NodeExecStats) GetScheduledNanos 

func (x *NodeExecStats) GetScheduledNanos() int64

func (*NodeExecStats) GetThreadId 

func (x *NodeExecStats) GetThreadId() uint32

func (*NodeExecStats) GetTimelineLabel 

func (x *NodeExecStats) GetTimelineLabel() string

func (*NodeExecStats) ProtoMessage 

func (*NodeExecStats) ProtoMessage()

func (*NodeExecStats) ProtoReflect 

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

func (*NodeExecStats) Reset 

func (x *NodeExecStats) Reset()

func (*NodeExecStats) String 

func (x *NodeExecStats) String() string

type NodeOutput 

type NodeOutput struct {
	Slot              int32              `protobuf:"varint,1,opt,name=slot,proto3" json:"slot,omitempty"`
	TensorDescription *TensorDescription `protobuf:"bytes,3,opt,name=tensor_description,json=tensorDescription,proto3" json:"tensor_description,omitempty"`
	// contains filtered or unexported fields
}

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

func (*NodeOutput) Descriptor deprecated 

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

Deprecated: Use NodeOutput.ProtoReflect.Descriptor instead.

func (*NodeOutput) GetSlot 

func (x *NodeOutput) GetSlot() int32

func (*NodeOutput) GetTensorDescription 

func (x *NodeOutput) GetTensorDescription() *TensorDescription

func (*NodeOutput) ProtoMessage 

func (*NodeOutput) ProtoMessage()

func (*NodeOutput) ProtoReflect 

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

func (*NodeOutput) Reset 

func (x *NodeOutput) Reset()

func (*NodeOutput) String 

func (x *NodeOutput) String() string

type OpDef 

type OpDef struct {

	// Op names starting with an underscore are reserved for internal use.
	// Names should be CamelCase and match the regexp "[A-Z][a-zA-Z0-9>_]*".
	Name string `protobuf:"bytes,1,opt,name=name,proto3" json:"name,omitempty"`
	// Description of the input(s).
	InputArg []*OpDef_ArgDef `protobuf:"bytes,2,rep,name=input_arg,json=inputArg,proto3" json:"input_arg,omitempty"`
	// Description of the output(s).
	OutputArg []*OpDef_ArgDef `protobuf:"bytes,3,rep,name=output_arg,json=outputArg,proto3" json:"output_arg,omitempty"`
	// Named control outputs for this operation. Useful only for composite
	// operations (i.e. functions) which want to name different control outputs.
	ControlOutput []string         `protobuf:"bytes,20,rep,name=control_output,json=controlOutput,proto3" json:"control_output,omitempty"`
	Attr          []*OpDef_AttrDef `protobuf:"bytes,4,rep,name=attr,proto3" json:"attr,omitempty"`
	// Optional deprecation based on GraphDef versions.
	Deprecation *OpDeprecation `protobuf:"bytes,8,opt,name=deprecation,proto3" json:"deprecation,omitempty"`
	// One-line human-readable description of what the Op does.
	Summary string `protobuf:"bytes,5,opt,name=summary,proto3" json:"summary,omitempty"`
	// Additional, longer human-readable description of what the Op does.
	Description string `protobuf:"bytes,6,opt,name=description,proto3" json:"description,omitempty"`
	// True if the operation is commutative ("op(a,b) == op(b,a)" for all inputs)
	IsCommutative bool `protobuf:"varint,18,opt,name=is_commutative,json=isCommutative,proto3" json:"is_commutative,omitempty"`
	// If is_aggregate is true, then this operation accepts N >= 2
	// inputs and produces 1 output all of the same type.  Should be
	// associative and commutative, and produce output with the same
	// shape as the input.  The optimizer may replace an aggregate op
	// taking input from multiple devices with a tree of aggregate ops
	// that aggregate locally within each device (and possibly within
	// groups of nearby devices) before communicating.
	// TODO(josh11b): Implement that optimization.
	IsAggregate bool `protobuf:"varint,16,opt,name=is_aggregate,json=isAggregate,proto3" json:"is_aggregate,omitempty"` // for things like add
	// Ops are marked as stateful if their behavior depends on some state beyond
	// their input tensors (e.g. variable reading op) or if they have
	// a side-effect (e.g. printing or asserting ops). Equivalently, stateless ops
	// must always produce the same output for the same input and have
	// no side-effects.
	//
	// By default Ops may be moved between devices.  Stateful ops should
	// either not be moved, or should only be moved if that state can also
	// be moved (e.g. via some sort of save / restore).
	// Stateful ops are guaranteed to never be optimized away by Common
	// Subexpression Elimination (CSE).
	IsStateful bool `protobuf:"varint,17,opt,name=is_stateful,json=isStateful,proto3" json:"is_stateful,omitempty"` // for things like variables, queue
	// By default, all inputs to an Op must be initialized Tensors.  Ops
	// that may initialize tensors for the first time should set this
	// field to true, to allow the Op to take an uninitialized Tensor as
	// input.
	AllowsUninitializedInput bool `` // for Assign, etc.
	/* 137-byte string literal not displayed */
	// Indicates whether the op implementation uses distributed communication.
	// If True, the op is allowed to return errors for network disconnection and
	// trigger TF network failure handling logics.
	IsDistributedCommunication bool `` /* 143-byte string literal not displayed */
	// contains filtered or unexported fields
}

Defines an operation. A NodeDef in a GraphDef specifies an Op by using the "op" field which should match the name of a OpDef. LINT.IfChange

func (*OpDef) Descriptor deprecated 

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

Deprecated: Use OpDef.ProtoReflect.Descriptor instead.

func (*OpDef) GetAllowsUninitializedInput 

func (x *OpDef) GetAllowsUninitializedInput() bool

func (*OpDef) GetAttr 

func (x *OpDef) GetAttr() []*OpDef_AttrDef

func (*OpDef) GetControlOutput 

func (x *OpDef) GetControlOutput() []string

func (*OpDef) GetDeprecation 

func (x *OpDef) GetDeprecation() *OpDeprecation

func (*OpDef) GetDescription 

func (x *OpDef) GetDescription() string

func (*OpDef) GetInputArg 

func (x *OpDef) GetInputArg() []*OpDef_ArgDef

func (*OpDef) GetIsAggregate 

func (x *OpDef) GetIsAggregate() bool

func (*OpDef) GetIsCommutative 

func (x *OpDef) GetIsCommutative() bool

func (*OpDef) GetIsDistributedCommunication 

func (x *OpDef) GetIsDistributedCommunication() bool

func (*OpDef) GetIsStateful 

func (x *OpDef) GetIsStateful() bool

func (*OpDef) GetName 

func (x *OpDef) GetName() string

func (*OpDef) GetOutputArg 

func (x *OpDef) GetOutputArg() []*OpDef_ArgDef

func (*OpDef) GetSummary 

func (x *OpDef) GetSummary() string

func (*OpDef) ProtoMessage 

func (*OpDef) ProtoMessage()

func (*OpDef) ProtoReflect 

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

func (*OpDef) Reset 

func (x *OpDef) Reset()

func (*OpDef) String 

func (x *OpDef) String() string

type OpDef_ArgDef 

type OpDef_ArgDef struct {

	// Name for the input/output.  Should match the regexp "[a-z][a-z0-9_]*".
	Name string `protobuf:"bytes,1,opt,name=name,proto3" json:"name,omitempty"`
	// Human readable description.
	Description string `protobuf:"bytes,2,opt,name=description,proto3" json:"description,omitempty"`
	// Describes the type of one or more tensors that are accepted/produced
	// by this input/output arg.  The only legal combinations are:
	// * For a single tensor: either the "type" field is set or the
	//   "type_attr" field is set to the name of an attr with type "type".
	// * For a sequence of tensors with the same type: the "number_attr"
	//   field will be set to the name of an attr with type "int", and
	//   either the "type" or "type_attr" field will be set as for
	//   single tensors.
	// * For a sequence of tensors, the "type_list_attr" field will be set
	//   to the name of an attr with type "list(type)".
	Type       DataType `protobuf:"varint,3,opt,name=type,proto3,enum=tensorflow.DataType" json:"type,omitempty"`
	TypeAttr   string   `protobuf:"bytes,4,opt,name=type_attr,json=typeAttr,proto3" json:"type_attr,omitempty"`       // if specified, attr must have type "type"
	NumberAttr string   `protobuf:"bytes,5,opt,name=number_attr,json=numberAttr,proto3" json:"number_attr,omitempty"` // if specified, attr must have type "int"
	// If specified, attr must have type "list(type)", and none of
	// type, type_attr, and number_attr may be specified.
	TypeListAttr string `protobuf:"bytes,6,opt,name=type_list_attr,json=typeListAttr,proto3" json:"type_list_attr,omitempty"`
	// The handle data for resource inputs.
	HandleData []*ResourceHandleProto_DtypeAndShape `protobuf:"bytes,7,rep,name=handle_data,json=handleData,proto3" json:"handle_data,omitempty"`
	// For inputs: if true, the inputs are required to be refs.
	//   By default, inputs can be either refs or non-refs.
	// For outputs: if true, outputs are refs, otherwise they are not.
	IsRef bool `protobuf:"varint,16,opt,name=is_ref,json=isRef,proto3" json:"is_ref,omitempty"`
	// Experimental. Full type declaration for this argument.
	// The full type specification combines type, type_attr, type_list_attr,
	// etc. into a unified representation.
	// This declaration may contain non-concrete types (for example,
	// Tensor<TypeVar<'T'>> is a valid type declaration.
	//
	// Note: this is a transient field. The long-term aim is to represent the
	// entire OpDef as a single type: a callable. In that context, this field is
	// just the type of a single argument.
	ExperimentalFullType *FullTypeDef `protobuf:"bytes,17,opt,name=experimental_full_type,json=experimentalFullType,proto3" json:"experimental_full_type,omitempty"`
	// contains filtered or unexported fields
}

For describing inputs and outputs.

func (*OpDef_ArgDef) Descriptor deprecated 

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

Deprecated: Use OpDef_ArgDef.ProtoReflect.Descriptor instead.

func (*OpDef_ArgDef) GetDescription 

func (x *OpDef_ArgDef) GetDescription() string

func (*OpDef_ArgDef) GetExperimentalFullType 

func (x *OpDef_ArgDef) GetExperimentalFullType() *FullTypeDef

func (*OpDef_ArgDef) GetHandleData 

func (x *OpDef_ArgDef) GetHandleData() []*ResourceHandleProto_DtypeAndShape

func (*OpDef_ArgDef) GetIsRef 

func (x *OpDef_ArgDef) GetIsRef() bool

func (*OpDef_ArgDef) GetName 

func (x *OpDef_ArgDef) GetName() string

func (*OpDef_ArgDef) GetNumberAttr 

func (x *OpDef_ArgDef) GetNumberAttr() string

func (*OpDef_ArgDef) GetType 

func (x *OpDef_ArgDef) GetType() DataType

func (*OpDef_ArgDef) GetTypeAttr 

func (x *OpDef_ArgDef) GetTypeAttr() string

func (*OpDef_ArgDef) GetTypeListAttr 

func (x *OpDef_ArgDef) GetTypeListAttr() string

func (*OpDef_ArgDef) ProtoMessage 

func (*OpDef_ArgDef) ProtoMessage()

func (*OpDef_ArgDef) ProtoReflect 

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

func (*OpDef_ArgDef) Reset 

func (x *OpDef_ArgDef) Reset()

func (*OpDef_ArgDef) String 

func (x *OpDef_ArgDef) String() string

type OpDef_AttrDef 

type OpDef_AttrDef struct {

	// A descriptive name for the argument.  May be used, e.g. by the
	// Python client, as a keyword argument name, and so should match
	// the regexp "[a-z][a-z0-9_]+".
	Name string `protobuf:"bytes,1,opt,name=name,proto3" json:"name,omitempty"`
	// One of the type names from attr_value.proto ("string", "list(string)",
	// "int", etc.).
	Type string `protobuf:"bytes,2,opt,name=type,proto3" json:"type,omitempty"`
	// A reasonable default for this attribute if the user does not supply
	// a value.  If not specified, the user must supply a value.
	DefaultValue *AttrValue `protobuf:"bytes,3,opt,name=default_value,json=defaultValue,proto3" json:"default_value,omitempty"`
	// Human-readable description.
	Description string `protobuf:"bytes,4,opt,name=description,proto3" json:"description,omitempty"`
	// For type == "int", this is a minimum value.  For "list(___)"
	// types, this is the minimum length.
	HasMinimum bool  `protobuf:"varint,5,opt,name=has_minimum,json=hasMinimum,proto3" json:"has_minimum,omitempty"`
	Minimum    int64 `protobuf:"varint,6,opt,name=minimum,proto3" json:"minimum,omitempty"`
	// The set of allowed values.  Has type that is the "list" version
	// of the "type" field above (uses the "list" field of AttrValue).
	// If type == "type" or "list(type)" above, then the "type" field
	// of "allowed_values.list" has the set of allowed DataTypes.
	// If type == "string" or "list(string)", then the "s" field of
	// "allowed_values.list" has the set of allowed strings.
	AllowedValues *AttrValue `protobuf:"bytes,7,opt,name=allowed_values,json=allowedValues,proto3" json:"allowed_values,omitempty"`
	// contains filtered or unexported fields
}

Description of the graph-construction-time configuration of this Op. That is to say, this describes the attr fields that will be specified in the NodeDef.

func (*OpDef_AttrDef) Descriptor deprecated 

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

Deprecated: Use OpDef_AttrDef.ProtoReflect.Descriptor instead.

func (*OpDef_AttrDef) GetAllowedValues 

func (x *OpDef_AttrDef) GetAllowedValues() *AttrValue

func (*OpDef_AttrDef) GetDefaultValue 

func (x *OpDef_AttrDef) GetDefaultValue() *AttrValue

func (*OpDef_AttrDef) GetDescription 

func (x *OpDef_AttrDef) GetDescription() string

func (*OpDef_AttrDef) GetHasMinimum 

func (x *OpDef_AttrDef) GetHasMinimum() bool

func (*OpDef_AttrDef) GetMinimum 

func (x *OpDef_AttrDef) GetMinimum() int64

func (*OpDef_AttrDef) GetName 

func (x *OpDef_AttrDef) GetName() string

func (*OpDef_AttrDef) GetType 

func (x *OpDef_AttrDef) GetType() string

func (*OpDef_AttrDef) ProtoMessage 

func (*OpDef_AttrDef) ProtoMessage()

func (*OpDef_AttrDef) ProtoReflect 

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

func (*OpDef_AttrDef) Reset 

func (x *OpDef_AttrDef) Reset()

func (*OpDef_AttrDef) String 

func (x *OpDef_AttrDef) String() string

type OpDeprecation 

type OpDeprecation struct {

	// First GraphDef version at which the op is disallowed.
	Version int32 `protobuf:"varint,1,opt,name=version,proto3" json:"version,omitempty"`
	// Explanation of why it was deprecated and what to use instead.
	Explanation string `protobuf:"bytes,2,opt,name=explanation,proto3" json:"explanation,omitempty"`
	// contains filtered or unexported fields
}

Information about version-dependent deprecation of an op

func (*OpDeprecation) Descriptor deprecated 

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

Deprecated: Use OpDeprecation.ProtoReflect.Descriptor instead.

func (*OpDeprecation) GetExplanation 

func (x *OpDeprecation) GetExplanation() string

func (*OpDeprecation) GetVersion 

func (x *OpDeprecation) GetVersion() int32

func (*OpDeprecation) ProtoMessage 

func (*OpDeprecation) ProtoMessage()

func (*OpDeprecation) ProtoReflect 

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

func (*OpDeprecation) Reset 

func (x *OpDeprecation) Reset()

func (*OpDeprecation) String 

func (x *OpDeprecation) String() string

type OpList 

type OpList struct {
	Op []*OpDef `protobuf:"bytes,1,rep,name=op,proto3" json:"op,omitempty"`
	// contains filtered or unexported fields
}

A collection of OpDefs

func (*OpList) Descriptor deprecated 

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

Deprecated: Use OpList.ProtoReflect.Descriptor instead.

func (*OpList) GetOp 

func (x *OpList) GetOp() []*OpDef

func (*OpList) ProtoMessage 

func (*OpList) ProtoMessage()

func (*OpList) ProtoReflect 

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

func (*OpList) Reset 

func (x *OpList) Reset()

func (*OpList) String 

func (x *OpList) String() string

type RegisteredGradient 

type RegisteredGradient struct {
	GradientFunc     string `protobuf:"bytes,1,opt,name=gradient_func,json=gradientFunc,proto3" json:"gradient_func,omitempty"`               // The gradient function's name.
	RegisteredOpType string `protobuf:"bytes,2,opt,name=registered_op_type,json=registeredOpType,proto3" json:"registered_op_type,omitempty"` // The gradient function's registered op type.
	// contains filtered or unexported fields
}

RegisteredGradient stores a gradient function that is registered in the gradients library and used in the ops of a function in the function library. Unlike GradientDef, these gradients are identified by op type, and not directly linked to any function.

func (*RegisteredGradient) Descriptor deprecated 

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

Deprecated: Use RegisteredGradient.ProtoReflect.Descriptor instead.

func (*RegisteredGradient) GetGradientFunc 

func (x *RegisteredGradient) GetGradientFunc() string

func (*RegisteredGradient) GetRegisteredOpType 

func (x *RegisteredGradient) GetRegisteredOpType() string

func (*RegisteredGradient) ProtoMessage 

func (*RegisteredGradient) ProtoMessage()

func (*RegisteredGradient) ProtoReflect 

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

func (*RegisteredGradient) Reset 

func (x *RegisteredGradient) Reset()

func (*RegisteredGradient) String 

func (x *RegisteredGradient) String() string

type ResourceHandleProto 

type ResourceHandleProto struct {

	// Unique name for the device containing the resource.
	Device string `protobuf:"bytes,1,opt,name=device,proto3" json:"device,omitempty"`
	// Container in which this resource is placed.
	Container string `protobuf:"bytes,2,opt,name=container,proto3" json:"container,omitempty"`
	// Unique name of this resource.
	Name string `protobuf:"bytes,3,opt,name=name,proto3" json:"name,omitempty"`
	// Hash code for the type of the resource. Is only valid in the same device
	// and in the same execution.
	HashCode uint64 `protobuf:"varint,4,opt,name=hash_code,json=hashCode,proto3" json:"hash_code,omitempty"`
	// For debug-only, the name of the type pointed to by this handle, if
	// available.
	MaybeTypeName string `protobuf:"bytes,5,opt,name=maybe_type_name,json=maybeTypeName,proto3" json:"maybe_type_name,omitempty"`
	// Data types and shapes for the underlying resource.
	DtypesAndShapes []*ResourceHandleProto_DtypeAndShape `protobuf:"bytes,6,rep,name=dtypes_and_shapes,json=dtypesAndShapes,proto3" json:"dtypes_and_shapes,omitempty"`
	// contains filtered or unexported fields
}

Protocol buffer representing a handle to a tensorflow resource. Handles are not valid across executions, but can be serialized back and forth from within a single run.

func (*ResourceHandleProto) Descriptor deprecated 

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

Deprecated: Use ResourceHandleProto.ProtoReflect.Descriptor instead.

func (*ResourceHandleProto) GetContainer 

func (x *ResourceHandleProto) GetContainer() string

func (*ResourceHandleProto) GetDevice 

func (x *ResourceHandleProto) GetDevice() string

func (*ResourceHandleProto) GetDtypesAndShapes 

func (x *ResourceHandleProto) GetDtypesAndShapes() []*ResourceHandleProto_DtypeAndShape

func (*ResourceHandleProto) GetHashCode 

func (x *ResourceHandleProto) GetHashCode() uint64

func (*ResourceHandleProto) GetMaybeTypeName 

func (x *ResourceHandleProto) GetMaybeTypeName() string

func (*ResourceHandleProto) GetName 

func (x *ResourceHandleProto) GetName() string

func (*ResourceHandleProto) ProtoMessage 

func (*ResourceHandleProto) ProtoMessage()

func (*ResourceHandleProto) ProtoReflect 

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

func (*ResourceHandleProto) Reset 

func (x *ResourceHandleProto) Reset()

func (*ResourceHandleProto) String 

func (x *ResourceHandleProto) String() string

type ResourceHandleProto_DtypeAndShape 

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

Protocol buffer representing a pair of (data type, tensor shape).

func (*ResourceHandleProto_DtypeAndShape) Descriptor deprecated 

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

Deprecated: Use ResourceHandleProto_DtypeAndShape.ProtoReflect.Descriptor instead.

func (*ResourceHandleProto_DtypeAndShape) GetDtype 

func (x *ResourceHandleProto_DtypeAndShape) GetDtype() DataType

func (*ResourceHandleProto_DtypeAndShape) GetShape 

func (x *ResourceHandleProto_DtypeAndShape) GetShape() *TensorShapeProto

func (*ResourceHandleProto_DtypeAndShape) ProtoMessage 

func (*ResourceHandleProto_DtypeAndShape) ProtoMessage()

func (*ResourceHandleProto_DtypeAndShape) ProtoReflect 

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

func (*ResourceHandleProto_DtypeAndShape) Reset 

func (x *ResourceHandleProto_DtypeAndShape) Reset()

func (*ResourceHandleProto_DtypeAndShape) String 

func (x *ResourceHandleProto_DtypeAndShape) String() string

type SaveSliceInfoDef 

type SaveSliceInfoDef struct {

	// Name of the full variable of which this is a slice.
	FullName string `protobuf:"bytes,1,opt,name=full_name,json=fullName,proto3" json:"full_name,omitempty"`
	// Shape of the full variable.
	FullShape []int64 `protobuf:"varint,2,rep,packed,name=full_shape,json=fullShape,proto3" json:"full_shape,omitempty"`
	// Offset of this variable into the full variable.
	VarOffset []int64 `protobuf:"varint,3,rep,packed,name=var_offset,json=varOffset,proto3" json:"var_offset,omitempty"`
	// Shape of this variable.
	VarShape []int64 `protobuf:"varint,4,rep,packed,name=var_shape,json=varShape,proto3" json:"var_shape,omitempty"`
	// contains filtered or unexported fields
}

func (*SaveSliceInfoDef) Descriptor deprecated 

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

Deprecated: Use SaveSliceInfoDef.ProtoReflect.Descriptor instead.

func (*SaveSliceInfoDef) GetFullName 

func (x *SaveSliceInfoDef) GetFullName() string

func (*SaveSliceInfoDef) GetFullShape 

func (x *SaveSliceInfoDef) GetFullShape() []int64

func (*SaveSliceInfoDef) GetVarOffset 

func (x *SaveSliceInfoDef) GetVarOffset() []int64

func (*SaveSliceInfoDef) GetVarShape 

func (x *SaveSliceInfoDef) GetVarShape() []int64

func (*SaveSliceInfoDef) ProtoMessage 

func (*SaveSliceInfoDef) ProtoMessage()

func (*SaveSliceInfoDef) ProtoReflect 

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

func (*SaveSliceInfoDef) Reset 

func (x *SaveSliceInfoDef) Reset()

func (*SaveSliceInfoDef) String 

func (x *SaveSliceInfoDef) String() string

type StepStats 

type StepStats struct {
	DevStats []*DeviceStepStats `protobuf:"bytes,1,rep,name=dev_stats,json=devStats,proto3" json:"dev_stats,omitempty"`
	// contains filtered or unexported fields
}

func (*StepStats) Descriptor deprecated 

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

Deprecated: Use StepStats.ProtoReflect.Descriptor instead.

func (*StepStats) GetDevStats 

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

func (*StepStats) ProtoMessage 

func (*StepStats) ProtoMessage()

func (*StepStats) ProtoReflect 

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

func (*StepStats) Reset 

func (x *StepStats) Reset()

func (*StepStats) String 

func (x *StepStats) String() string

type TensorDescription 

type TensorDescription struct {

	// Data type of tensor elements
	Dtype DataType `protobuf:"varint,1,opt,name=dtype,proto3,enum=tensorflow.DataType" json:"dtype,omitempty"`
	// Shape of the tensor.
	Shape *TensorShapeProto `protobuf:"bytes,2,opt,name=shape,proto3" json:"shape,omitempty"`
	// Information about the size and allocator used for the data
	AllocationDescription *AllocationDescription `protobuf:"bytes,4,opt,name=allocation_description,json=allocationDescription,proto3" json:"allocation_description,omitempty"`
	// contains filtered or unexported fields
}

func (*TensorDescription) Descriptor deprecated 

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

Deprecated: Use TensorDescription.ProtoReflect.Descriptor instead.

func (*TensorDescription) GetAllocationDescription 

func (x *TensorDescription) GetAllocationDescription() *AllocationDescription

func (*TensorDescription) GetDtype 

func (x *TensorDescription) GetDtype() DataType

func (*TensorDescription) GetShape 

func (x *TensorDescription) GetShape() *TensorShapeProto

func (*TensorDescription) ProtoMessage 

func (*TensorDescription) ProtoMessage()

func (*TensorDescription) ProtoReflect 

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

func (*TensorDescription) Reset 

func (x *TensorDescription) Reset()

func (*TensorDescription) String 

func (x *TensorDescription) String() string

type TensorProto 

type TensorProto struct {
	Dtype DataType `protobuf:"varint,1,opt,name=dtype,proto3,enum=tensorflow.DataType" json:"dtype,omitempty"`
	// Shape of the tensor.  TODO(touts): sort out the 0-rank issues.
	TensorShape *TensorShapeProto `protobuf:"bytes,2,opt,name=tensor_shape,json=tensorShape,proto3" json:"tensor_shape,omitempty"`
	// Version number.
	//
	// In version 0, if the "repeated xxx" representations contain only one
	// element, that element is repeated to fill the shape.  This makes it easy
	// to represent a constant Tensor with a single value.
	VersionNumber int32 `protobuf:"varint,3,opt,name=version_number,json=versionNumber,proto3" json:"version_number,omitempty"`
	// Serialized raw tensor content from either Tensor::AsProtoTensorContent or
	// memcpy in tensorflow::grpc::EncodeTensorToByteBuffer. This representation
	// can be used for all tensor types. The purpose of this representation is to
	// reduce serialization overhead during RPC call by avoiding serialization of
	// many repeated small items.
	TensorContent []byte `protobuf:"bytes,4,opt,name=tensor_content,json=tensorContent,proto3" json:"tensor_content,omitempty"`
	// DT_HALF, DT_BFLOAT16. Note that since protobuf has no int16 type, we'll
	// have some pointless zero padding for each value here.
	HalfVal []int32 `protobuf:"varint,13,rep,packed,name=half_val,json=halfVal,proto3" json:"half_val,omitempty"`
	// DT_FLOAT.
	FloatVal []float32 `protobuf:"fixed32,5,rep,packed,name=float_val,json=floatVal,proto3" json:"float_val,omitempty"`
	// DT_DOUBLE.
	DoubleVal []float64 `protobuf:"fixed64,6,rep,packed,name=double_val,json=doubleVal,proto3" json:"double_val,omitempty"`
	// DT_INT32, DT_INT16, DT_UINT16, DT_INT8, DT_UINT8.
	IntVal []int32 `protobuf:"varint,7,rep,packed,name=int_val,json=intVal,proto3" json:"int_val,omitempty"`
	// DT_STRING
	StringVal [][]byte `protobuf:"bytes,8,rep,name=string_val,json=stringVal,proto3" json:"string_val,omitempty"`
	// DT_COMPLEX64. scomplex_val(2*i) and scomplex_val(2*i+1) are real
	// and imaginary parts of i-th single precision complex.
	ScomplexVal []float32 `protobuf:"fixed32,9,rep,packed,name=scomplex_val,json=scomplexVal,proto3" json:"scomplex_val,omitempty"`
	// DT_INT64
	Int64Val []int64 `protobuf:"varint,10,rep,packed,name=int64_val,json=int64Val,proto3" json:"int64_val,omitempty"`
	// DT_BOOL
	BoolVal []bool `protobuf:"varint,11,rep,packed,name=bool_val,json=boolVal,proto3" json:"bool_val,omitempty"`
	// DT_COMPLEX128. dcomplex_val(2*i) and dcomplex_val(2*i+1) are real
	// and imaginary parts of i-th double precision complex.
	DcomplexVal []float64 `protobuf:"fixed64,12,rep,packed,name=dcomplex_val,json=dcomplexVal,proto3" json:"dcomplex_val,omitempty"`
	// DT_RESOURCE
	ResourceHandleVal []*ResourceHandleProto `protobuf:"bytes,14,rep,name=resource_handle_val,json=resourceHandleVal,proto3" json:"resource_handle_val,omitempty"`
	// DT_VARIANT
	VariantVal []*VariantTensorDataProto `protobuf:"bytes,15,rep,name=variant_val,json=variantVal,proto3" json:"variant_val,omitempty"`
	// DT_UINT32
	Uint32Val []uint32 `protobuf:"varint,16,rep,packed,name=uint32_val,json=uint32Val,proto3" json:"uint32_val,omitempty"`
	// DT_UINT64
	Uint64Val []uint64 `protobuf:"varint,17,rep,packed,name=uint64_val,json=uint64Val,proto3" json:"uint64_val,omitempty"`
	// contains filtered or unexported fields
}

Protocol buffer representing a tensor.

func (*TensorProto) Descriptor deprecated 

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

Deprecated: Use TensorProto.ProtoReflect.Descriptor instead.

func (*TensorProto) GetBoolVal 

func (x *TensorProto) GetBoolVal() []bool

func (*TensorProto) GetDcomplexVal 

func (x *TensorProto) GetDcomplexVal() []float64

func (*TensorProto) GetDoubleVal 

func (x *TensorProto) GetDoubleVal() []float64

func (*TensorProto) GetDtype 

func (x *TensorProto) GetDtype() DataType

func (*TensorProto) GetFloatVal 

func (x *TensorProto) GetFloatVal() []float32

func (*TensorProto) GetHalfVal 

func (x *TensorProto) GetHalfVal() []int32

func (*TensorProto) GetInt64Val 

func (x *TensorProto) GetInt64Val() []int64

func (*TensorProto) GetIntVal 

func (x *TensorProto) GetIntVal() []int32

func (*TensorProto) GetResourceHandleVal 

func (x *TensorProto) GetResourceHandleVal() []*ResourceHandleProto

func (*TensorProto) GetScomplexVal 

func (x *TensorProto) GetScomplexVal() []float32

func (*TensorProto) GetStringVal 

func (x *TensorProto) GetStringVal() [][]byte

func (*TensorProto) GetTensorContent 

func (x *TensorProto) GetTensorContent() []byte

func (*TensorProto) GetTensorShape 

func (x *TensorProto) GetTensorShape() *TensorShapeProto

func (*TensorProto) GetUint32Val 

func (x *TensorProto) GetUint32Val() []uint32

func (*TensorProto) GetUint64Val 

func (x *TensorProto) GetUint64Val() []uint64

func (*TensorProto) GetVariantVal 

func (x *TensorProto) GetVariantVal() []*VariantTensorDataProto

func (*TensorProto) GetVersionNumber 

func (x *TensorProto) GetVersionNumber() int32

func (*TensorProto) ProtoMessage 

func (*TensorProto) ProtoMessage()

func (*TensorProto) ProtoReflect 

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

func (*TensorProto) Reset 

func (x *TensorProto) Reset()

func (*TensorProto) String 

func (x *TensorProto) String() string

type TensorShapeProto 

type TensorShapeProto struct {

	// Dimensions of the tensor, such as {"input", 30}, {"output", 40}
	// for a 30 x 40 2D tensor.  If an entry has size -1, this
	// corresponds to a dimension of unknown size. The names are
	// optional.
	//
	// The order of entries in "dim" matters: It indicates the layout of the
	// values in the tensor in-memory representation.
	//
	// The first entry in "dim" is the outermost dimension used to layout the
	// values, the last entry is the innermost dimension.  This matches the
	// in-memory layout of RowMajor Eigen tensors.
	//
	// If "dim.size()" > 0, "unknown_rank" must be false.
	Dim []*TensorShapeProto_Dim `protobuf:"bytes,2,rep,name=dim,proto3" json:"dim,omitempty"`
	// If true, the number of dimensions in the shape is unknown.
	//
	// If true, "dim.size()" must be 0.
	UnknownRank bool `protobuf:"varint,3,opt,name=unknown_rank,json=unknownRank,proto3" json:"unknown_rank,omitempty"`
	// contains filtered or unexported fields
}

Dimensions of a tensor.

func (*TensorShapeProto) Descriptor deprecated 

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

Deprecated: Use TensorShapeProto.ProtoReflect.Descriptor instead.

func (*TensorShapeProto) GetDim 

func (x *TensorShapeProto) GetDim() []*TensorShapeProto_Dim

func (*TensorShapeProto) GetUnknownRank 

func (x *TensorShapeProto) GetUnknownRank() bool

func (*TensorShapeProto) ProtoMessage 

func (*TensorShapeProto) ProtoMessage()

func (*TensorShapeProto) ProtoReflect 

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

func (*TensorShapeProto) Reset 

func (x *TensorShapeProto) Reset()

func (*TensorShapeProto) String 

func (x *TensorShapeProto) String() string

type TensorShapeProto_Dim 

type TensorShapeProto_Dim struct {

	// Size of the tensor in that dimension.
	// This value must be >= -1, but values of -1 are reserved for "unknown"
	// shapes (values of -1 mean "unknown" dimension).  Certain wrappers
	// that work with TensorShapeProto may fail at runtime when deserializing
	// a TensorShapeProto containing a dim value of -1.
	Size int64 `protobuf:"varint,1,opt,name=size,proto3" json:"size,omitempty"`
	// Optional name of the tensor dimension.
	Name string `protobuf:"bytes,2,opt,name=name,proto3" json:"name,omitempty"`
	// contains filtered or unexported fields
}

One dimension of the tensor.

func (*TensorShapeProto_Dim) Descriptor deprecated 

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

Deprecated: Use TensorShapeProto_Dim.ProtoReflect.Descriptor instead.

func (*TensorShapeProto_Dim) GetName 

func (x *TensorShapeProto_Dim) GetName() string

func (*TensorShapeProto_Dim) GetSize 

func (x *TensorShapeProto_Dim) GetSize() int64

func (*TensorShapeProto_Dim) ProtoMessage 

func (*TensorShapeProto_Dim) ProtoMessage()

func (*TensorShapeProto_Dim) ProtoReflect 

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

func (*TensorShapeProto_Dim) Reset 

func (x *TensorShapeProto_Dim) Reset()

func (*TensorShapeProto_Dim) String 

func (x *TensorShapeProto_Dim) String() string

type VariableAggregation 

type VariableAggregation int32

Indicates how a distributed variable will be aggregated. 

const (
	// `NONE`: This is the default, giving an error if you use a
	// variable-update operation with multiple replicas.
	VariableAggregation_VARIABLE_AGGREGATION_NONE VariableAggregation = 0
	// `SUM`: Add the updates across replicas.
	VariableAggregation_VARIABLE_AGGREGATION_SUM VariableAggregation = 1
	// `MEAN`: Take the arithmetic mean ("average") of the updates across
	// replicas.
	VariableAggregation_VARIABLE_AGGREGATION_MEAN VariableAggregation = 2
	// `ONLY_FIRST_REPLICA`: This is for when every replica is performing the same
	// update, but we only want to perform the update once. Used, e.g., for the
	// global step counter.
	VariableAggregation_VARIABLE_AGGREGATION_ONLY_FIRST_REPLICA VariableAggregation = 3
)

func (VariableAggregation) Descriptor 

func (VariableAggregation) Descriptor() protoreflect.EnumDescriptor

func (VariableAggregation) Enum 

func (x VariableAggregation) Enum() *VariableAggregation

func (VariableAggregation) EnumDescriptor deprecated 

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

Deprecated: Use VariableAggregation.Descriptor instead.

func (VariableAggregation) Number 

func (x VariableAggregation) Number() protoreflect.EnumNumber

func (VariableAggregation) String 

func (x VariableAggregation) String() string

func (VariableAggregation) Type 

func (VariableAggregation) Type() protoreflect.EnumType

type VariableDef 

type VariableDef struct {

	// Name of the variable tensor.
	VariableName string `protobuf:"bytes,1,opt,name=variable_name,json=variableName,proto3" json:"variable_name,omitempty"`
	// Name of the tensor holding the variable's initial value.
	InitialValueName string `protobuf:"bytes,6,opt,name=initial_value_name,json=initialValueName,proto3" json:"initial_value_name,omitempty"`
	// Name of the initializer op.
	InitializerName string `protobuf:"bytes,2,opt,name=initializer_name,json=initializerName,proto3" json:"initializer_name,omitempty"`
	// Name of the snapshot tensor.
	SnapshotName string `protobuf:"bytes,3,opt,name=snapshot_name,json=snapshotName,proto3" json:"snapshot_name,omitempty"`
	// Support for saving variables as slices of a larger variable.
	SaveSliceInfoDef *SaveSliceInfoDef `protobuf:"bytes,4,opt,name=save_slice_info_def,json=saveSliceInfoDef,proto3" json:"save_slice_info_def,omitempty"`
	// Whether to represent this as a ResourceVariable.
	IsResource bool `protobuf:"varint,5,opt,name=is_resource,json=isResource,proto3" json:"is_resource,omitempty"`
	// Whether this variable should be trained.
	Trainable bool `protobuf:"varint,7,opt,name=trainable,proto3" json:"trainable,omitempty"`
	// Indicates when a distributed variable will be synced.
	Synchronization VariableSynchronization `protobuf:"varint,8,opt,name=synchronization,proto3,enum=tensorflow.VariableSynchronization" json:"synchronization,omitempty"`
	// Indicates how a distributed variable will be aggregated.
	Aggregation VariableAggregation `protobuf:"varint,9,opt,name=aggregation,proto3,enum=tensorflow.VariableAggregation" json:"aggregation,omitempty"`
	// contains filtered or unexported fields
}

Protocol buffer representing a Variable.

func (*VariableDef) Descriptor deprecated 

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

Deprecated: Use VariableDef.ProtoReflect.Descriptor instead.

func (*VariableDef) GetAggregation 

func (x *VariableDef) GetAggregation() VariableAggregation

func (*VariableDef) GetInitialValueName 

func (x *VariableDef) GetInitialValueName() string

func (*VariableDef) GetInitializerName 

func (x *VariableDef) GetInitializerName() string

func (*VariableDef) GetIsResource 

func (x *VariableDef) GetIsResource() bool

func (*VariableDef) GetSaveSliceInfoDef 

func (x *VariableDef) GetSaveSliceInfoDef() *SaveSliceInfoDef

func (*VariableDef) GetSnapshotName 

func (x *VariableDef) GetSnapshotName() string

func (*VariableDef) GetSynchronization 

func (x *VariableDef) GetSynchronization() VariableSynchronization

func (*VariableDef) GetTrainable 

func (x *VariableDef) GetTrainable() bool

func (*VariableDef) GetVariableName 

func (x *VariableDef) GetVariableName() string

func (*VariableDef) ProtoMessage 

func (*VariableDef) ProtoMessage()

func (*VariableDef) ProtoReflect 

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

func (*VariableDef) Reset 

func (x *VariableDef) Reset()

func (*VariableDef) String 

func (x *VariableDef) String() string

type VariableSynchronization 

type VariableSynchronization int32

Indicates when a distributed variable will be synced. 

const (
	// `AUTO`: Indicates that the synchronization will be determined by the
	// current `DistributionStrategy` (eg. With `MirroredStrategy` this would be
	// `ON_WRITE`).
	VariableSynchronization_VARIABLE_SYNCHRONIZATION_AUTO VariableSynchronization = 0
	// `NONE`: Indicates that there will only be one copy of the variable, so
	// there is no need to sync.
	VariableSynchronization_VARIABLE_SYNCHRONIZATION_NONE VariableSynchronization = 1
	// `ON_WRITE`: Indicates that the variable will be updated across devices
	// every time it is written.
	VariableSynchronization_VARIABLE_SYNCHRONIZATION_ON_WRITE VariableSynchronization = 2
	// `ON_READ`: Indicates that the variable will be aggregated across devices
	// when it is read (eg. when checkpointing or when evaluating an op that uses
	// the variable).
	VariableSynchronization_VARIABLE_SYNCHRONIZATION_ON_READ VariableSynchronization = 3
)

func (VariableSynchronization) Descriptor 

func (VariableSynchronization) Descriptor() protoreflect.EnumDescriptor

func (VariableSynchronization) Enum 

func (x VariableSynchronization) Enum() *VariableSynchronization

func (VariableSynchronization) EnumDescriptor deprecated 

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

Deprecated: Use VariableSynchronization.Descriptor instead.

func (VariableSynchronization) Number 

func (x VariableSynchronization) Number() protoreflect.EnumNumber

func (VariableSynchronization) String 

func (x VariableSynchronization) String() string

func (VariableSynchronization) Type 

func (VariableSynchronization) Type() protoreflect.EnumType

type VariantTensorDataProto 

type VariantTensorDataProto struct {

	// Name of the type of objects being serialized.
	TypeName string `protobuf:"bytes,1,opt,name=type_name,json=typeName,proto3" json:"type_name,omitempty"`
	// Portions of the object that are not Tensors.
	Metadata []byte `protobuf:"bytes,2,opt,name=metadata,proto3" json:"metadata,omitempty"`
	// Tensors contained within objects being serialized.
	Tensors []*TensorProto `protobuf:"bytes,3,rep,name=tensors,proto3" json:"tensors,omitempty"`
	// contains filtered or unexported fields
}

Protocol buffer representing the serialization format of DT_VARIANT tensors.

func (*VariantTensorDataProto) Descriptor deprecated 

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

Deprecated: Use VariantTensorDataProto.ProtoReflect.Descriptor instead.

func (*VariantTensorDataProto) GetMetadata 

func (x *VariantTensorDataProto) GetMetadata() []byte

func (*VariantTensorDataProto) GetTensors 

func (x *VariantTensorDataProto) GetTensors() []*TensorProto

func (*VariantTensorDataProto) GetTypeName 

func (x *VariantTensorDataProto) GetTypeName() string

func (*VariantTensorDataProto) ProtoMessage 

func (*VariantTensorDataProto) ProtoMessage()

func (*VariantTensorDataProto) ProtoReflect 

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

func (*VariantTensorDataProto) Reset 

func (x *VariantTensorDataProto) Reset()

func (*VariantTensorDataProto) String 

func (x *VariantTensorDataProto) String() string

type VersionDef 

type VersionDef struct {

	// The version of the code that produced this data.
	Producer int32 `protobuf:"varint,1,opt,name=producer,proto3" json:"producer,omitempty"`
	// Any consumer below this version is not allowed to consume this data.
	MinConsumer int32 `protobuf:"varint,2,opt,name=min_consumer,json=minConsumer,proto3" json:"min_consumer,omitempty"`
	// Specific consumer versions which are disallowed (e.g. due to bugs).
	BadConsumers []int32 `protobuf:"varint,3,rep,packed,name=bad_consumers,json=badConsumers,proto3" json:"bad_consumers,omitempty"`
	// contains filtered or unexported fields
}

Version information for a piece of serialized data

There are different types of versions for each type of data (GraphDef, etc.), but they all have the same common shape described here.

Each consumer has "consumer" and "min_producer" versions (specified elsewhere). A consumer is allowed to consume this data if 

producer >= min_producer
consumer >= min_consumer
consumer not in bad_consumers

func (*VersionDef) Descriptor deprecated 

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

Deprecated: Use VersionDef.ProtoReflect.Descriptor instead.

func (*VersionDef) GetBadConsumers 

func (x *VersionDef) GetBadConsumers() []int32

func (*VersionDef) GetMinConsumer 

func (x *VersionDef) GetMinConsumer() int32

func (*VersionDef) GetProducer 

func (x *VersionDef) GetProducer() int32

func (*VersionDef) ProtoMessage 

func (*VersionDef) ProtoMessage()

func (*VersionDef) ProtoReflect 

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

func (*VersionDef) Reset 

func (x *VersionDef) Reset()

func (*VersionDef) String 

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