framework

package
v2.12.0 Latest Latest
Warning

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

 Go to latest Published: Dec 8, 2023 License: BSD-3-Clause Imports: 5 Imported by: 0

Documentation

Index

Constants

This section is empty.

Variables

View Source 
var (
	ApiDef_Visibility_name = map[int32]string{
		0: "DEFAULT_VISIBILITY",
		1: "VISIBLE",
		2: "SKIP",
		3: "HIDDEN",
	}
	ApiDef_Visibility_value = map[string]int32{
		"DEFAULT_VISIBILITY": 0,
		"VISIBLE":            1,
		"SKIP":               2,
		"HIDDEN":             3,
	}
)

Enum value maps for ApiDef_Visibility.

View Source 
var (
	AutoShardPolicy_name = map[int32]string{
		0:  "AUTO",
		1:  "FILE",
		2:  "DATA",
		3:  "HINT",
		-1: "OFF",
	}
	AutoShardPolicy_value = map[string]int32{
		"AUTO": 0,
		"FILE": 1,
		"DATA": 2,
		"HINT": 3,
		"OFF":  -1,
	}
)

Enum value maps for AutoShardPolicy.

View Source 
var (
	ExternalStatePolicy_name = map[int32]string{
		0: "POLICY_WARN",
		1: "POLICY_IGNORE",
		2: "POLICY_FAIL",
	}
	ExternalStatePolicy_value = map[string]int32{
		"POLICY_WARN":   0,
		"POLICY_IGNORE": 1,
		"POLICY_FAIL":   2,
	}
)

Enum value maps for ExternalStatePolicy.

View Source 
var (
	CardinalityOptions_ComputeLevel_name = map[int32]string{
		0: "CARDINALITY_COMPUTE_UNSPECIFIED",
		1: "CARDINALITY_COMPUTE_LOW",
		2: "CARDINALITY_COMPUTE_MODERATE",
	}
	CardinalityOptions_ComputeLevel_value = map[string]int32{
		"CARDINALITY_COMPUTE_UNSPECIFIED": 0,
		"CARDINALITY_COMPUTE_LOW":         1,
		"CARDINALITY_COMPUTE_MODERATE":    2,
	}
)

Enum value maps for CardinalityOptions_ComputeLevel.

View Source 
var (
	FullTypeId_name = map[int32]string{
		0:     "TFT_UNSET",
		1:     "TFT_VAR",
		2:     "TFT_ANY",
		3:     "TFT_PRODUCT",
		4:     "TFT_NAMED",
		20:    "TFT_FOR_EACH",
		100:   "TFT_CALLABLE",
		1000:  "TFT_TENSOR",
		1001:  "TFT_ARRAY",
		1002:  "TFT_OPTIONAL",
		1003:  "TFT_LITERAL",
		1004:  "TFT_ENCODED",
		1005:  "TFT_SHAPE_TENSOR",
		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",
		10102: "TFT_DATASET",
		10103: "TFT_RAGGED",
		10104: "TFT_ITERATOR",
		10202: "TFT_MUTEX_LOCK",
		10203: "TFT_LEGACY_VARIANT",
	}
	FullTypeId_value = map[string]int32{
		"TFT_UNSET":          0,
		"TFT_VAR":            1,
		"TFT_ANY":            2,
		"TFT_PRODUCT":        3,
		"TFT_NAMED":          4,
		"TFT_FOR_EACH":       20,
		"TFT_CALLABLE":       100,
		"TFT_TENSOR":         1000,
		"TFT_ARRAY":          1001,
		"TFT_OPTIONAL":       1002,
		"TFT_LITERAL":        1003,
		"TFT_ENCODED":        1004,
		"TFT_SHAPE_TENSOR":   1005,
		"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,
		"TFT_DATASET":        10102,
		"TFT_RAGGED":         10103,
		"TFT_ITERATOR":       10104,
		"TFT_MUTEX_LOCK":     10202,
		"TFT_LEGACY_VARIANT": 10203,
	}
)

Enum value maps for FullTypeId.

View Source 
var (
	GraphTransferInfo_Destination_name = map[int32]string{
		0: "NOP",
		1: "HEXAGON",
	}
	GraphTransferInfo_Destination_value = map[string]int32{
		"NOP":     0,
		"HEXAGON": 1,
	}
)

Enum value maps for GraphTransferInfo_Destination.

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

Enum value maps for NodeClass.

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

Enum value maps for AutotuneAlgorithm.

View Source 
var (
	DataClass_name = map[int32]string{
		0: "DATA_CLASS_UNKNOWN",
		1: "DATA_CLASS_SCALAR",
		2: "DATA_CLASS_TENSOR",
		3: "DATA_CLASS_BLOB_SEQUENCE",
	}
	DataClass_value = map[string]int32{
		"DATA_CLASS_UNKNOWN":       0,
		"DATA_CLASS_SCALAR":        1,
		"DATA_CLASS_TENSOR":        2,
		"DATA_CLASS_BLOB_SEQUENCE": 3,
	}
)

Enum value maps for DataClass.

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",
		24:  "DT_FLOAT8_E5M2",
		25:  "DT_FLOAT8_E4M3FN",
		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",
		124: "DT_FLOAT8_E5M2_REF",
		125: "DT_FLOAT8_E4M3FN_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_FLOAT8_E5M2":       24,
		"DT_FLOAT8_E4M3FN":     25,
		"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,
		"DT_FLOAT8_E5M2_REF":   124,
		"DT_FLOAT8_E4M3FN_REF": 125,
	}
)

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_api_def_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_dataset_metadata_proto protoreflect.FileDescriptor
View Source 
var File_tensorflow_core_framework_dataset_options_proto protoreflect.FileDescriptor
View Source 
var File_tensorflow_core_framework_dataset_proto protoreflect.FileDescriptor
View Source 
var File_tensorflow_core_framework_device_attributes_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_graph_transfer_info_proto protoreflect.FileDescriptor
View Source 
var File_tensorflow_core_framework_kernel_def_proto protoreflect.FileDescriptor
View Source 
var File_tensorflow_core_framework_log_memory_proto protoreflect.FileDescriptor
View Source 
var File_tensorflow_core_framework_model_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_optimized_function_graph_proto protoreflect.FileDescriptor
View Source 
var File_tensorflow_core_framework_reader_base_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_summary_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_tensor_slice_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 ApiDef 

type ApiDef struct {

	// Name of the op (in the OpDef) to specify the API for.
	GraphOpName string `protobuf:"bytes,1,opt,name=graph_op_name,json=graphOpName,proto3" json:"graph_op_name,omitempty"`
	// If this op is deprecated, set deprecation message to the message
	// that should be logged when this op is used.
	// The message should indicate alternative op to use, if any.
	DeprecationMessage string `protobuf:"bytes,12,opt,name=deprecation_message,json=deprecationMessage,proto3" json:"deprecation_message,omitempty"`
	// Major version when the op will be deleted. For e.g. set this
	// value to 2 if op API should be removed in TensorFlow 2.0 and
	// deprecated in versions before that.
	DeprecationVersion int32              `protobuf:"varint,13,opt,name=deprecation_version,json=deprecationVersion,proto3" json:"deprecation_version,omitempty"`
	Visibility         ApiDef_Visibility  `protobuf:"varint,2,opt,name=visibility,proto3,enum=tensorflow.ApiDef_Visibility" json:"visibility,omitempty"`
	Endpoint           []*ApiDef_Endpoint `protobuf:"bytes,3,rep,name=endpoint,proto3" json:"endpoint,omitempty"`
	InArg              []*ApiDef_Arg      `protobuf:"bytes,4,rep,name=in_arg,json=inArg,proto3" json:"in_arg,omitempty"`
	OutArg             []*ApiDef_Arg      `protobuf:"bytes,5,rep,name=out_arg,json=outArg,proto3" json:"out_arg,omitempty"`
	// List of original in_arg names to specify new argument order.
	// Length of arg_order should be either empty to keep current order
	// or match size of in_arg.
	ArgOrder []string       `protobuf:"bytes,11,rep,name=arg_order,json=argOrder,proto3" json:"arg_order,omitempty"`
	Attr     []*ApiDef_Attr `protobuf:"bytes,6,rep,name=attr,proto3" json:"attr,omitempty"`
	// One-line human-readable description of what the Op does.
	Summary string `protobuf:"bytes,7,opt,name=summary,proto3" json:"summary,omitempty"`
	// Additional, longer human-readable description of what the Op does.
	Description string `protobuf:"bytes,8,opt,name=description,proto3" json:"description,omitempty"`
	// Modify an existing/inherited description by adding text to the beginning
	// or end.
	DescriptionPrefix string `protobuf:"bytes,9,opt,name=description_prefix,json=descriptionPrefix,proto3" json:"description_prefix,omitempty"`
	DescriptionSuffix string `protobuf:"bytes,10,opt,name=description_suffix,json=descriptionSuffix,proto3" json:"description_suffix,omitempty"`
	// contains filtered or unexported fields
}

Used to specify and override the default API & behavior in the generated code for client languages, from what you would get from the OpDef alone. There will be a set of ApiDefs that are common to all client languages, and another set per client language. The per-client-language ApiDefs will inherit values from the common ApiDefs which it can either replace or modify.

We separate the API definition from the OpDef so we can evolve the API while remaining backwards compatible when interpreting old graphs. Overrides go in an "api_def.pbtxt" file with a text-format ApiDefs message.

WARNING: Be *very* careful changing the API for any existing op -- you can change the semantics of existing code. These changes may need to wait until a major release of TensorFlow to avoid breaking our compatibility promises.

func (*ApiDef) Descriptor deprecated 

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

Deprecated: Use ApiDef.ProtoReflect.Descriptor instead.

func (*ApiDef) GetArgOrder 

func (x *ApiDef) GetArgOrder() []string

func (*ApiDef) GetAttr 

func (x *ApiDef) GetAttr() []*ApiDef_Attr

func (*ApiDef) GetDeprecationMessage 

func (x *ApiDef) GetDeprecationMessage() string

func (*ApiDef) GetDeprecationVersion 

func (x *ApiDef) GetDeprecationVersion() int32

func (*ApiDef) GetDescription 

func (x *ApiDef) GetDescription() string

func (*ApiDef) GetDescriptionPrefix 

func (x *ApiDef) GetDescriptionPrefix() string

func (*ApiDef) GetDescriptionSuffix 

func (x *ApiDef) GetDescriptionSuffix() string

func (*ApiDef) GetEndpoint 

func (x *ApiDef) GetEndpoint() []*ApiDef_Endpoint

func (*ApiDef) GetGraphOpName 

func (x *ApiDef) GetGraphOpName() string

func (*ApiDef) GetInArg 

func (x *ApiDef) GetInArg() []*ApiDef_Arg

func (*ApiDef) GetOutArg 

func (x *ApiDef) GetOutArg() []*ApiDef_Arg

func (*ApiDef) GetSummary 

func (x *ApiDef) GetSummary() string

func (*ApiDef) GetVisibility 

func (x *ApiDef) GetVisibility() ApiDef_Visibility

func (*ApiDef) ProtoMessage 

func (*ApiDef) ProtoMessage()

func (*ApiDef) ProtoReflect 

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

func (*ApiDef) Reset 

func (x *ApiDef) Reset()

func (*ApiDef) String 

func (x *ApiDef) String() string

type ApiDef_Arg 

type ApiDef_Arg struct {
	Name string `protobuf:"bytes,1,opt,name=name,proto3" json:"name,omitempty"`
	// Change the name used to access this arg in the API from what
	// is used in the GraphDef.  Note that these names in `backticks`
	// will also be replaced in the summary & description fields.
	RenameTo string `protobuf:"bytes,2,opt,name=rename_to,json=renameTo,proto3" json:"rename_to,omitempty"`
	// Note: this will replace any inherited arg doc. There is no
	// current way of modifying arg descriptions (other than replacing
	// them entirely) as can be done with op descriptions.
	Description string `protobuf:"bytes,3,opt,name=description,proto3" json:"description,omitempty"`
	// contains filtered or unexported fields
}

func (*ApiDef_Arg) Descriptor deprecated 

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

Deprecated: Use ApiDef_Arg.ProtoReflect.Descriptor instead.

func (*ApiDef_Arg) GetDescription 

func (x *ApiDef_Arg) GetDescription() string

func (*ApiDef_Arg) GetName 

func (x *ApiDef_Arg) GetName() string

func (*ApiDef_Arg) GetRenameTo 

func (x *ApiDef_Arg) GetRenameTo() string

func (*ApiDef_Arg) ProtoMessage 

func (*ApiDef_Arg) ProtoMessage()

func (*ApiDef_Arg) ProtoReflect 

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

func (*ApiDef_Arg) Reset 

func (x *ApiDef_Arg) Reset()

func (*ApiDef_Arg) String 

func (x *ApiDef_Arg) String() string

type ApiDef_Attr 

type ApiDef_Attr struct {
	Name string `protobuf:"bytes,1,opt,name=name,proto3" json:"name,omitempty"`
	// Change the name used to access this attr in the API from what
	// is used in the GraphDef.  Note that these names in `backticks`
	// will also be replaced in the summary & description fields.
	RenameTo string `protobuf:"bytes,2,opt,name=rename_to,json=renameTo,proto3" json:"rename_to,omitempty"`
	// Specify a new default value to use for this attr.  This default
	// will be used when creating new graphs, as opposed to the
	// default in the OpDef, which will be used when interpreting old
	// GraphDefs.
	DefaultValue *AttrValue `protobuf:"bytes,3,opt,name=default_value,json=defaultValue,proto3" json:"default_value,omitempty"`
	// Note: this will replace any inherited attr doc, there is no current
	// way of modifying attr descriptions as can be done with op descriptions.
	Description string `protobuf:"bytes,4,opt,name=description,proto3" json:"description,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 (*ApiDef_Attr) Descriptor deprecated 

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

Deprecated: Use ApiDef_Attr.ProtoReflect.Descriptor instead.

func (*ApiDef_Attr) GetDefaultValue 

func (x *ApiDef_Attr) GetDefaultValue() *AttrValue

func (*ApiDef_Attr) GetDescription 

func (x *ApiDef_Attr) GetDescription() string

func (*ApiDef_Attr) GetName 

func (x *ApiDef_Attr) GetName() string

func (*ApiDef_Attr) GetRenameTo 

func (x *ApiDef_Attr) GetRenameTo() string

func (*ApiDef_Attr) ProtoMessage 

func (*ApiDef_Attr) ProtoMessage()

func (*ApiDef_Attr) ProtoReflect 

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

func (*ApiDef_Attr) Reset 

func (x *ApiDef_Attr) Reset()

func (*ApiDef_Attr) String 

func (x *ApiDef_Attr) String() string

type ApiDef_Endpoint 

type ApiDef_Endpoint struct {

	// Name should be either like "CamelCaseName" or
	// "Package.CamelCaseName". Client-language-specific ApiDefs may
	// use a snake_case convention instead of CamelCase.
	Name string `protobuf:"bytes,1,opt,name=name,proto3" json:"name,omitempty"`
	// Set if this endpoint is deprecated. If set to true, a message suggesting
	// to use a non-deprecated endpoint instead will be printed. If all
	// endpoints are deprecated, set deprecation_message in ApiDef instead.
	Deprecated bool `protobuf:"varint,3,opt,name=deprecated,proto3" json:"deprecated,omitempty"`
	// Major version when an endpoint will be deleted. For e.g. set this
	// value to 2 if endpoint should be removed in TensorFlow 2.0 and
	// deprecated in versions before that.
	DeprecationVersion int32 `protobuf:"varint,4,opt,name=deprecation_version,json=deprecationVersion,proto3" json:"deprecation_version,omitempty"`
	// contains filtered or unexported fields
}

If you specify any endpoint, this will replace all of the inherited endpoints. The first endpoint should be the "canonical" endpoint, and should not be deprecated (unless all endpoints are deprecated).

func (*ApiDef_Endpoint) Descriptor deprecated 

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

Deprecated: Use ApiDef_Endpoint.ProtoReflect.Descriptor instead.

func (*ApiDef_Endpoint) GetDeprecated 

func (x *ApiDef_Endpoint) GetDeprecated() bool

func (*ApiDef_Endpoint) GetDeprecationVersion 

func (x *ApiDef_Endpoint) GetDeprecationVersion() int32

func (*ApiDef_Endpoint) GetName 

func (x *ApiDef_Endpoint) GetName() string

func (*ApiDef_Endpoint) ProtoMessage 

func (*ApiDef_Endpoint) ProtoMessage()

func (*ApiDef_Endpoint) ProtoReflect 

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

func (*ApiDef_Endpoint) Reset 

func (x *ApiDef_Endpoint) Reset()

func (*ApiDef_Endpoint) String 

func (x *ApiDef_Endpoint) String() string

type ApiDef_Visibility 

type ApiDef_Visibility int32
const (
	// Normally this is "VISIBLE" unless you are inheriting a
	// different value from another ApiDef.
	ApiDef_DEFAULT_VISIBILITY ApiDef_Visibility = 0
	// Publicly visible in the API.
	ApiDef_VISIBLE ApiDef_Visibility = 1
	// Do not include this op in the generated API. If visibility is
	// set to 'SKIP', other fields are ignored for this op.
	ApiDef_SKIP ApiDef_Visibility = 2
	// Hide this op by putting it into an internal namespace (or whatever
	// is appropriate in the target language).
	ApiDef_HIDDEN ApiDef_Visibility = 3
)

func (ApiDef_Visibility) Descriptor 

func (ApiDef_Visibility) Descriptor() protoreflect.EnumDescriptor

func (ApiDef_Visibility) Enum 

func (x ApiDef_Visibility) Enum() *ApiDef_Visibility

func (ApiDef_Visibility) EnumDescriptor deprecated 

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

Deprecated: Use ApiDef_Visibility.Descriptor instead.

func (ApiDef_Visibility) Number 

func (x ApiDef_Visibility) Number() protoreflect.EnumNumber

func (ApiDef_Visibility) String 

func (x ApiDef_Visibility) String() string

func (ApiDef_Visibility) Type 

func (ApiDef_Visibility) Type() protoreflect.EnumType

type ApiDefs 

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

func (*ApiDefs) Descriptor deprecated 

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

Deprecated: Use ApiDefs.ProtoReflect.Descriptor instead.

func (*ApiDefs) GetOp 

func (x *ApiDefs) GetOp() []*ApiDef

func (*ApiDefs) ProtoMessage 

func (*ApiDefs) ProtoMessage()

func (*ApiDefs) ProtoReflect 

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

func (*ApiDefs) Reset 

func (x *ApiDefs) Reset()

func (*ApiDefs) String 

func (x *ApiDefs) 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 AutoShardPolicy 

type AutoShardPolicy int32

Represents the type of auto-sharding we enable. 

const (
	// AUTO: Attempts FILE-based sharding, falling back to DATA-based sharding.
	AutoShardPolicy_AUTO AutoShardPolicy = 0
	// FILE: Shards by input files (i.e. each worker will get a set of files to
	// process). When this option is selected, make sure that there is at least as
	// many files as workers. If there are fewer input files than workers, a
	// runtime error will be raised.
	AutoShardPolicy_FILE AutoShardPolicy = 1
	// DATA: Shards by elements produced by the dataset. Each worker will process
	// the whole dataset and discard the portion that is not for itself. Note that
	// for this mode to correctly partitions the dataset elements, the dataset
	// needs to produce elements in a deterministic order.
	AutoShardPolicy_DATA AutoShardPolicy = 2
	// HINT: Looks for the presence of `shard(SHARD_HINT, ...)` which is treated
	// as a placeholder to replace with `shard(num_workers, worker_index)`.
	AutoShardPolicy_HINT AutoShardPolicy = 3
	// OFF: No sharding will be performed.
	AutoShardPolicy_OFF AutoShardPolicy = -1
)

func (AutoShardPolicy) Descriptor 

func (AutoShardPolicy) Descriptor() protoreflect.EnumDescriptor

func (AutoShardPolicy) Enum 

func (x AutoShardPolicy) Enum() *AutoShardPolicy

func (AutoShardPolicy) EnumDescriptor deprecated 

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

Deprecated: Use AutoShardPolicy.Descriptor instead.

func (AutoShardPolicy) Number 

func (x AutoShardPolicy) Number() protoreflect.EnumNumber

func (AutoShardPolicy) String 

func (x AutoShardPolicy) String() string

func (AutoShardPolicy) Type 

func (AutoShardPolicy) Type() protoreflect.EnumType

type AutotuneAlgorithm 

type AutotuneAlgorithm int32

Algorithm used for model autotuning optimization. 

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

func (AutotuneAlgorithm) Descriptor 

func (AutotuneAlgorithm) Descriptor() protoreflect.EnumDescriptor

func (AutotuneAlgorithm) Enum 

func (x AutotuneAlgorithm) Enum() *AutotuneAlgorithm

func (AutotuneAlgorithm) EnumDescriptor deprecated 

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

Deprecated: Use AutotuneAlgorithm.Descriptor instead.

func (AutotuneAlgorithm) Number 

func (x AutotuneAlgorithm) Number() protoreflect.EnumNumber

func (AutotuneAlgorithm) String 

func (x AutotuneAlgorithm) String() string

func (AutotuneAlgorithm) Type 

func (AutotuneAlgorithm) Type() protoreflect.EnumType

type AutotuneOptions 

type AutotuneOptions struct {

	// Whether to automatically tune performance knobs.
	//
	// Types that are assignable to OptionalEnabled:
	//
	//	*AutotuneOptions_Enabled
	OptionalEnabled isAutotuneOptions_OptionalEnabled `protobuf_oneof:"optional_enabled"`
	// When autotuning is enabled (through autotune), determines the CPU budget to
	// use. Values greater than the number of schedulable CPU cores are allowed
	// but may result in CPU contention.
	//
	// Types that are assignable to OptionalCpuBudget:
	//
	//	*AutotuneOptions_CpuBudget
	OptionalCpuBudget isAutotuneOptions_OptionalCpuBudget `protobuf_oneof:"optional_cpu_budget"`
	// When autotuning is enabled (through autotune), determines the RAM budget to
	// use. Values greater than the available RAM in bytes may result in OOM. If
	// 0, defaults to half of the available RAM in bytes.
	//
	// Types that are assignable to OptionalRamBudget:
	//
	//	*AutotuneOptions_RamBudget
	OptionalRamBudget isAutotuneOptions_OptionalRamBudget `protobuf_oneof:"optional_ram_budget"`
	// When autotuning is enabled (through autotune), determines the algorithm to
	// use. If not explicitly set by user, autotuning will follow HILL_CLIMB
	// algorithm but has more flexibility to tune parameters more aggressively,
	// in which case the behavior is implementation specific and may change over
	// time.
	//
	// Types that are assignable to OptionalAutotuneAlgorithm:
	//
	//	*AutotuneOptions_AutotuneAlgorithm
	OptionalAutotuneAlgorithm isAutotuneOptions_OptionalAutotuneAlgorithm `protobuf_oneof:"optional_autotune_algorithm"`
	// contains filtered or unexported fields
}

next: 5

func (*AutotuneOptions) Descriptor deprecated 

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

Deprecated: Use AutotuneOptions.ProtoReflect.Descriptor instead.

func (*AutotuneOptions) GetAutotuneAlgorithm 

func (x *AutotuneOptions) GetAutotuneAlgorithm() AutotuneAlgorithm

func (*AutotuneOptions) GetCpuBudget 

func (x *AutotuneOptions) GetCpuBudget() int32

func (*AutotuneOptions) GetEnabled 

func (x *AutotuneOptions) GetEnabled() bool

func (*AutotuneOptions) GetOptionalAutotuneAlgorithm 

func (m *AutotuneOptions) GetOptionalAutotuneAlgorithm() isAutotuneOptions_OptionalAutotuneAlgorithm

func (*AutotuneOptions) GetOptionalCpuBudget 

func (m *AutotuneOptions) GetOptionalCpuBudget() isAutotuneOptions_OptionalCpuBudget

func (*AutotuneOptions) GetOptionalEnabled 

func (m *AutotuneOptions) GetOptionalEnabled() isAutotuneOptions_OptionalEnabled

func (*AutotuneOptions) GetOptionalRamBudget 

func (m *AutotuneOptions) GetOptionalRamBudget() isAutotuneOptions_OptionalRamBudget

func (*AutotuneOptions) GetRamBudget 

func (x *AutotuneOptions) GetRamBudget() int64

func (*AutotuneOptions) ProtoMessage 

func (*AutotuneOptions) ProtoMessage()

func (*AutotuneOptions) ProtoReflect 

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

func (*AutotuneOptions) Reset 

func (x *AutotuneOptions) Reset()

func (*AutotuneOptions) String 

func (x *AutotuneOptions) String() string

type AutotuneOptions_AutotuneAlgorithm 

type AutotuneOptions_AutotuneAlgorithm struct {
	AutotuneAlgorithm AutotuneAlgorithm `` /* 128-byte string literal not displayed */
}

type AutotuneOptions_CpuBudget 

type AutotuneOptions_CpuBudget struct {
	CpuBudget int32 `protobuf:"varint,2,opt,name=cpu_budget,json=cpuBudget,proto3,oneof"`
}

type AutotuneOptions_Enabled 

type AutotuneOptions_Enabled struct {
	Enabled bool `protobuf:"varint,1,opt,name=enabled,proto3,oneof"`
}

type AutotuneOptions_RamBudget 

type AutotuneOptions_RamBudget struct {
	RamBudget int64 `protobuf:"varint,3,opt,name=ram_budget,json=ramBudget,proto3,oneof"`
}

type CardinalityOptions 

type CardinalityOptions struct {
	ComputeLevel CardinalityOptions_ComputeLevel `` /* 151-byte string literal not displayed */
	// contains filtered or unexported fields
}

next: 2

func (*CardinalityOptions) Descriptor deprecated 

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

Deprecated: Use CardinalityOptions.ProtoReflect.Descriptor instead.

func (*CardinalityOptions) GetComputeLevel 

func (x *CardinalityOptions) GetComputeLevel() CardinalityOptions_ComputeLevel

func (*CardinalityOptions) ProtoMessage 

func (*CardinalityOptions) ProtoMessage()

func (*CardinalityOptions) ProtoReflect 

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

func (*CardinalityOptions) Reset 

func (x *CardinalityOptions) Reset()

func (*CardinalityOptions) String 

func (x *CardinalityOptions) String() string

type CardinalityOptions_ComputeLevel 

type CardinalityOptions_ComputeLevel int32
const (
	CardinalityOptions_CARDINALITY_COMPUTE_UNSPECIFIED CardinalityOptions_ComputeLevel = 0
	// Cardinality will only be computed if it can be determined in a cheap
	// manner (ie. without reading from file sources). If the cardinality would
	// be nontrivial to compute, Cardinality() will return UNKNOWN_CARDINALITY.
	CardinalityOptions_CARDINALITY_COMPUTE_LOW CardinalityOptions_ComputeLevel = 1
	// Moderate effort will be made to determine cardinality, such as reading
	// index data from source files. If significant work is needed to compute
	// cardinality (e.g. reading entire source file contents or executing user
	// defined functions), Cardinality() will return UNKNOWN_CARDINALITY.
	CardinalityOptions_CARDINALITY_COMPUTE_MODERATE CardinalityOptions_ComputeLevel = 2
)

func (CardinalityOptions_ComputeLevel) Descriptor 

func (CardinalityOptions_ComputeLevel) Descriptor() protoreflect.EnumDescriptor

func (CardinalityOptions_ComputeLevel) Enum 

func (x CardinalityOptions_ComputeLevel) Enum() *CardinalityOptions_ComputeLevel

func (CardinalityOptions_ComputeLevel) EnumDescriptor deprecated 

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

Deprecated: Use CardinalityOptions_ComputeLevel.Descriptor instead.

func (CardinalityOptions_ComputeLevel) Number 

func (x CardinalityOptions_ComputeLevel) Number() protoreflect.EnumNumber

func (CardinalityOptions_ComputeLevel) String 

func (x CardinalityOptions_ComputeLevel) String() string

func (CardinalityOptions_ComputeLevel) Type 

func (CardinalityOptions_ComputeLevel) Type() protoreflect.EnumType

type CompressedComponentMetadata added in v2.12.0 

type CompressedComponentMetadata struct {

	// The dtype of the component tensor.
	Dtype DataType `protobuf:"varint,1,opt,name=dtype,proto3,enum=tensorflow.DataType" json:"dtype,omitempty"`
	// The shape of the component tensor.
	TensorShape *TensorShapeProto `protobuf:"bytes,2,opt,name=tensor_shape,json=tensorShape,proto3" json:"tensor_shape,omitempty"`
	// The amount of uncompressed tensor data.
	// - For string tensors, there is an element for each string indicating the
	// size of the string.
	// - For all other tensors, there is a single element indicating the size of
	// the tensor.
	UncompressedBytes []uint64 `protobuf:"varint,4,rep,packed,name=uncompressed_bytes,json=uncompressedBytes,proto3" json:"uncompressed_bytes,omitempty"`
	// contains filtered or unexported fields
}

Metadata describing a compressed component of a dataset element.

func (*CompressedComponentMetadata) Descriptor deprecated added in v2.12.0 

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

Deprecated: Use CompressedComponentMetadata.ProtoReflect.Descriptor instead.

func (*CompressedComponentMetadata) GetDtype added in v2.12.0 

func (x *CompressedComponentMetadata) GetDtype() DataType

func (*CompressedComponentMetadata) GetTensorShape added in v2.12.0 

func (x *CompressedComponentMetadata) GetTensorShape() *TensorShapeProto

func (*CompressedComponentMetadata) GetUncompressedBytes added in v2.12.0 

func (x *CompressedComponentMetadata) GetUncompressedBytes() []uint64

func (*CompressedComponentMetadata) ProtoMessage added in v2.12.0 

func (*CompressedComponentMetadata) ProtoMessage()

func (*CompressedComponentMetadata) ProtoReflect added in v2.12.0 

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

func (*CompressedComponentMetadata) Reset added in v2.12.0 

func (x *CompressedComponentMetadata) Reset()

func (*CompressedComponentMetadata) String added in v2.12.0 

func (x *CompressedComponentMetadata) String() string

type CompressedElement added in v2.12.0 

type CompressedElement struct {

	// Compressed tensor bytes for all components of the element.
	Data []byte `protobuf:"bytes,1,opt,name=data,proto3" json:"data,omitempty"`
	// Metadata for the components of the element.
	ComponentMetadata []*CompressedComponentMetadata `protobuf:"bytes,2,rep,name=component_metadata,json=componentMetadata,proto3" json:"component_metadata,omitempty"`
	// Version of the CompressedElement. CompressedElements may be stored on disk
	// and read back by later versions of code, so we store a version number to
	// help readers understand which version they are reading. When you add a new
	// field to this proto, you need to increment kCompressedElementVersion in
	// tensorflow/core/data/compression_utils.cc.
	Version int32 `protobuf:"varint,3,opt,name=version,proto3" json:"version,omitempty"`
	// contains filtered or unexported fields
}

func (*CompressedElement) Descriptor deprecated added in v2.12.0 

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

Deprecated: Use CompressedElement.ProtoReflect.Descriptor instead.

func (*CompressedElement) GetComponentMetadata added in v2.12.0 

func (x *CompressedElement) GetComponentMetadata() []*CompressedComponentMetadata

func (*CompressedElement) GetData added in v2.12.0 

func (x *CompressedElement) GetData() []byte

func (*CompressedElement) GetVersion added in v2.12.0 

func (x *CompressedElement) GetVersion() int32

func (*CompressedElement) ProtoMessage added in v2.12.0 

func (*CompressedElement) ProtoMessage()

func (*CompressedElement) ProtoReflect added in v2.12.0 

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

func (*CompressedElement) Reset added in v2.12.0 

func (x *CompressedElement) Reset()

func (*CompressedElement) String added in v2.12.0 

func (x *CompressedElement) String() string

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: Marked as deprecated in tensorflow/core/framework/cost_graph.proto.
	HostTempMemorySize int64 `protobuf:"varint,10,opt,name=host_temp_memory_size,json=hostTempMemorySize,proto3" json:"host_temp_memory_size,omitempty"`
	// Deprecated: Marked as deprecated in tensorflow/core/framework/cost_graph.proto.
	DeviceTempMemorySize int64 `` /* 127-byte string literal not displayed */
	// Deprecated: Marked as deprecated in tensorflow/core/framework/cost_graph.proto.
	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: Marked as deprecated in tensorflow/core/framework/cost_graph.proto.

func (*CostGraphDef_Node) GetDeviceTempMemorySize deprecated 

func (x *CostGraphDef_Node) GetDeviceTempMemorySize() int64

Deprecated: Marked as deprecated in tensorflow/core/framework/cost_graph.proto.

func (*CostGraphDef_Node) GetHostTempMemorySize deprecated 

func (x *CostGraphDef_Node) GetHostTempMemorySize() int64

Deprecated: Marked as deprecated in tensorflow/core/framework/cost_graph.proto.

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 DataClass 

type DataClass int32
const (
	// Unknown data class, used (implicitly) for legacy data. Will not be
	// processed by data ingestion pipelines.
	DataClass_DATA_CLASS_UNKNOWN DataClass = 0
	// Scalar time series. Each `Value` for the corresponding tag must have
	// `tensor` set to a rank-0 tensor of type `DT_FLOAT` (float32).
	DataClass_DATA_CLASS_SCALAR DataClass = 1
	// Tensor time series. Each `Value` for the corresponding tag must have
	// `tensor` set. The tensor value is arbitrary, but should be small to
	// accommodate direct storage in database backends: an upper bound of a few
	// kilobytes is a reasonable rule of thumb.
	DataClass_DATA_CLASS_TENSOR DataClass = 2
	// Blob sequence time series. Each `Value` for the corresponding tag must
	// have `tensor` set to a rank-1 tensor of bytestring dtype.
	DataClass_DATA_CLASS_BLOB_SEQUENCE DataClass = 3
)

func (DataClass) Descriptor 

func (DataClass) Descriptor() protoreflect.EnumDescriptor

func (DataClass) Enum 

func (x DataClass) Enum() *DataClass

func (DataClass) EnumDescriptor deprecated 

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

Deprecated: Use DataClass.Descriptor instead.

func (DataClass) Number 

func (x DataClass) Number() protoreflect.EnumNumber

func (DataClass) String 

func (x DataClass) String() string

func (DataClass) Type 

func (DataClass) Type() protoreflect.EnumType

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.
	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
	DataType_DT_FLOAT8_E5M2   DataType = 24 // 5 exponent bits, 2 mantissa bits.
	DataType_DT_FLOAT8_E4M3FN DataType = 25 // 4 exponent bits, 3 mantissa bits, finite-only, with
	// 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
	DataType_DT_FLOAT8_E5M2_REF   DataType = 124
	DataType_DT_FLOAT8_E4M3FN_REF DataType = 125
)

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 DeviceAttributes 

type DeviceAttributes struct {

	// Fully specified name of the device within a cluster.
	Name string `protobuf:"bytes,1,opt,name=name,proto3" json:"name,omitempty"`
	// String representation of device_type.
	DeviceType string `protobuf:"bytes,2,opt,name=device_type,json=deviceType,proto3" json:"device_type,omitempty"`
	// Memory capacity of device in bytes.
	MemoryLimit int64 `protobuf:"varint,4,opt,name=memory_limit,json=memoryLimit,proto3" json:"memory_limit,omitempty"`
	// Platform-specific data about device that may be useful
	// for supporting efficient data transfers.
	Locality *DeviceLocality `protobuf:"bytes,5,opt,name=locality,proto3" json:"locality,omitempty"`
	// A device is assigned a global unique number each time it is
	// initialized. "incarnation" should never be 0.
	Incarnation uint64 `protobuf:"fixed64,6,opt,name=incarnation,proto3" json:"incarnation,omitempty"`
	// String representation of the physical device that this device maps to.
	PhysicalDeviceDesc string `protobuf:"bytes,7,opt,name=physical_device_desc,json=physicalDeviceDesc,proto3" json:"physical_device_desc,omitempty"`
	// A physical device ID for use in XLA DeviceAssignments, unique across
	// clients in a multi-client setup. Set to -1 if unavailable, non-negative
	// otherwise.
	XlaGlobalId int64 `protobuf:"varint,8,opt,name=xla_global_id,json=xlaGlobalId,proto3" json:"xla_global_id,omitempty"`
	// contains filtered or unexported fields
}

func (*DeviceAttributes) Descriptor deprecated 

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

Deprecated: Use DeviceAttributes.ProtoReflect.Descriptor instead.

func (*DeviceAttributes) GetDeviceType 

func (x *DeviceAttributes) GetDeviceType() string

func (*DeviceAttributes) GetIncarnation 

func (x *DeviceAttributes) GetIncarnation() uint64

func (*DeviceAttributes) GetLocality 

func (x *DeviceAttributes) GetLocality() *DeviceLocality

func (*DeviceAttributes) GetMemoryLimit 

func (x *DeviceAttributes) GetMemoryLimit() int64

func (*DeviceAttributes) GetName 

func (x *DeviceAttributes) GetName() string

func (*DeviceAttributes) GetPhysicalDeviceDesc 

func (x *DeviceAttributes) GetPhysicalDeviceDesc() string

func (*DeviceAttributes) GetXlaGlobalId 

func (x *DeviceAttributes) GetXlaGlobalId() int64

func (*DeviceAttributes) ProtoMessage 

func (*DeviceAttributes) ProtoMessage()

func (*DeviceAttributes) ProtoReflect 

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

func (*DeviceAttributes) Reset 

func (x *DeviceAttributes) Reset()

func (*DeviceAttributes) String 

func (x *DeviceAttributes) String() string

type DeviceLocality 

type DeviceLocality struct {

	// Optional bus locality of device.  Default value of 0 means
	// no specific locality.  Specific localities are indexed from 1.
	BusId int32 `protobuf:"varint,1,opt,name=bus_id,json=busId,proto3" json:"bus_id,omitempty"`
	// Optional NUMA locality of device.
	NumaNode int32 `protobuf:"varint,2,opt,name=numa_node,json=numaNode,proto3" json:"numa_node,omitempty"`
	// Optional local interconnect links to other devices.
	Links *LocalLinks `protobuf:"bytes,3,opt,name=links,proto3" json:"links,omitempty"`
	// contains filtered or unexported fields
}

func (*DeviceLocality) Descriptor deprecated 

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

Deprecated: Use DeviceLocality.ProtoReflect.Descriptor instead.

func (*DeviceLocality) GetBusId 

func (x *DeviceLocality) GetBusId() int32
func (x *DeviceLocality) GetLinks() *LocalLinks

func (*DeviceLocality) GetNumaNode 

func (x *DeviceLocality) GetNumaNode() int32

func (*DeviceLocality) ProtoMessage 

func (*DeviceLocality) ProtoMessage()

func (*DeviceLocality) ProtoReflect 

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

func (*DeviceLocality) Reset 

func (x *DeviceLocality) Reset()

func (*DeviceLocality) String 

func (x *DeviceLocality) String() string

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 DistributeOptions 

type DistributeOptions struct {
	AutoShardPolicy AutoShardPolicy `` /* 146-byte string literal not displayed */
	// The number of devices attached to this input pipeline.
	//
	// Types that are assignable to OptionalNumDevices:
	//
	//	*DistributeOptions_NumDevices
	OptionalNumDevices isDistributeOptions_OptionalNumDevices `protobuf_oneof:"optional_num_devices"`
	// contains filtered or unexported fields
}

next: 3

func (*DistributeOptions) Descriptor deprecated 

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

Deprecated: Use DistributeOptions.ProtoReflect.Descriptor instead.

func (*DistributeOptions) GetAutoShardPolicy 

func (x *DistributeOptions) GetAutoShardPolicy() AutoShardPolicy

func (*DistributeOptions) GetNumDevices 

func (x *DistributeOptions) GetNumDevices() int32

func (*DistributeOptions) GetOptionalNumDevices 

func (m *DistributeOptions) GetOptionalNumDevices() isDistributeOptions_OptionalNumDevices

func (*DistributeOptions) ProtoMessage 

func (*DistributeOptions) ProtoMessage()

func (*DistributeOptions) ProtoReflect 

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

func (*DistributeOptions) Reset 

func (x *DistributeOptions) Reset()

func (*DistributeOptions) String 

func (x *DistributeOptions) String() string

type DistributeOptions_NumDevices 

type DistributeOptions_NumDevices struct {
	NumDevices int32 `protobuf:"varint,2,opt,name=num_devices,json=numDevices,proto3,oneof"`
}

type ExternalStatePolicy 

type ExternalStatePolicy int32

Represents how to handle external state during serialization. 

const (
	ExternalStatePolicy_POLICY_WARN   ExternalStatePolicy = 0
	ExternalStatePolicy_POLICY_IGNORE ExternalStatePolicy = 1
	ExternalStatePolicy_POLICY_FAIL   ExternalStatePolicy = 2
)

func (ExternalStatePolicy) Descriptor 

func (ExternalStatePolicy) Descriptor() protoreflect.EnumDescriptor

func (ExternalStatePolicy) Enum 

func (x ExternalStatePolicy) Enum() *ExternalStatePolicy

func (ExternalStatePolicy) EnumDescriptor deprecated 

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

Deprecated: Use ExternalStatePolicy.Descriptor instead.

func (ExternalStatePolicy) Number 

func (x ExternalStatePolicy) Number() protoreflect.EnumNumber

func (ExternalStatePolicy) String 

func (x ExternalStatePolicy) String() string

func (ExternalStatePolicy) Type 

func (ExternalStatePolicy) Type() protoreflect.EnumType

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 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
	//	*FullTypeDef_I
	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) GetI 

func (x *FullTypeDef) GetI() int64

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_I 

type FullTypeDef_I struct {
	I int64 `protobuf:"varint,4,opt,name=i,proto3,oneof"` // TODO(mdan): list/tensor, map? Need to reconcile with TFT_RECORD, etc.
}

type FullTypeDef_S 

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

type FullTypeId 

type FullTypeId int32

LINT.IfChange 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
	//	  independent 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
	// Represents a named field, with the name stored in the attribute.
	//
	// Parametrization:
	//
	//	TFT_NAMED[<type>]{<name>}
	//	* <type> is the type of the field
	//	* <name> is the field name, as string (thpugh can theoretically be an int
	//	  as well)
	//
	// Example:
	//
	//	TFT_RECORD[
	//	  TFT_NAMED[TFT_TENSOR[TFT_INT32]]{'foo'},
	//	  TFT_NAMED[TFT_TENSOR[TFT_FLOAT32]]{'bar'},
	//	]
	//	  is a structure with two fields, an int tensor "foo" and a float tensor
	//	  "bar".
	FullTypeId_TFT_NAMED FullTypeId = 4
	// Template definition. Expands the variables by repeating a template as
	// arguments of container.
	//
	// Parametrization:
	//
	//	TFT_FOR_EACH[<container_type>, <template>, <expansions>]
	//	* <container_type> is the type of the container that the template will be
	//	  expanded into
	//	* <template> is any type definition that potentially contains type
	//	  variables
	//	* <expansions> is a TFT_VAR and may include more types in the future
	//
	// Example:
	//
	//	TFT_FOR_EACH[
	//	      TFT_PRODUCT,
	//	      TFT_TENSOR[TFT_VAR["t"]],
	//	      TFT_VAR["t"]
	//	  ]
	//	  will substitute a T = TFT_INT32 to TFT_PRODUCT[TFT_TENSOR[TFT_INT32]]
	//	  and a T = (TFT_INT32, TFT_INT64) to
	//	  TFT_PRODUCT[TFT_TENSOR[TFT_INT32], TFT_TENSOR[TFT_INT64]].
	FullTypeId_TFT_FOR_EACH FullTypeId = 20
	// 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
	// Literal types describe compile-time constant values.
	// Literal types may also participate in dependent types.
	//
	// Parametrization:
	//
	//	TFT_LITERAL[<value type>]{<value>}
	//	* <value type> may be any concrete type compatible that can hold <value>
	//	* <value> is the type's attribute, and holds the actual literal value
	//
	// Examples:
	//
	//	TFT_LITERAL[TFT_INT32]{1} is the compile-time constant 1.
	FullTypeId_TFT_LITERAL FullTypeId = 1003
	// Encoding types describe a value of a certain type, encoded as a different
	// type.
	//
	// Parametrization:
	//
	//	TFT_ENCODED[<encoded type>, <encoding type>]
	//	* <encoded type> may be any type
	//	* <encoding type> may be any type
	//
	// Examples:
	//
	//	TFT_ENCODING[TFT_INT32, TFT_STRING] is an integer encoded as string.
	FullTypeId_TFT_ENCODED FullTypeId = 1004
	// The type of "shape tensors" where the runtime value is the shape of
	// some tensor(s), i.e. the output of tf.shape.
	// Shape tensors have special, host-only placement, in contrast to
	// TFT_TENSOR[TFT_INT32] which is the type of a normal numeric tensor
	// with no special placement.
	//
	// Examples:
	//
	//	TFT_SHAPE_TENSOR[TFT_INT32] is the most common
	//	TFT_SHAPE_TENSOR[TFT_INT64] is also allowed
	FullTypeId_TFT_SHAPE_TENSOR FullTypeId = 1005
	// 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
	// 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_DATASET[<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
	// A ragged tensor created by tf.ragged ops and APIs.
	//
	// Parametrization: TFT_RAGGED[<element_type>].
	FullTypeId_TFT_RAGGED FullTypeId = 10103
	// Iterators created by tf.data ops and APIs. Very similar to Datasets, except
	// they are mutable.
	//
	// Parametrization: TFT_ITERATOR[<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.
	FullTypeId_TFT_ITERATOR FullTypeId = 10104
	// A mutex lock tensor, produced by tf.raw_ops.MutexLock.
	// Unlike strict execution models, where ownership of a lock is denoted by
	// "running after the lock has been acquired", in non-strict mode, lock
	// ownership is in the true sense: "the op argument representing the lock is
	// available".
	// Mutex locks are the dynamic counterpart of control dependencies.
	// TODO(mdan): Properly document this thing.
	//
	// Parametrization: TFT_MUTEX_LOCK[].
	FullTypeId_TFT_MUTEX_LOCK FullTypeId = 10202
	// The equivalent of a Tensor with DT_VARIANT dtype, kept here to simplify
	// translation. This type should not normally appear after type inference.
	// Note that LEGACY_VARIANT != ANY: TENSOR[INT32] is a subtype of ANY, but is
	// not a subtype of LEGACY_VARIANT.
	FullTypeId_TFT_LEGACY_VARIANT FullTypeId = 10203
)

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: Marked as deprecated in tensorflow/core/framework/graph.proto.
	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: Marked as deprecated in tensorflow/core/framework/graph.proto.

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 GraphTransferConstNodeInfo 

type GraphTransferConstNodeInfo struct {
	Name   string   `protobuf:"bytes,1,opt,name=name,proto3" json:"name,omitempty"`
	NodeId int32    `protobuf:"varint,2,opt,name=node_id,json=nodeId,proto3" json:"node_id,omitempty"`
	Shape  []int64  `protobuf:"varint,3,rep,packed,name=shape,proto3" json:"shape,omitempty"`
	Data   []byte   `protobuf:"bytes,4,opt,name=data,proto3" json:"data,omitempty"`
	Dtype  DataType `protobuf:"varint,5,opt,name=dtype,proto3,enum=tensorflow.DataType" json:"dtype,omitempty"`
	// contains filtered or unexported fields
}

func (*GraphTransferConstNodeInfo) Descriptor deprecated 

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

Deprecated: Use GraphTransferConstNodeInfo.ProtoReflect.Descriptor instead.

func (*GraphTransferConstNodeInfo) GetData 

func (x *GraphTransferConstNodeInfo) GetData() []byte

func (*GraphTransferConstNodeInfo) GetDtype 

func (x *GraphTransferConstNodeInfo) GetDtype() DataType

func (*GraphTransferConstNodeInfo) GetName 

func (x *GraphTransferConstNodeInfo) GetName() string

func (*GraphTransferConstNodeInfo) GetNodeId 

func (x *GraphTransferConstNodeInfo) GetNodeId() int32

func (*GraphTransferConstNodeInfo) GetShape 

func (x *GraphTransferConstNodeInfo) GetShape() []int64

func (*GraphTransferConstNodeInfo) ProtoMessage 

func (*GraphTransferConstNodeInfo) ProtoMessage()

func (*GraphTransferConstNodeInfo) ProtoReflect 

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

func (*GraphTransferConstNodeInfo) Reset 

func (x *GraphTransferConstNodeInfo) Reset()

func (*GraphTransferConstNodeInfo) String 

func (x *GraphTransferConstNodeInfo) String() string

type GraphTransferGraphInputNodeInfo 

type GraphTransferGraphInputNodeInfo struct {
	Name  string   `protobuf:"bytes,1,opt,name=name,proto3" json:"name,omitempty"`
	Shape []int64  `protobuf:"varint,2,rep,packed,name=shape,proto3" json:"shape,omitempty"`
	Dtype DataType `protobuf:"varint,3,opt,name=dtype,proto3,enum=tensorflow.DataType" json:"dtype,omitempty"`
	// contains filtered or unexported fields
}

func (*GraphTransferGraphInputNodeInfo) Descriptor deprecated 

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

Deprecated: Use GraphTransferGraphInputNodeInfo.ProtoReflect.Descriptor instead.

func (*GraphTransferGraphInputNodeInfo) GetDtype 

func (x *GraphTransferGraphInputNodeInfo) GetDtype() DataType

func (*GraphTransferGraphInputNodeInfo) GetName 

func (x *GraphTransferGraphInputNodeInfo) GetName() string

func (*GraphTransferGraphInputNodeInfo) GetShape 

func (x *GraphTransferGraphInputNodeInfo) GetShape() []int64

func (*GraphTransferGraphInputNodeInfo) ProtoMessage 

func (*GraphTransferGraphInputNodeInfo) ProtoMessage()

func (*GraphTransferGraphInputNodeInfo) ProtoReflect 

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

func (*GraphTransferGraphInputNodeInfo) Reset 

func (x *GraphTransferGraphInputNodeInfo) Reset()

func (*GraphTransferGraphInputNodeInfo) String 

func (x *GraphTransferGraphInputNodeInfo) String() string

type GraphTransferGraphOutputNodeInfo 

type GraphTransferGraphOutputNodeInfo struct {
	Name  string   `protobuf:"bytes,1,opt,name=name,proto3" json:"name,omitempty"`
	Shape []int64  `protobuf:"varint,2,rep,packed,name=shape,proto3" json:"shape,omitempty"`
	Dtype DataType `protobuf:"varint,3,opt,name=dtype,proto3,enum=tensorflow.DataType" json:"dtype,omitempty"`
	// contains filtered or unexported fields
}

func (*GraphTransferGraphOutputNodeInfo) Descriptor deprecated 

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

Deprecated: Use GraphTransferGraphOutputNodeInfo.ProtoReflect.Descriptor instead.

func (*GraphTransferGraphOutputNodeInfo) GetDtype 

func (x *GraphTransferGraphOutputNodeInfo) GetDtype() DataType

func (*GraphTransferGraphOutputNodeInfo) GetName 

func (x *GraphTransferGraphOutputNodeInfo) GetName() string

func (*GraphTransferGraphOutputNodeInfo) GetShape 

func (x *GraphTransferGraphOutputNodeInfo) GetShape() []int64

func (*GraphTransferGraphOutputNodeInfo) ProtoMessage 

func (*GraphTransferGraphOutputNodeInfo) ProtoMessage()

func (*GraphTransferGraphOutputNodeInfo) ProtoReflect 

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

func (*GraphTransferGraphOutputNodeInfo) Reset 

func (x *GraphTransferGraphOutputNodeInfo) Reset()

func (*GraphTransferGraphOutputNodeInfo) String 

func (x *GraphTransferGraphOutputNodeInfo) String() string

type GraphTransferInfo 

type GraphTransferInfo struct {
	NodeInfo       []*GraphTransferNodeInfo       `protobuf:"bytes,1,rep,name=node_info,json=nodeInfo,proto3" json:"node_info,omitempty"`
	ConstNodeInfo  []*GraphTransferConstNodeInfo  `protobuf:"bytes,2,rep,name=const_node_info,json=constNodeInfo,proto3" json:"const_node_info,omitempty"`
	NodeInputInfo  []*GraphTransferNodeInputInfo  `protobuf:"bytes,3,rep,name=node_input_info,json=nodeInputInfo,proto3" json:"node_input_info,omitempty"`
	NodeOutputInfo []*GraphTransferNodeOutputInfo `protobuf:"bytes,4,rep,name=node_output_info,json=nodeOutputInfo,proto3" json:"node_output_info,omitempty"`
	// Input Node parameters of transferred graph
	GraphInputNodeInfo  []*GraphTransferGraphInputNodeInfo  `protobuf:"bytes,5,rep,name=graph_input_node_info,json=graphInputNodeInfo,proto3" json:"graph_input_node_info,omitempty"`
	GraphOutputNodeInfo []*GraphTransferGraphOutputNodeInfo `protobuf:"bytes,6,rep,name=graph_output_node_info,json=graphOutputNodeInfo,proto3" json:"graph_output_node_info,omitempty"`
	// Destination of graph transfer
	Destination GraphTransferInfo_Destination `protobuf:"varint,7,opt,name=destination,proto3,enum=tensorflow.GraphTransferInfo_Destination" json:"destination,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 (*GraphTransferInfo) Descriptor deprecated 

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

Deprecated: Use GraphTransferInfo.ProtoReflect.Descriptor instead.

func (*GraphTransferInfo) GetConstNodeInfo 

func (x *GraphTransferInfo) GetConstNodeInfo() []*GraphTransferConstNodeInfo

func (*GraphTransferInfo) GetDestination 

func (x *GraphTransferInfo) GetDestination() GraphTransferInfo_Destination

func (*GraphTransferInfo) GetGraphInputNodeInfo 

func (x *GraphTransferInfo) GetGraphInputNodeInfo() []*GraphTransferGraphInputNodeInfo

func (*GraphTransferInfo) GetGraphOutputNodeInfo 

func (x *GraphTransferInfo) GetGraphOutputNodeInfo() []*GraphTransferGraphOutputNodeInfo

func (*GraphTransferInfo) GetNodeInfo 

func (x *GraphTransferInfo) GetNodeInfo() []*GraphTransferNodeInfo

func (*GraphTransferInfo) GetNodeInputInfo 

func (x *GraphTransferInfo) GetNodeInputInfo() []*GraphTransferNodeInputInfo

func (*GraphTransferInfo) GetNodeOutputInfo 

func (x *GraphTransferInfo) GetNodeOutputInfo() []*GraphTransferNodeOutputInfo

func (*GraphTransferInfo) ProtoMessage 

func (*GraphTransferInfo) ProtoMessage()

func (*GraphTransferInfo) ProtoReflect 

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

func (*GraphTransferInfo) Reset 

func (x *GraphTransferInfo) Reset()

func (*GraphTransferInfo) String 

func (x *GraphTransferInfo) String() string

type GraphTransferInfo_Destination 

type GraphTransferInfo_Destination int32
const (
	GraphTransferInfo_NOP     GraphTransferInfo_Destination = 0
	GraphTransferInfo_HEXAGON GraphTransferInfo_Destination = 1
)

func (GraphTransferInfo_Destination) Descriptor 

func (GraphTransferInfo_Destination) Descriptor() protoreflect.EnumDescriptor

func (GraphTransferInfo_Destination) Enum 

func (x GraphTransferInfo_Destination) Enum() *GraphTransferInfo_Destination

func (GraphTransferInfo_Destination) EnumDescriptor deprecated 

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

Deprecated: Use GraphTransferInfo_Destination.Descriptor instead.

func (GraphTransferInfo_Destination) Number 

func (x GraphTransferInfo_Destination) Number() protoreflect.EnumNumber

func (GraphTransferInfo_Destination) String 

func (x GraphTransferInfo_Destination) String() string

func (GraphTransferInfo_Destination) Type 

func (GraphTransferInfo_Destination) Type() protoreflect.EnumType

type GraphTransferNodeInfo 

type GraphTransferNodeInfo struct {
	Name        string `protobuf:"bytes,1,opt,name=name,proto3" json:"name,omitempty"`
	NodeId      int32  `protobuf:"varint,2,opt,name=node_id,json=nodeId,proto3" json:"node_id,omitempty"`
	TypeName    string `protobuf:"bytes,3,opt,name=type_name,json=typeName,proto3" json:"type_name,omitempty"`
	SocOpId     int32  `protobuf:"varint,4,opt,name=soc_op_id,json=socOpId,proto3" json:"soc_op_id,omitempty"`
	PaddingId   int32  `protobuf:"varint,5,opt,name=padding_id,json=paddingId,proto3" json:"padding_id,omitempty"`
	InputCount  int32  `protobuf:"varint,6,opt,name=input_count,json=inputCount,proto3" json:"input_count,omitempty"`
	OutputCount int32  `protobuf:"varint,7,opt,name=output_count,json=outputCount,proto3" json:"output_count,omitempty"`
	// contains filtered or unexported fields
}

func (*GraphTransferNodeInfo) Descriptor deprecated 

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

Deprecated: Use GraphTransferNodeInfo.ProtoReflect.Descriptor instead.

func (*GraphTransferNodeInfo) GetInputCount 

func (x *GraphTransferNodeInfo) GetInputCount() int32

func (*GraphTransferNodeInfo) GetName 

func (x *GraphTransferNodeInfo) GetName() string

func (*GraphTransferNodeInfo) GetNodeId 

func (x *GraphTransferNodeInfo) GetNodeId() int32

func (*GraphTransferNodeInfo) GetOutputCount 

func (x *GraphTransferNodeInfo) GetOutputCount() int32

func (*GraphTransferNodeInfo) GetPaddingId 

func (x *GraphTransferNodeInfo) GetPaddingId() int32

func (*GraphTransferNodeInfo) GetSocOpId 

func (x *GraphTransferNodeInfo) GetSocOpId() int32

func (*GraphTransferNodeInfo) GetTypeName 

func (x *GraphTransferNodeInfo) GetTypeName() string

func (*GraphTransferNodeInfo) ProtoMessage 

func (*GraphTransferNodeInfo) ProtoMessage()

func (*GraphTransferNodeInfo) ProtoReflect 

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

func (*GraphTransferNodeInfo) Reset 

func (x *GraphTransferNodeInfo) Reset()

func (*GraphTransferNodeInfo) String 

func (x *GraphTransferNodeInfo) String() string

type GraphTransferNodeInput 

type GraphTransferNodeInput struct {
	NodeId     int32 `protobuf:"varint,1,opt,name=node_id,json=nodeId,proto3" json:"node_id,omitempty"`
	OutputPort int32 `protobuf:"varint,2,opt,name=output_port,json=outputPort,proto3" json:"output_port,omitempty"`
	// contains filtered or unexported fields
}

func (*GraphTransferNodeInput) Descriptor deprecated 

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

Deprecated: Use GraphTransferNodeInput.ProtoReflect.Descriptor instead.

func (*GraphTransferNodeInput) GetNodeId 

func (x *GraphTransferNodeInput) GetNodeId() int32

func (*GraphTransferNodeInput) GetOutputPort 

func (x *GraphTransferNodeInput) GetOutputPort() int32

func (*GraphTransferNodeInput) ProtoMessage 

func (*GraphTransferNodeInput) ProtoMessage()

func (*GraphTransferNodeInput) ProtoReflect 

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

func (*GraphTransferNodeInput) Reset 

func (x *GraphTransferNodeInput) Reset()

func (*GraphTransferNodeInput) String 

func (x *GraphTransferNodeInput) String() string

type GraphTransferNodeInputInfo 

type GraphTransferNodeInputInfo struct {
	NodeId    int32                     `protobuf:"varint,1,opt,name=node_id,json=nodeId,proto3" json:"node_id,omitempty"`
	NodeInput []*GraphTransferNodeInput `protobuf:"bytes,2,rep,name=node_input,json=nodeInput,proto3" json:"node_input,omitempty"`
	// contains filtered or unexported fields
}

func (*GraphTransferNodeInputInfo) Descriptor deprecated 

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

Deprecated: Use GraphTransferNodeInputInfo.ProtoReflect.Descriptor instead.

func (*GraphTransferNodeInputInfo) GetNodeId 

func (x *GraphTransferNodeInputInfo) GetNodeId() int32

func (*GraphTransferNodeInputInfo) GetNodeInput 

func (x *GraphTransferNodeInputInfo) GetNodeInput() []*GraphTransferNodeInput

func (*GraphTransferNodeInputInfo) ProtoMessage 

func (*GraphTransferNodeInputInfo) ProtoMessage()

func (*GraphTransferNodeInputInfo) ProtoReflect 

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

func (*GraphTransferNodeInputInfo) Reset 

func (x *GraphTransferNodeInputInfo) Reset()

func (*GraphTransferNodeInputInfo) String 

func (x *GraphTransferNodeInputInfo) String() string

type GraphTransferNodeOutputInfo 

type GraphTransferNodeOutputInfo struct {
	NodeId      int32   `protobuf:"varint,1,opt,name=node_id,json=nodeId,proto3" json:"node_id,omitempty"`
	MaxByteSize []int32 `protobuf:"varint,2,rep,packed,name=max_byte_size,json=maxByteSize,proto3" json:"max_byte_size,omitempty"`
	// contains filtered or unexported fields
}

func (*GraphTransferNodeOutputInfo) Descriptor deprecated 

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

Deprecated: Use GraphTransferNodeOutputInfo.ProtoReflect.Descriptor instead.

func (*GraphTransferNodeOutputInfo) GetMaxByteSize 

func (x *GraphTransferNodeOutputInfo) GetMaxByteSize() []int32

func (*GraphTransferNodeOutputInfo) GetNodeId 

func (x *GraphTransferNodeOutputInfo) GetNodeId() int32

func (*GraphTransferNodeOutputInfo) ProtoMessage 

func (*GraphTransferNodeOutputInfo) ProtoMessage()

func (*GraphTransferNodeOutputInfo) ProtoReflect 

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

func (*GraphTransferNodeOutputInfo) Reset 

func (x *GraphTransferNodeOutputInfo) Reset()

func (*GraphTransferNodeOutputInfo) String 

func (x *GraphTransferNodeOutputInfo) String() string

type HistogramProto 

type HistogramProto = protobuf.HistogramProto
type InterconnectLink struct {
	DeviceId int32  `protobuf:"varint,1,opt,name=device_id,json=deviceId,proto3" json:"device_id,omitempty"`
	Type     string `protobuf:"bytes,2,opt,name=type,proto3" json:"type,omitempty"`
	Strength int32  `protobuf:"varint,3,opt,name=strength,proto3" json:"strength,omitempty"`
	// contains filtered or unexported fields
}

func (*InterconnectLink) Descriptor deprecated 

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

Deprecated: Use InterconnectLink.ProtoReflect.Descriptor instead.

func (*InterconnectLink) GetDeviceId 

func (x *InterconnectLink) GetDeviceId() int32

func (*InterconnectLink) GetStrength 

func (x *InterconnectLink) GetStrength() int32

func (*InterconnectLink) GetType 

func (x *InterconnectLink) GetType() string

func (*InterconnectLink) ProtoMessage 

func (*InterconnectLink) ProtoMessage()

func (*InterconnectLink) ProtoReflect 

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

func (*InterconnectLink) Reset 

func (x *InterconnectLink) Reset()

func (*InterconnectLink) String 

func (x *InterconnectLink) String() string

type KernelDef 

type KernelDef struct {

	// Must match the name of an Op.
	Op string `protobuf:"bytes,1,opt,name=op,proto3" json:"op,omitempty"`
	// Type of device this kernel runs on.
	DeviceType string                      `protobuf:"bytes,2,opt,name=device_type,json=deviceType,proto3" json:"device_type,omitempty"`
	Constraint []*KernelDef_AttrConstraint `protobuf:"bytes,3,rep,name=constraint,proto3" json:"constraint,omitempty"`
	// Names of the Op's input_/output_args that reside in host memory
	// instead of device memory.
	HostMemoryArg []string `protobuf:"bytes,4,rep,name=host_memory_arg,json=hostMemoryArg,proto3" json:"host_memory_arg,omitempty"`
	// This allows experimental kernels to be registered for an op that
	// won't be used unless the user specifies a "_kernel" attr with
	// value matching this.
	Label string `protobuf:"bytes,5,opt,name=label,proto3" json:"label,omitempty"`
	// Prioritization of kernel amongst different devices. By default we assume
	// priority is 0. The higher the priority the better. By default (i.e. if
	// this is not set), we prefer GPU kernels over CPU.
	Priority int32 `protobuf:"varint,6,opt,name=priority,proto3" json:"priority,omitempty"`
	// contains filtered or unexported fields
}

func (*KernelDef) Descriptor deprecated 

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

Deprecated: Use KernelDef.ProtoReflect.Descriptor instead.

func (*KernelDef) GetConstraint 

func (x *KernelDef) GetConstraint() []*KernelDef_AttrConstraint

func (*KernelDef) GetDeviceType 

func (x *KernelDef) GetDeviceType() string

func (*KernelDef) GetHostMemoryArg 

func (x *KernelDef) GetHostMemoryArg() []string

func (*KernelDef) GetLabel 

func (x *KernelDef) GetLabel() string

func (*KernelDef) GetOp 

func (x *KernelDef) GetOp() string

func (*KernelDef) GetPriority 

func (x *KernelDef) GetPriority() int32

func (*KernelDef) ProtoMessage 

func (*KernelDef) ProtoMessage()

func (*KernelDef) ProtoReflect 

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

func (*KernelDef) Reset 

func (x *KernelDef) Reset()

func (*KernelDef) String 

func (x *KernelDef) String() string

type KernelDef_AttrConstraint 

type KernelDef_AttrConstraint struct {

	// Name of an attr from the Op.
	Name string `protobuf:"bytes,1,opt,name=name,proto3" json:"name,omitempty"`
	// A list of values that this kernel supports for this attr.
	// Like OpDef.AttrDef.allowed_values, except for kernels instead of Ops.
	AllowedValues *AttrValue `protobuf:"bytes,2,opt,name=allowed_values,json=allowedValues,proto3" json:"allowed_values,omitempty"`
	// contains filtered or unexported fields
}

func (*KernelDef_AttrConstraint) Descriptor deprecated 

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

Deprecated: Use KernelDef_AttrConstraint.ProtoReflect.Descriptor instead.

func (*KernelDef_AttrConstraint) GetAllowedValues 

func (x *KernelDef_AttrConstraint) GetAllowedValues() *AttrValue

func (*KernelDef_AttrConstraint) GetName 

func (x *KernelDef_AttrConstraint) GetName() string

func (*KernelDef_AttrConstraint) ProtoMessage 

func (*KernelDef_AttrConstraint) ProtoMessage()

func (*KernelDef_AttrConstraint) ProtoReflect 

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

func (*KernelDef_AttrConstraint) Reset 

func (x *KernelDef_AttrConstraint) Reset()

func (*KernelDef_AttrConstraint) String 

func (x *KernelDef_AttrConstraint) String() string

type KernelList 

type KernelList struct {
	Kernel []*KernelDef `protobuf:"bytes,1,rep,name=kernel,proto3" json:"kernel,omitempty"`
	// contains filtered or unexported fields
}

A collection of KernelDefs

func (*KernelList) Descriptor deprecated 

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

Deprecated: Use KernelList.ProtoReflect.Descriptor instead.

func (*KernelList) GetKernel 

func (x *KernelList) GetKernel() []*KernelDef

func (*KernelList) ProtoMessage 

func (*KernelList) ProtoMessage()

func (*KernelList) ProtoReflect 

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

func (*KernelList) Reset 

func (x *KernelList) Reset()

func (*KernelList) String 

func (x *KernelList) String() string
type LocalLinks struct {
	Link []*InterconnectLink `protobuf:"bytes,1,rep,name=link,proto3" json:"link,omitempty"`
	// contains filtered or unexported fields
}

func (*LocalLinks) Descriptor deprecated 

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

Deprecated: Use LocalLinks.ProtoReflect.Descriptor instead. 

func (x *LocalLinks) GetLink() []*InterconnectLink

func (*LocalLinks) ProtoMessage 

func (*LocalLinks) ProtoMessage()

func (*LocalLinks) ProtoReflect 

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

func (*LocalLinks) Reset 

func (x *LocalLinks) Reset()

func (*LocalLinks) String 

func (x *LocalLinks) String() string

type MemoryLogRawAllocation 

type MemoryLogRawAllocation struct {

	// Process-unique step id.
	StepId int64 `protobuf:"varint,1,opt,name=step_id,json=stepId,proto3" json:"step_id,omitempty"`
	// Name of the operation making the allocation.
	Operation string `protobuf:"bytes,2,opt,name=operation,proto3" json:"operation,omitempty"`
	// Number of bytes in the allocation.
	NumBytes int64 `protobuf:"varint,3,opt,name=num_bytes,json=numBytes,proto3" json:"num_bytes,omitempty"`
	// Address of the allocation.
	Ptr uint64 `protobuf:"varint,4,opt,name=ptr,proto3" json:"ptr,omitempty"`
	// Id of the tensor buffer being allocated, used to match to a
	// corresponding deallocation.
	AllocationId int64 `protobuf:"varint,5,opt,name=allocation_id,json=allocationId,proto3" json:"allocation_id,omitempty"`
	// Name of the allocator used.
	AllocatorName string `protobuf:"bytes,6,opt,name=allocator_name,json=allocatorName,proto3" json:"allocator_name,omitempty"`
	// contains filtered or unexported fields
}

func (*MemoryLogRawAllocation) Descriptor deprecated 

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

Deprecated: Use MemoryLogRawAllocation.ProtoReflect.Descriptor instead.

func (*MemoryLogRawAllocation) GetAllocationId 

func (x *MemoryLogRawAllocation) GetAllocationId() int64

func (*MemoryLogRawAllocation) GetAllocatorName 

func (x *MemoryLogRawAllocation) GetAllocatorName() string

func (*MemoryLogRawAllocation) GetNumBytes 

func (x *MemoryLogRawAllocation) GetNumBytes() int64

func (*MemoryLogRawAllocation) GetOperation 

func (x *MemoryLogRawAllocation) GetOperation() string

func (*MemoryLogRawAllocation) GetPtr 

func (x *MemoryLogRawAllocation) GetPtr() uint64

func (*MemoryLogRawAllocation) GetStepId 

func (x *MemoryLogRawAllocation) GetStepId() int64

func (*MemoryLogRawAllocation) ProtoMessage 

func (*MemoryLogRawAllocation) ProtoMessage()

func (*MemoryLogRawAllocation) ProtoReflect 

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

func (*MemoryLogRawAllocation) Reset 

func (x *MemoryLogRawAllocation) Reset()

func (*MemoryLogRawAllocation) String 

func (x *MemoryLogRawAllocation) String() string

type MemoryLogRawDeallocation 

type MemoryLogRawDeallocation struct {

	// Process-unique step id.
	StepId int64 `protobuf:"varint,1,opt,name=step_id,json=stepId,proto3" json:"step_id,omitempty"`
	// Name of the operation making the deallocation.
	Operation string `protobuf:"bytes,2,opt,name=operation,proto3" json:"operation,omitempty"`
	// Id of the tensor buffer being deallocated, used to match to a
	// corresponding allocation.
	AllocationId int64 `protobuf:"varint,3,opt,name=allocation_id,json=allocationId,proto3" json:"allocation_id,omitempty"`
	// Name of the allocator used.
	AllocatorName string `protobuf:"bytes,4,opt,name=allocator_name,json=allocatorName,proto3" json:"allocator_name,omitempty"`
	// True if the deallocation is queued and will be performed later,
	// e.g. for GPU lazy freeing of buffers.
	Deferred bool `protobuf:"varint,5,opt,name=deferred,proto3" json:"deferred,omitempty"`
	// contains filtered or unexported fields
}

func (*MemoryLogRawDeallocation) Descriptor deprecated 

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

Deprecated: Use MemoryLogRawDeallocation.ProtoReflect.Descriptor instead.

func (*MemoryLogRawDeallocation) GetAllocationId 

func (x *MemoryLogRawDeallocation) GetAllocationId() int64

func (*MemoryLogRawDeallocation) GetAllocatorName 

func (x *MemoryLogRawDeallocation) GetAllocatorName() string

func (*MemoryLogRawDeallocation) GetDeferred 

func (x *MemoryLogRawDeallocation) GetDeferred() bool

func (*MemoryLogRawDeallocation) GetOperation 

func (x *MemoryLogRawDeallocation) GetOperation() string

func (*MemoryLogRawDeallocation) GetStepId 

func (x *MemoryLogRawDeallocation) GetStepId() int64

func (*MemoryLogRawDeallocation) ProtoMessage 

func (*MemoryLogRawDeallocation) ProtoMessage()

func (*MemoryLogRawDeallocation) ProtoReflect 

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

func (*MemoryLogRawDeallocation) Reset 

func (x *MemoryLogRawDeallocation) Reset()

func (*MemoryLogRawDeallocation) String 

func (x *MemoryLogRawDeallocation) String() string

type MemoryLogStep 

type MemoryLogStep struct {

	// Process-unique step id.
	StepId int64 `protobuf:"varint,1,opt,name=step_id,json=stepId,proto3" json:"step_id,omitempty"`
	// Handle describing the feeds and fetches of the step.
	Handle string `protobuf:"bytes,2,opt,name=handle,proto3" json:"handle,omitempty"`
	// contains filtered or unexported fields
}

func (*MemoryLogStep) Descriptor deprecated 

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

Deprecated: Use MemoryLogStep.ProtoReflect.Descriptor instead.

func (*MemoryLogStep) GetHandle 

func (x *MemoryLogStep) GetHandle() string

func (*MemoryLogStep) GetStepId 

func (x *MemoryLogStep) GetStepId() int64

func (*MemoryLogStep) ProtoMessage 

func (*MemoryLogStep) ProtoMessage()

func (*MemoryLogStep) ProtoReflect 

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

func (*MemoryLogStep) Reset 

func (x *MemoryLogStep) Reset()

func (*MemoryLogStep) String 

func (x *MemoryLogStep) String() string

type MemoryLogTensorAllocation 

type MemoryLogTensorAllocation struct {

	// Process-unique step id.
	StepId int64 `protobuf:"varint,1,opt,name=step_id,json=stepId,proto3" json:"step_id,omitempty"`
	// Name of the kernel making the allocation as set in GraphDef,
	// e.g., "affine2/weights/Assign".
	KernelName string `protobuf:"bytes,2,opt,name=kernel_name,json=kernelName,proto3" json:"kernel_name,omitempty"`
	// Allocated tensor details.
	Tensor *TensorDescription `protobuf:"bytes,3,opt,name=tensor,proto3" json:"tensor,omitempty"`
	// contains filtered or unexported fields
}

func (*MemoryLogTensorAllocation) Descriptor deprecated 

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

Deprecated: Use MemoryLogTensorAllocation.ProtoReflect.Descriptor instead.

func (*MemoryLogTensorAllocation) GetKernelName 

func (x *MemoryLogTensorAllocation) GetKernelName() string

func (*MemoryLogTensorAllocation) GetStepId 

func (x *MemoryLogTensorAllocation) GetStepId() int64

func (*MemoryLogTensorAllocation) GetTensor 

func (x *MemoryLogTensorAllocation) GetTensor() *TensorDescription

func (*MemoryLogTensorAllocation) ProtoMessage 

func (*MemoryLogTensorAllocation) ProtoMessage()

func (*MemoryLogTensorAllocation) ProtoReflect 

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

func (*MemoryLogTensorAllocation) Reset 

func (x *MemoryLogTensorAllocation) Reset()

func (*MemoryLogTensorAllocation) String 

func (x *MemoryLogTensorAllocation) String() string

type MemoryLogTensorDeallocation 

type MemoryLogTensorDeallocation struct {

	// Id of the tensor buffer being deallocated, used to match to a
	// corresponding allocation.
	AllocationId int64 `protobuf:"varint,1,opt,name=allocation_id,json=allocationId,proto3" json:"allocation_id,omitempty"`
	// Name of the allocator used.
	AllocatorName string `protobuf:"bytes,2,opt,name=allocator_name,json=allocatorName,proto3" json:"allocator_name,omitempty"`
	// contains filtered or unexported fields
}

func (*MemoryLogTensorDeallocation) Descriptor deprecated 

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

Deprecated: Use MemoryLogTensorDeallocation.ProtoReflect.Descriptor instead.

func (*MemoryLogTensorDeallocation) GetAllocationId 

func (x *MemoryLogTensorDeallocation) GetAllocationId() int64

func (*MemoryLogTensorDeallocation) GetAllocatorName 

func (x *MemoryLogTensorDeallocation) GetAllocatorName() string

func (*MemoryLogTensorDeallocation) ProtoMessage 

func (*MemoryLogTensorDeallocation) ProtoMessage()

func (*MemoryLogTensorDeallocation) ProtoReflect 

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

func (*MemoryLogTensorDeallocation) Reset 

func (x *MemoryLogTensorDeallocation) Reset()

func (*MemoryLogTensorDeallocation) String 

func (x *MemoryLogTensorDeallocation) String() string

type MemoryLogTensorOutput 

type MemoryLogTensorOutput struct {

	// Process-unique step id.
	StepId int64 `protobuf:"varint,1,opt,name=step_id,json=stepId,proto3" json:"step_id,omitempty"`
	// Name of the kernel producing an output as set in GraphDef, e.g.,
	// "affine2/weights/Assign".
	KernelName string `protobuf:"bytes,2,opt,name=kernel_name,json=kernelName,proto3" json:"kernel_name,omitempty"`
	// Index of the output being set.
	Index int32 `protobuf:"varint,3,opt,name=index,proto3" json:"index,omitempty"`
	// Output tensor details.
	Tensor *TensorDescription `protobuf:"bytes,4,opt,name=tensor,proto3" json:"tensor,omitempty"`
	// contains filtered or unexported fields
}

func (*MemoryLogTensorOutput) Descriptor deprecated 

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

Deprecated: Use MemoryLogTensorOutput.ProtoReflect.Descriptor instead.

func (*MemoryLogTensorOutput) GetIndex 

func (x *MemoryLogTensorOutput) GetIndex() int32

func (*MemoryLogTensorOutput) GetKernelName 

func (x *MemoryLogTensorOutput) GetKernelName() string

func (*MemoryLogTensorOutput) GetStepId 

func (x *MemoryLogTensorOutput) GetStepId() int64

func (*MemoryLogTensorOutput) GetTensor 

func (x *MemoryLogTensorOutput) GetTensor() *TensorDescription

func (*MemoryLogTensorOutput) ProtoMessage 

func (*MemoryLogTensorOutput) ProtoMessage()

func (*MemoryLogTensorOutput) ProtoReflect 

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

func (*MemoryLogTensorOutput) Reset 

func (x *MemoryLogTensorOutput) Reset()

func (*MemoryLogTensorOutput) String 

func (x *MemoryLogTensorOutput) 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: Marked as deprecated in tensorflow/core/framework/step_stats.proto.
	DeviceTempMemorySize int64 `` /* 126-byte string literal not displayed */
	// Deprecated: Marked as deprecated in tensorflow/core/framework/step_stats.proto.
	DevicePersistentMemorySize int64 `` /* 144-byte string literal not displayed */
	// Deprecated: Marked as deprecated in tensorflow/core/framework/step_stats.proto.
	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: Marked as deprecated in tensorflow/core/framework/step_stats.proto.

func (*MemoryStats) GetDevicePersistentTensorAllocIds deprecated 

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

Deprecated: Marked as deprecated in tensorflow/core/framework/step_stats.proto.

func (*MemoryStats) GetDeviceTempMemorySize deprecated 

func (x *MemoryStats) GetDeviceTempMemorySize() int64

Deprecated: Marked as deprecated in tensorflow/core/framework/step_stats.proto.

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 Metadata 

type Metadata struct {
	Name []byte `protobuf:"bytes,1,opt,name=name,proto3" json:"name,omitempty"`
	// contains filtered or unexported fields
}

next: 2

func (*Metadata) Descriptor deprecated 

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

Deprecated: Use Metadata.ProtoReflect.Descriptor instead.

func (*Metadata) GetName 

func (x *Metadata) GetName() []byte

func (*Metadata) ProtoMessage 

func (*Metadata) ProtoMessage()

func (*Metadata) ProtoReflect 

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

func (*Metadata) Reset 

func (x *Metadata) Reset()

func (*Metadata) String 

func (x *Metadata) String() string

type ModelProto 

type ModelProto struct {

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

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

func (*ModelProto) Descriptor deprecated 

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

Deprecated: Use ModelProto.ProtoReflect.Descriptor instead.

func (*ModelProto) GetIdCounter 

func (x *ModelProto) GetIdCounter() int64

func (*ModelProto) GetNodes 

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

func (*ModelProto) GetOptimizationParams 

func (x *ModelProto) GetOptimizationParams() *ModelProto_OptimizationParams

func (*ModelProto) GetOutput 

func (x *ModelProto) GetOutput() int64

func (*ModelProto) ProtoMessage 

func (*ModelProto) ProtoMessage()

func (*ModelProto) ProtoReflect 

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

func (*ModelProto) Reset 

func (x *ModelProto) Reset()

func (*ModelProto) String 

func (x *ModelProto) String() string

type ModelProto_Node 

type ModelProto_Node struct {

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

General representation of a node in the model.

func (*ModelProto_Node) Descriptor deprecated 

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

Deprecated: Use ModelProto_Node.ProtoReflect.Descriptor instead.

func (*ModelProto_Node) GetAutotune 

func (x *ModelProto_Node) GetAutotune() bool

func (*ModelProto_Node) GetBufferedBytes 

func (x *ModelProto_Node) GetBufferedBytes() int64

func (*ModelProto_Node) GetBufferedElements 

func (x *ModelProto_Node) GetBufferedElements() int64

func (*ModelProto_Node) GetBytesConsumed 

func (x *ModelProto_Node) GetBytesConsumed() int64

func (*ModelProto_Node) GetBytesProduced 

func (x *ModelProto_Node) GetBytesProduced() int64

func (*ModelProto_Node) GetId 

func (x *ModelProto_Node) GetId() int64

func (*ModelProto_Node) GetInputProcessingTimeCount 

func (x *ModelProto_Node) GetInputProcessingTimeCount() int64

func (*ModelProto_Node) GetInputProcessingTimeSum 

func (x *ModelProto_Node) GetInputProcessingTimeSum() float64

func (*ModelProto_Node) GetInputs 

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

func (*ModelProto_Node) GetMemoryRatio 

func (x *ModelProto_Node) GetMemoryRatio() float64

func (*ModelProto_Node) GetName 

func (x *ModelProto_Node) GetName() string

func (*ModelProto_Node) GetNodeClass 

func (x *ModelProto_Node) GetNodeClass() NodeClass

func (*ModelProto_Node) GetNumElements 

func (x *ModelProto_Node) GetNumElements() int64

func (*ModelProto_Node) GetParameters 

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

func (*ModelProto_Node) GetProcessingTime 

func (x *ModelProto_Node) GetProcessingTime() int64

func (*ModelProto_Node) GetRatio 

func (x *ModelProto_Node) GetRatio() float64

func (*ModelProto_Node) GetRecordMetrics 

func (x *ModelProto_Node) GetRecordMetrics() bool

func (*ModelProto_Node) ProtoMessage 

func (*ModelProto_Node) ProtoMessage()

func (*ModelProto_Node) ProtoReflect 

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

func (*ModelProto_Node) Reset 

func (x *ModelProto_Node) Reset()

func (*ModelProto_Node) String 

func (x *ModelProto_Node) String() string

type ModelProto_Node_Parameter 

type ModelProto_Node_Parameter struct {

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

Represents a node parameter.

func (*ModelProto_Node_Parameter) Descriptor deprecated 

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

Deprecated: Use ModelProto_Node_Parameter.ProtoReflect.Descriptor instead.

func (*ModelProto_Node_Parameter) GetMax 

func (x *ModelProto_Node_Parameter) GetMax() float64

func (*ModelProto_Node_Parameter) GetMin 

func (x *ModelProto_Node_Parameter) GetMin() float64

func (*ModelProto_Node_Parameter) GetName 

func (x *ModelProto_Node_Parameter) GetName() string

func (*ModelProto_Node_Parameter) GetStateValue 

func (x *ModelProto_Node_Parameter) GetStateValue() float64

func (*ModelProto_Node_Parameter) GetTunable 

func (x *ModelProto_Node_Parameter) GetTunable() bool

func (*ModelProto_Node_Parameter) GetValue 

func (x *ModelProto_Node_Parameter) GetValue() float64

func (*ModelProto_Node_Parameter) ProtoMessage 

func (*ModelProto_Node_Parameter) ProtoMessage()

func (*ModelProto_Node_Parameter) ProtoReflect 

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

func (*ModelProto_Node_Parameter) Reset 

func (x *ModelProto_Node_Parameter) Reset()

func (*ModelProto_Node_Parameter) String 

func (x *ModelProto_Node_Parameter) String() string

type ModelProto_OptimizationParams 

type ModelProto_OptimizationParams struct {

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

Contains parameters of the model autotuning optimization.

func (*ModelProto_OptimizationParams) Descriptor deprecated 

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

Deprecated: Use ModelProto_OptimizationParams.ProtoReflect.Descriptor instead.

func (*ModelProto_OptimizationParams) GetAlgorithm 

func (x *ModelProto_OptimizationParams) GetAlgorithm() AutotuneAlgorithm

func (*ModelProto_OptimizationParams) GetCpuBudget 

func (x *ModelProto_OptimizationParams) GetCpuBudget() int64

func (*ModelProto_OptimizationParams) GetModelInputTime 

func (x *ModelProto_OptimizationParams) GetModelInputTime() float64

func (*ModelProto_OptimizationParams) GetRamBudget 

func (x *ModelProto_OptimizationParams) GetRamBudget() int64

func (*ModelProto_OptimizationParams) ProtoMessage 

func (*ModelProto_OptimizationParams) ProtoMessage()

func (*ModelProto_OptimizationParams) ProtoReflect 

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

func (*ModelProto_OptimizationParams) Reset 

func (x *ModelProto_OptimizationParams) Reset()

func (*ModelProto_OptimizationParams) String 

func (x *ModelProto_OptimizationParams) String() string

type 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 NodeClass 

type NodeClass int32

Class of a node in the performance model. 

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

func (NodeClass) Descriptor 

func (NodeClass) Descriptor() protoreflect.EnumDescriptor

func (NodeClass) Enum 

func (x NodeClass) Enum() *NodeClass

func (NodeClass) EnumDescriptor deprecated 

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

Deprecated: Use NodeClass.Descriptor instead.

func (NodeClass) Number 

func (x NodeClass) Number() protoreflect.EnumNumber

func (NodeClass) String 

func (x NodeClass) String() string

func (NodeClass) Type 

func (NodeClass) Type() protoreflect.EnumType

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 */
	// The complete type of this node. Experimental and subject to change.
	// Currently, the field only contains the return types of the node. That will
	// extend in the future to contain the entire signature of the node, as a
	// function type.
	ExperimentalType *FullTypeDef `protobuf:"bytes,7,opt,name=experimental_type,json=experimentalType,proto3" json:"experimental_type,omitempty"`
	// 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) GetExperimentalType 

func (x *NodeDef) GetExperimentalType() *FullTypeDef

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 OptimizationOptions 

type OptimizationOptions struct {

	// Whether to apply default graph optimizations. If False, only graph
	// optimizations that have been explicitly enabled will be applied.
	//
	// Types that are assignable to OptionalApplyDefaultOptimizations:
	//
	//	*OptimizationOptions_ApplyDefaultOptimizations
	OptionalApplyDefaultOptimizations isOptimizationOptions_OptionalApplyDefaultOptimizations `protobuf_oneof:"optional_apply_default_optimizations"`
	// Whether to fuse filter transformations.
	//
	// Types that are assignable to OptionalFilterFusion:
	//
	//	*OptimizationOptions_FilterFusion
	OptionalFilterFusion isOptimizationOptions_OptionalFilterFusion `protobuf_oneof:"optional_filter_fusion"`
	// Whether to fuse map and batch transformations.
	//
	// Types that are assignable to OptionalMapAndBatchFusion:
	//
	//	*OptimizationOptions_MapAndBatchFusion
	OptionalMapAndBatchFusion isOptimizationOptions_OptionalMapAndBatchFusion `protobuf_oneof:"optional_map_and_batch_fusion"`
	// Whether to fuse map and filter transformations.
	//
	// Types that are assignable to OptionalMapAndFilterFusion:
	//
	//	*OptimizationOptions_MapAndFilterFusion
	OptionalMapAndFilterFusion isOptimizationOptions_OptionalMapAndFilterFusion `protobuf_oneof:"optional_map_and_filter_fusion"`
	// Whether to fuse map transformations.
	//
	// Types that are assignable to OptionalMapFusion:
	//
	//	*OptimizationOptions_MapFusion
	OptionalMapFusion isOptimizationOptions_OptionalMapFusion `protobuf_oneof:"optional_map_fusion"`
	// Whether to parallelize stateless map transformations.
	//
	// Types that are assignable to OptionalMapParallelization:
	//
	//	*OptimizationOptions_MapParallelization
	OptionalMapParallelization isOptimizationOptions_OptionalMapParallelization `protobuf_oneof:"optional_map_parallelization"`
	// Whether to eliminate no-op transformations.
	//
	// Types that are assignable to OptionalNoopElimination:
	//
	//	*OptimizationOptions_NoopElimination
	OptionalNoopElimination isOptimizationOptions_OptionalNoopElimination `protobuf_oneof:"optional_noop_elimination"`
	// Whether to parallelize copying of batch elements. This optimization is
	// highly experimental and can cause performance degradation (e.g. when the
	// parallelization overhead exceeds the benefits of performing the data copies
	// in parallel). You should only enable this optimization if a) your input
	// pipeline is bottlenecked on batching and b) you have validated that this
	// optimization improves performance.
	//
	// Types that are assignable to OptionalParallelBatch:
	//
	//	*OptimizationOptions_ParallelBatch
	OptionalParallelBatch isOptimizationOptions_OptionalParallelBatch `protobuf_oneof:"optional_parallel_batch"`
	// Whether to fuse shuffle and repeat transformations.
	//
	// Types that are assignable to OptionalShuffleAndRepeatFusion:
	//
	//	*OptimizationOptions_ShuffleAndRepeatFusion
	OptionalShuffleAndRepeatFusion isOptimizationOptions_OptionalShuffleAndRepeatFusion `protobuf_oneof:"optional_shuffle_and_repeat_fusion"`
	// Whether to parallelize stateless filter transformations.
	//
	// Types that are assignable to OptionalFilterParallelization:
	//
	//	*OptimizationOptions_FilterParallelization
	OptionalFilterParallelization isOptimizationOptions_OptionalFilterParallelization `protobuf_oneof:"optional_filter_parallelization"`
	// Whether to inject 'Prefetch' as the last transformation. Only takes effect
	// if the last transformation is synchronous; otherwise does nothing.
	//
	// Types that are assignable to OptionalInjectPrefetch:
	//
	//	*OptimizationOptions_InjectPrefetch
	OptionalInjectPrefetch isOptimizationOptions_OptionalInjectPrefetch `protobuf_oneof:"optional_inject_prefetch"`
	// contains filtered or unexported fields
}

next: 20

func (*OptimizationOptions) Descriptor deprecated 

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

Deprecated: Use OptimizationOptions.ProtoReflect.Descriptor instead.

func (*OptimizationOptions) GetApplyDefaultOptimizations 

func (x *OptimizationOptions) GetApplyDefaultOptimizations() bool

func (*OptimizationOptions) GetFilterFusion 

func (x *OptimizationOptions) GetFilterFusion() bool

func (*OptimizationOptions) GetFilterParallelization 

func (x *OptimizationOptions) GetFilterParallelization() bool

func (*OptimizationOptions) GetInjectPrefetch 

func (x *OptimizationOptions) GetInjectPrefetch() bool

func (*OptimizationOptions) GetMapAndBatchFusion 

func (x *OptimizationOptions) GetMapAndBatchFusion() bool

func (*OptimizationOptions) GetMapAndFilterFusion 

func (x *OptimizationOptions) GetMapAndFilterFusion() bool

func (*OptimizationOptions) GetMapFusion 

func (x *OptimizationOptions) GetMapFusion() bool

func (*OptimizationOptions) GetMapParallelization 

func (x *OptimizationOptions) GetMapParallelization() bool

func (*OptimizationOptions) GetNoopElimination 

func (x *OptimizationOptions) GetNoopElimination() bool

func (*OptimizationOptions) GetOptionalApplyDefaultOptimizations 

func (m *OptimizationOptions) GetOptionalApplyDefaultOptimizations() isOptimizationOptions_OptionalApplyDefaultOptimizations

func (*OptimizationOptions) GetOptionalFilterFusion 

func (m *OptimizationOptions) GetOptionalFilterFusion() isOptimizationOptions_OptionalFilterFusion

func (*OptimizationOptions) GetOptionalFilterParallelization 

func (m *OptimizationOptions) GetOptionalFilterParallelization() isOptimizationOptions_OptionalFilterParallelization

func (*OptimizationOptions) GetOptionalInjectPrefetch 

func (m *OptimizationOptions) GetOptionalInjectPrefetch() isOptimizationOptions_OptionalInjectPrefetch

func (*OptimizationOptions) GetOptionalMapAndBatchFusion 

func (m *OptimizationOptions) GetOptionalMapAndBatchFusion() isOptimizationOptions_OptionalMapAndBatchFusion

func (*OptimizationOptions) GetOptionalMapAndFilterFusion 

func (m *OptimizationOptions) GetOptionalMapAndFilterFusion() isOptimizationOptions_OptionalMapAndFilterFusion

func (*OptimizationOptions) GetOptionalMapFusion 

func (m *OptimizationOptions) GetOptionalMapFusion() isOptimizationOptions_OptionalMapFusion

func (*OptimizationOptions) GetOptionalMapParallelization 

func (m *OptimizationOptions) GetOptionalMapParallelization() isOptimizationOptions_OptionalMapParallelization

func (*OptimizationOptions) GetOptionalNoopElimination 

func (m *OptimizationOptions) GetOptionalNoopElimination() isOptimizationOptions_OptionalNoopElimination

func (*OptimizationOptions) GetOptionalParallelBatch 

func (m *OptimizationOptions) GetOptionalParallelBatch() isOptimizationOptions_OptionalParallelBatch

func (*OptimizationOptions) GetOptionalShuffleAndRepeatFusion 

func (m *OptimizationOptions) GetOptionalShuffleAndRepeatFusion() isOptimizationOptions_OptionalShuffleAndRepeatFusion

func (*OptimizationOptions) GetParallelBatch 

func (x *OptimizationOptions) GetParallelBatch() bool

func (*OptimizationOptions) GetShuffleAndRepeatFusion 

func (x *OptimizationOptions) GetShuffleAndRepeatFusion() bool

func (*OptimizationOptions) ProtoMessage 

func (*OptimizationOptions) ProtoMessage()

func (*OptimizationOptions) ProtoReflect 

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

func (*OptimizationOptions) Reset 

func (x *OptimizationOptions) Reset()

func (*OptimizationOptions) String 

func (x *OptimizationOptions) String() string

type OptimizationOptions_ApplyDefaultOptimizations 

type OptimizationOptions_ApplyDefaultOptimizations struct {
	ApplyDefaultOptimizations bool `protobuf:"varint,1,opt,name=apply_default_optimizations,json=applyDefaultOptimizations,proto3,oneof"`
}

type OptimizationOptions_FilterFusion 

type OptimizationOptions_FilterFusion struct {
	FilterFusion bool `protobuf:"varint,6,opt,name=filter_fusion,json=filterFusion,proto3,oneof"`
}

type OptimizationOptions_FilterParallelization 

type OptimizationOptions_FilterParallelization struct {
	FilterParallelization bool `protobuf:"varint,18,opt,name=filter_parallelization,json=filterParallelization,proto3,oneof"`
}

type OptimizationOptions_InjectPrefetch 

type OptimizationOptions_InjectPrefetch struct {
	InjectPrefetch bool `protobuf:"varint,19,opt,name=inject_prefetch,json=injectPrefetch,proto3,oneof"`
}

type OptimizationOptions_MapAndBatchFusion 

type OptimizationOptions_MapAndBatchFusion struct {
	MapAndBatchFusion bool `protobuf:"varint,9,opt,name=map_and_batch_fusion,json=mapAndBatchFusion,proto3,oneof"`
}

type OptimizationOptions_MapAndFilterFusion 

type OptimizationOptions_MapAndFilterFusion struct {
	MapAndFilterFusion bool `protobuf:"varint,10,opt,name=map_and_filter_fusion,json=mapAndFilterFusion,proto3,oneof"`
}

type OptimizationOptions_MapFusion 

type OptimizationOptions_MapFusion struct {
	MapFusion bool `protobuf:"varint,11,opt,name=map_fusion,json=mapFusion,proto3,oneof"`
}

type OptimizationOptions_MapParallelization 

type OptimizationOptions_MapParallelization struct {
	MapParallelization bool `protobuf:"varint,12,opt,name=map_parallelization,json=mapParallelization,proto3,oneof"`
}

type OptimizationOptions_NoopElimination 

type OptimizationOptions_NoopElimination struct {
	NoopElimination bool `protobuf:"varint,14,opt,name=noop_elimination,json=noopElimination,proto3,oneof"`
}

type OptimizationOptions_ParallelBatch 

type OptimizationOptions_ParallelBatch struct {
	ParallelBatch bool `protobuf:"varint,15,opt,name=parallel_batch,json=parallelBatch,proto3,oneof"`
}

type OptimizationOptions_ShuffleAndRepeatFusion 

type OptimizationOptions_ShuffleAndRepeatFusion struct {
	ShuffleAndRepeatFusion bool `protobuf:"varint,17,opt,name=shuffle_and_repeat_fusion,json=shuffleAndRepeatFusion,proto3,oneof"`
}

type OptimizedFunctionGraph added in v2.12.0 

type OptimizedFunctionGraph struct {

	// Function name. It can be a human-readable SignatureDef's method name, or a
	// FunctionDef name.
	Name string `protobuf:"bytes,1,opt,name=name,proto3" json:"name,omitempty"`
	// Optimized function graph.
	FunctionGraph *GraphDef `protobuf:"bytes,2,opt,name=function_graph,json=functionGraph,proto3" json:"function_graph,omitempty"`
	// Maps from node name to control ret. This is an output from running TF/XLA
	// bridge.
	NodeNameToControlRet map[string]string `` /* 215-byte string literal not displayed */
	// Return node types of the function. This is an output of graph
	// preprocessing.
	RetTypes []DataType `protobuf:"varint,4,rep,packed,name=ret_types,json=retTypes,proto3,enum=tensorflow.DataType" json:"ret_types,omitempty"`
	// Number of return nodes. This is an output of graph preprocessing.
	NumReturnNodes uint32 `protobuf:"varint,5,opt,name=num_return_nodes,json=numReturnNodes,proto3" json:"num_return_nodes,omitempty"`
	// contains filtered or unexported fields
}

Optimized function graph after instantiation-related graph optimization passes (up till before graph partitioning). The first half of the proto is representing a GraphDef and the rest of the fields are extra information from graph optimizations.

func (*OptimizedFunctionGraph) Descriptor deprecated added in v2.12.0 

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

Deprecated: Use OptimizedFunctionGraph.ProtoReflect.Descriptor instead.

func (*OptimizedFunctionGraph) GetFunctionGraph added in v2.12.0 

func (x *OptimizedFunctionGraph) GetFunctionGraph() *GraphDef

func (*OptimizedFunctionGraph) GetName added in v2.12.0 

func (x *OptimizedFunctionGraph) GetName() string

func (*OptimizedFunctionGraph) GetNodeNameToControlRet added in v2.12.0 

func (x *OptimizedFunctionGraph) GetNodeNameToControlRet() map[string]string

func (*OptimizedFunctionGraph) GetNumReturnNodes added in v2.12.0 

func (x *OptimizedFunctionGraph) GetNumReturnNodes() uint32

func (*OptimizedFunctionGraph) GetRetTypes added in v2.12.0 

func (x *OptimizedFunctionGraph) GetRetTypes() []DataType

func (*OptimizedFunctionGraph) ProtoMessage added in v2.12.0 

func (*OptimizedFunctionGraph) ProtoMessage()

func (*OptimizedFunctionGraph) ProtoReflect added in v2.12.0 

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

func (*OptimizedFunctionGraph) Reset added in v2.12.0 

func (x *OptimizedFunctionGraph) Reset()

func (*OptimizedFunctionGraph) String added in v2.12.0 

func (x *OptimizedFunctionGraph) String() string

type Options 

type Options struct {

	// Whether the outputs need to be produced in deterministic order.
	//
	// Types that are assignable to OptionalDeterministic:
	//
	//	*Options_Deterministic
	OptionalDeterministic isOptions_OptionalDeterministic `protobuf_oneof:"optional_deterministic"`
	// The distribution strategy options associated with the dataset.
	AutotuneOptions *AutotuneOptions `protobuf:"bytes,7,opt,name=autotune_options,json=autotuneOptions,proto3" json:"autotune_options,omitempty"`
	// The distribution strategy options associated with the dataset.
	DistributeOptions *DistributeOptions `protobuf:"bytes,2,opt,name=distribute_options,json=distributeOptions,proto3" json:"distribute_options,omitempty"`
	// The optimization options associated with the dataset.
	OptimizationOptions *OptimizationOptions `protobuf:"bytes,3,opt,name=optimization_options,json=optimizationOptions,proto3" json:"optimization_options,omitempty"`
	// Whether to introduce 'slack' in the last `prefetch` of the input pipeline,
	// if it exists. This may reduce CPU contention with accelerator host-side
	// activity at the start of a step. The slack frequency is determined by the
	// number of devices attached to this input pipeline.
	//
	// Types that are assignable to OptionalSlack:
	//
	//	*Options_Slack
	OptionalSlack isOptions_OptionalSlack `protobuf_oneof:"optional_slack"`
	// The threading options associated with the dataset.
	ThreadingOptions *ThreadingOptions `protobuf:"bytes,5,opt,name=threading_options,json=threadingOptions,proto3" json:"threading_options,omitempty"`
	// This option can be used to override the default policy for how to handle
	// external state when serializing a dataset or checkpointing its iterator.
	// There are three settings available - IGNORE: External state is ignored
	// without a warning; WARN: External state is ignored and a warning is logged;
	// FAIL: External state results in an error.
	//
	// Types that are assignable to OptionalExternalStatePolicy:
	//
	//	*Options_ExternalStatePolicy
	OptionalExternalStatePolicy isOptions_OptionalExternalStatePolicy `protobuf_oneof:"optional_external_state_policy"`
	// This option indicates whether to checkpoint input pipeline state
	// "explicitly", by storing the internal state of iterators for each
	// tf.data operation, (the default), or "symbolically", by storing metadata
	// that captures the state of each tf.data operation at the time it processed
	// the last data seen by tf.data consumer.
	//
	// Symbolic checkpoints are expected to be much smaller but not all tf.data
	// operations are compatible with symbolic checkpointing. In particular,
	// symbolic checkpointing requires that data is processed in-order and
	// operations that reorder elements, such as `shuffle()` or non-deterministic
	// `map()`, are not compatible with symbolic checkpointing.
	//
	// Types that are assignable to OptionalSymbolicCheckpoint:
	//
	//	*Options_SymbolicCheckpoint
	OptionalSymbolicCheckpoint isOptions_OptionalSymbolicCheckpoint `protobuf_oneof:"optional_symbolic_checkpoint"`
	// contains filtered or unexported fields
}

Message stored with Dataset objects to control how datasets are processed and optimized.

next: 9

func (*Options) Descriptor deprecated 

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

Deprecated: Use Options.ProtoReflect.Descriptor instead.

func (*Options) GetAutotuneOptions 

func (x *Options) GetAutotuneOptions() *AutotuneOptions

func (*Options) GetDeterministic 

func (x *Options) GetDeterministic() bool

func (*Options) GetDistributeOptions 

func (x *Options) GetDistributeOptions() *DistributeOptions

func (*Options) GetExternalStatePolicy 

func (x *Options) GetExternalStatePolicy() ExternalStatePolicy

func (*Options) GetOptimizationOptions 

func (x *Options) GetOptimizationOptions() *OptimizationOptions

func (*Options) GetOptionalDeterministic 

func (m *Options) GetOptionalDeterministic() isOptions_OptionalDeterministic

func (*Options) GetOptionalExternalStatePolicy 

func (m *Options) GetOptionalExternalStatePolicy() isOptions_OptionalExternalStatePolicy

func (*Options) GetOptionalSlack 

func (m *Options) GetOptionalSlack() isOptions_OptionalSlack

func (*Options) GetOptionalSymbolicCheckpoint added in v2.12.0 

func (m *Options) GetOptionalSymbolicCheckpoint() isOptions_OptionalSymbolicCheckpoint

func (*Options) GetSlack 

func (x *Options) GetSlack() bool

func (*Options) GetSymbolicCheckpoint added in v2.12.0 

func (x *Options) GetSymbolicCheckpoint() bool

func (*Options) GetThreadingOptions 

func (x *Options) GetThreadingOptions() *ThreadingOptions

func (*Options) ProtoMessage 

func (*Options) ProtoMessage()

func (*Options) ProtoReflect 

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

func (*Options) Reset 

func (x *Options) Reset()

func (*Options) String 

func (x *Options) String() string

type Options_Deterministic 

type Options_Deterministic struct {
	Deterministic bool `protobuf:"varint,1,opt,name=deterministic,proto3,oneof"`
}

type Options_ExternalStatePolicy 

type Options_ExternalStatePolicy struct {
	ExternalStatePolicy ExternalStatePolicy `` /* 129-byte string literal not displayed */
}

type Options_Slack 

type Options_Slack struct {
	Slack bool `protobuf:"varint,4,opt,name=slack,proto3,oneof"`
}

type Options_SymbolicCheckpoint added in v2.12.0 

type Options_SymbolicCheckpoint struct {
	SymbolicCheckpoint bool `protobuf:"varint,8,opt,name=symbolic_checkpoint,json=symbolicCheckpoint,proto3,oneof"`
}

type ReaderBaseState 

type ReaderBaseState struct {
	WorkStarted        int64  `protobuf:"varint,1,opt,name=work_started,json=workStarted,proto3" json:"work_started,omitempty"`
	WorkFinished       int64  `protobuf:"varint,2,opt,name=work_finished,json=workFinished,proto3" json:"work_finished,omitempty"`
	NumRecordsProduced int64  `protobuf:"varint,3,opt,name=num_records_produced,json=numRecordsProduced,proto3" json:"num_records_produced,omitempty"`
	CurrentWork        []byte `protobuf:"bytes,4,opt,name=current_work,json=currentWork,proto3" json:"current_work,omitempty"`
	// contains filtered or unexported fields
}

For serializing and restoring the state of ReaderBase, see reader_base.h for details.

func (*ReaderBaseState) Descriptor deprecated 

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

Deprecated: Use ReaderBaseState.ProtoReflect.Descriptor instead.

func (*ReaderBaseState) GetCurrentWork 

func (x *ReaderBaseState) GetCurrentWork() []byte

func (*ReaderBaseState) GetNumRecordsProduced 

func (x *ReaderBaseState) GetNumRecordsProduced() int64

func (*ReaderBaseState) GetWorkFinished 

func (x *ReaderBaseState) GetWorkFinished() int64

func (*ReaderBaseState) GetWorkStarted 

func (x *ReaderBaseState) GetWorkStarted() int64

func (*ReaderBaseState) ProtoMessage 

func (*ReaderBaseState) ProtoMessage()

func (*ReaderBaseState) ProtoReflect 

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

func (*ReaderBaseState) Reset 

func (x *ReaderBaseState) Reset()

func (*ReaderBaseState) String 

func (x *ReaderBaseState) 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 SerializedDType 

type SerializedDType struct {
	Datatype DataType `protobuf:"varint,1,opt,name=datatype,proto3,enum=tensorflow.DataType" json:"datatype,omitempty"`
	// contains filtered or unexported fields
}

Represents a serialized tf.dtypes.Dtype

func (*SerializedDType) Descriptor deprecated 

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

Deprecated: Use SerializedDType.ProtoReflect.Descriptor instead.

func (*SerializedDType) GetDatatype 

func (x *SerializedDType) GetDatatype() DataType

func (*SerializedDType) ProtoMessage 

func (*SerializedDType) ProtoMessage()

func (*SerializedDType) ProtoReflect 

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

func (*SerializedDType) Reset 

func (x *SerializedDType) Reset()

func (*SerializedDType) String 

func (x *SerializedDType) 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 Summary 

type Summary struct {

	// Set of values for the summary.
	Value []*Summary_Value `protobuf:"bytes,1,rep,name=value,proto3" json:"value,omitempty"`
	// contains filtered or unexported fields
}

A Summary is a set of named values to be displayed by the visualizer.

Summaries are produced regularly during training, as controlled by the "summary_interval_secs" attribute of the training operation. Summaries are also produced at the end of an evaluation.

func (*Summary) Descriptor deprecated 

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

Deprecated: Use Summary.ProtoReflect.Descriptor instead.

func (*Summary) GetValue 

func (x *Summary) GetValue() []*Summary_Value

func (*Summary) ProtoMessage 

func (*Summary) ProtoMessage()

func (*Summary) ProtoReflect 

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

func (*Summary) Reset 

func (x *Summary) Reset()

func (*Summary) String 

func (x *Summary) String() string

type SummaryDescription 

type SummaryDescription struct {

	// Hint on how plugins should process the data in this series.
	// Supported values include "scalar", "histogram", "image", "audio"
	TypeHint string `protobuf:"bytes,1,opt,name=type_hint,json=typeHint,proto3" json:"type_hint,omitempty"`
	// contains filtered or unexported fields
}

Metadata associated with a series of Summary data

func (*SummaryDescription) Descriptor deprecated 

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

Deprecated: Use SummaryDescription.ProtoReflect.Descriptor instead.

func (*SummaryDescription) GetTypeHint 

func (x *SummaryDescription) GetTypeHint() string

func (*SummaryDescription) ProtoMessage 

func (*SummaryDescription) ProtoMessage()

func (*SummaryDescription) ProtoReflect 

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

func (*SummaryDescription) Reset 

func (x *SummaryDescription) Reset()

func (*SummaryDescription) String 

func (x *SummaryDescription) String() string

type SummaryMetadata 

type SummaryMetadata struct {

	// Data that associates a summary with a certain plugin.
	PluginData *SummaryMetadata_PluginData `protobuf:"bytes,1,opt,name=plugin_data,json=pluginData,proto3" json:"plugin_data,omitempty"`
	// Display name for viewing in TensorBoard.
	DisplayName string `protobuf:"bytes,2,opt,name=display_name,json=displayName,proto3" json:"display_name,omitempty"`
	// Longform readable description of the summary sequence. Markdown supported.
	SummaryDescription string `protobuf:"bytes,3,opt,name=summary_description,json=summaryDescription,proto3" json:"summary_description,omitempty"`
	// Class of data stored in this time series. Required for compatibility with
	// TensorBoard's generic data facilities (`DataProvider`, et al.). This value
	// imposes constraints on the dtype and shape of the corresponding tensor
	// values. See `DataClass` docs for details.
	DataClass DataClass `protobuf:"varint,4,opt,name=data_class,json=dataClass,proto3,enum=tensorflow.DataClass" json:"data_class,omitempty"`
	// contains filtered or unexported fields
}

A SummaryMetadata encapsulates information on which plugins are able to make use of a certain summary value.

func (*SummaryMetadata) Descriptor deprecated 

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

Deprecated: Use SummaryMetadata.ProtoReflect.Descriptor instead.

func (*SummaryMetadata) GetDataClass 

func (x *SummaryMetadata) GetDataClass() DataClass

func (*SummaryMetadata) GetDisplayName 

func (x *SummaryMetadata) GetDisplayName() string

func (*SummaryMetadata) GetPluginData 

func (x *SummaryMetadata) GetPluginData() *SummaryMetadata_PluginData

func (*SummaryMetadata) GetSummaryDescription 

func (x *SummaryMetadata) GetSummaryDescription() string

func (*SummaryMetadata) ProtoMessage 

func (*SummaryMetadata) ProtoMessage()

func (*SummaryMetadata) ProtoReflect 

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

func (*SummaryMetadata) Reset 

func (x *SummaryMetadata) Reset()

func (*SummaryMetadata) String 

func (x *SummaryMetadata) String() string

type SummaryMetadata_PluginData 

type SummaryMetadata_PluginData struct {

	// The name of the plugin this data pertains to.
	PluginName string `protobuf:"bytes,1,opt,name=plugin_name,json=pluginName,proto3" json:"plugin_name,omitempty"`
	// The content to store for the plugin. The best practice is for this to be
	// a binary serialized protocol buffer.
	Content []byte `protobuf:"bytes,2,opt,name=content,proto3" json:"content,omitempty"`
	// contains filtered or unexported fields
}

func (*SummaryMetadata_PluginData) Descriptor deprecated 

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

Deprecated: Use SummaryMetadata_PluginData.ProtoReflect.Descriptor instead.

func (*SummaryMetadata_PluginData) GetContent 

func (x *SummaryMetadata_PluginData) GetContent() []byte

func (*SummaryMetadata_PluginData) GetPluginName 

func (x *SummaryMetadata_PluginData) GetPluginName() string

func (*SummaryMetadata_PluginData) ProtoMessage 

func (*SummaryMetadata_PluginData) ProtoMessage()

func (*SummaryMetadata_PluginData) ProtoReflect 

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

func (*SummaryMetadata_PluginData) Reset 

func (x *SummaryMetadata_PluginData) Reset()

func (*SummaryMetadata_PluginData) String 

func (x *SummaryMetadata_PluginData) String() string

type Summary_Audio 

type Summary_Audio struct {

	// Sample rate of the audio in Hz.
	SampleRate float32 `protobuf:"fixed32,1,opt,name=sample_rate,json=sampleRate,proto3" json:"sample_rate,omitempty"`
	// Number of channels of audio.
	NumChannels int64 `protobuf:"varint,2,opt,name=num_channels,json=numChannels,proto3" json:"num_channels,omitempty"`
	// Length of the audio in frames (samples per channel).
	LengthFrames int64 `protobuf:"varint,3,opt,name=length_frames,json=lengthFrames,proto3" json:"length_frames,omitempty"`
	// Encoded audio data and its associated RFC 2045 content type (e.g.
	// "audio/wav").
	EncodedAudioString []byte `protobuf:"bytes,4,opt,name=encoded_audio_string,json=encodedAudioString,proto3" json:"encoded_audio_string,omitempty"`
	ContentType        string `protobuf:"bytes,5,opt,name=content_type,json=contentType,proto3" json:"content_type,omitempty"`
	// contains filtered or unexported fields
}

func (*Summary_Audio) Descriptor deprecated 

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

Deprecated: Use Summary_Audio.ProtoReflect.Descriptor instead.

func (*Summary_Audio) GetContentType 

func (x *Summary_Audio) GetContentType() string

func (*Summary_Audio) GetEncodedAudioString 

func (x *Summary_Audio) GetEncodedAudioString() []byte

func (*Summary_Audio) GetLengthFrames 

func (x *Summary_Audio) GetLengthFrames() int64

func (*Summary_Audio) GetNumChannels 

func (x *Summary_Audio) GetNumChannels() int64

func (*Summary_Audio) GetSampleRate 

func (x *Summary_Audio) GetSampleRate() float32

func (*Summary_Audio) ProtoMessage 

func (*Summary_Audio) ProtoMessage()

func (*Summary_Audio) ProtoReflect 

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

func (*Summary_Audio) Reset 

func (x *Summary_Audio) Reset()

func (*Summary_Audio) String 

func (x *Summary_Audio) String() string

type Summary_Image 

type Summary_Image struct {

	// Dimensions of the image.
	Height int32 `protobuf:"varint,1,opt,name=height,proto3" json:"height,omitempty"`
	Width  int32 `protobuf:"varint,2,opt,name=width,proto3" json:"width,omitempty"`
	// Valid colorspace values are
	//
	//	1 - grayscale
	//	2 - grayscale + alpha
	//	3 - RGB
	//	4 - RGBA
	//	5 - DIGITAL_YUV
	//	6 - BGRA
	Colorspace int32 `protobuf:"varint,3,opt,name=colorspace,proto3" json:"colorspace,omitempty"`
	// Image data in encoded format.  All image formats supported by
	// image_codec::CoderUtil can be stored here.
	EncodedImageString []byte `protobuf:"bytes,4,opt,name=encoded_image_string,json=encodedImageString,proto3" json:"encoded_image_string,omitempty"`
	// contains filtered or unexported fields
}

func (*Summary_Image) Descriptor deprecated 

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

Deprecated: Use Summary_Image.ProtoReflect.Descriptor instead.

func (*Summary_Image) GetColorspace 

func (x *Summary_Image) GetColorspace() int32

func (*Summary_Image) GetEncodedImageString 

func (x *Summary_Image) GetEncodedImageString() []byte

func (*Summary_Image) GetHeight 

func (x *Summary_Image) GetHeight() int32

func (*Summary_Image) GetWidth 

func (x *Summary_Image) GetWidth() int32

func (*Summary_Image) ProtoMessage 

func (*Summary_Image) ProtoMessage()

func (*Summary_Image) ProtoReflect 

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

func (*Summary_Image) Reset 

func (x *Summary_Image) Reset()

func (*Summary_Image) String 

func (x *Summary_Image) String() string

type Summary_Value 

type Summary_Value struct {

	// This field is deprecated and will not be set.
	NodeName string `protobuf:"bytes,7,opt,name=node_name,json=nodeName,proto3" json:"node_name,omitempty"`
	// Tag name for the data. Used by TensorBoard plugins to organize data. Tags
	// are often organized by scope (which contains slashes to convey
	// hierarchy). For example: foo/bar/0
	Tag string `protobuf:"bytes,1,opt,name=tag,proto3" json:"tag,omitempty"`
	// Contains metadata on the summary value such as which plugins may use it.
	// Take note that many summary values may lack a metadata field. This is
	// because the FileWriter only keeps a metadata object on the first summary
	// value with a certain tag for each tag. TensorBoard then remembers which
	// tags are associated with which plugins. This saves space.
	Metadata *SummaryMetadata `protobuf:"bytes,9,opt,name=metadata,proto3" json:"metadata,omitempty"`
	// Value associated with the tag.
	//
	// Types that are assignable to Value:
	//
	//	*Summary_Value_SimpleValue
	//	*Summary_Value_ObsoleteOldStyleHistogram
	//	*Summary_Value_Image
	//	*Summary_Value_Histo
	//	*Summary_Value_Audio
	//	*Summary_Value_Tensor
	Value isSummary_Value_Value `protobuf_oneof:"value"`
	// contains filtered or unexported fields
}

func (*Summary_Value) Descriptor deprecated 

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

Deprecated: Use Summary_Value.ProtoReflect.Descriptor instead.

func (*Summary_Value) GetAudio 

func (x *Summary_Value) GetAudio() *Summary_Audio

func (*Summary_Value) GetHisto 

func (x *Summary_Value) GetHisto() *protobuf.HistogramProto

func (*Summary_Value) GetImage 

func (x *Summary_Value) GetImage() *Summary_Image

func (*Summary_Value) GetMetadata 

func (x *Summary_Value) GetMetadata() *SummaryMetadata

func (*Summary_Value) GetNodeName 

func (x *Summary_Value) GetNodeName() string

func (*Summary_Value) GetObsoleteOldStyleHistogram 

func (x *Summary_Value) GetObsoleteOldStyleHistogram() []byte

func (*Summary_Value) GetSimpleValue 

func (x *Summary_Value) GetSimpleValue() float32

func (*Summary_Value) GetTag 

func (x *Summary_Value) GetTag() string

func (*Summary_Value) GetTensor 

func (x *Summary_Value) GetTensor() *TensorProto

func (*Summary_Value) GetValue 

func (m *Summary_Value) GetValue() isSummary_Value_Value

func (*Summary_Value) ProtoMessage 

func (*Summary_Value) ProtoMessage()

func (*Summary_Value) ProtoReflect 

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

func (*Summary_Value) Reset 

func (x *Summary_Value) Reset()

func (*Summary_Value) String 

func (x *Summary_Value) String() string

type Summary_Value_Audio 

type Summary_Value_Audio struct {
	Audio *Summary_Audio `protobuf:"bytes,6,opt,name=audio,proto3,oneof"`
}

type Summary_Value_Histo 

type Summary_Value_Histo struct {
	Histo *protobuf.HistogramProto `protobuf:"bytes,5,opt,name=histo,proto3,oneof"`
}

type Summary_Value_Image 

type Summary_Value_Image struct {
	Image *Summary_Image `protobuf:"bytes,4,opt,name=image,proto3,oneof"`
}

type Summary_Value_ObsoleteOldStyleHistogram 

type Summary_Value_ObsoleteOldStyleHistogram struct {
	ObsoleteOldStyleHistogram []byte `protobuf:"bytes,3,opt,name=obsolete_old_style_histogram,json=obsoleteOldStyleHistogram,proto3,oneof"`
}

type Summary_Value_SimpleValue 

type Summary_Value_SimpleValue struct {
	SimpleValue float32 `protobuf:"fixed32,2,opt,name=simple_value,json=simpleValue,proto3,oneof"`
}

type Summary_Value_Tensor 

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

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"`
	// DT_FLOAT8_*, use variable-sized set of bytes
	// (i.e. the equivalent of repeated uint8, if such a thing existed).
	Float8Val []byte `protobuf:"bytes,18,opt,name=float8_val,json=float8Val,proto3" json:"float8_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) GetFloat8Val added in v2.12.0 

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

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 TensorSliceProto 

type TensorSliceProto struct {

	// Extent of the slice in all tensor dimensions.
	//
	// Must have one entry for each of the dimension of the tensor that this
	// slice belongs to.  The order of sizes is the same as the order of
	// dimensions in the TensorShape.
	Extent []*TensorSliceProto_Extent `protobuf:"bytes,1,rep,name=extent,proto3" json:"extent,omitempty"`
	// contains filtered or unexported fields
}

Can only be interpreted if you know the corresponding TensorShape.

func (*TensorSliceProto) Descriptor deprecated 

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

Deprecated: Use TensorSliceProto.ProtoReflect.Descriptor instead.

func (*TensorSliceProto) GetExtent 

func (x *TensorSliceProto) GetExtent() []*TensorSliceProto_Extent

func (*TensorSliceProto) ProtoMessage 

func (*TensorSliceProto) ProtoMessage()

func (*TensorSliceProto) ProtoReflect 

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

func (*TensorSliceProto) Reset 

func (x *TensorSliceProto) Reset()

func (*TensorSliceProto) String 

func (x *TensorSliceProto) String() string

type TensorSliceProto_Extent 

type TensorSliceProto_Extent struct {

	// Start index of the slice, starting at 0.
	Start int64 `protobuf:"varint,1,opt,name=start,proto3" json:"start,omitempty"`
	// Length of the slice: if the length is missing or -1 we will
	// interpret this as "everything in this dimension".  We use
	// "oneof" to preserve information about whether the length is
	// present without changing the serialization format from the
	// prior proto2 version of this proto.
	//
	// Types that are assignable to HasLength:
	//
	//	*TensorSliceProto_Extent_Length
	HasLength isTensorSliceProto_Extent_HasLength `protobuf_oneof:"has_length"`
	// contains filtered or unexported fields
}

Extent of the slice in one dimension.

func (*TensorSliceProto_Extent) Descriptor deprecated 

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

Deprecated: Use TensorSliceProto_Extent.ProtoReflect.Descriptor instead.

func (*TensorSliceProto_Extent) GetHasLength 

func (m *TensorSliceProto_Extent) GetHasLength() isTensorSliceProto_Extent_HasLength

func (*TensorSliceProto_Extent) GetLength 

func (x *TensorSliceProto_Extent) GetLength() int64

func (*TensorSliceProto_Extent) GetStart 

func (x *TensorSliceProto_Extent) GetStart() int64

func (*TensorSliceProto_Extent) ProtoMessage 

func (*TensorSliceProto_Extent) ProtoMessage()

func (*TensorSliceProto_Extent) ProtoReflect 

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

func (*TensorSliceProto_Extent) Reset 

func (x *TensorSliceProto_Extent) Reset()

func (*TensorSliceProto_Extent) String 

func (x *TensorSliceProto_Extent) String() string

type TensorSliceProto_Extent_Length 

type TensorSliceProto_Extent_Length struct {
	Length int64 `protobuf:"varint,2,opt,name=length,proto3,oneof"`
}

type ThreadingOptions 

type ThreadingOptions struct {

	// If set, it overrides the maximum degree of intra-op parallelism.
	//
	// Types that are assignable to OptionalMaxIntraOpParallelism:
	//
	//	*ThreadingOptions_MaxIntraOpParallelism
	OptionalMaxIntraOpParallelism isThreadingOptions_OptionalMaxIntraOpParallelism `protobuf_oneof:"optional_max_intra_op_parallelism"`
	// If set, the dataset will use a private threadpool of the given size.
	//
	// Types that are assignable to OptionalPrivateThreadpoolSize:
	//
	//	*ThreadingOptions_PrivateThreadpoolSize
	OptionalPrivateThreadpoolSize isThreadingOptions_OptionalPrivateThreadpoolSize `protobuf_oneof:"optional_private_threadpool_size"`
	// contains filtered or unexported fields
}

next: 3

func (*ThreadingOptions) Descriptor deprecated 

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

Deprecated: Use ThreadingOptions.ProtoReflect.Descriptor instead.

func (*ThreadingOptions) GetMaxIntraOpParallelism 

func (x *ThreadingOptions) GetMaxIntraOpParallelism() int32

func (*ThreadingOptions) GetOptionalMaxIntraOpParallelism 

func (m *ThreadingOptions) GetOptionalMaxIntraOpParallelism() isThreadingOptions_OptionalMaxIntraOpParallelism

func (*ThreadingOptions) GetOptionalPrivateThreadpoolSize 

func (m *ThreadingOptions) GetOptionalPrivateThreadpoolSize() isThreadingOptions_OptionalPrivateThreadpoolSize

func (*ThreadingOptions) GetPrivateThreadpoolSize 

func (x *ThreadingOptions) GetPrivateThreadpoolSize() int32

func (*ThreadingOptions) ProtoMessage 

func (*ThreadingOptions) ProtoMessage()

func (*ThreadingOptions) ProtoReflect 

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

func (*ThreadingOptions) Reset 

func (x *ThreadingOptions) Reset()

func (*ThreadingOptions) String 

func (x *ThreadingOptions) String() string

type ThreadingOptions_MaxIntraOpParallelism 

type ThreadingOptions_MaxIntraOpParallelism struct {
	MaxIntraOpParallelism int32 `protobuf:"varint,1,opt,name=max_intra_op_parallelism,json=maxIntraOpParallelism,proto3,oneof"`
}

type ThreadingOptions_PrivateThreadpoolSize 

type ThreadingOptions_PrivateThreadpoolSize struct {
	PrivateThreadpoolSize int32 `protobuf:"varint,2,opt,name=private_threadpool_size,json=privateThreadpoolSize,proto3,oneof"`
}

type UncompressedElement added in v2.12.0 

type UncompressedElement struct {
	Components []*TensorProto `protobuf:"bytes,1,rep,name=components,proto3" json:"components,omitempty"`
	// contains filtered or unexported fields
}

An uncompressed dataset element.

func (*UncompressedElement) Descriptor deprecated added in v2.12.0 

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

Deprecated: Use UncompressedElement.ProtoReflect.Descriptor instead.

func (*UncompressedElement) GetComponents added in v2.12.0 

func (x *UncompressedElement) GetComponents() []*TensorProto

func (*UncompressedElement) ProtoMessage added in v2.12.0 

func (*UncompressedElement) ProtoMessage()

func (*UncompressedElement) ProtoReflect added in v2.12.0 

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

func (*UncompressedElement) Reset added in v2.12.0 

func (x *UncompressedElement) Reset()

func (*UncompressedElement) String added in v2.12.0 

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