pbs

package
v0.0.0-...-a925eb1 Latest Latest
Warning

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

 Go to latest Published: Oct 2, 2023 License: Apache-2.0, BSD-2-Clause Imports: 8 Imported by: 0

Documentation

Overview

Package pbs contains generated go code for tensorflow protocol buffer objects.

WARNING: The API in this package has not been finalized and can change without notice.

Index

Examples

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 (
	AutotuneResult_FailureKind_name = map[int32]string{
		0: "UNKNOWN",
		1: "REDZONE_MODIFIED",
		2: "WRONG_RESULT",
		3: "DISQUALIFIED",
	}
	AutotuneResult_FailureKind_value = map[string]int32{
		"UNKNOWN":          0,
		"REDZONE_MODIFIED": 1,
		"WRONG_RESULT":     2,
		"DISQUALIFIED":     3,
	}
)

Enum value maps for AutotuneResult_FailureKind.

View Source 
var (
	OptimizerOptions_Level_name = map[int32]string{
		0:  "L1",
		-1: "L0",
	}
	OptimizerOptions_Level_value = map[string]int32{
		"L1": 0,
		"L0": -1,
	}
)

Enum value maps for OptimizerOptions_Level.

View Source 
var (
	OptimizerOptions_GlobalJitLevel_name = map[int32]string{
		0:  "DEFAULT",
		-1: "OFF",
		1:  "ON_1",
		2:  "ON_2",
	}
	OptimizerOptions_GlobalJitLevel_value = map[string]int32{
		"DEFAULT": 0,
		"OFF":     -1,
		"ON_1":    1,
		"ON_2":    2,
	}
)

Enum value maps for OptimizerOptions_GlobalJitLevel.

View Source 
var (
	ConfigProto_Experimental_MlirBridgeRollout_name = map[int32]string{
		0: "MLIR_BRIDGE_ROLLOUT_UNSPECIFIED",
		1: "MLIR_BRIDGE_ROLLOUT_ENABLED",
		2: "MLIR_BRIDGE_ROLLOUT_DISABLED",
		3: "MLIR_BRIDGE_ROLLOUT_SAFE_MODE_ENABLED",
		4: "MLIR_BRIDGE_ROLLOUT_SAFE_MODE_FALLBACK_ENABLED",
	}
	ConfigProto_Experimental_MlirBridgeRollout_value = map[string]int32{
		"MLIR_BRIDGE_ROLLOUT_UNSPECIFIED":                0,
		"MLIR_BRIDGE_ROLLOUT_ENABLED":                    1,
		"MLIR_BRIDGE_ROLLOUT_DISABLED":                   2,
		"MLIR_BRIDGE_ROLLOUT_SAFE_MODE_ENABLED":          3,
		"MLIR_BRIDGE_ROLLOUT_SAFE_MODE_FALLBACK_ENABLED": 4,
	}
)

Enum value maps for ConfigProto_Experimental_MlirBridgeRollout.

View Source 
var (
	RunOptions_TraceLevel_name = map[int32]string{
		0: "NO_TRACE",
		1: "SOFTWARE_TRACE",
		2: "HARDWARE_TRACE",
		3: "FULL_TRACE",
	}
	RunOptions_TraceLevel_value = map[string]int32{
		"NO_TRACE":       0,
		"SOFTWARE_TRACE": 1,
		"HARDWARE_TRACE": 2,
		"FULL_TRACE":     3,
	}
)

Enum value maps for RunOptions_TraceLevel.

View Source 
var (
	CoordinatedTaskState_name = map[int32]string{
		0: "TASKSTATE_UNSPECIFIED",
		1: "TASKSTATE_UNINITIALIZED",
		2: "TASKSTATE_DISCONNECTED",
		3: "TASKSTATE_CONNECTED",
		4: "TASKSTATE_ERROR",
	}
	CoordinatedTaskState_value = map[string]int32{
		"TASKSTATE_UNSPECIFIED":   0,
		"TASKSTATE_UNINITIALIZED": 1,
		"TASKSTATE_DISCONNECTED":  2,
		"TASKSTATE_CONNECTED":     3,
		"TASKSTATE_ERROR":         4,
	}
)

Enum value maps for CoordinatedTaskState.

View Source 
var (
	ErrorSourceProto_ErrorSource_name = map[int32]string{
		0: "UNKNOWN",
		1: "TPU_COMPILE_OP",
		2: "TF_XLA_BRIDGE",
		3: "MLIR_BRIDGE_PHASE_1",
		4: "MLIR_BRIDGE_PHASE_2",
		5: "EAGER_REMOTE_MGR",
	}
	ErrorSourceProto_ErrorSource_value = map[string]int32{
		"UNKNOWN":             0,
		"TPU_COMPILE_OP":      1,
		"TF_XLA_BRIDGE":       2,
		"MLIR_BRIDGE_PHASE_1": 3,
		"MLIR_BRIDGE_PHASE_2": 4,
		"EAGER_REMOTE_MGR":    5,
	}
)

Enum value maps for ErrorSourceProto_ErrorSource.

View Source 
var (
	DeploymentMode_name = map[int32]string{
		0: "DEPLOYMENT_MODE_UNSPECIFIED",
		1: "DEPLOYMENT_MODE_COLOCATED",
		2: "DEPLOYMENT_MODE_REMOTE",
		3: "DEPLOYMENT_MODE_HYBRID",
	}
	DeploymentMode_value = map[string]int32{
		"DEPLOYMENT_MODE_UNSPECIFIED": 0,
		"DEPLOYMENT_MODE_COLOCATED":   1,
		"DEPLOYMENT_MODE_REMOTE":      2,
		"DEPLOYMENT_MODE_HYBRID":      3,
	}
)

Enum value maps for DeploymentMode.

View Source 
var (
	ProcessingModeDef_ShardingPolicy_name = map[int32]string{
		0: "OFF",
		1: "DYNAMIC",
		2: "FILE",
		3: "DATA",
		4: "FILE_OR_DATA",
		5: "HINT",
	}
	ProcessingModeDef_ShardingPolicy_value = map[string]int32{
		"OFF":          0,
		"DYNAMIC":      1,
		"FILE":         2,
		"DATA":         3,
		"FILE_OR_DATA": 4,
		"HINT":         5,
	}
)

Enum value maps for ProcessingModeDef_ShardingPolicy.

View Source 
var (
	DataServiceMetadata_Compression_name = map[int32]string{
		0: "COMPRESSION_UNSPECIFIED",
		1: "COMPRESSION_OFF",
		2: "COMPRESSION_SNAPPY",
	}
	DataServiceMetadata_Compression_value = map[string]int32{
		"COMPRESSION_UNSPECIFIED": 0,
		"COMPRESSION_OFF":         1,
		"COMPRESSION_SNAPPY":      2,
	}
)

Enum value maps for DataServiceMetadata_Compression.

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 (
	TensorDebugMode_name = map[int32]string{
		0: "UNSPECIFIED",
		1: "NO_TENSOR",
		2: "CURT_HEALTH",
		3: "CONCISE_HEALTH",
		4: "FULL_HEALTH",
		5: "SHAPE",
		6: "FULL_NUMERICS",
		7: "FULL_TENSOR",
		8: "REDUCE_INF_NAN_THREE_SLOTS",
	}
	TensorDebugMode_value = map[string]int32{
		"UNSPECIFIED":                0,
		"NO_TENSOR":                  1,
		"CURT_HEALTH":                2,
		"CONCISE_HEALTH":             3,
		"FULL_HEALTH":                4,
		"SHAPE":                      5,
		"FULL_NUMERICS":              6,
		"FULL_TENSOR":                7,
		"REDUCE_INF_NAN_THREE_SLOTS": 8,
	}
)

Enum value maps for TensorDebugMode.

View Source 
var (
	DnnDataType_name = map[int32]string{
		0: "kFloat",
		1: "kDouble",
		2: "kHalf",
		3: "kInt8",
		4: "kInt32",
		5: "kComplexFloat",
		6: "kComplexDouble",
		7: "kBF16",
	}
	DnnDataType_value = map[string]int32{
		"kFloat":         0,
		"kDouble":        1,
		"kHalf":          2,
		"kInt8":          3,
		"kInt32":         4,
		"kComplexFloat":  5,
		"kComplexDouble": 6,
		"kBF16":          7,
	}
)

Enum value maps for DnnDataType.

View Source 
var (
	DataLayout_name = map[int32]string{
		0: "kYXDepthBatch",
		1: "kYXBatchDepth",
		2: "kBatchYXDepth",
		3: "kBatchDepthYX",
		4: "kBatchDepthYX4",
		5: "kBatchDepthYX32",
	}
	DataLayout_value = map[string]int32{
		"kYXDepthBatch":   0,
		"kYXBatchDepth":   1,
		"kBatchYXDepth":   2,
		"kBatchDepthYX":   3,
		"kBatchDepthYX4":  4,
		"kBatchDepthYX32": 5,
	}
)

Enum value maps for DataLayout.

View Source 
var (
	FilterLayout_name = map[int32]string{
		0: "kOutputInputYX",
		1: "kOutputYXInput",
		2: "kOutputInputYX4",
		5: "kOutputInputYX32",
		3: "kInputYXOutput",
		4: "kYXInputOutput",
	}
	FilterLayout_value = map[string]int32{
		"kOutputInputYX":   0,
		"kOutputYXInput":   1,
		"kOutputInputYX4":  2,
		"kOutputInputYX32": 5,
		"kInputYXOutput":   3,
		"kYXInputOutput":   4,
	}
)

Enum value maps for FilterLayout.

View Source 
var (
	ActivationMode_name = map[int32]string{
		0: "kNone",
		1: "kSigmoid",
		2: "kRelu",
		3: "kRelu6",
		4: "kReluX",
		5: "kTanh",
		6: "kBandPass",
	}
	ActivationMode_value = map[string]int32{
		"kNone":     0,
		"kSigmoid":  1,
		"kRelu":     2,
		"kRelu6":    3,
		"kReluX":    4,
		"kTanh":     5,
		"kBandPass": 6,
	}
)

Enum value maps for ActivationMode.

View Source 
var (
	ConvolutionMode_name = map[int32]string{
		0: "CROSS_CORRELATION",
		1: "CONVOLUTION",
	}
	ConvolutionMode_value = map[string]int32{
		"CROSS_CORRELATION": 0,
		"CONVOLUTION":       1,
	}
)

Enum value maps for ConvolutionMode.

View Source 
var (
	ConvolutionKind_name = map[int32]string{
		0: "INVALID",
		1: "FORWARD",
		2: "BACKWARD_FILTER",
		3: "BACKWARD_DATA",
		4: "FORWARD_BIAS_ACTIVATION",
	}
	ConvolutionKind_value = map[string]int32{
		"INVALID":                 0,
		"FORWARD":                 1,
		"BACKWARD_FILTER":         2,
		"BACKWARD_DATA":           3,
		"FORWARD_BIAS_ACTIVATION": 4,
	}
)

Enum value maps for ConvolutionKind.

View Source 
var (
	AlgorithmProto_MathType_name = map[int32]string{
		0: "DEFAULT_MATH",
		1: "TENSOR_OP_MATH",
	}
	AlgorithmProto_MathType_value = map[string]int32{
		"DEFAULT_MATH":   0,
		"TENSOR_OP_MATH": 1,
	}
)

Enum value maps for AlgorithmProto_MathType.

View Source 
var (
	Code_name = map[int32]string{
		0:  "OK",
		1:  "CANCELLED",
		2:  "UNKNOWN",
		3:  "INVALID_ARGUMENT",
		4:  "DEADLINE_EXCEEDED",
		5:  "NOT_FOUND",
		6:  "ALREADY_EXISTS",
		7:  "PERMISSION_DENIED",
		16: "UNAUTHENTICATED",
		8:  "RESOURCE_EXHAUSTED",
		9:  "FAILED_PRECONDITION",
		10: "ABORTED",
		11: "OUT_OF_RANGE",
		12: "UNIMPLEMENTED",
		13: "INTERNAL",
		14: "UNAVAILABLE",
		15: "DATA_LOSS",
		20: "DO_NOT_USE_RESERVED_FOR_FUTURE_EXPANSION_USE_DEFAULT_IN_SWITCH_INSTEAD_",
	}
	Code_value = map[string]int32{
		"OK":                  0,
		"CANCELLED":           1,
		"UNKNOWN":             2,
		"INVALID_ARGUMENT":    3,
		"DEADLINE_EXCEEDED":   4,
		"NOT_FOUND":           5,
		"ALREADY_EXISTS":      6,
		"PERMISSION_DENIED":   7,
		"UNAUTHENTICATED":     16,
		"RESOURCE_EXHAUSTED":  8,
		"FAILED_PRECONDITION": 9,
		"ABORTED":             10,
		"OUT_OF_RANGE":        11,
		"UNIMPLEMENTED":       12,
		"INTERNAL":            13,
		"UNAVAILABLE":         14,
		"DATA_LOSS":           15,
		"DO_NOT_USE_RESERVED_FOR_FUTURE_EXPANSION_USE_DEFAULT_IN_SWITCH_INSTEAD_": 20,
	}
)

Enum value maps for Code.

View Source 
var (
	WorkerHealth_name = map[int32]string{
		0: "OK",
		1: "RECEIVED_SHUTDOWN_SIGNAL",
		2: "INTERNAL_ERROR",
		3: "SHUTTING_DOWN",
	}
	WorkerHealth_value = map[string]int32{
		"OK":                       0,
		"RECEIVED_SHUTDOWN_SIGNAL": 1,
		"INTERNAL_ERROR":           2,
		"SHUTTING_DOWN":            3,
	}
)

Enum value maps for WorkerHealth.

View Source 
var (
	WorkerShutdownMode_name = map[int32]string{
		0: "DEFAULT",
		1: "NOT_CONFIGURED",
		2: "WAIT_FOR_COORDINATOR",
		3: "SHUTDOWN_AFTER_TIMEOUT",
	}
	WorkerShutdownMode_value = map[string]int32{
		"DEFAULT":                0,
		"NOT_CONFIGURED":         1,
		"WAIT_FOR_COORDINATOR":   2,
		"SHUTDOWN_AFTER_TIMEOUT": 3,
	}
)

Enum value maps for WorkerShutdownMode.

View Source 
var (
	LogMessage_Level_name = map[int32]string{
		0:  "UNKNOWN",
		10: "DEBUGGING",
		20: "INFO",
		30: "WARN",
		40: "ERROR",
		50: "FATAL",
	}
	LogMessage_Level_value = map[string]int32{
		"UNKNOWN":   0,
		"DEBUGGING": 10,
		"INFO":      20,
		"WARN":      30,
		"ERROR":     40,
		"FATAL":     50,
	}
)

Enum value maps for LogMessage_Level.

View Source 
var (
	SessionLog_SessionStatus_name = map[int32]string{
		0: "STATUS_UNSPECIFIED",
		1: "START",
		2: "STOP",
		3: "CHECKPOINT",
	}
	SessionLog_SessionStatus_value = map[string]int32{
		"STATUS_UNSPECIFIED": 0,
		"START":              1,
		"STOP":               2,
		"CHECKPOINT":         3,
	}
)

Enum value maps for SessionLog_SessionStatus.

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",
		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_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 (
	RewriterConfig_Toggle_name = map[int32]string{
		0: "DEFAULT",
		1: "ON",
		2: "OFF",
		3: "AGGRESSIVE",
		4: "EXPERIMENTAL_MLIR",
		5: "EXPERIMENTAL_BOTH",
	}
	RewriterConfig_Toggle_value = map[string]int32{
		"DEFAULT":           0,
		"ON":                1,
		"OFF":               2,
		"AGGRESSIVE":        3,
		"EXPERIMENTAL_MLIR": 4,
		"EXPERIMENTAL_BOTH": 5,
	}
)

Enum value maps for RewriterConfig_Toggle.

View Source 
var (
	RewriterConfig_CpuLayout_name = map[int32]string{
		0: "NO_CONVERSION_ON_CPU",
		1: "NCHW_TO_NHWC",
		2: "NHWC_TO_NCHW",
	}
	RewriterConfig_CpuLayout_value = map[string]int32{
		"NO_CONVERSION_ON_CPU": 0,
		"NCHW_TO_NHWC":         1,
		"NHWC_TO_NCHW":         2,
	}
)

Enum value maps for RewriterConfig_CpuLayout.

View Source 
var (
	RewriterConfig_NumIterationsType_name = map[int32]string{
		0: "DEFAULT_NUM_ITERS",
		1: "ONE",
		2: "TWO",
	}
	RewriterConfig_NumIterationsType_value = map[string]int32{
		"DEFAULT_NUM_ITERS": 0,
		"ONE":               1,
		"TWO":               2,
	}
)

Enum value maps for RewriterConfig_NumIterationsType.

View Source 
var (
	RewriterConfig_MemOptType_name = map[int32]string{
		0: "DEFAULT_MEM_OPT",
		1: "NO_MEM_OPT",
		2: "MANUAL",
		4: "SWAPPING_HEURISTICS",
		5: "RECOMPUTATION_HEURISTICS",
		6: "SCHEDULING_HEURISTICS",
		3: "HEURISTICS",
	}
	RewriterConfig_MemOptType_value = map[string]int32{
		"DEFAULT_MEM_OPT":          0,
		"NO_MEM_OPT":               1,
		"MANUAL":                   2,
		"SWAPPING_HEURISTICS":      4,
		"RECOMPUTATION_HEURISTICS": 5,
		"SCHEDULING_HEURISTICS":    6,
		"HEURISTICS":               3,
	}
)

Enum value maps for RewriterConfig_MemOptType.

View Source 
var (
	FunctionSpec_JitCompile_name = map[int32]string{
		0: "DEFAULT",
		1: "ON",
		2: "OFF",
	}
	FunctionSpec_JitCompile_value = map[string]int32{
		"DEFAULT": 0,
		"ON":      1,
		"OFF":     2,
	}
)

Enum value maps for FunctionSpec_JitCompile.

View Source 
var (
	SaverDef_CheckpointFormatVersion_name = map[int32]string{
		0: "LEGACY",
		1: "V1",
		2: "V2",
	}
	SaverDef_CheckpointFormatVersion_value = map[string]int32{
		"LEGACY": 0,
		"V1":     1,
		"V2":     2,
	}
)

Enum value maps for SaverDef_CheckpointFormatVersion.

View Source 
var (
	TypeSpecProto_TypeSpecClass_name = map[int32]string{
		0:  "UNKNOWN",
		1:  "SPARSE_TENSOR_SPEC",
		2:  "INDEXED_SLICES_SPEC",
		3:  "RAGGED_TENSOR_SPEC",
		4:  "TENSOR_ARRAY_SPEC",
		5:  "DATA_DATASET_SPEC",
		6:  "DATA_ITERATOR_SPEC",
		7:  "OPTIONAL_SPEC",
		8:  "PER_REPLICA_SPEC",
		9:  "VARIABLE_SPEC",
		10: "ROW_PARTITION_SPEC",
		12: "REGISTERED_TYPE_SPEC",
		13: "EXTENSION_TYPE_SPEC",
	}
	TypeSpecProto_TypeSpecClass_value = map[string]int32{
		"UNKNOWN":              0,
		"SPARSE_TENSOR_SPEC":   1,
		"INDEXED_SLICES_SPEC":  2,
		"RAGGED_TENSOR_SPEC":   3,
		"TENSOR_ARRAY_SPEC":    4,
		"DATA_DATASET_SPEC":    5,
		"DATA_ITERATOR_SPEC":   6,
		"OPTIONAL_SPEC":        7,
		"PER_REPLICA_SPEC":     8,
		"VARIABLE_SPEC":        9,
		"ROW_PARTITION_SPEC":   10,
		"REGISTERED_TYPE_SPEC": 12,
		"EXTENSION_TYPE_SPEC":  13,
	}
)

Enum value maps for TypeSpecProto_TypeSpecClass.

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 (
	BundleHeaderProto_Endianness_name = map[int32]string{
		0: "LITTLE",
		1: "BIG",
	}
	BundleHeaderProto_Endianness_value = map[string]int32{
		"LITTLE": 0,
		"BIG":    1,
	}
)

Enum value maps for BundleHeaderProto_Endianness.

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

Enum value maps for DataType.

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

Enum value maps for VariableSynchronization.

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

Enum value maps for VariableAggregation.

View Source 
var (
	VerifierConfig_Toggle_name = map[int32]string{
		0: "DEFAULT",
		1: "ON",
		2: "OFF",
	}
	VerifierConfig_Toggle_value = map[string]int32{
		"DEFAULT": 0,
		"ON":      1,
		"OFF":     2,
	}
)

Enum value maps for VerifierConfig_Toggle.

View Source 
var File_tensorflow_compiler_xla_pjrt_distributed_protocol_proto protoreflect.FileDescriptor
View Source 
var File_tensorflow_core_example_example_parser_configuration_proto protoreflect.FileDescriptor
View Source 
var File_tensorflow_core_example_example_proto protoreflect.FileDescriptor
View Source 
var File_tensorflow_core_example_feature_proto protoreflect.FileDescriptor
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_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_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
View Source 
var File_tensorflow_core_protobuf_autotuning_proto protoreflect.FileDescriptor
View Source 
var File_tensorflow_core_protobuf_bfc_memory_map_proto protoreflect.FileDescriptor
View Source 
var File_tensorflow_core_protobuf_cluster_proto protoreflect.FileDescriptor
View Source 
var File_tensorflow_core_protobuf_composite_tensor_variant_proto protoreflect.FileDescriptor
View Source 
var File_tensorflow_core_protobuf_config_proto protoreflect.FileDescriptor
View Source 
var File_tensorflow_core_protobuf_control_flow_proto protoreflect.FileDescriptor
View Source 
var File_tensorflow_core_protobuf_conv_autotuning_proto protoreflect.FileDescriptor
View Source 
var File_tensorflow_core_protobuf_coordination_config_proto protoreflect.FileDescriptor
View Source 
var File_tensorflow_core_protobuf_coordination_service_proto protoreflect.FileDescriptor
View Source 
var File_tensorflow_core_protobuf_core_platform_payloads_proto protoreflect.FileDescriptor
View Source 
var File_tensorflow_core_protobuf_critical_section_proto protoreflect.FileDescriptor
View Source 
var File_tensorflow_core_protobuf_data_service_proto protoreflect.FileDescriptor
View Source 
var File_tensorflow_core_protobuf_debug_event_proto protoreflect.FileDescriptor
View Source 
var File_tensorflow_core_protobuf_debug_proto protoreflect.FileDescriptor
View Source 
var File_tensorflow_core_protobuf_device_filters_proto protoreflect.FileDescriptor
View Source 
var File_tensorflow_core_protobuf_device_properties_proto protoreflect.FileDescriptor
View Source 
var File_tensorflow_core_protobuf_distributed_runtime_payloads_proto protoreflect.FileDescriptor
View Source 
var File_tensorflow_core_protobuf_eager_service_proto protoreflect.FileDescriptor
View Source 
var File_tensorflow_core_protobuf_error_codes_proto protoreflect.FileDescriptor
View Source 
var File_tensorflow_core_protobuf_fingerprint_proto protoreflect.FileDescriptor
View Source 
var File_tensorflow_core_protobuf_graph_debug_info_proto protoreflect.FileDescriptor
View Source 
var File_tensorflow_core_protobuf_master_proto protoreflect.FileDescriptor
View Source 
var File_tensorflow_core_protobuf_master_service_proto protoreflect.FileDescriptor
View Source 
var File_tensorflow_core_protobuf_meta_graph_proto protoreflect.FileDescriptor
View Source 
var File_tensorflow_core_protobuf_named_tensor_proto protoreflect.FileDescriptor
View Source 
var File_tensorflow_core_protobuf_queue_runner_proto protoreflect.FileDescriptor
View Source 
var File_tensorflow_core_protobuf_remote_tensor_handle_proto protoreflect.FileDescriptor
View Source 
var File_tensorflow_core_protobuf_replay_log_proto protoreflect.FileDescriptor
View Source 
var File_tensorflow_core_protobuf_rewriter_config_proto protoreflect.FileDescriptor
View Source 
var File_tensorflow_core_protobuf_saved_model_proto protoreflect.FileDescriptor
View Source 
var File_tensorflow_core_protobuf_saved_object_graph_proto protoreflect.FileDescriptor
View Source 
var File_tensorflow_core_protobuf_saver_proto protoreflect.FileDescriptor
View Source 
var File_tensorflow_core_protobuf_service_config_proto protoreflect.FileDescriptor
View Source 
var File_tensorflow_core_protobuf_snapshot_proto protoreflect.FileDescriptor
View Source 
var File_tensorflow_core_protobuf_status_proto protoreflect.FileDescriptor
View Source 
var File_tensorflow_core_protobuf_struct_proto protoreflect.FileDescriptor
View Source 
var File_tensorflow_core_protobuf_tensor_bundle_proto protoreflect.FileDescriptor
View Source 
var File_tensorflow_core_protobuf_tensorflow_server_proto protoreflect.FileDescriptor
View Source 
var File_tensorflow_core_protobuf_trackable_object_graph_proto protoreflect.FileDescriptor
View Source 
var File_tensorflow_core_protobuf_transport_options_proto protoreflect.FileDescriptor
View Source 
var File_tensorflow_core_protobuf_verifier_config_proto protoreflect.FileDescriptor
View Source 
var File_tensorflow_core_protobuf_worker_proto protoreflect.FileDescriptor
View Source 
var File_tensorflow_core_protobuf_worker_service_proto protoreflect.FileDescriptor
View Source 
var File_tensorflow_core_util_event_proto protoreflect.FileDescriptor
View Source 
var File_tensorflow_stream_executor_dnn_proto protoreflect.FileDescriptor

Functions

func MustMarshal 

func MustMarshal(m protoreflect.ProtoMessage) []byte

MustMarshal returns the wire-format encoding of m. It panics if this is not possible.

Example 
m := ConfigProto{
	AllowSoftPlacement:   true,
	OperationTimeoutInMs: 10000,
	GraphOptions: &GraphOptions{
		OptimizerOptions: &OptimizerOptions{
			OptLevel: OptimizerOptions_L1,
		},
	},
	Experimental: &ConfigProto_Experimental{
		OptimizeForStaticGraph: true,
		UseTfrt:                false,
	},
	GpuOptions: &GPUOptions{
		Experimental: &GPUOptions_Experimental{
			VirtualDevices: []*GPUOptions_Experimental_VirtualDevices{
				{MemoryLimitMb: []float32{2000}},
			},
		},
	},
}
b := MustMarshal(&m)
fmt.Printf("[]byte(%q)", b)

Output:

[]byte("2\nJ\b\n\x06\n\x04\x00\x00\xfaD8\x01R\x02\x1a\x00X\x90N\x82\x01\x02`\x01")

func MustUnmarshal 

func MustUnmarshal[T protoreflect.ProtoMessage](b []byte, m T) T

MustUnmarshal parses the wire-format message in b into m and also returns it. It panics if unmarshalling is not possible. To get e.g. a ConfigProto from its byte encoding use 

e.g. MustUnmarshal(b, &ConfigProto{})
Example 
b := []byte("8\x01R\x02\x1a\x00X\x90N\x82\x01\x00")
m := MustUnmarshal(b, &ConfigProto{})
s := strings.ReplaceAll(m.String(), "  ", " ") // work around spaces intentionally injected by protobuf stringer
fmt.Printf("%T:\n%v", m, s)

Output:

*pbs.ConfigProto:
allow_soft_placement:true graph_options:{optimizer_options:{}} operation_timeout_in_ms:10000 experimental:{}

Types

type ActivationMode 

type ActivationMode int32

Describes a kind of non-linearity (threshold-like mathematical function). 

const (
	ActivationMode_kNone    ActivationMode = 0
	ActivationMode_kSigmoid ActivationMode = 1
	// Rectified linear activation: f(x) = x < 0 ? 0 : x
	ActivationMode_kRelu ActivationMode = 2
	// Rectified linear activation; where upper maximum is 6.0.
	ActivationMode_kRelu6 ActivationMode = 3
	// Rectified linear activation; where upper maximum specified by
	// BatchDescriptor::value_max().
	ActivationMode_kReluX ActivationMode = 4
	ActivationMode_kTanh  ActivationMode = 5
	// Like ReluX; but passes all values in the range [-X,X].
	ActivationMode_kBandPass ActivationMode = 6
)

func (ActivationMode) Descriptor 

func (ActivationMode) Descriptor() protoreflect.EnumDescriptor

func (ActivationMode) Enum 

func (x ActivationMode) Enum() *ActivationMode

func (ActivationMode) EnumDescriptor deprecated 

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

Deprecated: Use ActivationMode.Descriptor instead.

func (ActivationMode) Number 

func (x ActivationMode) Number() protoreflect.EnumNumber

func (ActivationMode) String 

func (x ActivationMode) String() string

func (ActivationMode) Type 

func (ActivationMode) Type() protoreflect.EnumType

type AlgorithmConfigProto 

type AlgorithmConfigProto struct {

	// Use oneof to emulate optional semantics in proto2 since older
	// version of proto3 cannot distinguish "unset field" and "default field".
	//
	// Types that are assignable to OptionalAlgorithm:
	//	*AlgorithmConfigProto_Algorithm
	OptionalAlgorithm isAlgorithmConfigProto_OptionalAlgorithm `protobuf_oneof:"optional_algorithm"`
	// Types that are assignable to OptionalAlgorithmNoScratch:
	//	*AlgorithmConfigProto_AlgorithmNoScratch
	OptionalAlgorithmNoScratch isAlgorithmConfigProto_OptionalAlgorithmNoScratch `protobuf_oneof:"optional_algorithm_no_scratch"`
	// Types that are assignable to OptionalScratchSize:
	//	*AlgorithmConfigProto_ScratchSize
	OptionalScratchSize isAlgorithmConfigProto_OptionalScratchSize `protobuf_oneof:"optional_scratch_size"`
	// contains filtered or unexported fields
}

Proto definition of AlgorithmConfig in "dnn.h". TODO(ruochengw): After cl/380702564 is submitted, add support for algorithm configs with cuDNN Frontend APIs.

func (*AlgorithmConfigProto) Descriptor deprecated 

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

Deprecated: Use AlgorithmConfigProto.ProtoReflect.Descriptor instead.

func (*AlgorithmConfigProto) GetAlgorithm 

func (x *AlgorithmConfigProto) GetAlgorithm() *AlgorithmProto

func (*AlgorithmConfigProto) GetAlgorithmNoScratch 

func (x *AlgorithmConfigProto) GetAlgorithmNoScratch() *AlgorithmProto

func (*AlgorithmConfigProto) GetOptionalAlgorithm 

func (m *AlgorithmConfigProto) GetOptionalAlgorithm() isAlgorithmConfigProto_OptionalAlgorithm

func (*AlgorithmConfigProto) GetOptionalAlgorithmNoScratch 

func (m *AlgorithmConfigProto) GetOptionalAlgorithmNoScratch() isAlgorithmConfigProto_OptionalAlgorithmNoScratch

func (*AlgorithmConfigProto) GetOptionalScratchSize 

func (m *AlgorithmConfigProto) GetOptionalScratchSize() isAlgorithmConfigProto_OptionalScratchSize

func (*AlgorithmConfigProto) GetScratchSize 

func (x *AlgorithmConfigProto) GetScratchSize() int64

func (*AlgorithmConfigProto) ProtoMessage 

func (*AlgorithmConfigProto) ProtoMessage()

func (*AlgorithmConfigProto) ProtoReflect 

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

func (*AlgorithmConfigProto) Reset 

func (x *AlgorithmConfigProto) Reset()

func (*AlgorithmConfigProto) String 

func (x *AlgorithmConfigProto) String() string

type AlgorithmConfigProto_Algorithm 

type AlgorithmConfigProto_Algorithm struct {
	Algorithm *AlgorithmProto `protobuf:"bytes,1,opt,name=algorithm,proto3,oneof"`
}

type AlgorithmConfigProto_AlgorithmNoScratch 

type AlgorithmConfigProto_AlgorithmNoScratch struct {
	AlgorithmNoScratch *AlgorithmProto `protobuf:"bytes,2,opt,name=algorithm_no_scratch,json=algorithmNoScratch,proto3,oneof"`
}

type AlgorithmConfigProto_ScratchSize 

type AlgorithmConfigProto_ScratchSize struct {
	ScratchSize int64 `protobuf:"varint,3,opt,name=scratch_size,json=scratchSize,proto3,oneof"`
}

type AlgorithmProto 

type AlgorithmProto struct {
	AlgoId      int64                   `protobuf:"varint,1,opt,name=algo_id,json=algoId,proto3" json:"algo_id,omitempty"`
	MathType    AlgorithmProto_MathType `` /* 135-byte string literal not displayed */
	TuningKnobs map[int64]int64         `` /* 184-byte string literal not displayed */
	// Legacy algorithm enums and cuDNN Frontend engine numbers need to coexist in
	// the same proto medium-term, until we can be confident of no longer needing
	// the legacy cuDNN convolution API.  Once the migration is complete, we can
	// stop producing legacy algorithm enums and remove this field.
	IsCudnnFrontend bool `protobuf:"varint,5,opt,name=is_cudnn_frontend,json=isCudnnFrontend,proto3" json:"is_cudnn_frontend,omitempty"`
	// For ROCm only, it's impossible to re-query the required workspace size
	// after running the algorithm search, so we must store the workspace size
	// along with the choice of algorithm.  For consistency and convenience,
	// cuDNN uses this field in the same way, even though it would be possible to
	// re-query the workspace size from cuDNN at each use.
	//
	// Since this message is persisted in files, we need to be able to distinguish
	// 0 workspace size from unknown workspace size in an old message, so this is
	// a message field.
	WorkspaceSize *wrappers.UInt64Value `protobuf:"bytes,6,opt,name=workspace_size,json=workspaceSize,proto3" json:"workspace_size,omitempty"`
	// contains filtered or unexported fields
}

Generic algorithm representation.

func (*AlgorithmProto) Descriptor deprecated 

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

Deprecated: Use AlgorithmProto.ProtoReflect.Descriptor instead.

func (*AlgorithmProto) GetAlgoId 

func (x *AlgorithmProto) GetAlgoId() int64

func (*AlgorithmProto) GetIsCudnnFrontend 

func (x *AlgorithmProto) GetIsCudnnFrontend() bool

func (*AlgorithmProto) GetMathType 

func (x *AlgorithmProto) GetMathType() AlgorithmProto_MathType

func (*AlgorithmProto) GetTuningKnobs 

func (x *AlgorithmProto) GetTuningKnobs() map[int64]int64

func (*AlgorithmProto) GetWorkspaceSize 

func (x *AlgorithmProto) GetWorkspaceSize() *wrappers.UInt64Value

func (*AlgorithmProto) ProtoMessage 

func (*AlgorithmProto) ProtoMessage()

func (*AlgorithmProto) ProtoReflect 

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

func (*AlgorithmProto) Reset 

func (x *AlgorithmProto) Reset()

func (*AlgorithmProto) String 

func (x *AlgorithmProto) String() string

type AlgorithmProto_MathType 

type AlgorithmProto_MathType int32
const (
	AlgorithmProto_DEFAULT_MATH AlgorithmProto_MathType = 0
	// The GPU may operate 4x4 matrix FMA.
	// See cuDNN's documentation for CUDNN_TENSOR_OP_MATH.
	AlgorithmProto_TENSOR_OP_MATH AlgorithmProto_MathType = 1
)

func (AlgorithmProto_MathType) Descriptor 

func (AlgorithmProto_MathType) Descriptor() protoreflect.EnumDescriptor

func (AlgorithmProto_MathType) Enum 

func (x AlgorithmProto_MathType) Enum() *AlgorithmProto_MathType

func (AlgorithmProto_MathType) EnumDescriptor deprecated 

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

Deprecated: Use AlgorithmProto_MathType.Descriptor instead.

func (AlgorithmProto_MathType) Number 

func (x AlgorithmProto_MathType) Number() protoreflect.EnumNumber

func (AlgorithmProto_MathType) String 

func (x AlgorithmProto_MathType) String() string

func (AlgorithmProto_MathType) Type 

func (AlgorithmProto_MathType) Type() protoreflect.EnumType

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 AssetFileDef 

type AssetFileDef struct {

	// The tensor to bind the asset filename to.
	TensorInfo *TensorInfo `protobuf:"bytes,1,opt,name=tensor_info,json=tensorInfo,proto3" json:"tensor_info,omitempty"`
	// The filename within an assets directory. Note: does not include the path
	// prefix, i.e. directories. For an asset at /tmp/path/vocab.txt, the filename
	// would be "vocab.txt".
	Filename string `protobuf:"bytes,2,opt,name=filename,proto3" json:"filename,omitempty"`
	// contains filtered or unexported fields
}

An asset file def for a single file or a set of sharded files with the same name.

func (*AssetFileDef) Descriptor deprecated 

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

Deprecated: Use AssetFileDef.ProtoReflect.Descriptor instead.

func (*AssetFileDef) GetFilename 

func (x *AssetFileDef) GetFilename() string

func (*AssetFileDef) GetTensorInfo 

func (x *AssetFileDef) GetTensorInfo() *TensorInfo

func (*AssetFileDef) ProtoMessage 

func (*AssetFileDef) ProtoMessage()

func (*AssetFileDef) ProtoReflect 

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

func (*AssetFileDef) Reset 

func (x *AssetFileDef) Reset()

func (*AssetFileDef) String 

func (x *AssetFileDef) 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 AutoParallelOptions 

type AutoParallelOptions struct {
	Enable      bool  `protobuf:"varint,1,opt,name=enable,proto3" json:"enable,omitempty"`
	NumReplicas int32 `protobuf:"varint,2,opt,name=num_replicas,json=numReplicas,proto3" json:"num_replicas,omitempty"`
	// contains filtered or unexported fields
}

func (*AutoParallelOptions) Descriptor deprecated 

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

Deprecated: Use AutoParallelOptions.ProtoReflect.Descriptor instead.

func (*AutoParallelOptions) GetEnable 

func (x *AutoParallelOptions) GetEnable() bool

func (*AutoParallelOptions) GetNumReplicas 

func (x *AutoParallelOptions) GetNumReplicas() int32

func (*AutoParallelOptions) ProtoMessage 

func (*AutoParallelOptions) ProtoMessage()

func (*AutoParallelOptions) ProtoReflect 

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

func (*AutoParallelOptions) Reset 

func (x *AutoParallelOptions) Reset()

func (*AutoParallelOptions) String 

func (x *AutoParallelOptions) String() string

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 AutotuneResult 

type AutotuneResult struct {
	ScratchBytes int64                         `protobuf:"varint,8,opt,name=scratch_bytes,json=scratchBytes,proto3" json:"scratch_bytes,omitempty"`
	RunTime      *duration.Duration            `protobuf:"bytes,9,opt,name=run_time,json=runTime,proto3" json:"run_time,omitempty"`
	Failure      *AutotuneResult_FailureResult `protobuf:"bytes,7,opt,name=failure,proto3" json:"failure,omitempty"`
	// Types that are assignable to Key:
	//	*AutotuneResult_Conv
	//	*AutotuneResult_Gemm
	//	*AutotuneResult_CudaConvPlan
	//	*AutotuneResult_Algorithm
	Key isAutotuneResult_Key `protobuf_oneof:"key"`
	// contains filtered or unexported fields
}

func (*AutotuneResult) Descriptor deprecated 

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

Deprecated: Use AutotuneResult.ProtoReflect.Descriptor instead.

func (*AutotuneResult) GetAlgorithm 

func (x *AutotuneResult) GetAlgorithm() *AlgorithmProto

func (*AutotuneResult) GetConv 

func (x *AutotuneResult) GetConv() *AutotuneResult_ConvKey

func (*AutotuneResult) GetCudaConvPlan 

func (x *AutotuneResult) GetCudaConvPlan() *AutotuneResult_CudaConvPlanKey

func (*AutotuneResult) GetFailure 

func (x *AutotuneResult) GetFailure() *AutotuneResult_FailureResult

func (*AutotuneResult) GetGemm 

func (x *AutotuneResult) GetGemm() *AutotuneResult_GemmKey

func (*AutotuneResult) GetKey 

func (m *AutotuneResult) GetKey() isAutotuneResult_Key

func (*AutotuneResult) GetRunTime 

func (x *AutotuneResult) GetRunTime() *duration.Duration

func (*AutotuneResult) GetScratchBytes 

func (x *AutotuneResult) GetScratchBytes() int64

func (*AutotuneResult) ProtoMessage 

func (*AutotuneResult) ProtoMessage()

func (*AutotuneResult) ProtoReflect 

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

func (*AutotuneResult) Reset 

func (x *AutotuneResult) Reset()

func (*AutotuneResult) String 

func (x *AutotuneResult) String() string

type AutotuneResult_Algorithm 

type AutotuneResult_Algorithm struct {
	Algorithm *AlgorithmProto `protobuf:"bytes,16,opt,name=algorithm,proto3,oneof"`
}

type AutotuneResult_Conv 

type AutotuneResult_Conv struct {
	Conv *AutotuneResult_ConvKey `protobuf:"bytes,5,opt,name=conv,proto3,oneof"`
}

type AutotuneResult_ConvKey 

type AutotuneResult_ConvKey struct {
	Algorithm        int64 `protobuf:"varint,1,opt,name=algorithm,proto3" json:"algorithm,omitempty"`
	TensorOpsEnabled bool  `protobuf:"varint,2,opt,name=tensor_ops_enabled,json=tensorOpsEnabled,proto3" json:"tensor_ops_enabled,omitempty"`
	// contains filtered or unexported fields
}

Legacy and unused in new data; superseded by AlgorithmProto.

func (*AutotuneResult_ConvKey) Descriptor deprecated 

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

Deprecated: Use AutotuneResult_ConvKey.ProtoReflect.Descriptor instead.

func (*AutotuneResult_ConvKey) GetAlgorithm 

func (x *AutotuneResult_ConvKey) GetAlgorithm() int64

func (*AutotuneResult_ConvKey) GetTensorOpsEnabled 

func (x *AutotuneResult_ConvKey) GetTensorOpsEnabled() bool

func (*AutotuneResult_ConvKey) ProtoMessage 

func (*AutotuneResult_ConvKey) ProtoMessage()

func (*AutotuneResult_ConvKey) ProtoReflect 

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

func (*AutotuneResult_ConvKey) Reset 

func (x *AutotuneResult_ConvKey) Reset()

func (*AutotuneResult_ConvKey) String 

func (x *AutotuneResult_ConvKey) String() string

type AutotuneResult_CudaConvPlan 

type AutotuneResult_CudaConvPlan struct {
	CudaConvPlan *AutotuneResult_CudaConvPlanKey `protobuf:"bytes,15,opt,name=cuda_conv_plan,json=cudaConvPlan,proto3,oneof"`
}

type AutotuneResult_CudaConvPlanKey 

type AutotuneResult_CudaConvPlanKey struct {
	ExecPlanId string `protobuf:"bytes,1,opt,name=exec_plan_id,json=execPlanId,proto3" json:"exec_plan_id,omitempty"`
	// contains filtered or unexported fields
}

Legacy and unused in new data; superseded by AlgorithmProto.

func (*AutotuneResult_CudaConvPlanKey) Descriptor deprecated 

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

Deprecated: Use AutotuneResult_CudaConvPlanKey.ProtoReflect.Descriptor instead.

func (*AutotuneResult_CudaConvPlanKey) GetExecPlanId 

func (x *AutotuneResult_CudaConvPlanKey) GetExecPlanId() string

func (*AutotuneResult_CudaConvPlanKey) ProtoMessage 

func (*AutotuneResult_CudaConvPlanKey) ProtoMessage()

func (*AutotuneResult_CudaConvPlanKey) ProtoReflect 

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

func (*AutotuneResult_CudaConvPlanKey) Reset 

func (x *AutotuneResult_CudaConvPlanKey) Reset()

func (*AutotuneResult_CudaConvPlanKey) String 

func (x *AutotuneResult_CudaConvPlanKey) String() string

type AutotuneResult_FailureKind 

type AutotuneResult_FailureKind int32
const (
	AutotuneResult_UNKNOWN AutotuneResult_FailureKind = 0
	// Algorithm wrote memory outside its output buffers.
	AutotuneResult_REDZONE_MODIFIED AutotuneResult_FailureKind = 1
	// Algorithm gave a different result from a reference algorithm.
	AutotuneResult_WRONG_RESULT AutotuneResult_FailureKind = 2
	// Algorithm was rejected for failing to run or for known bugs.
	AutotuneResult_DISQUALIFIED AutotuneResult_FailureKind = 3
)

func (AutotuneResult_FailureKind) Descriptor 

func (AutotuneResult_FailureKind) Descriptor() protoreflect.EnumDescriptor

func (AutotuneResult_FailureKind) Enum 

func (x AutotuneResult_FailureKind) Enum() *AutotuneResult_FailureKind

func (AutotuneResult_FailureKind) EnumDescriptor deprecated 

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

Deprecated: Use AutotuneResult_FailureKind.Descriptor instead.

func (AutotuneResult_FailureKind) Number 

func (x AutotuneResult_FailureKind) Number() protoreflect.EnumNumber

func (AutotuneResult_FailureKind) String 

func (x AutotuneResult_FailureKind) String() string

func (AutotuneResult_FailureKind) Type 

func (AutotuneResult_FailureKind) Type() protoreflect.EnumType

type AutotuneResult_FailureResult 

type AutotuneResult_FailureResult struct {
	Kind AutotuneResult_FailureKind `protobuf:"varint,1,opt,name=kind,proto3,enum=tensorflow.AutotuneResult_FailureKind" json:"kind,omitempty"`
	Msg  string                     `protobuf:"bytes,2,opt,name=msg,proto3" json:"msg,omitempty"`
	// For failure_kind == WRONG_RESULT, this field indicates the reference
	// configuration that we compared against.
	//
	// Note that the reference algorithm isn't always correct.  However,
	// empirically it's more correct, as it's "algo 0", less fancy than the
	// compared one.
	//
	// Types that are assignable to Key:
	//	*AutotuneResult_FailureResult_ReferenceConv
	//	*AutotuneResult_FailureResult_ReferenceGemm
	//	*AutotuneResult_FailureResult_ReferenceCudaConvPlan
	//	*AutotuneResult_FailureResult_ReferenceAlgorithm
	Key           isAutotuneResult_FailureResult_Key `protobuf_oneof:"key"`
	BufferAddress int64                              `protobuf:"varint,13,opt,name=buffer_address,json=bufferAddress,proto3" json:"buffer_address,omitempty"`
	// contains filtered or unexported fields
}

func (*AutotuneResult_FailureResult) Descriptor deprecated 

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

Deprecated: Use AutotuneResult_FailureResult.ProtoReflect.Descriptor instead.

func (*AutotuneResult_FailureResult) GetBufferAddress 

func (x *AutotuneResult_FailureResult) GetBufferAddress() int64

func (*AutotuneResult_FailureResult) GetKey 

func (m *AutotuneResult_FailureResult) GetKey() isAutotuneResult_FailureResult_Key

func (*AutotuneResult_FailureResult) GetKind 

func (x *AutotuneResult_FailureResult) GetKind() AutotuneResult_FailureKind

func (*AutotuneResult_FailureResult) GetMsg 

func (x *AutotuneResult_FailureResult) GetMsg() string

func (*AutotuneResult_FailureResult) GetReferenceAlgorithm 

func (x *AutotuneResult_FailureResult) GetReferenceAlgorithm() *AlgorithmProto

func (*AutotuneResult_FailureResult) GetReferenceConv 

func (x *AutotuneResult_FailureResult) GetReferenceConv() *AutotuneResult_ConvKey

func (*AutotuneResult_FailureResult) GetReferenceCudaConvPlan 

func (x *AutotuneResult_FailureResult) GetReferenceCudaConvPlan() *AutotuneResult_CudaConvPlanKey

func (*AutotuneResult_FailureResult) GetReferenceGemm 

func (x *AutotuneResult_FailureResult) GetReferenceGemm() *AutotuneResult_GemmKey

func (*AutotuneResult_FailureResult) ProtoMessage 

func (*AutotuneResult_FailureResult) ProtoMessage()

func (*AutotuneResult_FailureResult) ProtoReflect 

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

func (*AutotuneResult_FailureResult) Reset 

func (x *AutotuneResult_FailureResult) Reset()

func (*AutotuneResult_FailureResult) String 

func (x *AutotuneResult_FailureResult) String() string

type AutotuneResult_FailureResult_ReferenceAlgorithm 

type AutotuneResult_FailureResult_ReferenceAlgorithm struct {
	ReferenceAlgorithm *AlgorithmProto `protobuf:"bytes,15,opt,name=reference_algorithm,json=referenceAlgorithm,proto3,oneof"`
}

type AutotuneResult_FailureResult_ReferenceConv 

type AutotuneResult_FailureResult_ReferenceConv struct {
	ReferenceConv *AutotuneResult_ConvKey `protobuf:"bytes,11,opt,name=reference_conv,json=referenceConv,proto3,oneof"`
}

type AutotuneResult_FailureResult_ReferenceCudaConvPlan 

type AutotuneResult_FailureResult_ReferenceCudaConvPlan struct {
	ReferenceCudaConvPlan *AutotuneResult_CudaConvPlanKey `protobuf:"bytes,14,opt,name=reference_cuda_conv_plan,json=referenceCudaConvPlan,proto3,oneof"`
}

type AutotuneResult_FailureResult_ReferenceGemm 

type AutotuneResult_FailureResult_ReferenceGemm struct {
	ReferenceGemm *AutotuneResult_GemmKey `protobuf:"bytes,12,opt,name=reference_gemm,json=referenceGemm,proto3,oneof"`
}

type AutotuneResult_Gemm 

type AutotuneResult_Gemm struct {
	Gemm *AutotuneResult_GemmKey `protobuf:"bytes,6,opt,name=gemm,proto3,oneof"`
}

type AutotuneResult_GemmKey 

type AutotuneResult_GemmKey struct {
	Algorithm int64 `protobuf:"varint,1,opt,name=algorithm,proto3" json:"algorithm,omitempty"`
	// contains filtered or unexported fields
}

func (*AutotuneResult_GemmKey) Descriptor deprecated 

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

Deprecated: Use AutotuneResult_GemmKey.ProtoReflect.Descriptor instead.

func (*AutotuneResult_GemmKey) GetAlgorithm 

func (x *AutotuneResult_GemmKey) GetAlgorithm() int64

func (*AutotuneResult_GemmKey) ProtoMessage 

func (*AutotuneResult_GemmKey) ProtoMessage()

func (*AutotuneResult_GemmKey) ProtoReflect 

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

func (*AutotuneResult_GemmKey) Reset 

func (x *AutotuneResult_GemmKey) Reset()

func (*AutotuneResult_GemmKey) String 

func (x *AutotuneResult_GemmKey) String() string

type AutotuningLog 

type AutotuningLog struct {
	Instr *any.Any `protobuf:"bytes,1,opt,name=instr,proto3" json:"instr,omitempty"`
	// Records all auto-tuning results per algorithm.
	Results           []*AutotuneResult  `protobuf:"bytes,2,rep,name=results,proto3" json:"results,omitempty"`
	CudnnVersion      *CudnnVersion      `protobuf:"bytes,3,opt,name=cudnn_version,json=cudnnVersion,proto3" json:"cudnn_version,omitempty"`
	ComputeCapability *ComputeCapability `protobuf:"bytes,4,opt,name=compute_capability,json=computeCapability,proto3" json:"compute_capability,omitempty"`
	// stream_executor::DeviceDescription::pci_bus_id.
	DevicePciBusId string `protobuf:"bytes,5,opt,name=device_pci_bus_id,json=devicePciBusId,proto3" json:"device_pci_bus_id,omitempty"`
	BlasVersion    string `protobuf:"bytes,6,opt,name=blas_version,json=blasVersion,proto3" json:"blas_version,omitempty"`
	// contains filtered or unexported fields
}

func (*AutotuningLog) Descriptor deprecated 

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

Deprecated: Use AutotuningLog.ProtoReflect.Descriptor instead.

func (*AutotuningLog) GetBlasVersion 

func (x *AutotuningLog) GetBlasVersion() string

func (*AutotuningLog) GetComputeCapability 

func (x *AutotuningLog) GetComputeCapability() *ComputeCapability

func (*AutotuningLog) GetCudnnVersion 

func (x *AutotuningLog) GetCudnnVersion() *CudnnVersion

func (*AutotuningLog) GetDevicePciBusId 

func (x *AutotuningLog) GetDevicePciBusId() string

func (*AutotuningLog) GetInstr 

func (x *AutotuningLog) GetInstr() *any.Any

func (*AutotuningLog) GetResults 

func (x *AutotuningLog) GetResults() []*AutotuneResult

func (*AutotuningLog) ProtoMessage 

func (*AutotuningLog) ProtoMessage()

func (*AutotuningLog) ProtoReflect 

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

func (*AutotuningLog) Reset 

func (x *AutotuningLog) Reset()

func (*AutotuningLog) String 

func (x *AutotuningLog) String() string

type BarrierRequest 

type BarrierRequest struct {
	BarrierId          string `protobuf:"bytes,1,opt,name=barrier_id,json=barrierId,proto3" json:"barrier_id,omitempty"`
	BarrierTimeoutInMs int64  `protobuf:"varint,2,opt,name=barrier_timeout_in_ms,json=barrierTimeoutInMs,proto3" json:"barrier_timeout_in_ms,omitempty"`
	// Denotes list of tasks that will wait for the barrier. If unspecified, it
	// implies that the entire cluster is participating in the barrier.
	Tasks []*CoordinatedTask `protobuf:"bytes,3,rep,name=tasks,proto3" json:"tasks,omitempty"`
	// Task that is making the request.
	SourceTask *CoordinatedTask `protobuf:"bytes,4,opt,name=source_task,json=sourceTask,proto3" json:"source_task,omitempty"`
	// contains filtered or unexported fields
}

Request and response messages for generic sync barriers.

func (*BarrierRequest) Descriptor deprecated 

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

Deprecated: Use BarrierRequest.ProtoReflect.Descriptor instead.

func (*BarrierRequest) GetBarrierId 

func (x *BarrierRequest) GetBarrierId() string

func (*BarrierRequest) GetBarrierTimeoutInMs 

func (x *BarrierRequest) GetBarrierTimeoutInMs() int64

func (*BarrierRequest) GetSourceTask 

func (x *BarrierRequest) GetSourceTask() *CoordinatedTask

func (*BarrierRequest) GetTasks 

func (x *BarrierRequest) GetTasks() []*CoordinatedTask

func (*BarrierRequest) ProtoMessage 

func (*BarrierRequest) ProtoMessage()

func (*BarrierRequest) ProtoReflect 

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

func (*BarrierRequest) Reset 

func (x *BarrierRequest) Reset()

func (*BarrierRequest) String 

func (x *BarrierRequest) String() string

type BarrierResponse 

type BarrierResponse struct {
	// contains filtered or unexported fields
}

func (*BarrierResponse) Descriptor deprecated 

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

Deprecated: Use BarrierResponse.ProtoReflect.Descriptor instead.

func (*BarrierResponse) ProtoMessage 

func (*BarrierResponse) ProtoMessage()

func (*BarrierResponse) ProtoReflect 

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

func (*BarrierResponse) Reset 

func (x *BarrierResponse) Reset()

func (*BarrierResponse) String 

func (x *BarrierResponse) String() string

type BinSummary 

type BinSummary struct {
	Bin              int32 `protobuf:"varint,1,opt,name=bin,proto3" json:"bin,omitempty"`
	TotalBytesInUse  int64 `protobuf:"varint,2,opt,name=total_bytes_in_use,json=totalBytesInUse,proto3" json:"total_bytes_in_use,omitempty"`
	TotalBytesInBin  int64 `protobuf:"varint,3,opt,name=total_bytes_in_bin,json=totalBytesInBin,proto3" json:"total_bytes_in_bin,omitempty"`
	TotalChunksInUse int64 `protobuf:"varint,4,opt,name=total_chunks_in_use,json=totalChunksInUse,proto3" json:"total_chunks_in_use,omitempty"`
	TotalChunksInBin int64 `protobuf:"varint,5,opt,name=total_chunks_in_bin,json=totalChunksInBin,proto3" json:"total_chunks_in_bin,omitempty"`
	// contains filtered or unexported fields
}

func (*BinSummary) Descriptor deprecated 

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

Deprecated: Use BinSummary.ProtoReflect.Descriptor instead.

func (*BinSummary) GetBin 

func (x *BinSummary) GetBin() int32

func (*BinSummary) GetTotalBytesInBin 

func (x *BinSummary) GetTotalBytesInBin() int64

func (*BinSummary) GetTotalBytesInUse 

func (x *BinSummary) GetTotalBytesInUse() int64

func (*BinSummary) GetTotalChunksInBin 

func (x *BinSummary) GetTotalChunksInBin() int64

func (*BinSummary) GetTotalChunksInUse 

func (x *BinSummary) GetTotalChunksInUse() int64

func (*BinSummary) ProtoMessage 

func (*BinSummary) ProtoMessage()

func (*BinSummary) ProtoReflect 

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

func (*BinSummary) Reset 

func (x *BinSummary) Reset()

func (*BinSummary) String 

func (x *BinSummary) String() string

type BoundedTensorSpecProto 

type BoundedTensorSpecProto struct {
	Name    string            `protobuf:"bytes,1,opt,name=name,proto3" json:"name,omitempty"`
	Shape   *TensorShapeProto `protobuf:"bytes,2,opt,name=shape,proto3" json:"shape,omitempty"`
	Dtype   DataType          `protobuf:"varint,3,opt,name=dtype,proto3,enum=tensorflow.DataType" json:"dtype,omitempty"`
	Minimum *TensorProto      `protobuf:"bytes,4,opt,name=minimum,proto3" json:"minimum,omitempty"`
	Maximum *TensorProto      `protobuf:"bytes,5,opt,name=maximum,proto3" json:"maximum,omitempty"`
	// contains filtered or unexported fields
}

A protobuf to represent tf.BoundedTensorSpec.

func (*BoundedTensorSpecProto) Descriptor deprecated 

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

Deprecated: Use BoundedTensorSpecProto.ProtoReflect.Descriptor instead.

func (*BoundedTensorSpecProto) GetDtype 

func (x *BoundedTensorSpecProto) GetDtype() DataType

func (*BoundedTensorSpecProto) GetMaximum 

func (x *BoundedTensorSpecProto) GetMaximum() *TensorProto

func (*BoundedTensorSpecProto) GetMinimum 

func (x *BoundedTensorSpecProto) GetMinimum() *TensorProto

func (*BoundedTensorSpecProto) GetName 

func (x *BoundedTensorSpecProto) GetName() string

func (*BoundedTensorSpecProto) GetShape 

func (x *BoundedTensorSpecProto) GetShape() *TensorShapeProto

func (*BoundedTensorSpecProto) ProtoMessage 

func (*BoundedTensorSpecProto) ProtoMessage()

func (*BoundedTensorSpecProto) ProtoReflect 

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

func (*BoundedTensorSpecProto) Reset 

func (x *BoundedTensorSpecProto) Reset()

func (*BoundedTensorSpecProto) String 

func (x *BoundedTensorSpecProto) String() string

type BundleEntryProto 

type BundleEntryProto struct {

	// The tensor dtype and shape.
	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"`
	// The binary content of the tensor lies in:
	//   File "shard_id": bytes [offset, offset + size).
	ShardId int32 `protobuf:"varint,3,opt,name=shard_id,json=shardId,proto3" json:"shard_id,omitempty"`
	Offset  int64 `protobuf:"varint,4,opt,name=offset,proto3" json:"offset,omitempty"`
	Size    int64 `protobuf:"varint,5,opt,name=size,proto3" json:"size,omitempty"`
	// The CRC32C checksum of the tensor bytes.
	Crc32C uint32 `protobuf:"fixed32,6,opt,name=crc32c,proto3" json:"crc32c,omitempty"`
	// Iff present, this entry represents a partitioned tensor.  The previous
	// fields are interpreted as follows:
	//
	//   "dtype", "shape": describe the full tensor.
	//   "shard_id", "offset", "size", "crc32c": all IGNORED.
	//      These information for each slice can be looked up in their own
	//      BundleEntryProto, keyed by each "slice_name".
	Slices []*TensorSliceProto `protobuf:"bytes,7,rep,name=slices,proto3" json:"slices,omitempty"`
	// contains filtered or unexported fields
}

Describes the metadata related to a checkpointed tensor.

func (*BundleEntryProto) Descriptor deprecated 

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

Deprecated: Use BundleEntryProto.ProtoReflect.Descriptor instead.

func (*BundleEntryProto) GetCrc32C 

func (x *BundleEntryProto) GetCrc32C() uint32

func (*BundleEntryProto) GetDtype 

func (x *BundleEntryProto) GetDtype() DataType

func (*BundleEntryProto) GetOffset 

func (x *BundleEntryProto) GetOffset() int64

func (*BundleEntryProto) GetShape 

func (x *BundleEntryProto) GetShape() *TensorShapeProto

func (*BundleEntryProto) GetShardId 

func (x *BundleEntryProto) GetShardId() int32

func (*BundleEntryProto) GetSize 

func (x *BundleEntryProto) GetSize() int64

func (*BundleEntryProto) GetSlices 

func (x *BundleEntryProto) GetSlices() []*TensorSliceProto

func (*BundleEntryProto) ProtoMessage 

func (*BundleEntryProto) ProtoMessage()

func (*BundleEntryProto) ProtoReflect 

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

func (*BundleEntryProto) Reset 

func (x *BundleEntryProto) Reset()

func (*BundleEntryProto) String 

func (x *BundleEntryProto) String() string

type BundleHeaderProto 

type BundleHeaderProto struct {

	// Number of data files in the bundle.
	NumShards  int32                        `protobuf:"varint,1,opt,name=num_shards,json=numShards,proto3" json:"num_shards,omitempty"`
	Endianness BundleHeaderProto_Endianness `protobuf:"varint,2,opt,name=endianness,proto3,enum=tensorflow.BundleHeaderProto_Endianness" json:"endianness,omitempty"`
	// Versioning of the tensor bundle format.
	Version *VersionDef `protobuf:"bytes,3,opt,name=version,proto3" json:"version,omitempty"`
	// contains filtered or unexported fields
}

Special header that is associated with a bundle.

TODO(zongheng,zhifengc): maybe in the future, we can add information about which binary produced this checkpoint, timestamp, etc. Sometime, these can be valuable debugging information. And if needed, these can be used as defensive information ensuring reader (binary version) of the checkpoint and the writer (binary version) must match within certain range, etc.

func (*BundleHeaderProto) Descriptor deprecated 

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

Deprecated: Use BundleHeaderProto.ProtoReflect.Descriptor instead.

func (*BundleHeaderProto) GetEndianness 

func (x *BundleHeaderProto) GetEndianness() BundleHeaderProto_Endianness

func (*BundleHeaderProto) GetNumShards 

func (x *BundleHeaderProto) GetNumShards() int32

func (*BundleHeaderProto) GetVersion 

func (x *BundleHeaderProto) GetVersion() *VersionDef

func (*BundleHeaderProto) ProtoMessage 

func (*BundleHeaderProto) ProtoMessage()

func (*BundleHeaderProto) ProtoReflect 

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

func (*BundleHeaderProto) Reset 

func (x *BundleHeaderProto) Reset()

func (*BundleHeaderProto) String 

func (x *BundleHeaderProto) String() string

type BundleHeaderProto_Endianness 

type BundleHeaderProto_Endianness int32

An enum indicating the endianness of the platform that produced this bundle. A bundle can only be read by a platform with matching endianness. Defaults to LITTLE, as most modern platforms are little-endian.

Affects the binary tensor data bytes only, not the metadata in protobufs. 

const (
	BundleHeaderProto_LITTLE BundleHeaderProto_Endianness = 0
	BundleHeaderProto_BIG    BundleHeaderProto_Endianness = 1
)

func (BundleHeaderProto_Endianness) Descriptor 

func (BundleHeaderProto_Endianness) Descriptor() protoreflect.EnumDescriptor

func (BundleHeaderProto_Endianness) Enum 

func (x BundleHeaderProto_Endianness) Enum() *BundleHeaderProto_Endianness

func (BundleHeaderProto_Endianness) EnumDescriptor deprecated 

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

Deprecated: Use BundleHeaderProto_Endianness.Descriptor instead.

func (BundleHeaderProto_Endianness) Number 

func (x BundleHeaderProto_Endianness) Number() protoreflect.EnumNumber

func (BundleHeaderProto_Endianness) String 

func (x BundleHeaderProto_Endianness) String() string

func (BundleHeaderProto_Endianness) Type 

func (BundleHeaderProto_Endianness) Type() protoreflect.EnumType

type BytesList 

type BytesList struct {
	Value [][]byte `protobuf:"bytes,1,rep,name=value,proto3" json:"value,omitempty"`
	// contains filtered or unexported fields
}

LINT.IfChange Containers to hold repeated fundamental values.

func (*BytesList) Descriptor deprecated 

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

Deprecated: Use BytesList.ProtoReflect.Descriptor instead.

func (*BytesList) GetValue 

func (x *BytesList) GetValue() [][]byte

func (*BytesList) ProtoMessage 

func (*BytesList) ProtoMessage()

func (*BytesList) ProtoReflect 

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

func (*BytesList) Reset 

func (x *BytesList) Reset()

func (*BytesList) String 

func (x *BytesList) String() string

type CallableOptions 

type CallableOptions struct {

	// Tensors to be fed in the callable. Each feed is the name of a tensor.
	Feed []string `protobuf:"bytes,1,rep,name=feed,proto3" json:"feed,omitempty"`
	// Fetches. A list of tensor names. The caller of the callable expects a
	// tensor to be returned for each fetch[i] (see RunStepResponse.tensor). The
	// order of specified fetches does not change the execution order.
	Fetch []string `protobuf:"bytes,2,rep,name=fetch,proto3" json:"fetch,omitempty"`
	// Target Nodes. A list of node names. The named nodes will be run by the
	// callable but their outputs will not be returned.
	Target []string `protobuf:"bytes,3,rep,name=target,proto3" json:"target,omitempty"`
	// Options that will be applied to each run.
	RunOptions *RunOptions `protobuf:"bytes,4,opt,name=run_options,json=runOptions,proto3" json:"run_options,omitempty"`
	// Tensors to be connected in the callable. Each TensorConnection denotes
	// a pair of tensors in the graph, between which an edge will be created
	// in the callable.
	TensorConnection []*TensorConnection `protobuf:"bytes,5,rep,name=tensor_connection,json=tensorConnection,proto3" json:"tensor_connection,omitempty"`
	// The Tensor objects fed in the callable and fetched from the callable
	// are expected to be backed by host (CPU) memory by default.
	//
	// The options below allow changing that - feeding tensors backed by
	// device memory, or returning tensors that are backed by device memory.
	//
	// The maps below map the name of a feed/fetch tensor (which appears in
	// 'feed' or 'fetch' fields above), to the fully qualified name of the device
	// owning the memory backing the contents of the tensor.
	//
	// For example, creating a callable with the following options:
	//
	// CallableOptions {
	//   feed: "a:0"
	//   feed: "b:0"
	//
	//   fetch: "x:0"
	//   fetch: "y:0"
	//
	//   feed_devices: {
	//     "a:0": "/job:localhost/replica:0/task:0/device:GPU:0"
	//   }
	//
	//   fetch_devices: {
	//     "y:0": "/job:localhost/replica:0/task:0/device:GPU:0"
	//  }
	// }
	//
	// means that the Callable expects:
	// - The first argument ("a:0") is a Tensor backed by GPU memory.
	// - The second argument ("b:0") is a Tensor backed by host memory.
	// and of its return values:
	// - The first output ("x:0") will be backed by host memory.
	// - The second output ("y:0") will be backed by GPU memory.
	//
	// FEEDS:
	// It is the responsibility of the caller to ensure that the memory of the fed
	// tensors will be correctly initialized and synchronized before it is
	// accessed by operations executed during the call to Session::RunCallable().
	//
	// This is typically ensured by using the TensorFlow memory allocators
	// (Device::GetAllocator()) to create the Tensor to be fed.
	//
	// Alternatively, for CUDA-enabled GPU devices, this typically means that the
	// operation that produced the contents of the tensor has completed, i.e., the
	// CUDA stream has been synchronized (e.g., via cuCtxSynchronize() or
	// cuStreamSynchronize()).
	FeedDevices  map[string]string `` /* 182-byte string literal not displayed */
	FetchDevices map[string]string `` /* 185-byte string literal not displayed */
	// By default, RunCallable() will synchronize the GPU stream before returning
	// fetched tensors on a GPU device, to ensure that the values in those tensors
	// have been produced. This simplifies interacting with the tensors, but
	// potentially incurs a performance hit.
	//
	// If this options is set to true, the caller is responsible for ensuring
	// that the values in the fetched tensors have been produced before they are
	// used. The caller can do this by invoking `Device::Sync()` on the underlying
	// device(s), or by feeding the tensors back to the same Session using
	// `feed_devices` with the same corresponding device name.
	FetchSkipSync bool `protobuf:"varint,8,opt,name=fetch_skip_sync,json=fetchSkipSync,proto3" json:"fetch_skip_sync,omitempty"`
	// contains filtered or unexported fields
}

Defines a subgraph in another `GraphDef` as a set of feed points and nodes to be fetched or executed.

Compare with the arguments to `Session::Run()`.

func (*CallableOptions) Descriptor deprecated 

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

Deprecated: Use CallableOptions.ProtoReflect.Descriptor instead.

func (*CallableOptions) GetFeed 

func (x *CallableOptions) GetFeed() []string

func (*CallableOptions) GetFeedDevices 

func (x *CallableOptions) GetFeedDevices() map[string]string

func (*CallableOptions) GetFetch 

func (x *CallableOptions) GetFetch() []string

func (*CallableOptions) GetFetchDevices 

func (x *CallableOptions) GetFetchDevices() map[string]string

func (*CallableOptions) GetFetchSkipSync 

func (x *CallableOptions) GetFetchSkipSync() bool

func (*CallableOptions) GetRunOptions 

func (x *CallableOptions) GetRunOptions() *RunOptions

func (*CallableOptions) GetTarget 

func (x *CallableOptions) GetTarget() []string

func (*CallableOptions) GetTensorConnection 

func (x *CallableOptions) GetTensorConnection() []*TensorConnection

func (*CallableOptions) ProtoMessage 

func (*CallableOptions) ProtoMessage()

func (*CallableOptions) ProtoReflect 

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

func (*CallableOptions) Reset 

func (x *CallableOptions) Reset()

func (*CallableOptions) String 

func (x *CallableOptions) String() string

type CancelBarrierRequest 

type CancelBarrierRequest struct {
	BarrierId string `protobuf:"bytes,1,opt,name=barrier_id,json=barrierId,proto3" json:"barrier_id,omitempty"`
	// Task that is making the request.
	SourceTask *CoordinatedTask `protobuf:"bytes,2,opt,name=source_task,json=sourceTask,proto3" json:"source_task,omitempty"`
	// contains filtered or unexported fields
}

Request and response messages for cancelling generic sync barriers.

func (*CancelBarrierRequest) Descriptor deprecated 

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

Deprecated: Use CancelBarrierRequest.ProtoReflect.Descriptor instead.

func (*CancelBarrierRequest) GetBarrierId 

func (x *CancelBarrierRequest) GetBarrierId() string

func (*CancelBarrierRequest) GetSourceTask 

func (x *CancelBarrierRequest) GetSourceTask() *CoordinatedTask

func (*CancelBarrierRequest) ProtoMessage 

func (*CancelBarrierRequest) ProtoMessage()

func (*CancelBarrierRequest) ProtoReflect 

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

func (*CancelBarrierRequest) Reset 

func (x *CancelBarrierRequest) Reset()

func (*CancelBarrierRequest) String 

func (x *CancelBarrierRequest) String() string

type CancelBarrierResponse 

type CancelBarrierResponse struct {
	// contains filtered or unexported fields
}

func (*CancelBarrierResponse) Descriptor deprecated 

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

Deprecated: Use CancelBarrierResponse.ProtoReflect.Descriptor instead.

func (*CancelBarrierResponse) ProtoMessage 

func (*CancelBarrierResponse) ProtoMessage()

func (*CancelBarrierResponse) ProtoReflect 

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

func (*CancelBarrierResponse) Reset 

func (x *CancelBarrierResponse) Reset()

func (*CancelBarrierResponse) String 

func (x *CancelBarrierResponse) String() string

type CapturedTensor 

type CapturedTensor struct {

	// Name of captured tensor
	Name string `protobuf:"bytes,1,opt,name=name,proto3" json:"name,omitempty"`
	// Name of concrete function which contains the computed graph tensor.
	ConcreteFunction string `protobuf:"bytes,2,opt,name=concrete_function,json=concreteFunction,proto3" json:"concrete_function,omitempty"`
	// contains filtered or unexported fields
}

func (*CapturedTensor) Descriptor deprecated 

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

Deprecated: Use CapturedTensor.ProtoReflect.Descriptor instead.

func (*CapturedTensor) GetConcreteFunction 

func (x *CapturedTensor) GetConcreteFunction() string

func (*CapturedTensor) GetName 

func (x *CapturedTensor) GetName() string

func (*CapturedTensor) ProtoMessage 

func (*CapturedTensor) ProtoMessage()

func (*CapturedTensor) ProtoReflect 

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

func (*CapturedTensor) Reset 

func (x *CapturedTensor) Reset()

func (*CapturedTensor) String 

func (x *CapturedTensor) String() string

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 CleanupAllRequest 

type CleanupAllRequest struct {

	// A list of container names.
	//
	// If 'container' is not empty, releases resources in the given
	// containers in all devices.
	//
	// If 'container' is empty, releases resources in the default
	// container in all devices.
	Container []string `protobuf:"bytes,1,rep,name=container,proto3" json:"container,omitempty"`
	// contains filtered or unexported fields
}

func (*CleanupAllRequest) Descriptor deprecated 

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

Deprecated: Use CleanupAllRequest.ProtoReflect.Descriptor instead.

func (*CleanupAllRequest) GetContainer 

func (x *CleanupAllRequest) GetContainer() []string

func (*CleanupAllRequest) ProtoMessage 

func (*CleanupAllRequest) ProtoMessage()

func (*CleanupAllRequest) ProtoReflect 

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

func (*CleanupAllRequest) Reset 

func (x *CleanupAllRequest) Reset()

func (*CleanupAllRequest) String 

func (x *CleanupAllRequest) String() string

type CleanupAllResponse 

type CleanupAllResponse struct {
	// contains filtered or unexported fields
}

func (*CleanupAllResponse) Descriptor deprecated 

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

Deprecated: Use CleanupAllResponse.ProtoReflect.Descriptor instead.

func (*CleanupAllResponse) ProtoMessage 

func (*CleanupAllResponse) ProtoMessage()

func (*CleanupAllResponse) ProtoReflect 

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

func (*CleanupAllResponse) Reset 

func (x *CleanupAllResponse) Reset()

func (*CleanupAllResponse) String 

func (x *CleanupAllResponse) String() string

type CleanupFunctionOp 

type CleanupFunctionOp struct {
	StepId int64 `protobuf:"varint,1,opt,name=step_id,json=stepId,proto3" json:"step_id,omitempty"`
	// contains filtered or unexported fields
}

Cleanup the step state of a multi-device function (e.g. tensors buffered by a `Send` op but not picked up by its corresponding `Recv` op).

func (*CleanupFunctionOp) Descriptor deprecated 

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

Deprecated: Use CleanupFunctionOp.ProtoReflect.Descriptor instead.

func (*CleanupFunctionOp) GetStepId 

func (x *CleanupFunctionOp) GetStepId() int64

func (*CleanupFunctionOp) ProtoMessage 

func (*CleanupFunctionOp) ProtoMessage()

func (*CleanupFunctionOp) ProtoReflect 

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

func (*CleanupFunctionOp) Reset 

func (x *CleanupFunctionOp) Reset()

func (*CleanupFunctionOp) String 

func (x *CleanupFunctionOp) String() string

type CleanupGraphRequest 

type CleanupGraphRequest struct {
	StepId int64 `protobuf:"varint,1,opt,name=step_id,json=stepId,proto3" json:"step_id,omitempty"`
	// contains filtered or unexported fields
}

func (*CleanupGraphRequest) Descriptor deprecated 

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

Deprecated: Use CleanupGraphRequest.ProtoReflect.Descriptor instead.

func (*CleanupGraphRequest) GetStepId 

func (x *CleanupGraphRequest) GetStepId() int64

func (*CleanupGraphRequest) ProtoMessage 

func (*CleanupGraphRequest) ProtoMessage()

func (*CleanupGraphRequest) ProtoReflect 

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

func (*CleanupGraphRequest) Reset 

func (x *CleanupGraphRequest) Reset()

func (*CleanupGraphRequest) String 

func (x *CleanupGraphRequest) String() string

type CleanupGraphResponse 

type CleanupGraphResponse struct {
	// contains filtered or unexported fields
}

func (*CleanupGraphResponse) Descriptor deprecated 

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

Deprecated: Use CleanupGraphResponse.ProtoReflect.Descriptor instead.

func (*CleanupGraphResponse) ProtoMessage 

func (*CleanupGraphResponse) ProtoMessage()

func (*CleanupGraphResponse) ProtoReflect 

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

func (*CleanupGraphResponse) Reset 

func (x *CleanupGraphResponse) Reset()

func (*CleanupGraphResponse) String 

func (x *CleanupGraphResponse) String() string

type CloseContextRequest 

type CloseContextRequest struct {
	ContextId     uint64 `protobuf:"fixed64,1,opt,name=context_id,json=contextId,proto3" json:"context_id,omitempty"`
	ContextViewId uint64 `protobuf:"fixed64,2,opt,name=context_view_id,json=contextViewId,proto3" json:"context_view_id,omitempty"`
	// contains filtered or unexported fields
}

func (*CloseContextRequest) Descriptor deprecated 

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

Deprecated: Use CloseContextRequest.ProtoReflect.Descriptor instead.

func (*CloseContextRequest) GetContextId 

func (x *CloseContextRequest) GetContextId() uint64

func (*CloseContextRequest) GetContextViewId 

func (x *CloseContextRequest) GetContextViewId() uint64

func (*CloseContextRequest) ProtoMessage 

func (*CloseContextRequest) ProtoMessage()

func (*CloseContextRequest) ProtoReflect 

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

func (*CloseContextRequest) Reset 

func (x *CloseContextRequest) Reset()

func (*CloseContextRequest) String 

func (x *CloseContextRequest) String() string

type CloseContextResponse 

type CloseContextResponse struct {
	// contains filtered or unexported fields
}

func (*CloseContextResponse) Descriptor deprecated 

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

Deprecated: Use CloseContextResponse.ProtoReflect.Descriptor instead.

func (*CloseContextResponse) ProtoMessage 

func (*CloseContextResponse) ProtoMessage()

func (*CloseContextResponse) ProtoReflect 

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

func (*CloseContextResponse) Reset 

func (x *CloseContextResponse) Reset()

func (*CloseContextResponse) String 

func (x *CloseContextResponse) String() string

type CloseSessionRequest 

type CloseSessionRequest struct {

	// REQUIRED: session_handle must be returned by a CreateSession call
	// to the same master service.
	SessionHandle string `protobuf:"bytes,1,opt,name=session_handle,json=sessionHandle,proto3" json:"session_handle,omitempty"`
	// contains filtered or unexported fields
}

func (*CloseSessionRequest) Descriptor deprecated 

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

Deprecated: Use CloseSessionRequest.ProtoReflect.Descriptor instead.

func (*CloseSessionRequest) GetSessionHandle 

func (x *CloseSessionRequest) GetSessionHandle() string

func (*CloseSessionRequest) ProtoMessage 

func (*CloseSessionRequest) ProtoMessage()

func (*CloseSessionRequest) ProtoReflect 

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

func (*CloseSessionRequest) Reset 

func (x *CloseSessionRequest) Reset()

func (*CloseSessionRequest) String 

func (x *CloseSessionRequest) String() string

type CloseSessionResponse 

type CloseSessionResponse struct {
	// contains filtered or unexported fields
}

func (*CloseSessionResponse) Descriptor deprecated 

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

Deprecated: Use CloseSessionResponse.ProtoReflect.Descriptor instead.

func (*CloseSessionResponse) ProtoMessage 

func (*CloseSessionResponse) ProtoMessage()

func (*CloseSessionResponse) ProtoReflect 

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

func (*CloseSessionResponse) Reset 

func (x *CloseSessionResponse) Reset()

func (*CloseSessionResponse) String 

func (x *CloseSessionResponse) String() string

type ClusterDef 

type ClusterDef struct {

	// The jobs that comprise the cluster.
	Job []*JobDef `protobuf:"bytes,1,rep,name=job,proto3" json:"job,omitempty"`
	// contains filtered or unexported fields
}

Defines a TensorFlow cluster as a set of jobs.

func (*ClusterDef) Descriptor deprecated 

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

Deprecated: Use ClusterDef.ProtoReflect.Descriptor instead.

func (*ClusterDef) GetJob 

func (x *ClusterDef) GetJob() []*JobDef

func (*ClusterDef) ProtoMessage 

func (*ClusterDef) ProtoMessage()

func (*ClusterDef) ProtoReflect 

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

func (*ClusterDef) Reset 

func (x *ClusterDef) Reset()

func (*ClusterDef) String 

func (x *ClusterDef) String() string

type ClusterDeviceFilters 

type ClusterDeviceFilters struct {
	Jobs []*JobDeviceFilters `protobuf:"bytes,1,rep,name=jobs,proto3" json:"jobs,omitempty"`
	// contains filtered or unexported fields
}

Defines the device filters for jobs in a cluster.

func (*ClusterDeviceFilters) Descriptor deprecated 

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

Deprecated: Use ClusterDeviceFilters.ProtoReflect.Descriptor instead.

func (*ClusterDeviceFilters) GetJobs 

func (x *ClusterDeviceFilters) GetJobs() []*JobDeviceFilters

func (*ClusterDeviceFilters) ProtoMessage 

func (*ClusterDeviceFilters) ProtoMessage()

func (*ClusterDeviceFilters) ProtoReflect 

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

func (*ClusterDeviceFilters) Reset 

func (x *ClusterDeviceFilters) Reset()

func (*ClusterDeviceFilters) String 

func (x *ClusterDeviceFilters) String() string

type Code 

type Code int32

The canonical error codes for TensorFlow APIs.

Warnings:

  • Do not change any numeric assignments.
  • Changes to this list should only be made if there is a compelling need that can't be satisfied in another way. Such changes must be approved by at least two OWNERS.
  • These error codes must match gRPC and protobuf error codes (except for DO_NOT_USE_RESERVED_FOR_FUTURE_EXPANSION_USE_DEFAULT_IN_SWITCH_INSTEAD_).

Sometimes multiple error codes may apply. Services should return the most specific error code that applies. For example, prefer OUT_OF_RANGE over FAILED_PRECONDITION if both codes apply. Similarly prefer NOT_FOUND or ALREADY_EXISTS over FAILED_PRECONDITION. 

const (
	// Not an error; returned on success
	Code_OK Code = 0
	// The operation was cancelled (typically by the caller).
	Code_CANCELLED Code = 1
	// Unknown error.  An example of where this error may be returned is
	// if a Status value received from another address space belongs to
	// an error-space that is not known in this address space.  Also
	// errors raised by APIs that do not return enough error information
	// may be converted to this error.
	Code_UNKNOWN Code = 2
	// Client specified an invalid argument.  Note that this differs
	// from FAILED_PRECONDITION.  INVALID_ARGUMENT indicates arguments
	// that are problematic regardless of the state of the system
	// (e.g., a malformed file name).
	Code_INVALID_ARGUMENT Code = 3
	// Deadline expired before operation could complete.  For operations
	// that change the state of the system, this error may be returned
	// even if the operation has completed successfully.  For example, a
	// successful response from a server could have been delayed long
	// enough for the deadline to expire.
	Code_DEADLINE_EXCEEDED Code = 4
	// Some requested entity (e.g., file or directory) was not found.
	// For privacy reasons, this code *may* be returned when the client
	// does not have the access right to the entity.
	Code_NOT_FOUND Code = 5
	// Some entity that we attempted to create (e.g., file or directory)
	// already exists.
	Code_ALREADY_EXISTS Code = 6
	// The caller does not have permission to execute the specified
	// operation.  PERMISSION_DENIED must not be used for rejections
	// caused by exhausting some resource (use RESOURCE_EXHAUSTED
	// instead for those errors).  PERMISSION_DENIED must not be
	// used if the caller can not be identified (use UNAUTHENTICATED
	// instead for those errors).
	Code_PERMISSION_DENIED Code = 7
	// The request does not have valid authentication credentials for the
	// operation.
	Code_UNAUTHENTICATED Code = 16
	// Some resource has been exhausted, perhaps a per-user quota, or
	// perhaps the entire file system is out of space.
	Code_RESOURCE_EXHAUSTED Code = 8
	// Operation was rejected because the system is not in a state
	// required for the operation's execution.  For example, directory
	// to be deleted may be non-empty, an rmdir operation is applied to
	// a non-directory, etc.
	//
	// A litmus test that may help a service implementor in deciding
	// between FAILED_PRECONDITION, ABORTED, and UNAVAILABLE:
	//  (a) Use UNAVAILABLE if the client can retry just the failing call.
	//  (b) Use ABORTED if the client should retry at a higher-level
	//      (e.g., restarting a read-modify-write sequence).
	//  (c) Use FAILED_PRECONDITION if the client should not retry until
	//      the system state has been explicitly fixed.  E.g., if an "rmdir"
	//      fails because the directory is non-empty, FAILED_PRECONDITION
	//      should be returned since the client should not retry unless
	//      they have first fixed up the directory by deleting files from it.
	//  (d) Use FAILED_PRECONDITION if the client performs conditional
	//      REST Get/Update/Delete on a resource and the resource on the
	//      server does not match the condition. E.g., conflicting
	//      read-modify-write on the same resource.
	Code_FAILED_PRECONDITION Code = 9
	// The operation was aborted, typically due to a concurrency issue
	// like sequencer check failures, transaction aborts, etc.
	//
	// See litmus test above for deciding between FAILED_PRECONDITION,
	// ABORTED, and UNAVAILABLE.
	Code_ABORTED Code = 10
	// Operation tried to iterate past the valid input range.  E.g., seeking or
	// reading past end of file.
	//
	// Unlike INVALID_ARGUMENT, this error indicates a problem that may
	// be fixed if the system state changes. For example, a 32-bit file
	// system will generate INVALID_ARGUMENT if asked to read at an
	// offset that is not in the range [0,2^32-1], but it will generate
	// OUT_OF_RANGE if asked to read from an offset past the current
	// file size.
	//
	// There is a fair bit of overlap between FAILED_PRECONDITION and
	// OUT_OF_RANGE.  We recommend using OUT_OF_RANGE (the more specific
	// error) when it applies so that callers who are iterating through
	// a space can easily look for an OUT_OF_RANGE error to detect when
	// they are done.
	Code_OUT_OF_RANGE Code = 11
	// Operation is not implemented or not supported/enabled in this service.
	Code_UNIMPLEMENTED Code = 12
	// Internal errors.  Means some invariant expected by the underlying
	// system has been broken.  If you see one of these errors,
	// something is very broken.
	Code_INTERNAL Code = 13
	// The service is currently unavailable.  This is a most likely a
	// transient condition and may be corrected by retrying with
	// a backoff.
	//
	// See litmus test above for deciding between FAILED_PRECONDITION,
	// ABORTED, and UNAVAILABLE.
	Code_UNAVAILABLE Code = 14
	// Unrecoverable data loss or corruption.
	Code_DATA_LOSS Code = 15
	// An extra enum entry to prevent people from writing code that
	// fails to compile when a new code is added.
	//
	// Nobody should ever reference this enumeration entry. In particular,
	// if you write C++ code that switches on this enumeration, add a default:
	// case instead of a case that mentions this enumeration entry.
	//
	// Nobody should rely on the value (currently 20) listed here.  It
	// may change in the future.
	Code_DO_NOT_USE_RESERVED_FOR_FUTURE_EXPANSION_USE_DEFAULT_IN_SWITCH_INSTEAD_ Code = 20
)

func (Code) Descriptor 

func (Code) Descriptor() protoreflect.EnumDescriptor

func (Code) Enum 

func (x Code) Enum() *Code

func (Code) EnumDescriptor deprecated 

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

Deprecated: Use Code.Descriptor instead.

func (Code) Number 

func (x Code) Number() protoreflect.EnumNumber

func (Code) String 

func (x Code) String() string

func (Code) Type 

func (Code) Type() protoreflect.EnumType

type CodeLocation 

type CodeLocation struct {

	// Host name on which the source files are located.
	HostName string `protobuf:"bytes,1,opt,name=host_name,json=hostName,proto3" json:"host_name,omitempty"`
	// ID to a stack frame, each of which is pointed to
	// by a unique ID. The ordering of the frames is consistent with Python's
	// `traceback.extract_tb()`.
	StackFrameIds []string `protobuf:"bytes,2,rep,name=stack_frame_ids,json=stackFrameIds,proto3" json:"stack_frame_ids,omitempty"`
	// contains filtered or unexported fields
}

Code location information: A stack trace with host-name information. Instead of encoding the detailed stack trace, this proto refers to IDs of stack frames stored as `StackFrameWithId` protos.

func (*CodeLocation) Descriptor deprecated 

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

Deprecated: Use CodeLocation.ProtoReflect.Descriptor instead.

func (*CodeLocation) GetHostName 

func (x *CodeLocation) GetHostName() string

func (*CodeLocation) GetStackFrameIds 

func (x *CodeLocation) GetStackFrameIds() []string

func (*CodeLocation) ProtoMessage 

func (*CodeLocation) ProtoMessage()

func (*CodeLocation) ProtoReflect 

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

func (*CodeLocation) Reset 

func (x *CodeLocation) Reset()

func (*CodeLocation) String 

func (x *CodeLocation) String() string

type CollectionDef 

type CollectionDef struct {

	// Types that are assignable to Kind:
	//	*CollectionDef_NodeList_
	//	*CollectionDef_BytesList_
	//	*CollectionDef_Int64List_
	//	*CollectionDef_FloatList_
	//	*CollectionDef_AnyList_
	Kind isCollectionDef_Kind `protobuf_oneof:"kind"`
	// contains filtered or unexported fields
}

CollectionDef should cover most collections. To add a user-defined collection, do one of the following:

  1. For simple data types, such as string, int, float: tf.add_to_collection("your_collection_name", your_simple_value) strings will be stored as bytes_list.

2. For Protobuf types, there are three ways to add them:

  1. tf.add_to_collection("your_collection_name", your_proto.SerializeToString())

    collection_def { key: "user_defined_bytes_collection" value { bytes_list { value: "queue_name: \"test_queue\"\n" } } }

    or

  2. tf.add_to_collection("your_collection_name", str(your_proto))

    collection_def { key: "user_defined_string_collection" value { bytes_list { value: "\n\ntest_queue" } } }

    or

  3. any_buf = any_pb2.Any() tf.add_to_collection("your_collection_name", any_buf.Pack(your_proto))

    collection_def { key: "user_defined_any_collection" value { any_list { value { type_url: "type.googleapis.com/tensorflow.QueueRunnerDef" value: "\n\ntest_queue" } } } }

  4. For Python objects, implement to_proto() and from_proto(), and register them in the following manner: ops.register_proto_function("your_collection_name", proto_type, to_proto=YourPythonObject.to_proto, from_proto=YourPythonObject.from_proto) These functions will be invoked to serialize and de-serialize the collection. For example, ops.register_proto_function(ops.GraphKeys.GLOBAL_VARIABLES, proto_type=variable_pb2.VariableDef, to_proto=Variable.to_proto, from_proto=Variable.from_proto)

func (*CollectionDef) Descriptor deprecated 

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

Deprecated: Use CollectionDef.ProtoReflect.Descriptor instead.

func (*CollectionDef) GetAnyList 

func (x *CollectionDef) GetAnyList() *CollectionDef_AnyList

func (*CollectionDef) GetBytesList 

func (x *CollectionDef) GetBytesList() *CollectionDef_BytesList

func (*CollectionDef) GetFloatList 

func (x *CollectionDef) GetFloatList() *CollectionDef_FloatList

func (*CollectionDef) GetInt64List 

func (x *CollectionDef) GetInt64List() *CollectionDef_Int64List

func (*CollectionDef) GetKind 

func (m *CollectionDef) GetKind() isCollectionDef_Kind

func (*CollectionDef) GetNodeList 

func (x *CollectionDef) GetNodeList() *CollectionDef_NodeList

func (*CollectionDef) ProtoMessage 

func (*CollectionDef) ProtoMessage()

func (*CollectionDef) ProtoReflect 

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

func (*CollectionDef) Reset 

func (x *CollectionDef) Reset()

func (*CollectionDef) String 

func (x *CollectionDef) String() string

type CollectionDef_AnyList 

type CollectionDef_AnyList struct {
	Value []*any.Any `protobuf:"bytes,1,rep,name=value,proto3" json:"value,omitempty"`
	// contains filtered or unexported fields
}

AnyList is used for collecting Any protos.

func (*CollectionDef_AnyList) Descriptor deprecated 

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

Deprecated: Use CollectionDef_AnyList.ProtoReflect.Descriptor instead.

func (*CollectionDef_AnyList) GetValue 

func (x *CollectionDef_AnyList) GetValue() []*any.Any

func (*CollectionDef_AnyList) ProtoMessage 

func (*CollectionDef_AnyList) ProtoMessage()

func (*CollectionDef_AnyList) ProtoReflect 

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

func (*CollectionDef_AnyList) Reset 

func (x *CollectionDef_AnyList) Reset()

func (*CollectionDef_AnyList) String 

func (x *CollectionDef_AnyList) String() string

type CollectionDef_AnyList_ 

type CollectionDef_AnyList_ struct {
	AnyList *CollectionDef_AnyList `protobuf:"bytes,5,opt,name=any_list,json=anyList,proto3,oneof"`
}

type CollectionDef_BytesList 

type CollectionDef_BytesList struct {
	Value [][]byte `protobuf:"bytes,1,rep,name=value,proto3" json:"value,omitempty"`
	// contains filtered or unexported fields
}

BytesList is used for collecting strings and serialized protobufs. For example: 

collection_def {
  key: "trainable_variables"
  value {
    bytes_list {
      value: "\n\017conv1/weights:0\022\024conv1/weights/Assign
             \032\024conv1/weights/read:0"
      value: "\n\016conv1/biases:0\022\023conv1/biases/Assign\032
             \023conv1/biases/read:0"
    }
  }
}

func (*CollectionDef_BytesList) Descriptor deprecated 

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

Deprecated: Use CollectionDef_BytesList.ProtoReflect.Descriptor instead.

func (*CollectionDef_BytesList) GetValue 

func (x *CollectionDef_BytesList) GetValue() [][]byte

func (*CollectionDef_BytesList) ProtoMessage 

func (*CollectionDef_BytesList) ProtoMessage()

func (*CollectionDef_BytesList) ProtoReflect 

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

func (*CollectionDef_BytesList) Reset 

func (x *CollectionDef_BytesList) Reset()

func (*CollectionDef_BytesList) String 

func (x *CollectionDef_BytesList) String() string

type CollectionDef_BytesList_ 

type CollectionDef_BytesList_ struct {
	BytesList *CollectionDef_BytesList `protobuf:"bytes,2,opt,name=bytes_list,json=bytesList,proto3,oneof"`
}

type CollectionDef_FloatList 

type CollectionDef_FloatList struct {
	Value []float32 `protobuf:"fixed32,1,rep,packed,name=value,proto3" json:"value,omitempty"`
	// contains filtered or unexported fields
}

FloatList is used for collecting float values.

func (*CollectionDef_FloatList) Descriptor deprecated 

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

Deprecated: Use CollectionDef_FloatList.ProtoReflect.Descriptor instead.

func (*CollectionDef_FloatList) GetValue 

func (x *CollectionDef_FloatList) GetValue() []float32

func (*CollectionDef_FloatList) ProtoMessage 

func (*CollectionDef_FloatList) ProtoMessage()

func (*CollectionDef_FloatList) ProtoReflect 

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

func (*CollectionDef_FloatList) Reset 

func (x *CollectionDef_FloatList) Reset()

func (*CollectionDef_FloatList) String 

func (x *CollectionDef_FloatList) String() string

type CollectionDef_FloatList_ 

type CollectionDef_FloatList_ struct {
	FloatList *CollectionDef_FloatList `protobuf:"bytes,4,opt,name=float_list,json=floatList,proto3,oneof"`
}

type CollectionDef_Int64List 

type CollectionDef_Int64List struct {
	Value []int64 `protobuf:"varint,1,rep,packed,name=value,proto3" json:"value,omitempty"`
	// contains filtered or unexported fields
}

Int64List is used for collecting int, int64 and long values.

func (*CollectionDef_Int64List) Descriptor deprecated 

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

Deprecated: Use CollectionDef_Int64List.ProtoReflect.Descriptor instead.

func (*CollectionDef_Int64List) GetValue 

func (x *CollectionDef_Int64List) GetValue() []int64

func (*CollectionDef_Int64List) ProtoMessage 

func (*CollectionDef_Int64List) ProtoMessage()

func (*CollectionDef_Int64List) ProtoReflect 

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

func (*CollectionDef_Int64List) Reset 

func (x *CollectionDef_Int64List) Reset()

func (*CollectionDef_Int64List) String 

func (x *CollectionDef_Int64List) String() string

type CollectionDef_Int64List_ 

type CollectionDef_Int64List_ struct {
	Int64List *CollectionDef_Int64List `protobuf:"bytes,3,opt,name=int64_list,json=int64List,proto3,oneof"`
}

type CollectionDef_NodeList 

type CollectionDef_NodeList struct {
	Value []string `protobuf:"bytes,1,rep,name=value,proto3" json:"value,omitempty"`
	// contains filtered or unexported fields
}

NodeList is used for collecting nodes in graph. For example 

collection_def {
  key: "summaries"
  value {
    node_list {
      value: "input_producer/ScalarSummary:0"
      value: "shuffle_batch/ScalarSummary:0"
      value: "ImageSummary:0"
    }
  }

func (*CollectionDef_NodeList) Descriptor deprecated 

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

Deprecated: Use CollectionDef_NodeList.ProtoReflect.Descriptor instead.

func (*CollectionDef_NodeList) GetValue 

func (x *CollectionDef_NodeList) GetValue() []string

func (*CollectionDef_NodeList) ProtoMessage 

func (*CollectionDef_NodeList) ProtoMessage()

func (*CollectionDef_NodeList) ProtoReflect 

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

func (*CollectionDef_NodeList) Reset 

func (x *CollectionDef_NodeList) Reset()

func (*CollectionDef_NodeList) String 

func (x *CollectionDef_NodeList) String() string

type CollectionDef_NodeList_ 

type CollectionDef_NodeList_ struct {
	NodeList *CollectionDef_NodeList `protobuf:"bytes,1,opt,name=node_list,json=nodeList,proto3,oneof"`
}

type CompleteGroupRequest 

type CompleteGroupRequest struct {
	GroupKey         int32             `protobuf:"varint,1,opt,name=group_key,json=groupKey,proto3" json:"group_key,omitempty"`
	GroupSize        int32             `protobuf:"varint,2,opt,name=group_size,json=groupSize,proto3" json:"group_size,omitempty"`
	DeviceType       string            `protobuf:"bytes,3,opt,name=device_type,json=deviceType,proto3" json:"device_type,omitempty"`
	CollectiveType   int32             `protobuf:"varint,5,opt,name=collective_type,json=collectiveType,proto3" json:"collective_type,omitempty"`
	DeviceAttributes *DeviceAttributes `protobuf:"bytes,6,opt,name=device_attributes,json=deviceAttributes,proto3" json:"device_attributes,omitempty"`
	// contains filtered or unexported fields
}

Supplies one or more device names as members of the group identified by group_key. Service will respond when all group_size devices become known. All devices in group must have same type.

func (*CompleteGroupRequest) Descriptor deprecated 

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

Deprecated: Use CompleteGroupRequest.ProtoReflect.Descriptor instead.

func (*CompleteGroupRequest) GetCollectiveType 

func (x *CompleteGroupRequest) GetCollectiveType() int32

func (*CompleteGroupRequest) GetDeviceAttributes 

func (x *CompleteGroupRequest) GetDeviceAttributes() *DeviceAttributes

func (*CompleteGroupRequest) GetDeviceType 

func (x *CompleteGroupRequest) GetDeviceType() string

func (*CompleteGroupRequest) GetGroupKey 

func (x *CompleteGroupRequest) GetGroupKey() int32

func (*CompleteGroupRequest) GetGroupSize 

func (x *CompleteGroupRequest) GetGroupSize() int32

func (*CompleteGroupRequest) ProtoMessage 

func (*CompleteGroupRequest) ProtoMessage()

func (*CompleteGroupRequest) ProtoReflect 

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

func (*CompleteGroupRequest) Reset 

func (x *CompleteGroupRequest) Reset()

func (*CompleteGroupRequest) String 

func (x *CompleteGroupRequest) String() string

type CompleteGroupResponse 

type CompleteGroupResponse struct {
	GroupKey         int32               `protobuf:"varint,1,opt,name=group_key,json=groupKey,proto3" json:"group_key,omitempty"`
	GroupSize        int32               `protobuf:"varint,2,opt,name=group_size,json=groupSize,proto3" json:"group_size,omitempty"`
	DeviceType       string              `protobuf:"bytes,3,opt,name=device_type,json=deviceType,proto3" json:"device_type,omitempty"`
	NumTasks         int32               `protobuf:"varint,4,opt,name=num_tasks,json=numTasks,proto3" json:"num_tasks,omitempty"` // number of distinct tasks hosting the devices
	CommunicatorKey  []byte              `protobuf:"bytes,7,opt,name=communicator_key,json=communicatorKey,proto3" json:"communicator_key,omitempty"`
	DeviceAttributes []*DeviceAttributes `protobuf:"bytes,8,rep,name=device_attributes,json=deviceAttributes,proto3" json:"device_attributes,omitempty"`
	// contains filtered or unexported fields
}

Gives the complete membership of the group identified by group_key.

func (*CompleteGroupResponse) Descriptor deprecated 

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

Deprecated: Use CompleteGroupResponse.ProtoReflect.Descriptor instead.

func (*CompleteGroupResponse) GetCommunicatorKey 

func (x *CompleteGroupResponse) GetCommunicatorKey() []byte

func (*CompleteGroupResponse) GetDeviceAttributes 

func (x *CompleteGroupResponse) GetDeviceAttributes() []*DeviceAttributes

func (*CompleteGroupResponse) GetDeviceType 

func (x *CompleteGroupResponse) GetDeviceType() string

func (*CompleteGroupResponse) GetGroupKey 

func (x *CompleteGroupResponse) GetGroupKey() int32

func (*CompleteGroupResponse) GetGroupSize 

func (x *CompleteGroupResponse) GetGroupSize() int32

func (*CompleteGroupResponse) GetNumTasks 

func (x *CompleteGroupResponse) GetNumTasks() int32

func (*CompleteGroupResponse) ProtoMessage 

func (*CompleteGroupResponse) ProtoMessage()

func (*CompleteGroupResponse) ProtoReflect 

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

func (*CompleteGroupResponse) Reset 

func (x *CompleteGroupResponse) Reset()

func (*CompleteGroupResponse) String 

func (x *CompleteGroupResponse) String() string

type CompleteInstanceRequest 

type CompleteInstanceRequest struct {
	Name         string            `protobuf:"bytes,1,opt,name=name,proto3" json:"name,omitempty"`
	Type         int32             `protobuf:"varint,2,opt,name=type,proto3" json:"type,omitempty"`
	DataType     DataType          `protobuf:"varint,3,opt,name=data_type,json=dataType,proto3,enum=tensorflow.DataType" json:"data_type,omitempty"`
	Shape        *TensorShapeProto `protobuf:"bytes,4,opt,name=shape,proto3" json:"shape,omitempty"`
	GroupKey     int32             `protobuf:"varint,5,opt,name=group_key,json=groupKey,proto3" json:"group_key,omitempty"`
	GroupSize    int32             `protobuf:"varint,6,opt,name=group_size,json=groupSize,proto3" json:"group_size,omitempty"`
	InstanceKey  int32             `protobuf:"varint,7,opt,name=instance_key,json=instanceKey,proto3" json:"instance_key,omitempty"`
	DeviceType   string            `protobuf:"bytes,8,opt,name=device_type,json=deviceType,proto3" json:"device_type,omitempty"`
	SubdivOffset []int32           `protobuf:"varint,9,rep,packed,name=subdiv_offset,json=subdivOffset,proto3" json:"subdiv_offset,omitempty"`
	Device       string            `protobuf:"bytes,10,opt,name=device,proto3" json:"device,omitempty"`
	IsSource     bool              `protobuf:"varint,11,opt,name=is_source,json=isSource,proto3" json:"is_source,omitempty"`
	// contains filtered or unexported fields
}

Supplies data about one collective op belonging to the instance identified by instance_key. Service will respond when all group_size ops have become known. Most of the data being sent is for correctness checking, to ensure that all ops in the instance share common attributes.

func (*CompleteInstanceRequest) Descriptor deprecated 

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

Deprecated: Use CompleteInstanceRequest.ProtoReflect.Descriptor instead.

func (*CompleteInstanceRequest) GetDataType 

func (x *CompleteInstanceRequest) GetDataType() DataType

func (*CompleteInstanceRequest) GetDevice 

func (x *CompleteInstanceRequest) GetDevice() string

func (*CompleteInstanceRequest) GetDeviceType 

func (x *CompleteInstanceRequest) GetDeviceType() string

func (*CompleteInstanceRequest) GetGroupKey 

func (x *CompleteInstanceRequest) GetGroupKey() int32

func (*CompleteInstanceRequest) GetGroupSize 

func (x *CompleteInstanceRequest) GetGroupSize() int32

func (*CompleteInstanceRequest) GetInstanceKey 

func (x *CompleteInstanceRequest) GetInstanceKey() int32

func (*CompleteInstanceRequest) GetIsSource 

func (x *CompleteInstanceRequest) GetIsSource() bool

func (*CompleteInstanceRequest) GetName 

func (x *CompleteInstanceRequest) GetName() string

func (*CompleteInstanceRequest) GetShape 

func (x *CompleteInstanceRequest) GetShape() *TensorShapeProto

func (*CompleteInstanceRequest) GetSubdivOffset 

func (x *CompleteInstanceRequest) GetSubdivOffset() []int32

func (*CompleteInstanceRequest) GetType 

func (x *CompleteInstanceRequest) GetType() int32

func (*CompleteInstanceRequest) ProtoMessage 

func (*CompleteInstanceRequest) ProtoMessage()

func (*CompleteInstanceRequest) ProtoReflect 

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

func (*CompleteInstanceRequest) Reset 

func (x *CompleteInstanceRequest) Reset()

func (*CompleteInstanceRequest) String 

func (x *CompleteInstanceRequest) String() string

type CompleteInstanceResponse 

type CompleteInstanceResponse struct {
	InstanceKey int32 `protobuf:"varint,1,opt,name=instance_key,json=instanceKey,proto3" json:"instance_key,omitempty"`
	SourceRank  int32 `protobuf:"varint,2,opt,name=source_rank,json=sourceRank,proto3" json:"source_rank,omitempty"`
	// contains filtered or unexported fields
}

Confirms that every op in the instance has consistently declared itself. Also gives the source_rank in case of broadcast.

func (*CompleteInstanceResponse) Descriptor deprecated 

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

Deprecated: Use CompleteInstanceResponse.ProtoReflect.Descriptor instead.

func (*CompleteInstanceResponse) GetInstanceKey 

func (x *CompleteInstanceResponse) GetInstanceKey() int32

func (*CompleteInstanceResponse) GetSourceRank 

func (x *CompleteInstanceResponse) GetSourceRank() int32

func (*CompleteInstanceResponse) ProtoMessage 

func (*CompleteInstanceResponse) ProtoMessage()

func (*CompleteInstanceResponse) ProtoReflect 

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

func (*CompleteInstanceResponse) Reset 

func (x *CompleteInstanceResponse) Reset()

func (*CompleteInstanceResponse) String 

func (x *CompleteInstanceResponse) String() string

type CompositeTensorVariantMetadata 

type CompositeTensorVariantMetadata struct {
	TypeSpecProto *TypeSpecProto `protobuf:"bytes,1,opt,name=type_spec_proto,json=typeSpecProto,proto3" json:"type_spec_proto,omitempty"`
	// contains filtered or unexported fields
}

Metadata for CompositeTensorVariant, used when serializing as Variant.

We define a new message here (rather than directly using TypeSpecProto for the metadata string) to retain flexibility to change the metadata encoding to support additional features.

func (*CompositeTensorVariantMetadata) Descriptor deprecated 

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

Deprecated: Use CompositeTensorVariantMetadata.ProtoReflect.Descriptor instead.

func (*CompositeTensorVariantMetadata) GetTypeSpecProto 

func (x *CompositeTensorVariantMetadata) GetTypeSpecProto() *TypeSpecProto

func (*CompositeTensorVariantMetadata) ProtoMessage 

func (*CompositeTensorVariantMetadata) ProtoMessage()

func (*CompositeTensorVariantMetadata) ProtoReflect 

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

func (*CompositeTensorVariantMetadata) Reset 

func (x *CompositeTensorVariantMetadata) Reset()

func (*CompositeTensorVariantMetadata) String 

func (x *CompositeTensorVariantMetadata) String() string

type ComputeCapability 

type ComputeCapability struct {
	Major int32 `protobuf:"varint,1,opt,name=major,proto3" json:"major,omitempty"`
	Minor int32 `protobuf:"varint,2,opt,name=minor,proto3" json:"minor,omitempty"`
	// contains filtered or unexported fields
}

func (*ComputeCapability) Descriptor deprecated 

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

Deprecated: Use ComputeCapability.ProtoReflect.Descriptor instead.

func (*ComputeCapability) GetMajor 

func (x *ComputeCapability) GetMajor() int32

func (*ComputeCapability) GetMinor 

func (x *ComputeCapability) GetMinor() int32

func (*ComputeCapability) ProtoMessage 

func (*ComputeCapability) ProtoMessage()

func (*ComputeCapability) ProtoReflect 

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

func (*ComputeCapability) Reset 

func (x *ComputeCapability) Reset()

func (*ComputeCapability) String 

func (x *ComputeCapability) String() string

type CondContextDef 

type CondContextDef struct {

	// Name of the context.
	ContextName string `protobuf:"bytes,1,opt,name=context_name,json=contextName,proto3" json:"context_name,omitempty"`
	// Name of the pred tensor.
	PredName string `protobuf:"bytes,2,opt,name=pred_name,json=predName,proto3" json:"pred_name,omitempty"`
	// Name of the pivot tensor.
	PivotName string `protobuf:"bytes,3,opt,name=pivot_name,json=pivotName,proto3" json:"pivot_name,omitempty"`
	// Branch prediction. 0 or 1.
	Branch int32 `protobuf:"varint,4,opt,name=branch,proto3" json:"branch,omitempty"`
	// Values and external values in control flow context.
	ValuesDef *ValuesDef `protobuf:"bytes,5,opt,name=values_def,json=valuesDef,proto3" json:"values_def,omitempty"`
	// Contexts contained inside this context (e.g. nested conds).
	NestedContexts []*ControlFlowContextDef `protobuf:"bytes,6,rep,name=nested_contexts,json=nestedContexts,proto3" json:"nested_contexts,omitempty"`
	// contains filtered or unexported fields
}

Protocol buffer representing a CondContext object.

func (*CondContextDef) Descriptor deprecated 

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

Deprecated: Use CondContextDef.ProtoReflect.Descriptor instead.

func (*CondContextDef) GetBranch 

func (x *CondContextDef) GetBranch() int32

func (*CondContextDef) GetContextName 

func (x *CondContextDef) GetContextName() string

func (*CondContextDef) GetNestedContexts 

func (x *CondContextDef) GetNestedContexts() []*ControlFlowContextDef

func (*CondContextDef) GetPivotName 

func (x *CondContextDef) GetPivotName() string

func (*CondContextDef) GetPredName 

func (x *CondContextDef) GetPredName() string

func (*CondContextDef) GetValuesDef 

func (x *CondContextDef) GetValuesDef() *ValuesDef

func (*CondContextDef) ProtoMessage 

func (*CondContextDef) ProtoMessage()

func (*CondContextDef) ProtoReflect 

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

func (*CondContextDef) Reset 

func (x *CondContextDef) Reset()

func (*CondContextDef) String 

func (x *CondContextDef) String() string

type ConfigProto 

type ConfigProto struct {

	// Map from device type name (e.g., "CPU" or "GPU" ) to maximum
	// number of devices of that type to use.  If a particular device
	// type is not found in the map, the system picks an appropriate
	// number.
	DeviceCount map[string]int32 `` /* 183-byte string literal not displayed */
	// The execution of an individual op (for some op types) can be
	// parallelized on a pool of intra_op_parallelism_threads.
	// 0 means the system picks an appropriate number.
	//
	// If you create an ordinary session, e.g., from Python or C++,
	// then there is exactly one intra op thread pool per process.
	// The first session created determines the number of threads in this pool.
	// All subsequent sessions reuse/share this one global pool.
	//
	// There are notable exceptions to the default behavior described above:
	// 1. There is an environment variable  for overriding this thread pool,
	//    named TF_OVERRIDE_GLOBAL_THREADPOOL.
	// 2. When connecting to a server, such as a remote `tf.train.Server`
	//    instance, then this option will be ignored altogether.
	IntraOpParallelismThreads int32 `` /* 141-byte string literal not displayed */
	// Nodes that perform blocking operations are enqueued on a pool of
	// inter_op_parallelism_threads available in each process.
	//
	// 0 means the system picks an appropriate number.
	// Negative means all operations are performed in caller's thread.
	//
	// Note that the first Session created in the process sets the
	// number of threads for all future sessions unless use_per_session_threads is
	// true or session_inter_op_thread_pool is configured.
	InterOpParallelismThreads int32 `` /* 141-byte string literal not displayed */
	// If true, use a new set of threads for this session rather than the global
	// pool of threads. Only supported by direct sessions.
	//
	// If false, use the global threads created by the first session, or the
	// per-session thread pools configured by session_inter_op_thread_pool.
	//
	// This option is deprecated. The same effect can be achieved by setting
	// session_inter_op_thread_pool to have one element, whose num_threads equals
	// inter_op_parallelism_threads.
	UsePerSessionThreads bool `` /* 126-byte string literal not displayed */
	// This option is experimental - it may be replaced with a different mechanism
	// in the future.
	//
	// Configures session thread pools. If this is configured, then RunOptions for
	// a Run call can select the thread pool to use.
	//
	// The intended use is for when some session invocations need to run in a
	// background pool limited to a small number of threads:
	// - For example, a session may be configured to have one large pool (for
	// regular compute) and one small pool (for periodic, low priority work);
	// using the small pool is currently the mechanism for limiting the inter-op
	// parallelism of the low priority work.  Note that it does not limit the
	// parallelism of work spawned by a single op kernel implementation.
	// - Using this setting is normally not needed in training, but may help some
	// serving use cases.
	// - It is also generally recommended to set the global_name field of this
	// proto, to avoid creating multiple large pools. It is typically better to
	// run the non-low-priority work, even across sessions, in a single large
	// pool.
	SessionInterOpThreadPool []*ThreadPoolOptionProto `` /* 140-byte string literal not displayed */
	// Assignment of Nodes to Devices is recomputed every placement_period
	// steps until the system warms up (at which point the recomputation
	// typically slows down automatically).
	PlacementPeriod int32 `protobuf:"varint,3,opt,name=placement_period,json=placementPeriod,proto3" json:"placement_period,omitempty"`
	// When any filters are present sessions will ignore all devices which do not
	// match the filters. Each filter can be partially specified, e.g. "/job:ps"
	// "/job:worker/replica:3", etc.
	DeviceFilters []string `protobuf:"bytes,4,rep,name=device_filters,json=deviceFilters,proto3" json:"device_filters,omitempty"`
	// Options that apply to all GPUs.
	GpuOptions *GPUOptions `protobuf:"bytes,6,opt,name=gpu_options,json=gpuOptions,proto3" json:"gpu_options,omitempty"`
	// Whether soft placement is allowed. If allow_soft_placement is true,
	// an op will be placed on CPU if
	//   1. there's no GPU implementation for the OP
	// or
	//   2. no GPU devices are known or registered
	// or
	//   3. need to co-locate with reftype input(s) which are from CPU.
	AllowSoftPlacement bool `protobuf:"varint,7,opt,name=allow_soft_placement,json=allowSoftPlacement,proto3" json:"allow_soft_placement,omitempty"`
	// Whether device placements should be logged.
	LogDevicePlacement bool `protobuf:"varint,8,opt,name=log_device_placement,json=logDevicePlacement,proto3" json:"log_device_placement,omitempty"`
	// Options that apply to all graphs.
	GraphOptions *GraphOptions `protobuf:"bytes,10,opt,name=graph_options,json=graphOptions,proto3" json:"graph_options,omitempty"`
	// Global timeout for all blocking operations in this session.  If non-zero,
	// and not overridden on a per-operation basis, this value will be used as the
	// deadline for all blocking operations.
	OperationTimeoutInMs int64 `` /* 127-byte string literal not displayed */
	// Options that apply when this session uses the distributed runtime.
	RpcOptions *RPCOptions `protobuf:"bytes,13,opt,name=rpc_options,json=rpcOptions,proto3" json:"rpc_options,omitempty"`
	// Optional list of all workers to use in this session.
	ClusterDef *ClusterDef `protobuf:"bytes,14,opt,name=cluster_def,json=clusterDef,proto3" json:"cluster_def,omitempty"`
	// If true, any resources such as Variables used in the session will not be
	// shared with other sessions. However, when clusterspec propagation is
	// enabled, this field is ignored and sessions are always isolated.
	IsolateSessionState bool `protobuf:"varint,15,opt,name=isolate_session_state,json=isolateSessionState,proto3" json:"isolate_session_state,omitempty"`
	// When true, WorkerSessions are created with device attributes from the
	// full cluster.
	// This is helpful when a worker wants to partition a graph
	// (for example during a PartitionedCallOp).
	ShareClusterDevicesInSession bool                      `` /* 153-byte string literal not displayed */
	Experimental                 *ConfigProto_Experimental `protobuf:"bytes,16,opt,name=experimental,proto3" json:"experimental,omitempty"`
	// contains filtered or unexported fields
}

Session configuration parameters. The system picks appropriate values for fields that are not set.

func (*ConfigProto) Descriptor deprecated 

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

Deprecated: Use ConfigProto.ProtoReflect.Descriptor instead.

func (*ConfigProto) GetAllowSoftPlacement 

func (x *ConfigProto) GetAllowSoftPlacement() bool

func (*ConfigProto) GetClusterDef 

func (x *ConfigProto) GetClusterDef() *ClusterDef

func (*ConfigProto) GetDeviceCount 

func (x *ConfigProto) GetDeviceCount() map[string]int32

func (*ConfigProto) GetDeviceFilters 

func (x *ConfigProto) GetDeviceFilters() []string

func (*ConfigProto) GetExperimental 

func (x *ConfigProto) GetExperimental() *ConfigProto_Experimental

func (*ConfigProto) GetGpuOptions 

func (x *ConfigProto) GetGpuOptions() *GPUOptions

func (*ConfigProto) GetGraphOptions 

func (x *ConfigProto) GetGraphOptions() *GraphOptions

func (*ConfigProto) GetInterOpParallelismThreads 

func (x *ConfigProto) GetInterOpParallelismThreads() int32

func (*ConfigProto) GetIntraOpParallelismThreads 

func (x *ConfigProto) GetIntraOpParallelismThreads() int32

func (*ConfigProto) GetIsolateSessionState 

func (x *ConfigProto) GetIsolateSessionState() bool

func (*ConfigProto) GetLogDevicePlacement 

func (x *ConfigProto) GetLogDevicePlacement() bool

func (*ConfigProto) GetOperationTimeoutInMs 

func (x *ConfigProto) GetOperationTimeoutInMs() int64

func (*ConfigProto) GetPlacementPeriod 

func (x *ConfigProto) GetPlacementPeriod() int32

func (*ConfigProto) GetRpcOptions 

func (x *ConfigProto) GetRpcOptions() *RPCOptions

func (*ConfigProto) GetSessionInterOpThreadPool 

func (x *ConfigProto) GetSessionInterOpThreadPool() []*ThreadPoolOptionProto

func (*ConfigProto) GetShareClusterDevicesInSession 

func (x *ConfigProto) GetShareClusterDevicesInSession() bool

func (*ConfigProto) GetUsePerSessionThreads 

func (x *ConfigProto) GetUsePerSessionThreads() bool

func (*ConfigProto) ProtoMessage 

func (*ConfigProto) ProtoMessage()

func (*ConfigProto) ProtoReflect 

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

func (*ConfigProto) Reset 

func (x *ConfigProto) Reset()

func (*ConfigProto) String 

func (x *ConfigProto) String() string

type ConfigProto_Experimental 

type ConfigProto_Experimental struct {

	// Task name for group resolution.
	CollectiveGroupLeader string `` /* 126-byte string literal not displayed */
	// Which executor to use, the default executor will be used
	// if it is an empty string or "DEFAULT"
	ExecutorType string `protobuf:"bytes,3,opt,name=executor_type,json=executorType,proto3" json:"executor_type,omitempty"`
	// Guidance to formatting of large RecvBuf fields for transfer.
	// Any positive value sets the max chunk size.  0 defaults to 4096.
	// Any negative value indicates no max, i.e. one chunk only.
	RecvBufMaxChunk int32 `protobuf:"varint,4,opt,name=recv_buf_max_chunk,json=recvBufMaxChunk,proto3" json:"recv_buf_max_chunk,omitempty"`
	// If true, and supported by the platform, the runtime will attempt to
	// use NUMA affinity where applicable.  One consequence will be the
	// existence of as many CPU devices as there are available NUMA nodes.
	UseNumaAffinity bool `protobuf:"varint,5,opt,name=use_numa_affinity,json=useNumaAffinity,proto3" json:"use_numa_affinity,omitempty"`
	// If true, make collective op execution order sequential and deterministic
	// for potentially concurrent collective instances.
	CollectiveDeterministicSequentialExecution bool `` /* 192-byte string literal not displayed */
	// If true, use NCCL for CollectiveOps.  This feature is highly
	// experimental.
	CollectiveNccl bool `protobuf:"varint,7,opt,name=collective_nccl,json=collectiveNccl,proto3" json:"collective_nccl,omitempty"`
	// In the following, session state means the value of a variable, elements
	// in a hash table, or any other resource, accessible by worker sessions
	// held by a TF server.
	//
	// When ClusterSpec propagation is enabled, the value of
	// isolate_session_state is ignored when deciding whether to share session
	// states in a TF server (for backwards compatibility reasons).
	// - If share_session_state_in_clusterspec_propagation is true, the session
	// states are shared.
	// - If share_session_state_in_clusterspec_propagation is false, session
	// states are isolated.
	//
	// When clusterspec propagation is not used, the value of
	// share_session_state_in_clusterspec_propagation is ignored when deciding
	// whether to share session states in a TF server.
	// - If isolate_session_state is true, session states are isolated.
	// - If isolate_session_state is false, session states are shared.
	//
	// TODO(b/129330037): Add a single API that consistently treats
	// isolate_session_state and ClusterSpec propagation.
	ShareSessionStateInClusterspecPropagation bool `` /* 193-byte string literal not displayed */
	// If using a direct session, disable spinning while waiting for work in
	// the thread pool. This may result in higher latency for completing ops,
	// but in the case where there is a lot of spinning may result in lower
	// CPU usage.
	DisableThreadSpinning bool `` /* 127-byte string literal not displayed */
	// This was promoted to a non-experimental API. Please use
	// ConfigProto.share_cluster_devices_in_session instead.
	ShareClusterDevicesInSession bool `` /* 153-byte string literal not displayed */
	// Metadata about the session.
	//
	// If set, this can be used by the runtime and the Ops for debugging,
	// monitoring, etc.
	//
	// NOTE: This is currently used and propagated only by the direct session.
	SessionMetadata *SessionMetadata `protobuf:"bytes,11,opt,name=session_metadata,json=sessionMetadata,proto3" json:"session_metadata,omitempty"`
	// If true, the session may treat the graph as being static for optimization
	// purposes.
	//
	// If this option is set to true when a session is created, the full
	// GraphDef must be passed in a single call to Session::Create(), and
	// Session::Extend() may not be supported.
	OptimizeForStaticGraph bool `` /* 133-byte string literal not displayed */
	// This field will eventually be deprecated and replaced by
	// mlir_bridge_rollout (b/166038521).
	//
	// Whether to enable the MLIR-based TF->XLA bridge.
	//
	// This is a replacement to the existing bridge, and not ready for
	// production usage yet.
	// If this option is set to true when a session is created, MLIR is used to
	// perform the set of graph transformations to put the graph in a form that
	// can be executed with delegation of some computations to an accelerator.
	// This builds on the model of XLA where a subset of the graph is
	// encapsulated and attached to a "compile" operation, whose result is fed
	// to an "execute" operation. The kernel for these operations is responsible
	// to lower the encapsulated graph to a particular device.
	EnableMlirBridge bool `protobuf:"varint,13,opt,name=enable_mlir_bridge,json=enableMlirBridge,proto3" json:"enable_mlir_bridge,omitempty"`
	// This field is underdevelopment, for now use enable_mlir_bridge
	// (b/166038521).
	//
	// Whether to enable the MLIR-based TF->XLA bridge.
	MlirBridgeRollout ConfigProto_Experimental_MlirBridgeRollout `` /* 175-byte string literal not displayed */
	// Whether to enable the MLIR-based Graph optimizations.
	//
	// This will become a part of standard Tensorflow graph optimization
	// pipeline, currently this is only used for gradual migration and testing
	// new passes that are replacing existing optimizations in Grappler.
	EnableMlirGraphOptimization bool `` /* 148-byte string literal not displayed */
	// If true, the session will not store an additional copy of the graph for
	// each subgraph.
	//
	// If this option is set to true when a session is created, the
	// `RunOptions.output_partition_graphs` options must not be set.
	DisableOutputPartitionGraphs bool `` /* 151-byte string literal not displayed */
	// Minimum number of batches run through the XLA graph before XLA fusion
	// autotuner is enabled. Default value of zero disables the autotuner.
	//
	// The XLA fusion autotuner can improve performance by executing a heuristic
	// search on the compiler parameters.
	XlaFusionAutotunerThresh int64 `` /* 139-byte string literal not displayed */
	// Whether runtime execution uses TFRT.
	UseTfrt bool `protobuf:"varint,18,opt,name=use_tfrt,json=useTfrt,proto3" json:"use_tfrt,omitempty"`
	// Whether functional control flow op lowering should be disabled. This is
	// useful when executing within a portable runtime where control flow op
	// kernels may not be loaded due to selective registration.
	DisableFunctionalOpsLowering bool `` /* 151-byte string literal not displayed */
	// Provides a hint to XLA auto clustering to prefer forming a single large
	// cluster that encompases most of the graph.
	XlaPreferSingleGraphCluster bool `` /* 150-byte string literal not displayed */
	// Distributed coordination service configurations.
	CoordinationConfig *CoordinationServiceConfig `protobuf:"bytes,23,opt,name=coordination_config,json=coordinationConfig,proto3" json:"coordination_config,omitempty"`
	// contains filtered or unexported fields
}

Everything inside Experimental is subject to change and is not subject to API stability guarantees in https://www.tensorflow.org/guide/version_compat.

func (*ConfigProto_Experimental) Descriptor deprecated 

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

Deprecated: Use ConfigProto_Experimental.ProtoReflect.Descriptor instead.

func (*ConfigProto_Experimental) GetCollectiveDeterministicSequentialExecution 

func (x *ConfigProto_Experimental) GetCollectiveDeterministicSequentialExecution() bool

func (*ConfigProto_Experimental) GetCollectiveGroupLeader 

func (x *ConfigProto_Experimental) GetCollectiveGroupLeader() string

func (*ConfigProto_Experimental) GetCollectiveNccl 

func (x *ConfigProto_Experimental) GetCollectiveNccl() bool

func (*ConfigProto_Experimental) GetCoordinationConfig 

func (x *ConfigProto_Experimental) GetCoordinationConfig() *CoordinationServiceConfig

func (*ConfigProto_Experimental) GetDisableFunctionalOpsLowering 

func (x *ConfigProto_Experimental) GetDisableFunctionalOpsLowering() bool

func (*ConfigProto_Experimental) GetDisableOutputPartitionGraphs 

func (x *ConfigProto_Experimental) GetDisableOutputPartitionGraphs() bool

func (*ConfigProto_Experimental) GetDisableThreadSpinning 

func (x *ConfigProto_Experimental) GetDisableThreadSpinning() bool

func (*ConfigProto_Experimental) GetEnableMlirBridge 

func (x *ConfigProto_Experimental) GetEnableMlirBridge() bool

func (*ConfigProto_Experimental) GetEnableMlirGraphOptimization 

func (x *ConfigProto_Experimental) GetEnableMlirGraphOptimization() bool

func (*ConfigProto_Experimental) GetExecutorType 

func (x *ConfigProto_Experimental) GetExecutorType() string

func (*ConfigProto_Experimental) GetMlirBridgeRollout 

func (x *ConfigProto_Experimental) GetMlirBridgeRollout() ConfigProto_Experimental_MlirBridgeRollout

func (*ConfigProto_Experimental) GetOptimizeForStaticGraph 

func (x *ConfigProto_Experimental) GetOptimizeForStaticGraph() bool

func (*ConfigProto_Experimental) GetRecvBufMaxChunk 

func (x *ConfigProto_Experimental) GetRecvBufMaxChunk() int32

func (*ConfigProto_Experimental) GetSessionMetadata 

func (x *ConfigProto_Experimental) GetSessionMetadata() *SessionMetadata

func (*ConfigProto_Experimental) GetShareClusterDevicesInSession 

func (x *ConfigProto_Experimental) GetShareClusterDevicesInSession() bool

func (*ConfigProto_Experimental) GetShareSessionStateInClusterspecPropagation 

func (x *ConfigProto_Experimental) GetShareSessionStateInClusterspecPropagation() bool

func (*ConfigProto_Experimental) GetUseNumaAffinity 

func (x *ConfigProto_Experimental) GetUseNumaAffinity() bool

func (*ConfigProto_Experimental) GetUseTfrt 

func (x *ConfigProto_Experimental) GetUseTfrt() bool

func (*ConfigProto_Experimental) GetXlaFusionAutotunerThresh 

func (x *ConfigProto_Experimental) GetXlaFusionAutotunerThresh() int64

func (*ConfigProto_Experimental) GetXlaPreferSingleGraphCluster 

func (x *ConfigProto_Experimental) GetXlaPreferSingleGraphCluster() bool

func (*ConfigProto_Experimental) ProtoMessage 

func (*ConfigProto_Experimental) ProtoMessage()

func (*ConfigProto_Experimental) ProtoReflect 

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

func (*ConfigProto_Experimental) Reset 

func (x *ConfigProto_Experimental) Reset()

func (*ConfigProto_Experimental) String 

func (x *ConfigProto_Experimental) String() string

type ConfigProto_Experimental_MlirBridgeRollout 

type ConfigProto_Experimental_MlirBridgeRollout int32

An enum that describes the state of the MLIR bridge rollout. 

const (
	// If this field is left unspecified, the MLIR bridge may be selectively
	// enabled on a per graph basis.
	ConfigProto_Experimental_MLIR_BRIDGE_ROLLOUT_UNSPECIFIED ConfigProto_Experimental_MlirBridgeRollout = 0
	// Enabling the MLIR bridge enables it for all graphs in this session.
	ConfigProto_Experimental_MLIR_BRIDGE_ROLLOUT_ENABLED ConfigProto_Experimental_MlirBridgeRollout = 1
	// Disabling the MLIR bridge disables it for all graphs in this session.
	ConfigProto_Experimental_MLIR_BRIDGE_ROLLOUT_DISABLED ConfigProto_Experimental_MlirBridgeRollout = 2
	// Enable the MLIR bridge on a per graph basis based on an analysis of
	// the features used in the graph. If the features used by the graph are
	// supported by the MLIR bridge, the MLIR bridge will be used to run the
	// graph.
	ConfigProto_Experimental_MLIR_BRIDGE_ROLLOUT_SAFE_MODE_ENABLED ConfigProto_Experimental_MlirBridgeRollout = 3
	// Enable the MLIR bridge in a fallback mode on a per graph basis based
	// on an analysis of the features used in the graph.
	// Running the MLIR bridge in the fallback mode means that it is
	// executed and it commits all the changes to the TF graph in case
	// of success. And it does not in case of failures and let the old bridge
	// to process the TF graph.
	ConfigProto_Experimental_MLIR_BRIDGE_ROLLOUT_SAFE_MODE_FALLBACK_ENABLED ConfigProto_Experimental_MlirBridgeRollout = 4
)

func (ConfigProto_Experimental_MlirBridgeRollout) Descriptor 

func (ConfigProto_Experimental_MlirBridgeRollout) Descriptor() protoreflect.EnumDescriptor

func (ConfigProto_Experimental_MlirBridgeRollout) Enum 

func (x ConfigProto_Experimental_MlirBridgeRollout) Enum() *ConfigProto_Experimental_MlirBridgeRollout

func (ConfigProto_Experimental_MlirBridgeRollout) EnumDescriptor deprecated 

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

Deprecated: Use ConfigProto_Experimental_MlirBridgeRollout.Descriptor instead.

func (ConfigProto_Experimental_MlirBridgeRollout) Number 

func (x ConfigProto_Experimental_MlirBridgeRollout) Number() protoreflect.EnumNumber

func (ConfigProto_Experimental_MlirBridgeRollout) String 

func (x ConfigProto_Experimental_MlirBridgeRollout) String() string

func (ConfigProto_Experimental_MlirBridgeRollout) Type 

func (ConfigProto_Experimental_MlirBridgeRollout) Type() protoreflect.EnumType

type ConnectRequest 

type ConnectRequest struct {
	ProtocolVersion     int32 `protobuf:"varint,1,opt,name=protocol_version,json=protocolVersion,proto3" json:"protocol_version,omitempty"`
	TimeoutMilliseconds int32 `protobuf:"varint,2,opt,name=timeout_milliseconds,json=timeoutMilliseconds,proto3" json:"timeout_milliseconds,omitempty"`
	// We assume that each node knows its globally-unique node ID, provided by
	// whatever mechanism launches the tasks. Node IDs should form a dense range
	// of integers [0, num_nodes).
	NodeId int32 `protobuf:"varint,3,opt,name=node_id,json=nodeId,proto3" json:"node_id,omitempty"`
	// A unique ID number for the client.
	ClientId uint64 `protobuf:"varint,4,opt,name=client_id,json=clientId,proto3" json:"client_id,omitempty"`
	// contains filtered or unexported fields
}

func (*ConnectRequest) Descriptor deprecated 

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

Deprecated: Use ConnectRequest.ProtoReflect.Descriptor instead.

func (*ConnectRequest) GetClientId 

func (x *ConnectRequest) GetClientId() uint64

func (*ConnectRequest) GetNodeId 

func (x *ConnectRequest) GetNodeId() int32

func (*ConnectRequest) GetProtocolVersion 

func (x *ConnectRequest) GetProtocolVersion() int32

func (*ConnectRequest) GetTimeoutMilliseconds 

func (x *ConnectRequest) GetTimeoutMilliseconds() int32

func (*ConnectRequest) ProtoMessage 

func (*ConnectRequest) ProtoMessage()

func (*ConnectRequest) ProtoReflect 

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

func (*ConnectRequest) Reset 

func (x *ConnectRequest) Reset()

func (*ConnectRequest) String 

func (x *ConnectRequest) String() string

type ConnectResponse 

type ConnectResponse struct {
	SessionId uint64 `protobuf:"varint,1,opt,name=session_id,json=sessionId,proto3" json:"session_id,omitempty"`
	// contains filtered or unexported fields
}

func (*ConnectResponse) Descriptor deprecated 

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

Deprecated: Use ConnectResponse.ProtoReflect.Descriptor instead.

func (*ConnectResponse) GetSessionId 

func (x *ConnectResponse) GetSessionId() uint64

func (*ConnectResponse) ProtoMessage 

func (*ConnectResponse) ProtoMessage()

func (*ConnectResponse) ProtoReflect 

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

func (*ConnectResponse) Reset 

func (x *ConnectResponse) Reset()

func (*ConnectResponse) String 

func (x *ConnectResponse) String() string

type ControlFlowContextDef 

type ControlFlowContextDef struct {

	// Types that are assignable to Ctxt:
	//	*ControlFlowContextDef_CondCtxt
	//	*ControlFlowContextDef_WhileCtxt
	Ctxt isControlFlowContextDef_Ctxt `protobuf_oneof:"ctxt"`
	// contains filtered or unexported fields
}

Container for any kind of control flow context. Any other control flow contexts that are added below should also be added here.

func (*ControlFlowContextDef) Descriptor deprecated 

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

Deprecated: Use ControlFlowContextDef.ProtoReflect.Descriptor instead.

func (*ControlFlowContextDef) GetCondCtxt 

func (x *ControlFlowContextDef) GetCondCtxt() *CondContextDef

func (*ControlFlowContextDef) GetCtxt 

func (m *ControlFlowContextDef) GetCtxt() isControlFlowContextDef_Ctxt

func (*ControlFlowContextDef) GetWhileCtxt 

func (x *ControlFlowContextDef) GetWhileCtxt() *WhileContextDef

func (*ControlFlowContextDef) ProtoMessage 

func (*ControlFlowContextDef) ProtoMessage()

func (*ControlFlowContextDef) ProtoReflect 

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

func (*ControlFlowContextDef) Reset 

func (x *ControlFlowContextDef) Reset()

func (*ControlFlowContextDef) String 

func (x *ControlFlowContextDef) String() string

type ControlFlowContextDef_CondCtxt 

type ControlFlowContextDef_CondCtxt struct {
	CondCtxt *CondContextDef `protobuf:"bytes,1,opt,name=cond_ctxt,json=condCtxt,proto3,oneof"`
}

type ControlFlowContextDef_WhileCtxt 

type ControlFlowContextDef_WhileCtxt struct {
	WhileCtxt *WhileContextDef `protobuf:"bytes,2,opt,name=while_ctxt,json=whileCtxt,proto3,oneof"`
}

type ConvolutionDescriptorProto 

type ConvolutionDescriptorProto struct {
	Paddings  []int64 `protobuf:"varint,1,rep,packed,name=paddings,proto3" json:"paddings,omitempty"`
	Strides   []int64 `protobuf:"varint,2,rep,packed,name=strides,proto3" json:"strides,omitempty"`
	Dilations []int64 `protobuf:"varint,3,rep,packed,name=dilations,proto3" json:"dilations,omitempty"`
	// The "accumulator" type. For example, use F32 as an accumulator for F16
	// convolutions.
	// See cuDNN's cudnnConvolutionMode_t.
	ComputeMode DnnDataType `` /* 129-byte string literal not displayed */
	// See cuDNN's group count.
	GroupCount      int32           `protobuf:"varint,5,opt,name=group_count,json=groupCount,proto3" json:"group_count,omitempty"`
	ConvolutionMode ConvolutionMode `` /* 148-byte string literal not displayed */
	// Tensorflow node name, same as in NodeDef, for debugging purposes.
	Name string `protobuf:"bytes,7,opt,name=name,proto3" json:"name,omitempty"`
	// contains filtered or unexported fields
}

Convolution-specific parameters.

func (*ConvolutionDescriptorProto) Descriptor deprecated 

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

Deprecated: Use ConvolutionDescriptorProto.ProtoReflect.Descriptor instead.

func (*ConvolutionDescriptorProto) GetComputeMode 

func (x *ConvolutionDescriptorProto) GetComputeMode() DnnDataType

func (*ConvolutionDescriptorProto) GetConvolutionMode 

func (x *ConvolutionDescriptorProto) GetConvolutionMode() ConvolutionMode

func (*ConvolutionDescriptorProto) GetDilations 

func (x *ConvolutionDescriptorProto) GetDilations() []int64

func (*ConvolutionDescriptorProto) GetGroupCount 

func (x *ConvolutionDescriptorProto) GetGroupCount() int32

func (*ConvolutionDescriptorProto) GetName 

func (x *ConvolutionDescriptorProto) GetName() string

func (*ConvolutionDescriptorProto) GetPaddings 

func (x *ConvolutionDescriptorProto) GetPaddings() []int64

func (*ConvolutionDescriptorProto) GetStrides 

func (x *ConvolutionDescriptorProto) GetStrides() []int64

func (*ConvolutionDescriptorProto) ProtoMessage 

func (*ConvolutionDescriptorProto) ProtoMessage()

func (*ConvolutionDescriptorProto) ProtoReflect 

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

func (*ConvolutionDescriptorProto) Reset 

func (x *ConvolutionDescriptorProto) Reset()

func (*ConvolutionDescriptorProto) String 

func (x *ConvolutionDescriptorProto) String() string

type ConvolutionKind 

type ConvolutionKind int32
const (
	ConvolutionKind_INVALID                 ConvolutionKind = 0
	ConvolutionKind_FORWARD                 ConvolutionKind = 1
	ConvolutionKind_BACKWARD_FILTER         ConvolutionKind = 2
	ConvolutionKind_BACKWARD_DATA           ConvolutionKind = 3
	ConvolutionKind_FORWARD_BIAS_ACTIVATION ConvolutionKind = 4
)

func (ConvolutionKind) Descriptor 

func (ConvolutionKind) Descriptor() protoreflect.EnumDescriptor

func (ConvolutionKind) Enum 

func (x ConvolutionKind) Enum() *ConvolutionKind

func (ConvolutionKind) EnumDescriptor deprecated 

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

Deprecated: Use ConvolutionKind.Descriptor instead.

func (ConvolutionKind) Number 

func (x ConvolutionKind) Number() protoreflect.EnumNumber

func (ConvolutionKind) String 

func (x ConvolutionKind) String() string

func (ConvolutionKind) Type 

func (ConvolutionKind) Type() protoreflect.EnumType

type ConvolutionMode 

type ConvolutionMode int32

Describe the math definition for the conv op. The popular behavior is actually called cross-correlation in math, despite the operation is often referred as convolution. See cuDNN cudnnConvolutionMode_t. 

const (
	ConvolutionMode_CROSS_CORRELATION ConvolutionMode = 0
	ConvolutionMode_CONVOLUTION       ConvolutionMode = 1
)

func (ConvolutionMode) Descriptor 

func (ConvolutionMode) Descriptor() protoreflect.EnumDescriptor

func (ConvolutionMode) Enum 

func (x ConvolutionMode) Enum() *ConvolutionMode

func (ConvolutionMode) EnumDescriptor deprecated 

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

Deprecated: Use ConvolutionMode.Descriptor instead.

func (ConvolutionMode) Number 

func (x ConvolutionMode) Number() protoreflect.EnumNumber

func (ConvolutionMode) String 

func (x ConvolutionMode) String() string

func (ConvolutionMode) Type 

func (ConvolutionMode) Type() protoreflect.EnumType

type ConvolutionProto 

type ConvolutionProto struct {
	Kind     ConvolutionKind             `protobuf:"varint,1,opt,name=kind,proto3,enum=dnn.ConvolutionKind" json:"kind,omitempty"`
	Input    *TensorDescriptorProto      `protobuf:"bytes,2,opt,name=input,proto3" json:"input,omitempty"`
	Filter   *TensorDescriptorProto      `protobuf:"bytes,3,opt,name=filter,proto3" json:"filter,omitempty"`
	Output   *TensorDescriptorProto      `protobuf:"bytes,4,opt,name=output,proto3" json:"output,omitempty"`
	ConvDesc *ConvolutionDescriptorProto `protobuf:"bytes,5,opt,name=conv_desc,json=convDesc,proto3" json:"conv_desc,omitempty"`
	// result = conv_scale * conv(...) + side_value_scale * side_value.
	// side_value is an arbitrary buffer if activation is not none. Otherwise, it
	// has to be the result buffer (using its old values).
	ConvScale        float64        `protobuf:"fixed64,6,opt,name=conv_scale,json=convScale,proto3" json:"conv_scale,omitempty"`
	SideValueScale   float64        `protobuf:"fixed64,7,opt,name=side_value_scale,json=sideValueScale,proto3" json:"side_value_scale,omitempty"`
	Activation       ActivationMode `protobuf:"varint,8,opt,name=activation,proto3,enum=dnn.ActivationMode" json:"activation,omitempty"`
	InputAddress     int64          `protobuf:"varint,9,opt,name=input_address,json=inputAddress,proto3" json:"input_address,omitempty"`
	FilterAddress    int64          `protobuf:"varint,10,opt,name=filter_address,json=filterAddress,proto3" json:"filter_address,omitempty"`
	OutputAddress    int64          `protobuf:"varint,11,opt,name=output_address,json=outputAddress,proto3" json:"output_address,omitempty"`
	BiasAddress      int64          `protobuf:"varint,12,opt,name=bias_address,json=biasAddress,proto3" json:"bias_address,omitempty"`
	SideInputAddress int64          `protobuf:"varint,13,opt,name=side_input_address,json=sideInputAddress,proto3" json:"side_input_address,omitempty"`
	// contains filtered or unexported fields
}

A convolution. Currently it's only used for logging. In the future, we may want to use it in the API as well.

func (*ConvolutionProto) Descriptor deprecated 

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

Deprecated: Use ConvolutionProto.ProtoReflect.Descriptor instead.

func (*ConvolutionProto) GetActivation 

func (x *ConvolutionProto) GetActivation() ActivationMode

func (*ConvolutionProto) GetBiasAddress 

func (x *ConvolutionProto) GetBiasAddress() int64

func (*ConvolutionProto) GetConvDesc 

func (x *ConvolutionProto) GetConvDesc() *ConvolutionDescriptorProto

func (*ConvolutionProto) GetConvScale 

func (x *ConvolutionProto) GetConvScale() float64

func (*ConvolutionProto) GetFilter 

func (x *ConvolutionProto) GetFilter() *TensorDescriptorProto

func (*ConvolutionProto) GetFilterAddress 

func (x *ConvolutionProto) GetFilterAddress() int64

func (*ConvolutionProto) GetInput 

func (x *ConvolutionProto) GetInput() *TensorDescriptorProto

func (*ConvolutionProto) GetInputAddress 

func (x *ConvolutionProto) GetInputAddress() int64

func (*ConvolutionProto) GetKind 

func (x *ConvolutionProto) GetKind() ConvolutionKind

func (*ConvolutionProto) GetOutput 

func (x *ConvolutionProto) GetOutput() *TensorDescriptorProto

func (*ConvolutionProto) GetOutputAddress 

func (x *ConvolutionProto) GetOutputAddress() int64

func (*ConvolutionProto) GetSideInputAddress 

func (x *ConvolutionProto) GetSideInputAddress() int64

func (*ConvolutionProto) GetSideValueScale 

func (x *ConvolutionProto) GetSideValueScale() float64

func (*ConvolutionProto) ProtoMessage 

func (*ConvolutionProto) ProtoMessage()

func (*ConvolutionProto) ProtoReflect 

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

func (*ConvolutionProto) Reset 

func (x *ConvolutionProto) Reset()

func (*ConvolutionProto) String 

func (x *ConvolutionProto) String() string

type CoordinatedTask 

type CoordinatedTask struct {
	JobName string `protobuf:"bytes,1,opt,name=job_name,json=jobName,proto3" json:"job_name,omitempty"`
	TaskId  int32  `protobuf:"varint,2,opt,name=task_id,json=taskId,proto3" json:"task_id,omitempty"`
	// contains filtered or unexported fields
}

Represents a remote worker task, specified by job name and task id.

func (*CoordinatedTask) Descriptor deprecated 

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

Deprecated: Use CoordinatedTask.ProtoReflect.Descriptor instead.

func (*CoordinatedTask) GetJobName 

func (x *CoordinatedTask) GetJobName() string

func (*CoordinatedTask) GetTaskId 

func (x *CoordinatedTask) GetTaskId() int32

func (*CoordinatedTask) ProtoMessage 

func (*CoordinatedTask) ProtoMessage()

func (*CoordinatedTask) ProtoReflect 

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

func (*CoordinatedTask) Reset 

func (x *CoordinatedTask) Reset()

func (*CoordinatedTask) String 

func (x *CoordinatedTask) String() string

type CoordinatedTaskState 

type CoordinatedTaskState int32

Represents the state of a remote worker 

const (
	// TASKSTATE_UNSPECIFIED is an invalid state such that indicates a bug.
	CoordinatedTaskState_TASKSTATE_UNSPECIFIED CoordinatedTaskState = 0
	// TASKSTATE_UNINITIALIZED is an agent-only state. While the agent is
	// disconnected, the service has no way of knowing if the task is
	// initialized/uninitialized.
	CoordinatedTaskState_TASKSTATE_UNINITIALIZED CoordinatedTaskState = 1
	CoordinatedTaskState_TASKSTATE_DISCONNECTED  CoordinatedTaskState = 2
	CoordinatedTaskState_TASKSTATE_CONNECTED     CoordinatedTaskState = 3
	CoordinatedTaskState_TASKSTATE_ERROR         CoordinatedTaskState = 4
)

func (CoordinatedTaskState) Descriptor 

func (CoordinatedTaskState) Descriptor() protoreflect.EnumDescriptor

func (CoordinatedTaskState) Enum 

func (x CoordinatedTaskState) Enum() *CoordinatedTaskState

func (CoordinatedTaskState) EnumDescriptor deprecated 

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

Deprecated: Use CoordinatedTaskState.Descriptor instead.

func (CoordinatedTaskState) Number 

func (x CoordinatedTaskState) Number() protoreflect.EnumNumber

func (CoordinatedTaskState) String 

func (x CoordinatedTaskState) String() string

func (CoordinatedTaskState) Type 

func (CoordinatedTaskState) Type() protoreflect.EnumType

type CoordinationServiceConfig 

type CoordinationServiceConfig struct {

	// Type of coordination service implementation to enable.
	// For example, setting the service type as "standalone" starts a service
	// instance on the leader task to provide the coordination services such as
	// heartbeats and consistent key-value store.
	ServiceType string `protobuf:"bytes,1,opt,name=service_type,json=serviceType,proto3" json:"service_type,omitempty"`
	// Address where the coordination service instance is hosted.
	ServiceLeader string `protobuf:"bytes,2,opt,name=service_leader,json=serviceLeader,proto3" json:"service_leader,omitempty"`
	// Whether to enable the health check mechanism.
	EnableHealthCheck bool `protobuf:"varint,3,opt,name=enable_health_check,json=enableHealthCheck,proto3" json:"enable_health_check,omitempty"`
	// Maximum wait time for all members in the cluster to be registered.
	ClusterRegisterTimeoutInMs int64 `` /* 146-byte string literal not displayed */
	// Heartbeat timeout, if a task does not record heartbeat in this time
	// window, it will be considered disconnected.
	// Note: This is also used as a grace period to accept any heartbeats after
	// the agent has disconnected, to account for the lag time between the service
	// recording the state change and the agent stopping heartbeats.
	HeartbeatTimeoutInMs int64 `` /* 126-byte string literal not displayed */
	// The list of jobs that partipate in the coordination service. If empty, all
	// jobs will be included in the coordination service by default.
	CoordinatedJobs []string `protobuf:"bytes,6,rep,name=coordinated_jobs,json=coordinatedJobs,proto3" json:"coordinated_jobs,omitempty"`
	// Denotes how long to wait for all coordination agents to reach the barriers
	// (after the first shutdown request) before disconnecting together. If
	// set to 0, no barrier is imposed upon shutdown and each worker can
	// disconnect individually.
	ShutdownBarrierTimeoutInMs int64 `` /* 146-byte string literal not displayed */
	// If set, agents do not make an explicit Shutdown() call. Service will only
	// find out about the disconnecte agent via stale heartbeats. Used for
	// testing.
	AgentDestructionWithoutShutdown bool `` /* 159-byte string literal not displayed */
	// The list of jobs which are recoverable. If a task in this list fails,
	// it will not propagate error to other tasks.
	// If empty, no jobs will be recoverable and every task failure will cause
	// error propagation to other tasks.
	RecoverableJobs []string `protobuf:"bytes,9,rep,name=recoverable_jobs,json=recoverableJobs,proto3" json:"recoverable_jobs,omitempty"`
	// contains filtered or unexported fields
}

Coordination service configuration parameters. The system picks appropriate values for fields that are not set.

func (*CoordinationServiceConfig) Descriptor deprecated 

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

Deprecated: Use CoordinationServiceConfig.ProtoReflect.Descriptor instead.

func (*CoordinationServiceConfig) GetAgentDestructionWithoutShutdown 

func (x *CoordinationServiceConfig) GetAgentDestructionWithoutShutdown() bool

func (*CoordinationServiceConfig) GetClusterRegisterTimeoutInMs 

func (x *CoordinationServiceConfig) GetClusterRegisterTimeoutInMs() int64

func (*CoordinationServiceConfig) GetCoordinatedJobs 

func (x *CoordinationServiceConfig) GetCoordinatedJobs() []string

func (*CoordinationServiceConfig) GetEnableHealthCheck 

func (x *CoordinationServiceConfig) GetEnableHealthCheck() bool

func (*CoordinationServiceConfig) GetHeartbeatTimeoutInMs 

func (x *CoordinationServiceConfig) GetHeartbeatTimeoutInMs() int64

func (*CoordinationServiceConfig) GetRecoverableJobs 

func (x *CoordinationServiceConfig) GetRecoverableJobs() []string

func (*CoordinationServiceConfig) GetServiceLeader 

func (x *CoordinationServiceConfig) GetServiceLeader() string

func (*CoordinationServiceConfig) GetServiceType 

func (x *CoordinationServiceConfig) GetServiceType() string

func (*CoordinationServiceConfig) GetShutdownBarrierTimeoutInMs 

func (x *CoordinationServiceConfig) GetShutdownBarrierTimeoutInMs() int64

func (*CoordinationServiceConfig) ProtoMessage 

func (*CoordinationServiceConfig) ProtoMessage()

func (*CoordinationServiceConfig) ProtoReflect 

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

func (*CoordinationServiceConfig) Reset 

func (x *CoordinationServiceConfig) Reset()

func (*CoordinationServiceConfig) String 

func (x *CoordinationServiceConfig) String() string

type CoordinationServiceDeviceInfo 

type CoordinationServiceDeviceInfo struct {

	// Types that are assignable to Type:
	//	*CoordinationServiceDeviceInfo_Tf
	//	*CoordinationServiceDeviceInfo_Xla
	Type isCoordinationServiceDeviceInfo_Type `protobuf_oneof:"type"`
	// contains filtered or unexported fields
}

func (*CoordinationServiceDeviceInfo) Descriptor deprecated 

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

Deprecated: Use CoordinationServiceDeviceInfo.ProtoReflect.Descriptor instead.

func (*CoordinationServiceDeviceInfo) GetTf 

func (x *CoordinationServiceDeviceInfo) GetTf() *TfDeviceList

func (*CoordinationServiceDeviceInfo) GetType 

func (m *CoordinationServiceDeviceInfo) GetType() isCoordinationServiceDeviceInfo_Type

func (*CoordinationServiceDeviceInfo) GetXla 

func (x *CoordinationServiceDeviceInfo) GetXla() *XlaDeviceList

func (*CoordinationServiceDeviceInfo) ProtoMessage 

func (*CoordinationServiceDeviceInfo) ProtoMessage()

func (*CoordinationServiceDeviceInfo) ProtoReflect 

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

func (*CoordinationServiceDeviceInfo) Reset 

func (x *CoordinationServiceDeviceInfo) Reset()

func (*CoordinationServiceDeviceInfo) String 

func (x *CoordinationServiceDeviceInfo) String() string

type CoordinationServiceDeviceInfo_Tf 

type CoordinationServiceDeviceInfo_Tf struct {
	Tf *TfDeviceList `protobuf:"bytes,1,opt,name=tf,proto3,oneof"`
}

type CoordinationServiceDeviceInfo_Xla 

type CoordinationServiceDeviceInfo_Xla struct {
	Xla *XlaDeviceList `protobuf:"bytes,2,opt,name=xla,proto3,oneof"`
}

type CoordinationServiceError 

type CoordinationServiceError struct {

	// If true, error is reported via the agent API by the user (and not an
	// internal service error).
	IsReportedError bool `protobuf:"varint,3,opt,name=is_reported_error,json=isReportedError,proto3" json:"is_reported_error,omitempty"`
	// Denotes which task hit the error. If unset, the error originated from the
	// same task that is processing this error.
	SourceTask *CoordinatedTask `protobuf:"bytes,4,opt,name=source_task,json=sourceTask,proto3" json:"source_task,omitempty"`
	// contains filtered or unexported fields
}

Status payload for all coordination service errors. Note: an empty proto may be set if the error is triggered by the task's own agent calls (i.e. not propagated by the service from another remote task).

func (*CoordinationServiceError) Descriptor deprecated 

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

Deprecated: Use CoordinationServiceError.ProtoReflect.Descriptor instead.

func (*CoordinationServiceError) GetIsReportedError 

func (x *CoordinationServiceError) GetIsReportedError() bool

func (*CoordinationServiceError) GetSourceTask 

func (x *CoordinationServiceError) GetSourceTask() *CoordinatedTask

func (*CoordinationServiceError) ProtoMessage 

func (*CoordinationServiceError) ProtoMessage()

func (*CoordinationServiceError) ProtoReflect 

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

func (*CoordinationServiceError) Reset 

func (x *CoordinationServiceError) Reset()

func (*CoordinationServiceError) String 

func (x *CoordinationServiceError) 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: Do not use.
	HostTempMemorySize int64 `protobuf:"varint,10,opt,name=host_temp_memory_size,json=hostTempMemorySize,proto3" json:"host_temp_memory_size,omitempty"`
	// Deprecated: Do not use.
	DeviceTempMemorySize int64 `` /* 127-byte string literal not displayed */
	// Deprecated: Do not use.
	DevicePersistentMemorySize int64 `` /* 145-byte string literal not displayed */
	// Estimate of the computational cost of this node, in microseconds.
	ComputeCost int64 `protobuf:"varint,9,opt,name=compute_cost,json=computeCost,proto3" json:"compute_cost,omitempty"`
	// Analytical estimate of the computational cost of this node, in
	// microseconds.
	ComputeTime int64 `protobuf:"varint,14,opt,name=compute_time,json=computeTime,proto3" json:"compute_time,omitempty"`
	// Analytical estimate of the memory access cost of this node, in
	// microseconds.
	MemoryTime int64 `protobuf:"varint,15,opt,name=memory_time,json=memoryTime,proto3" json:"memory_time,omitempty"`
	// If true, the output is permanent: it can't be discarded, because this
	// node is part of the "final output". Nodes may depend on final nodes.
	IsFinal bool `protobuf:"varint,7,opt,name=is_final,json=isFinal,proto3" json:"is_final,omitempty"`
	// Ids of the control inputs for this node.
	ControlInput []int32 `protobuf:"varint,8,rep,packed,name=control_input,json=controlInput,proto3" json:"control_input,omitempty"`
	// Are the costs inaccurate?
	Inaccurate bool `protobuf:"varint,17,opt,name=inaccurate,proto3" json:"inaccurate,omitempty"`
	// contains filtered or unexported fields
}

func (*CostGraphDef_Node) Descriptor deprecated 

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

Deprecated: Use CostGraphDef_Node.ProtoReflect.Descriptor instead.

func (*CostGraphDef_Node) GetComputeCost 

func (x *CostGraphDef_Node) GetComputeCost() int64

func (*CostGraphDef_Node) GetComputeTime 

func (x *CostGraphDef_Node) GetComputeTime() int64

func (*CostGraphDef_Node) GetControlInput 

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

func (*CostGraphDef_Node) GetDevice 

func (x *CostGraphDef_Node) GetDevice() string

func (*CostGraphDef_Node) GetDevicePersistentMemorySize deprecated 

func (x *CostGraphDef_Node) GetDevicePersistentMemorySize() int64

Deprecated: Do not use.

func (*CostGraphDef_Node) GetDeviceTempMemorySize deprecated 

func (x *CostGraphDef_Node) GetDeviceTempMemorySize() int64

Deprecated: Do not use.

func (*CostGraphDef_Node) GetHostTempMemorySize deprecated 

func (x *CostGraphDef_Node) GetHostTempMemorySize() int64

Deprecated: Do not use.

func (*CostGraphDef_Node) GetId 

func (x *CostGraphDef_Node) GetId() int32

func (*CostGraphDef_Node) GetInaccurate 

func (x *CostGraphDef_Node) GetInaccurate() bool

func (*CostGraphDef_Node) GetInputInfo 

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

func (*CostGraphDef_Node) GetIsFinal 

func (x *CostGraphDef_Node) GetIsFinal() bool

func (*CostGraphDef_Node) GetMemoryTime 

func (x *CostGraphDef_Node) GetMemoryTime() int64

func (*CostGraphDef_Node) GetName 

func (x *CostGraphDef_Node) GetName() string

func (*CostGraphDef_Node) GetOutputInfo 

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

func (*CostGraphDef_Node) GetPersistentMemorySize 

func (x *CostGraphDef_Node) GetPersistentMemorySize() int64

func (*CostGraphDef_Node) GetTemporaryMemorySize 

func (x *CostGraphDef_Node) GetTemporaryMemorySize() int64

func (*CostGraphDef_Node) ProtoMessage 

func (*CostGraphDef_Node) ProtoMessage()

func (*CostGraphDef_Node) ProtoReflect 

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

func (*CostGraphDef_Node) Reset 

func (x *CostGraphDef_Node) Reset()

func (*CostGraphDef_Node) String 

func (x *CostGraphDef_Node) String() string

type CostGraphDef_Node_InputInfo 

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

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

func (*CostGraphDef_Node_InputInfo) Descriptor deprecated 

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

Deprecated: Use CostGraphDef_Node_InputInfo.ProtoReflect.Descriptor instead.

func (*CostGraphDef_Node_InputInfo) GetPrecedingNode 

func (x *CostGraphDef_Node_InputInfo) GetPrecedingNode() int32

func (*CostGraphDef_Node_InputInfo) GetPrecedingPort 

func (x *CostGraphDef_Node_InputInfo) GetPrecedingPort() int32

func (*CostGraphDef_Node_InputInfo) ProtoMessage 

func (*CostGraphDef_Node_InputInfo) ProtoMessage()

func (*CostGraphDef_Node_InputInfo) ProtoReflect 

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

func (*CostGraphDef_Node_InputInfo) Reset 

func (x *CostGraphDef_Node_InputInfo) Reset()

func (*CostGraphDef_Node_InputInfo) String 

func (x *CostGraphDef_Node_InputInfo) String() string

type CostGraphDef_Node_OutputInfo 

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

Outputs of this node.

func (*CostGraphDef_Node_OutputInfo) Descriptor deprecated 

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

Deprecated: Use CostGraphDef_Node_OutputInfo.ProtoReflect.Descriptor instead.

func (*CostGraphDef_Node_OutputInfo) GetAliasInputPort 

func (x *CostGraphDef_Node_OutputInfo) GetAliasInputPort() int64

func (*CostGraphDef_Node_OutputInfo) GetDtype 

func (x *CostGraphDef_Node_OutputInfo) GetDtype() DataType

func (*CostGraphDef_Node_OutputInfo) GetShape 

func (x *CostGraphDef_Node_OutputInfo) GetShape() *TensorShapeProto

func (*CostGraphDef_Node_OutputInfo) GetSize 

func (x *CostGraphDef_Node_OutputInfo) GetSize() int64

func (*CostGraphDef_Node_OutputInfo) ProtoMessage 

func (*CostGraphDef_Node_OutputInfo) ProtoMessage()

func (*CostGraphDef_Node_OutputInfo) ProtoReflect 

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

func (*CostGraphDef_Node_OutputInfo) Reset 

func (x *CostGraphDef_Node_OutputInfo) Reset()

func (*CostGraphDef_Node_OutputInfo) String 

func (x *CostGraphDef_Node_OutputInfo) String() string

type CreateContextRequest 

type CreateContextRequest struct {

	// Identifies the full cluster, and this particular worker's position within.
	ServerDef *ServerDef `protobuf:"bytes,1,opt,name=server_def,json=serverDef,proto3" json:"server_def,omitempty"`
	// Whether the ops on the worker should be executed synchronously or
	// asynchronously. By default, ops are executed synchronously.
	Async bool `protobuf:"varint,2,opt,name=async,proto3" json:"async,omitempty"`
	// Number of seconds to keep the context alive. If more than keep_alive_secs
	// has passed since a particular context has been communicated with, it will
	// be garbage collected.
	KeepAliveSecs int64 `protobuf:"varint,3,opt,name=keep_alive_secs,json=keepAliveSecs,proto3" json:"keep_alive_secs,omitempty"`
	// This is the version for all the ops that will be enqueued by the client.
	VersionDef *VersionDef `protobuf:"bytes,4,opt,name=version_def,json=versionDef,proto3" json:"version_def,omitempty"`
	// Device attributes in the cluster
	ClusterDeviceAttributes []*DeviceAttributes `` /* 132-byte string literal not displayed */
	// The ID of the created context. This is usually a randomly generated number,
	// that will be used to identify the context in future requests to the
	// service. Contexts are not persisted through server restarts.
	// This ID will be used for all future communications as well. It is essential
	// that both ends use this ID for selecting a rendezvous to get everything to
	// match.
	ContextId uint64 `protobuf:"fixed64,7,opt,name=context_id,json=contextId,proto3" json:"context_id,omitempty"`
	// The view ID of the context.
	ContextViewId uint64 `protobuf:"fixed64,8,opt,name=context_view_id,json=contextViewId,proto3" json:"context_view_id,omitempty"`
	// For a multi device function, if false, eagerly copy all remote inputs to
	// the default function device; if true, lazily copy remote inputs to their
	// target devices after function instantiation to avoid redundant copies.
	LazyCopyRemoteFunctionInputs bool `` /* 152-byte string literal not displayed */
	// contains filtered or unexported fields
}

func (*CreateContextRequest) Descriptor deprecated 

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

Deprecated: Use CreateContextRequest.ProtoReflect.Descriptor instead.

func (*CreateContextRequest) GetAsync 

func (x *CreateContextRequest) GetAsync() bool

func (*CreateContextRequest) GetClusterDeviceAttributes 

func (x *CreateContextRequest) GetClusterDeviceAttributes() []*DeviceAttributes

func (*CreateContextRequest) GetContextId 

func (x *CreateContextRequest) GetContextId() uint64

func (*CreateContextRequest) GetContextViewId 

func (x *CreateContextRequest) GetContextViewId() uint64

func (*CreateContextRequest) GetKeepAliveSecs 

func (x *CreateContextRequest) GetKeepAliveSecs() int64

func (*CreateContextRequest) GetLazyCopyRemoteFunctionInputs 

func (x *CreateContextRequest) GetLazyCopyRemoteFunctionInputs() bool

func (*CreateContextRequest) GetServerDef 

func (x *CreateContextRequest) GetServerDef() *ServerDef

func (*CreateContextRequest) GetVersionDef 

func (x *CreateContextRequest) GetVersionDef() *VersionDef

func (*CreateContextRequest) ProtoMessage 

func (*CreateContextRequest) ProtoMessage()

func (*CreateContextRequest) ProtoReflect 

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

func (*CreateContextRequest) Reset 

func (x *CreateContextRequest) Reset()

func (*CreateContextRequest) String 

func (x *CreateContextRequest) String() string

type CreateContextResponse 

type CreateContextResponse struct {

	// List of devices that are locally accessible to the worker.
	DeviceAttributes []*DeviceAttributes `protobuf:"bytes,2,rep,name=device_attributes,json=deviceAttributes,proto3" json:"device_attributes,omitempty"`
	// contains filtered or unexported fields
}

func (*CreateContextResponse) Descriptor deprecated 

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

Deprecated: Use CreateContextResponse.ProtoReflect.Descriptor instead.

func (*CreateContextResponse) GetDeviceAttributes 

func (x *CreateContextResponse) GetDeviceAttributes() []*DeviceAttributes

func (*CreateContextResponse) ProtoMessage 

func (*CreateContextResponse) ProtoMessage()

func (*CreateContextResponse) ProtoReflect 

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

func (*CreateContextResponse) Reset 

func (x *CreateContextResponse) Reset()

func (*CreateContextResponse) String 

func (x *CreateContextResponse) String() string

type CreateSessionRequest 

type CreateSessionRequest struct {

	// The initial graph definition.
	GraphDef *GraphDef `protobuf:"bytes,1,opt,name=graph_def,json=graphDef,proto3" json:"graph_def,omitempty"`
	// Configuration options.
	Config *ConfigProto `protobuf:"bytes,2,opt,name=config,proto3" json:"config,omitempty"`
	// The target string used from the client's perspective.
	Target string `protobuf:"bytes,3,opt,name=target,proto3" json:"target,omitempty"`
	// contains filtered or unexported fields
}

func (*CreateSessionRequest) Descriptor deprecated 

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

Deprecated: Use CreateSessionRequest.ProtoReflect.Descriptor instead.

func (*CreateSessionRequest) GetConfig 

func (x *CreateSessionRequest) GetConfig() *ConfigProto

func (*CreateSessionRequest) GetGraphDef 

func (x *CreateSessionRequest) GetGraphDef() *GraphDef

func (*CreateSessionRequest) GetTarget 

func (x *CreateSessionRequest) GetTarget() string

func (*CreateSessionRequest) ProtoMessage 

func (*CreateSessionRequest) ProtoMessage()

func (*CreateSessionRequest) ProtoReflect 

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

func (*CreateSessionRequest) Reset 

func (x *CreateSessionRequest) Reset()

func (*CreateSessionRequest) String 

func (x *CreateSessionRequest) String() string

type CreateSessionResponse 

type CreateSessionResponse struct {

	// The session handle to be used in subsequent calls for the created session.
	//
	// The client must arrange to call CloseSession with this returned
	// session handle to close the session.
	SessionHandle string `protobuf:"bytes,1,opt,name=session_handle,json=sessionHandle,proto3" json:"session_handle,omitempty"`
	// The initial version number for the graph, to be used in the next call
	// to ExtendSession.
	GraphVersion int64 `protobuf:"varint,2,opt,name=graph_version,json=graphVersion,proto3" json:"graph_version,omitempty"`
	// contains filtered or unexported fields
}

func (*CreateSessionResponse) Descriptor deprecated 

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

Deprecated: Use CreateSessionResponse.ProtoReflect.Descriptor instead.

func (*CreateSessionResponse) GetGraphVersion 

func (x *CreateSessionResponse) GetGraphVersion() int64

func (*CreateSessionResponse) GetSessionHandle 

func (x *CreateSessionResponse) GetSessionHandle() string

func (*CreateSessionResponse) ProtoMessage 

func (*CreateSessionResponse) ProtoMessage()

func (*CreateSessionResponse) ProtoReflect 

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

func (*CreateSessionResponse) Reset 

func (x *CreateSessionResponse) Reset()

func (*CreateSessionResponse) String 

func (x *CreateSessionResponse) String() string

type CreateWorkerSessionRequest 

type CreateWorkerSessionRequest struct {

	// Sessions are identified by a given handle.
	SessionHandle string `protobuf:"bytes,1,opt,name=session_handle,json=sessionHandle,proto3" json:"session_handle,omitempty"`
	// Defines the configuration of a TensorFlow worker.
	ServerDef *ServerDef `protobuf:"bytes,2,opt,name=server_def,json=serverDef,proto3" json:"server_def,omitempty"`
	// If true, any resources such as Variables used in the session will not be
	// shared with other sessions.
	IsolateSessionState bool `protobuf:"varint,3,opt,name=isolate_session_state,json=isolateSessionState,proto3" json:"isolate_session_state,omitempty"`
	// The device attributes of all the devices in the cluster.
	ClusterDeviceAttributes []*DeviceAttributes `` /* 132-byte string literal not displayed */
	// The master task name from which the request is sent.
	MasterTask string `protobuf:"bytes,5,opt,name=master_task,json=masterTask,proto3" json:"master_task,omitempty"`
	// The incarnation ID of the master task local CPU device.
	// If the target worker already has a WorkerSession created previously with
	// the same master task name but a different incarnation, it usually indicates
	// that the previous master failed before deleting the WorkerSession on the
	// worker. To prevent memory leaks, the worker should garbage collect the old
	// WorkerSessions.
	MasterIncarnation int64 `protobuf:"varint,6,opt,name=master_incarnation,json=masterIncarnation,proto3" json:"master_incarnation,omitempty"`
	// contains filtered or unexported fields
}

func (*CreateWorkerSessionRequest) Descriptor deprecated 

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

Deprecated: Use CreateWorkerSessionRequest.ProtoReflect.Descriptor instead.

func (*CreateWorkerSessionRequest) GetClusterDeviceAttributes 

func (x *CreateWorkerSessionRequest) GetClusterDeviceAttributes() []*DeviceAttributes

func (*CreateWorkerSessionRequest) GetIsolateSessionState 

func (x *CreateWorkerSessionRequest) GetIsolateSessionState() bool

func (*CreateWorkerSessionRequest) GetMasterIncarnation 

func (x *CreateWorkerSessionRequest) GetMasterIncarnation() int64

func (*CreateWorkerSessionRequest) GetMasterTask 

func (x *CreateWorkerSessionRequest) GetMasterTask() string

func (*CreateWorkerSessionRequest) GetServerDef 

func (x *CreateWorkerSessionRequest) GetServerDef() *ServerDef

func (*CreateWorkerSessionRequest) GetSessionHandle 

func (x *CreateWorkerSessionRequest) GetSessionHandle() string

func (*CreateWorkerSessionRequest) ProtoMessage 

func (*CreateWorkerSessionRequest) ProtoMessage()

func (*CreateWorkerSessionRequest) ProtoReflect 

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

func (*CreateWorkerSessionRequest) Reset 

func (x *CreateWorkerSessionRequest) Reset()

func (*CreateWorkerSessionRequest) String 

func (x *CreateWorkerSessionRequest) String() string

type CreateWorkerSessionResponse 

type CreateWorkerSessionResponse struct {
	// contains filtered or unexported fields
}

func (*CreateWorkerSessionResponse) Descriptor deprecated 

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

Deprecated: Use CreateWorkerSessionResponse.ProtoReflect.Descriptor instead.

func (*CreateWorkerSessionResponse) ProtoMessage 

func (*CreateWorkerSessionResponse) ProtoMessage()

func (*CreateWorkerSessionResponse) ProtoReflect 

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

func (*CreateWorkerSessionResponse) Reset 

func (x *CreateWorkerSessionResponse) Reset()

func (*CreateWorkerSessionResponse) String 

func (x *CreateWorkerSessionResponse) String() string

type CriticalSectionDef 

type CriticalSectionDef struct {

	// Name of the critical section handle.
	CriticalSectionName string `protobuf:"bytes,1,opt,name=critical_section_name,json=criticalSectionName,proto3" json:"critical_section_name,omitempty"`
	// contains filtered or unexported fields
}

Protocol buffer representing a CriticalSection.

func (*CriticalSectionDef) Descriptor deprecated 

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

Deprecated: Use CriticalSectionDef.ProtoReflect.Descriptor instead.

func (*CriticalSectionDef) GetCriticalSectionName 

func (x *CriticalSectionDef) GetCriticalSectionName() string

func (*CriticalSectionDef) ProtoMessage 

func (*CriticalSectionDef) ProtoMessage()

func (*CriticalSectionDef) ProtoReflect 

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

func (*CriticalSectionDef) Reset 

func (x *CriticalSectionDef) Reset()

func (*CriticalSectionDef) String 

func (x *CriticalSectionDef) String() string

type CriticalSectionExecutionDef 

type CriticalSectionExecutionDef struct {

	// Name of the critical section handle.
	ExecuteInCriticalSectionName string `` /* 151-byte string literal not displayed */
	// Whether this operation requires exclusive access to its resources,
	// (i.e., no other CriticalSections may request the same resources).
	ExclusiveResourceAccess bool `` /* 133-byte string literal not displayed */
	// contains filtered or unexported fields
}

Protocol buffer representing a CriticalSection execution.

func (*CriticalSectionExecutionDef) Descriptor deprecated 

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

Deprecated: Use CriticalSectionExecutionDef.ProtoReflect.Descriptor instead.

func (*CriticalSectionExecutionDef) GetExclusiveResourceAccess 

func (x *CriticalSectionExecutionDef) GetExclusiveResourceAccess() bool

func (*CriticalSectionExecutionDef) GetExecuteInCriticalSectionName 

func (x *CriticalSectionExecutionDef) GetExecuteInCriticalSectionName() string

func (*CriticalSectionExecutionDef) ProtoMessage 

func (*CriticalSectionExecutionDef) ProtoMessage()

func (*CriticalSectionExecutionDef) ProtoReflect 

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

func (*CriticalSectionExecutionDef) Reset 

func (x *CriticalSectionExecutionDef) Reset()

func (*CriticalSectionExecutionDef) String 

func (x *CriticalSectionExecutionDef) String() string

type CrossTrainerCacheOptions 

type CrossTrainerCacheOptions struct {
	TrainerId string `protobuf:"bytes,1,opt,name=trainer_id,json=trainerId,proto3" json:"trainer_id,omitempty"`
	// contains filtered or unexported fields
}

func (*CrossTrainerCacheOptions) Descriptor deprecated 

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

Deprecated: Use CrossTrainerCacheOptions.ProtoReflect.Descriptor instead.

func (*CrossTrainerCacheOptions) GetTrainerId 

func (x *CrossTrainerCacheOptions) GetTrainerId() string

func (*CrossTrainerCacheOptions) ProtoMessage 

func (*CrossTrainerCacheOptions) ProtoMessage()

func (*CrossTrainerCacheOptions) ProtoReflect 

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

func (*CrossTrainerCacheOptions) Reset 

func (x *CrossTrainerCacheOptions) Reset()

func (*CrossTrainerCacheOptions) String 

func (x *CrossTrainerCacheOptions) String() string

type CudnnVersion 

type CudnnVersion struct {
	Major int32 `protobuf:"varint,1,opt,name=major,proto3" json:"major,omitempty"`
	Minor int32 `protobuf:"varint,2,opt,name=minor,proto3" json:"minor,omitempty"`
	Patch int32 `protobuf:"varint,3,opt,name=patch,proto3" json:"patch,omitempty"`
	// contains filtered or unexported fields
}

func (*CudnnVersion) Descriptor deprecated 

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

Deprecated: Use CudnnVersion.ProtoReflect.Descriptor instead.

func (*CudnnVersion) GetMajor 

func (x *CudnnVersion) GetMajor() int32

func (*CudnnVersion) GetMinor 

func (x *CudnnVersion) GetMinor() int32

func (*CudnnVersion) GetPatch 

func (x *CudnnVersion) GetPatch() int32

func (*CudnnVersion) ProtoMessage 

func (*CudnnVersion) ProtoMessage()

func (*CudnnVersion) ProtoReflect 

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

func (*CudnnVersion) Reset 

func (x *CudnnVersion) Reset()

func (*CudnnVersion) String 

func (x *CudnnVersion) 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 DataLayout 

type DataLayout int32

Describes how a convolution input or output layer's data is formatted. 

const (
	// Naming convention:
	// Y <-> row or height
	// X <-> column or width
	// Batch <-> batch, or N
	// Depth <-> feature, or channel
	// TODO(timshen): turn them into cuDNN names, e.g. kNCHW.
	//
	// Note: In cudnn, kBatchDepthYX4 and kBatchDepthYX32 are the same layout
	// (namely, NCHW_VECT_C).  It differentiates between these two by using a
	// different data type (int8x4 vs int8x32).  In StreamExecutor we use
	// different layouts for these, because we don't usually pass an explicit data
	// type to StreamExecutor functions.
	DataLayout_kYXDepthBatch   DataLayout = 0
	DataLayout_kYXBatchDepth   DataLayout = 1
	DataLayout_kBatchYXDepth   DataLayout = 2 // cuDNN's NHWC layout
	DataLayout_kBatchDepthYX   DataLayout = 3 // cuDNN's NCHW layout
	DataLayout_kBatchDepthYX4  DataLayout = 4 // cuDNN's NCHW_VECT_C with 4-elem vectors (e.g. int8x4)
	DataLayout_kBatchDepthYX32 DataLayout = 5 // cuDNN's NCHW_VECT_C with 32-elem vects (e.g. int8x32)
)

func (DataLayout) Descriptor 

func (DataLayout) Descriptor() protoreflect.EnumDescriptor

func (DataLayout) Enum 

func (x DataLayout) Enum() *DataLayout

func (DataLayout) EnumDescriptor deprecated 

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

Deprecated: Use DataLayout.Descriptor instead.

func (DataLayout) Number 

func (x DataLayout) Number() protoreflect.EnumNumber

func (DataLayout) String 

func (x DataLayout) String() string

func (DataLayout) Type 

func (DataLayout) Type() protoreflect.EnumType

type DataServiceConfig 

type DataServiceConfig struct {
	DeploymentMode DeploymentMode `` /* 140-byte string literal not displayed */
	// contains filtered or unexported fields
}

Data service config available to the client through GetDataServiceConfig RPC. Next tag: 2

func (*DataServiceConfig) Descriptor deprecated 

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

Deprecated: Use DataServiceConfig.ProtoReflect.Descriptor instead.

func (*DataServiceConfig) GetDeploymentMode 

func (x *DataServiceConfig) GetDeploymentMode() DeploymentMode

func (*DataServiceConfig) ProtoMessage 

func (*DataServiceConfig) ProtoMessage()

func (*DataServiceConfig) ProtoReflect 

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

func (*DataServiceConfig) Reset 

func (x *DataServiceConfig) Reset()

func (*DataServiceConfig) String 

func (x *DataServiceConfig) String() string

type DataServiceMetadata 

type DataServiceMetadata struct {

	// Types that are assignable to OptionalElementSpec:
	//	*DataServiceMetadata_ElementSpec
	OptionalElementSpec isDataServiceMetadata_OptionalElementSpec `protobuf_oneof:"optional_element_spec"`
	Compression         DataServiceMetadata_Compression           `` /* 129-byte string literal not displayed */
	// Cardinality of the dataset.
	Cardinality int64 `protobuf:"varint,3,opt,name=cardinality,proto3" json:"cardinality,omitempty"`
	// contains filtered or unexported fields
}

Metadata related to tf.data service datasets. Next tag: 4

func (*DataServiceMetadata) Descriptor deprecated 

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

Deprecated: Use DataServiceMetadata.ProtoReflect.Descriptor instead.

func (*DataServiceMetadata) GetCardinality 

func (x *DataServiceMetadata) GetCardinality() int64

func (*DataServiceMetadata) GetCompression 

func (x *DataServiceMetadata) GetCompression() DataServiceMetadata_Compression

func (*DataServiceMetadata) GetElementSpec 

func (x *DataServiceMetadata) GetElementSpec() []byte

func (*DataServiceMetadata) GetOptionalElementSpec 

func (m *DataServiceMetadata) GetOptionalElementSpec() isDataServiceMetadata_OptionalElementSpec

func (*DataServiceMetadata) ProtoMessage 

func (*DataServiceMetadata) ProtoMessage()

func (*DataServiceMetadata) ProtoReflect 

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

func (*DataServiceMetadata) Reset 

func (x *DataServiceMetadata) Reset()

func (*DataServiceMetadata) String 

func (x *DataServiceMetadata) String() string

type DataServiceMetadata_Compression 

type DataServiceMetadata_Compression int32
const (
	DataServiceMetadata_COMPRESSION_UNSPECIFIED DataServiceMetadata_Compression = 0
	// No compression.
	DataServiceMetadata_COMPRESSION_OFF DataServiceMetadata_Compression = 1
	// Snappy compression as defined in tensorflow/core/platform/snappy.h.
	DataServiceMetadata_COMPRESSION_SNAPPY DataServiceMetadata_Compression = 2
)

func (DataServiceMetadata_Compression) Descriptor 

func (DataServiceMetadata_Compression) Descriptor() protoreflect.EnumDescriptor

func (DataServiceMetadata_Compression) Enum 

func (x DataServiceMetadata_Compression) Enum() *DataServiceMetadata_Compression

func (DataServiceMetadata_Compression) EnumDescriptor deprecated 

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

Deprecated: Use DataServiceMetadata_Compression.Descriptor instead.

func (DataServiceMetadata_Compression) Number 

func (x DataServiceMetadata_Compression) Number() protoreflect.EnumNumber

func (DataServiceMetadata_Compression) String 

func (x DataServiceMetadata_Compression) String() string

func (DataServiceMetadata_Compression) Type 

func (DataServiceMetadata_Compression) Type() protoreflect.EnumType

type DataServiceMetadata_ElementSpec 

type DataServiceMetadata_ElementSpec struct {
	// Serialized element spec.
	ElementSpec []byte `protobuf:"bytes,1,opt,name=element_spec,json=elementSpec,proto3,oneof"`
}

type DataType 

type DataType int32

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

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

func (DataType) Descriptor 

func (DataType) Descriptor() protoreflect.EnumDescriptor

func (DataType) Enum 

func (x DataType) Enum() *DataType

func (DataType) EnumDescriptor deprecated 

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

Deprecated: Use DataType.Descriptor instead.

func (DataType) Number 

func (x DataType) Number() protoreflect.EnumNumber

func (DataType) String 

func (x DataType) String() string

func (DataType) Type 

func (DataType) Type() protoreflect.EnumType

type DebugEvent 

type DebugEvent struct {

	// Timestamp in seconds (with microsecond precision).
	WallTime float64 `protobuf:"fixed64,1,opt,name=wall_time,json=wallTime,proto3" json:"wall_time,omitempty"`
	// Step of training (if available).
	Step int64 `protobuf:"varint,2,opt,name=step,proto3" json:"step,omitempty"`
	// Types that are assignable to What:
	//	*DebugEvent_DebugMetadata
	//	*DebugEvent_SourceFile
	//	*DebugEvent_StackFrameWithId
	//	*DebugEvent_GraphOpCreation
	//	*DebugEvent_DebuggedGraph
	//	*DebugEvent_Execution
	//	*DebugEvent_GraphExecutionTrace
	//	*DebugEvent_GraphId
	//	*DebugEvent_DebuggedDevice
	What isDebugEvent_What `protobuf_oneof:"what"`
	// contains filtered or unexported fields
}

An Event related to the debugging of a TensorFlow program.

func (*DebugEvent) Descriptor deprecated 

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

Deprecated: Use DebugEvent.ProtoReflect.Descriptor instead.

func (*DebugEvent) GetDebugMetadata 

func (x *DebugEvent) GetDebugMetadata() *DebugMetadata

func (*DebugEvent) GetDebuggedDevice 

func (x *DebugEvent) GetDebuggedDevice() *DebuggedDevice

func (*DebugEvent) GetDebuggedGraph 

func (x *DebugEvent) GetDebuggedGraph() *DebuggedGraph

func (*DebugEvent) GetExecution 

func (x *DebugEvent) GetExecution() *Execution

func (*DebugEvent) GetGraphExecutionTrace 

func (x *DebugEvent) GetGraphExecutionTrace() *GraphExecutionTrace

func (*DebugEvent) GetGraphId 

func (x *DebugEvent) GetGraphId() string

func (*DebugEvent) GetGraphOpCreation 

func (x *DebugEvent) GetGraphOpCreation() *GraphOpCreation

func (*DebugEvent) GetSourceFile 

func (x *DebugEvent) GetSourceFile() *SourceFile

func (*DebugEvent) GetStackFrameWithId 

func (x *DebugEvent) GetStackFrameWithId() *StackFrameWithId

func (*DebugEvent) GetStep 

func (x *DebugEvent) GetStep() int64

func (*DebugEvent) GetWallTime 

func (x *DebugEvent) GetWallTime() float64

func (*DebugEvent) GetWhat 

func (m *DebugEvent) GetWhat() isDebugEvent_What

func (*DebugEvent) ProtoMessage 

func (*DebugEvent) ProtoMessage()

func (*DebugEvent) ProtoReflect 

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

func (*DebugEvent) Reset 

func (x *DebugEvent) Reset()

func (*DebugEvent) String 

func (x *DebugEvent) String() string

type DebugEvent_DebugMetadata 

type DebugEvent_DebugMetadata struct {
	// Metadata related to this debugging data.
	DebugMetadata *DebugMetadata `protobuf:"bytes,3,opt,name=debug_metadata,json=debugMetadata,proto3,oneof"`
}

type DebugEvent_DebuggedDevice 

type DebugEvent_DebuggedDevice struct {
	// A device on which debugger-instrumented ops and/or tensors reside.
	DebuggedDevice *DebuggedDevice `protobuf:"bytes,12,opt,name=debugged_device,json=debuggedDevice,proto3,oneof"`
}

type DebugEvent_DebuggedGraph 

type DebugEvent_DebuggedGraph struct {
	// Information about a debugged graph.
	DebuggedGraph *DebuggedGraph `protobuf:"bytes,8,opt,name=debugged_graph,json=debuggedGraph,proto3,oneof"`
}

type DebugEvent_Execution 

type DebugEvent_Execution struct {
	// Execution of an op or a Graph (e.g., a tf.function).
	Execution *Execution `protobuf:"bytes,9,opt,name=execution,proto3,oneof"`
}

type DebugEvent_GraphExecutionTrace 

type DebugEvent_GraphExecutionTrace struct {
	// A graph execution trace: Contains information about the intermediate
	// tensors computed during the graph execution.
	GraphExecutionTrace *GraphExecutionTrace `protobuf:"bytes,10,opt,name=graph_execution_trace,json=graphExecutionTrace,proto3,oneof"`
}

type DebugEvent_GraphId 

type DebugEvent_GraphId struct {
	// The ID of the graph (i.e., FuncGraph) executed here: applicable only
	// to the execution of a FuncGraph.
	GraphId string `protobuf:"bytes,11,opt,name=graph_id,json=graphId,proto3,oneof"`
}

type DebugEvent_GraphOpCreation 

type DebugEvent_GraphOpCreation struct {
	// The creation of an op within a graph (e.g., a FuncGraph compiled from
	// a Python function).
	GraphOpCreation *GraphOpCreation `protobuf:"bytes,7,opt,name=graph_op_creation,json=graphOpCreation,proto3,oneof"`
}

type DebugEvent_SourceFile 

type DebugEvent_SourceFile struct {
	// The content of a source file.
	SourceFile *SourceFile `protobuf:"bytes,4,opt,name=source_file,json=sourceFile,proto3,oneof"`
}

type DebugEvent_StackFrameWithId 

type DebugEvent_StackFrameWithId struct {
	// A stack frame (filename, line number and column number, function name and
	// code string) with ID.
	StackFrameWithId *StackFrameWithId `protobuf:"bytes,6,opt,name=stack_frame_with_id,json=stackFrameWithId,proto3,oneof"`
}

type DebugMetadata 

type DebugMetadata struct {

	// Version of TensorFlow.
	TensorflowVersion string `protobuf:"bytes,1,opt,name=tensorflow_version,json=tensorflowVersion,proto3" json:"tensorflow_version,omitempty"`
	// Version of the DebugEvent file format.
	// Has a format of "debug.Event:<number>", e.g., "debug.Event:1".
	FileVersion string `protobuf:"bytes,2,opt,name=file_version,json=fileVersion,proto3" json:"file_version,omitempty"`
	// A unique ID for the current run of tfdbg.
	// A run of tfdbg is defined as a TensorFlow job instrumented by tfdbg.
	// Multiple hosts in a distributed TensorFlow job instrumented by tfdbg
	// have the same ID.
	TfdbgRunId string `protobuf:"bytes,3,opt,name=tfdbg_run_id,json=tfdbgRunId,proto3" json:"tfdbg_run_id,omitempty"`
	// contains filtered or unexported fields
}

Metadata about the debugger and the debugged TensorFlow program.

func (*DebugMetadata) Descriptor deprecated 

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

Deprecated: Use DebugMetadata.ProtoReflect.Descriptor instead.

func (*DebugMetadata) GetFileVersion 

func (x *DebugMetadata) GetFileVersion() string

func (*DebugMetadata) GetTensorflowVersion 

func (x *DebugMetadata) GetTensorflowVersion() string

func (*DebugMetadata) GetTfdbgRunId 

func (x *DebugMetadata) GetTfdbgRunId() string

func (*DebugMetadata) ProtoMessage 

func (*DebugMetadata) ProtoMessage()

func (*DebugMetadata) ProtoReflect 

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

func (*DebugMetadata) Reset 

func (x *DebugMetadata) Reset()

func (*DebugMetadata) String 

func (x *DebugMetadata) String() string

type DebugOptions 

type DebugOptions struct {

	// Debugging options
	DebugTensorWatchOpts []*DebugTensorWatch `protobuf:"bytes,4,rep,name=debug_tensor_watch_opts,json=debugTensorWatchOpts,proto3" json:"debug_tensor_watch_opts,omitempty"`
	// Caller-specified global step count.
	// Note that this is distinct from the session run count and the executor
	// step count.
	GlobalStep int64 `protobuf:"varint,10,opt,name=global_step,json=globalStep,proto3" json:"global_step,omitempty"`
	// Whether the total disk usage of tfdbg is to be reset to zero
	// in this Session.run call. This is used by wrappers and hooks
	// such as the local CLI ones to indicate that the dumped tensors
	// are cleaned up from the disk after each Session.run.
	ResetDiskByteUsage bool `protobuf:"varint,11,opt,name=reset_disk_byte_usage,json=resetDiskByteUsage,proto3" json:"reset_disk_byte_usage,omitempty"`
	// contains filtered or unexported fields
}

Options for initializing DebuggerState in TensorFlow Debugger (tfdbg).

func (*DebugOptions) Descriptor deprecated 

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

Deprecated: Use DebugOptions.ProtoReflect.Descriptor instead.

func (*DebugOptions) GetDebugTensorWatchOpts 

func (x *DebugOptions) GetDebugTensorWatchOpts() []*DebugTensorWatch

func (*DebugOptions) GetGlobalStep 

func (x *DebugOptions) GetGlobalStep() int64

func (*DebugOptions) GetResetDiskByteUsage 

func (x *DebugOptions) GetResetDiskByteUsage() bool

func (*DebugOptions) ProtoMessage 

func (*DebugOptions) ProtoMessage()

func (*DebugOptions) ProtoReflect 

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

func (*DebugOptions) Reset 

func (x *DebugOptions) Reset()

func (*DebugOptions) String 

func (x *DebugOptions) String() string

type DebugTensorWatch 

type DebugTensorWatch struct {

	// Name of the node to watch.
	// Use "*" for wildcard. But note: currently, regex is not supported in
	// general.
	NodeName string `protobuf:"bytes,1,opt,name=node_name,json=nodeName,proto3" json:"node_name,omitempty"`
	// Output slot to watch.
	// The semantics of output_slot == -1 is that all outputs of the node
	// will be watched (i.e., a wildcard).
	// Other negative values of output_slot are invalid and will lead to
	// errors currently.
	OutputSlot int32 `protobuf:"varint,2,opt,name=output_slot,json=outputSlot,proto3" json:"output_slot,omitempty"`
	// Name(s) of the debugging op(s).
	// One or more than one probes on a tensor.
	// e.g., {"DebugIdentity", "DebugNanCount"}
	DebugOps []string `protobuf:"bytes,3,rep,name=debug_ops,json=debugOps,proto3" json:"debug_ops,omitempty"`
	// URL(s) for debug targets(s).
	//
	// Supported URL formats are:
	//   - file:///foo/tfdbg_dump: Writes out Event content to file
	//     /foo/tfdbg_dump.  Assumes all directories can be created if they don't
	//     already exist.
	//   - grpc://localhost:11011: Sends an RPC request to an EventListener
	//     service running at localhost:11011 with the event.
	//   - memcbk:///event_key: Routes tensors to clients using the
	//     callback registered with the DebugCallbackRegistry for event_key.
	//
	// Each debug op listed in debug_ops will publish its output tensor (debug
	// signal) to all URLs in debug_urls.
	//
	// N.B. Session::Run() supports concurrent invocations of the same inputs
	// (feed keys), outputs and target nodes. If such concurrent invocations
	// are to be debugged, the callers of Session::Run() must use distinct
	// debug_urls to make sure that the streamed or dumped events do not overlap
	// among the invocations.
	// TODO(cais): More visible documentation of this in g3docs.
	DebugUrls []string `protobuf:"bytes,4,rep,name=debug_urls,json=debugUrls,proto3" json:"debug_urls,omitempty"`
	// Do not error out if debug op creation fails (e.g., due to dtype
	// incompatibility). Instead, just log the failure.
	TolerateDebugOpCreationFailures bool `` /* 161-byte string literal not displayed */
	// contains filtered or unexported fields
}

Option for watching a node in TensorFlow Debugger (tfdbg).

func (*DebugTensorWatch) Descriptor deprecated 

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

Deprecated: Use DebugTensorWatch.ProtoReflect.Descriptor instead.

func (*DebugTensorWatch) GetDebugOps 

func (x *DebugTensorWatch) GetDebugOps() []string

func (*DebugTensorWatch) GetDebugUrls 

func (x *DebugTensorWatch) GetDebugUrls() []string

func (*DebugTensorWatch) GetNodeName 

func (x *DebugTensorWatch) GetNodeName() string

func (*DebugTensorWatch) GetOutputSlot 

func (x *DebugTensorWatch) GetOutputSlot() int32

func (*DebugTensorWatch) GetTolerateDebugOpCreationFailures 

func (x *DebugTensorWatch) GetTolerateDebugOpCreationFailures() bool

func (*DebugTensorWatch) ProtoMessage 

func (*DebugTensorWatch) ProtoMessage()

func (*DebugTensorWatch) ProtoReflect 

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

func (*DebugTensorWatch) Reset 

func (x *DebugTensorWatch) Reset()

func (*DebugTensorWatch) String 

func (x *DebugTensorWatch) String() string

type DebuggedDevice 

type DebuggedDevice struct {

	// Name of the device.
	DeviceName string `protobuf:"bytes,1,opt,name=device_name,json=deviceName,proto3" json:"device_name,omitempty"`
	// A debugger-generated ID for the device. Guaranteed to be unique within
	// the scope of the debugged TensorFlow program, including single-host and
	// multi-host settings.
	// TODO(cais): Test the uniqueness guarantee in multi-host settings.
	DeviceId int32 `protobuf:"varint,2,opt,name=device_id,json=deviceId,proto3" json:"device_id,omitempty"`
	// contains filtered or unexported fields
}

A device on which ops and/or tensors are instrumented by the debugger.

func (*DebuggedDevice) Descriptor deprecated 

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

Deprecated: Use DebuggedDevice.ProtoReflect.Descriptor instead.

func (*DebuggedDevice) GetDeviceId 

func (x *DebuggedDevice) GetDeviceId() int32

func (*DebuggedDevice) GetDeviceName 

func (x *DebuggedDevice) GetDeviceName() string

func (*DebuggedDevice) ProtoMessage 

func (*DebuggedDevice) ProtoMessage()

func (*DebuggedDevice) ProtoReflect 

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

func (*DebuggedDevice) Reset 

func (x *DebuggedDevice) Reset()

func (*DebuggedDevice) String 

func (x *DebuggedDevice) String() string

type DebuggedGraph 

type DebuggedGraph struct {

	// An ID for the graph.
	// This can be used up to look up graph names. Generated by the debugger.
	GraphId string `protobuf:"bytes,1,opt,name=graph_id,json=graphId,proto3" json:"graph_id,omitempty"`
	// Name of the graph (if available).
	GraphName string `protobuf:"bytes,2,opt,name=graph_name,json=graphName,proto3" json:"graph_name,omitempty"`
	// Names of the instrumented ops. This can be used to look up op name
	// based on the numeric-summary tensors (2nd column).
	InstrumentedOps []string `protobuf:"bytes,3,rep,name=instrumented_ops,json=instrumentedOps,proto3" json:"instrumented_ops,omitempty"`
	// Original (uninstrumented) GraphDef (if available).
	OriginalGraphDef []byte `protobuf:"bytes,4,opt,name=original_graph_def,json=originalGraphDef,proto3" json:"original_graph_def,omitempty"`
	// An encoded version of a GraphDef.
	// This graph may include the debugger-inserted ops.
	InstrumentedGraphDef []byte `protobuf:"bytes,5,opt,name=instrumented_graph_def,json=instrumentedGraphDef,proto3" json:"instrumented_graph_def,omitempty"`
	// IDs of the immediate enclosing context (graph), if any.
	OuterContextId string `protobuf:"bytes,6,opt,name=outer_context_id,json=outerContextId,proto3" json:"outer_context_id,omitempty"`
	// contains filtered or unexported fields
}

A debugger-instrumented graph.

func (*DebuggedGraph) Descriptor deprecated 

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

Deprecated: Use DebuggedGraph.ProtoReflect.Descriptor instead.

func (*DebuggedGraph) GetGraphId 

func (x *DebuggedGraph) GetGraphId() string

func (*DebuggedGraph) GetGraphName 

func (x *DebuggedGraph) GetGraphName() string

func (*DebuggedGraph) GetInstrumentedGraphDef 

func (x *DebuggedGraph) GetInstrumentedGraphDef() []byte

func (*DebuggedGraph) GetInstrumentedOps 

func (x *DebuggedGraph) GetInstrumentedOps() []string

func (*DebuggedGraph) GetOriginalGraphDef 

func (x *DebuggedGraph) GetOriginalGraphDef() []byte

func (*DebuggedGraph) GetOuterContextId 

func (x *DebuggedGraph) GetOuterContextId() string

func (*DebuggedGraph) ProtoMessage 

func (*DebuggedGraph) ProtoMessage()

func (*DebuggedGraph) ProtoReflect 

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

func (*DebuggedGraph) Reset 

func (x *DebuggedGraph) Reset()

func (*DebuggedGraph) String 

func (x *DebuggedGraph) String() string

type DebuggedSourceFile 

type DebuggedSourceFile struct {

	// The host name on which a source code file is located.
	Host string `protobuf:"bytes,1,opt,name=host,proto3" json:"host,omitempty"`
	// Path to the source code file.
	FilePath string `protobuf:"bytes,2,opt,name=file_path,json=filePath,proto3" json:"file_path,omitempty"`
	// The timestamp at which the source code file is last modified.
	LastModified int64 `protobuf:"varint,3,opt,name=last_modified,json=lastModified,proto3" json:"last_modified,omitempty"`
	// Byte size of the file.
	Bytes int64 `protobuf:"varint,4,opt,name=bytes,proto3" json:"bytes,omitempty"`
	// Line-by-line content of the source code file.
	Lines []string `protobuf:"bytes,5,rep,name=lines,proto3" json:"lines,omitempty"`
	// contains filtered or unexported fields
}

func (*DebuggedSourceFile) Descriptor deprecated 

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

Deprecated: Use DebuggedSourceFile.ProtoReflect.Descriptor instead.

func (*DebuggedSourceFile) GetBytes 

func (x *DebuggedSourceFile) GetBytes() int64

func (*DebuggedSourceFile) GetFilePath 

func (x *DebuggedSourceFile) GetFilePath() string

func (*DebuggedSourceFile) GetHost 

func (x *DebuggedSourceFile) GetHost() string

func (*DebuggedSourceFile) GetLastModified 

func (x *DebuggedSourceFile) GetLastModified() int64

func (*DebuggedSourceFile) GetLines 

func (x *DebuggedSourceFile) GetLines() []string

func (*DebuggedSourceFile) ProtoMessage 

func (*DebuggedSourceFile) ProtoMessage()

func (*DebuggedSourceFile) ProtoReflect 

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

func (*DebuggedSourceFile) Reset 

func (x *DebuggedSourceFile) Reset()

func (*DebuggedSourceFile) String 

func (x *DebuggedSourceFile) String() string

type DebuggedSourceFiles 

type DebuggedSourceFiles struct {

	// A collection of source code files.
	SourceFiles []*DebuggedSourceFile `protobuf:"bytes,1,rep,name=source_files,json=sourceFiles,proto3" json:"source_files,omitempty"`
	// contains filtered or unexported fields
}

func (*DebuggedSourceFiles) Descriptor deprecated 

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

Deprecated: Use DebuggedSourceFiles.ProtoReflect.Descriptor instead.

func (*DebuggedSourceFiles) GetSourceFiles 

func (x *DebuggedSourceFiles) GetSourceFiles() []*DebuggedSourceFile

func (*DebuggedSourceFiles) ProtoMessage 

func (*DebuggedSourceFiles) ProtoMessage()

func (*DebuggedSourceFiles) ProtoReflect 

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

func (*DebuggedSourceFiles) Reset 

func (x *DebuggedSourceFiles) Reset()

func (*DebuggedSourceFiles) String 

func (x *DebuggedSourceFiles) String() string

type DeleteKeyValueRequest 

type DeleteKeyValueRequest struct {
	Key         string `protobuf:"bytes,1,opt,name=key,proto3" json:"key,omitempty"`
	IsDirectory bool   `protobuf:"varint,2,opt,name=is_directory,json=isDirectory,proto3" json:"is_directory,omitempty"`
	// contains filtered or unexported fields
}

Request and response messages for deleting configuration key-value data. When is_directory is true, delete key-values recursively under `key`.

func (*DeleteKeyValueRequest) Descriptor deprecated 

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

Deprecated: Use DeleteKeyValueRequest.ProtoReflect.Descriptor instead.

func (*DeleteKeyValueRequest) GetIsDirectory 

func (x *DeleteKeyValueRequest) GetIsDirectory() bool

func (*DeleteKeyValueRequest) GetKey 

func (x *DeleteKeyValueRequest) GetKey() string

func (*DeleteKeyValueRequest) ProtoMessage 

func (*DeleteKeyValueRequest) ProtoMessage()

func (*DeleteKeyValueRequest) ProtoReflect 

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

func (*DeleteKeyValueRequest) Reset 

func (x *DeleteKeyValueRequest) Reset()

func (*DeleteKeyValueRequest) String 

func (x *DeleteKeyValueRequest) String() string

type DeleteKeyValueResponse 

type DeleteKeyValueResponse struct {
	// contains filtered or unexported fields
}

func (*DeleteKeyValueResponse) Descriptor deprecated 

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

Deprecated: Use DeleteKeyValueResponse.ProtoReflect.Descriptor instead.

func (*DeleteKeyValueResponse) ProtoMessage 

func (*DeleteKeyValueResponse) ProtoMessage()

func (*DeleteKeyValueResponse) ProtoReflect 

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

func (*DeleteKeyValueResponse) Reset 

func (x *DeleteKeyValueResponse) Reset()

func (*DeleteKeyValueResponse) String 

func (x *DeleteKeyValueResponse) String() string

type DeleteWorkerSessionRequest 

type DeleteWorkerSessionRequest struct {

	// Sessions are identified by a given handle.
	SessionHandle string `protobuf:"bytes,1,opt,name=session_handle,json=sessionHandle,proto3" json:"session_handle,omitempty"`
	// contains filtered or unexported fields
}

func (*DeleteWorkerSessionRequest) Descriptor deprecated 

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

Deprecated: Use DeleteWorkerSessionRequest.ProtoReflect.Descriptor instead.

func (*DeleteWorkerSessionRequest) GetSessionHandle 

func (x *DeleteWorkerSessionRequest) GetSessionHandle() string

func (*DeleteWorkerSessionRequest) ProtoMessage 

func (*DeleteWorkerSessionRequest) ProtoMessage()

func (*DeleteWorkerSessionRequest) ProtoReflect 

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

func (*DeleteWorkerSessionRequest) Reset 

func (x *DeleteWorkerSessionRequest) Reset()

func (*DeleteWorkerSessionRequest) String 

func (x *DeleteWorkerSessionRequest) String() string

type DeleteWorkerSessionResponse 

type DeleteWorkerSessionResponse struct {
	// contains filtered or unexported fields
}

func (*DeleteWorkerSessionResponse) Descriptor deprecated 

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

Deprecated: Use DeleteWorkerSessionResponse.ProtoReflect.Descriptor instead.

func (*DeleteWorkerSessionResponse) ProtoMessage 

func (*DeleteWorkerSessionResponse) ProtoMessage()

func (*DeleteWorkerSessionResponse) ProtoReflect 

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

func (*DeleteWorkerSessionResponse) Reset 

func (x *DeleteWorkerSessionResponse) Reset()

func (*DeleteWorkerSessionResponse) String 

func (x *DeleteWorkerSessionResponse) String() string

type DeploymentMode 

type DeploymentMode int32

tf.data service deployment mode. 

const (
	DeploymentMode_DEPLOYMENT_MODE_UNSPECIFIED DeploymentMode = 0
	// tf.data service workers colocate with TF workers.
	DeploymentMode_DEPLOYMENT_MODE_COLOCATED DeploymentMode = 1
	// tf.data service workers run in dedicated tf.data hosts.
	DeploymentMode_DEPLOYMENT_MODE_REMOTE DeploymentMode = 2
	// tf.data service workers run in colocated TF hosts and dedicated tf.data
	// hosts.
	DeploymentMode_DEPLOYMENT_MODE_HYBRID DeploymentMode = 3
)

func (DeploymentMode) Descriptor 

func (DeploymentMode) Descriptor() protoreflect.EnumDescriptor

func (DeploymentMode) Enum 

func (x DeploymentMode) Enum() *DeploymentMode

func (DeploymentMode) EnumDescriptor deprecated 

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

Deprecated: Use DeploymentMode.Descriptor instead.

func (DeploymentMode) Number 

func (x DeploymentMode) Number() protoreflect.EnumNumber

func (DeploymentMode) String 

func (x DeploymentMode) String() string

func (DeploymentMode) Type 

func (DeploymentMode) Type() protoreflect.EnumType

type DeregisterGraphRequest 

type DeregisterGraphRequest struct {

	// The session_handle used when registering the graph. If session_handle is
	// empty, a single global namespace is used.
	SessionHandle string `protobuf:"bytes,2,opt,name=session_handle,json=sessionHandle,proto3" json:"session_handle,omitempty"`
	// Set to true if `CreateWorkerSession` was called for `session_handle`.
	CreateWorkerSessionCalled bool `` /* 141-byte string literal not displayed */
	// REQUIRED: graph_handle must be returned by a RegisterGraph call
	// to the same WorkerService.
	GraphHandle string `protobuf:"bytes,1,opt,name=graph_handle,json=graphHandle,proto3" json:"graph_handle,omitempty"`
	// contains filtered or unexported fields
}

func (*DeregisterGraphRequest) Descriptor deprecated 

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

Deprecated: Use DeregisterGraphRequest.ProtoReflect.Descriptor instead.

func (*DeregisterGraphRequest) GetCreateWorkerSessionCalled 

func (x *DeregisterGraphRequest) GetCreateWorkerSessionCalled() bool

func (*DeregisterGraphRequest) GetGraphHandle 

func (x *DeregisterGraphRequest) GetGraphHandle() string

func (*DeregisterGraphRequest) GetSessionHandle 

func (x *DeregisterGraphRequest) GetSessionHandle() string

func (*DeregisterGraphRequest) ProtoMessage 

func (*DeregisterGraphRequest) ProtoMessage()

func (*DeregisterGraphRequest) ProtoReflect 

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

func (*DeregisterGraphRequest) Reset 

func (x *DeregisterGraphRequest) Reset()

func (*DeregisterGraphRequest) String 

func (x *DeregisterGraphRequest) String() string

type DeregisterGraphResponse 

type DeregisterGraphResponse struct {
	// contains filtered or unexported fields
}

func (*DeregisterGraphResponse) Descriptor deprecated 

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

Deprecated: Use DeregisterGraphResponse.ProtoReflect.Descriptor instead.

func (*DeregisterGraphResponse) ProtoMessage 

func (*DeregisterGraphResponse) ProtoMessage()

func (*DeregisterGraphResponse) ProtoReflect 

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

func (*DeregisterGraphResponse) Reset 

func (x *DeregisterGraphResponse) Reset()

func (*DeregisterGraphResponse) String 

func (x *DeregisterGraphResponse) String() string

type DerivedStatus 

type DerivedStatus struct {
	// contains filtered or unexported fields
}

If included as a payload, this message flags the Status to be a "derived" Status. Used by StatusGroup to ignore certain Statuses when reporting errors to end users.

func (*DerivedStatus) Descriptor deprecated 

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

Deprecated: Use DerivedStatus.ProtoReflect.Descriptor instead.

func (*DerivedStatus) ProtoMessage 

func (*DerivedStatus) ProtoMessage()

func (*DerivedStatus) ProtoReflect 

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

func (*DerivedStatus) Reset 

func (x *DerivedStatus) Reset()

func (*DerivedStatus) String 

func (x *DerivedStatus) String() string

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 DeviceProperties 

type DeviceProperties struct {

	// Device type (CPU, GPU, ...)
	Type string `protobuf:"bytes,1,opt,name=type,proto3" json:"type,omitempty"`
	// Vendor (Intel, nvidia, ...)
	Vendor string `protobuf:"bytes,2,opt,name=vendor,proto3" json:"vendor,omitempty"`
	// Model (Haswell, K40, ...)
	Model string `protobuf:"bytes,3,opt,name=model,proto3" json:"model,omitempty"`
	// Core Frequency in Mhz
	Frequency int64 `protobuf:"varint,4,opt,name=frequency,proto3" json:"frequency,omitempty"`
	// Number of cores
	NumCores int64 `protobuf:"varint,5,opt,name=num_cores,json=numCores,proto3" json:"num_cores,omitempty"`
	// Version of the tools and libraries used with this device (e.g. gcc 4.9,
	// cudnn 5.1)
	Environment map[string]string `` /* 163-byte string literal not displayed */
	// Number of registers per core.
	NumRegisters int64 `protobuf:"varint,7,opt,name=num_registers,json=numRegisters,proto3" json:"num_registers,omitempty"`
	// L1 cache size in bytes
	L1CacheSize int64 `protobuf:"varint,8,opt,name=l1_cache_size,json=l1CacheSize,proto3" json:"l1_cache_size,omitempty"`
	// L2 cache size in bytes
	L2CacheSize int64 `protobuf:"varint,9,opt,name=l2_cache_size,json=l2CacheSize,proto3" json:"l2_cache_size,omitempty"`
	// L3 cache size in bytes
	L3CacheSize int64 `protobuf:"varint,10,opt,name=l3_cache_size,json=l3CacheSize,proto3" json:"l3_cache_size,omitempty"`
	// Shared memory size per multiprocessor in bytes. This field is
	// applicable to GPUs only.
	SharedMemorySizePerMultiprocessor int64 `` /* 168-byte string literal not displayed */
	// Memory size in bytes
	MemorySize int64 `protobuf:"varint,12,opt,name=memory_size,json=memorySize,proto3" json:"memory_size,omitempty"`
	// Memory bandwidth in KB/s
	Bandwidth int64 `protobuf:"varint,13,opt,name=bandwidth,proto3" json:"bandwidth,omitempty"`
	// contains filtered or unexported fields
}

func (*DeviceProperties) Descriptor deprecated 

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

Deprecated: Use DeviceProperties.ProtoReflect.Descriptor instead.

func (*DeviceProperties) GetBandwidth 

func (x *DeviceProperties) GetBandwidth() int64

func (*DeviceProperties) GetEnvironment 

func (x *DeviceProperties) GetEnvironment() map[string]string

func (*DeviceProperties) GetFrequency 

func (x *DeviceProperties) GetFrequency() int64

func (*DeviceProperties) GetL1CacheSize 

func (x *DeviceProperties) GetL1CacheSize() int64

func (*DeviceProperties) GetL2CacheSize 

func (x *DeviceProperties) GetL2CacheSize() int64

func (*DeviceProperties) GetL3CacheSize 

func (x *DeviceProperties) GetL3CacheSize() int64

func (*DeviceProperties) GetMemorySize 

func (x *DeviceProperties) GetMemorySize() int64

func (*DeviceProperties) GetModel 

func (x *DeviceProperties) GetModel() string

func (*DeviceProperties) GetNumCores 

func (x *DeviceProperties) GetNumCores() int64

func (*DeviceProperties) GetNumRegisters 

func (x *DeviceProperties) GetNumRegisters() int64

func (*DeviceProperties) GetSharedMemorySizePerMultiprocessor 

func (x *DeviceProperties) GetSharedMemorySizePerMultiprocessor() int64

func (*DeviceProperties) GetType 

func (x *DeviceProperties) GetType() string

func (*DeviceProperties) GetVendor 

func (x *DeviceProperties) GetVendor() string

func (*DeviceProperties) ProtoMessage 

func (*DeviceProperties) ProtoMessage()

func (*DeviceProperties) ProtoReflect 

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

func (*DeviceProperties) Reset 

func (x *DeviceProperties) Reset()

func (*DeviceProperties) String 

func (x *DeviceProperties) String() string

type DeviceProto 

type DeviceProto struct {
	LocalDeviceOrdinal int32  `protobuf:"varint,1,opt,name=local_device_ordinal,json=localDeviceOrdinal,proto3" json:"local_device_ordinal,omitempty"`
	Name               string `protobuf:"bytes,2,opt,name=name,proto3" json:"name,omitempty"`
	Vendor             string `protobuf:"bytes,3,opt,name=vendor,proto3" json:"vendor,omitempty"`
	// The following fields are present in the GlobalTopologyProto message
	// returned by EnumerateDevices() but not in the LocalTopologyProto messages
	// passed to EnumerateDevices(). In other words, the coordinator node
	// determines the global device IDs during EnumerateDevices().
	GlobalDeviceId int32 `protobuf:"varint,4,opt,name=global_device_id,json=globalDeviceId,proto3" json:"global_device_id,omitempty"` // Globally unique ID number.
	// contains filtered or unexported fields
}

Describes a device local to a host.

func (*DeviceProto) Descriptor deprecated 

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

Deprecated: Use DeviceProto.ProtoReflect.Descriptor instead.

func (*DeviceProto) GetGlobalDeviceId 

func (x *DeviceProto) GetGlobalDeviceId() int32

func (*DeviceProto) GetLocalDeviceOrdinal 

func (x *DeviceProto) GetLocalDeviceOrdinal() int32

func (*DeviceProto) GetName 

func (x *DeviceProto) GetName() string

func (*DeviceProto) GetVendor 

func (x *DeviceProto) GetVendor() string

func (*DeviceProto) ProtoMessage 

func (*DeviceProto) ProtoMessage()

func (*DeviceProto) ProtoReflect 

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

func (*DeviceProto) Reset 

func (x *DeviceProto) Reset()

func (*DeviceProto) String 

func (x *DeviceProto) 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 DictValue 

type DictValue struct {
	Fields map[string]*StructuredValue `` /* 153-byte string literal not displayed */
	// contains filtered or unexported fields
}

Represents a Python dict keyed by `str`. The comment on Unicode from Value.string_value applies analogously.

func (*DictValue) Descriptor deprecated 

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

Deprecated: Use DictValue.ProtoReflect.Descriptor instead.

func (*DictValue) GetFields 

func (x *DictValue) GetFields() map[string]*StructuredValue

func (*DictValue) ProtoMessage 

func (*DictValue) ProtoMessage()

func (*DictValue) ProtoReflect 

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

func (*DictValue) Reset 

func (x *DictValue) Reset()

func (*DictValue) String 

func (x *DictValue) String() string

type DispatcherConfig 

type DispatcherConfig struct {

	// The port for the dispatcher to bind to. A value of 0 indicates that the
	// dispatcher may bind to any available port.
	Port int64 `protobuf:"varint,1,opt,name=port,proto3" json:"port,omitempty"`
	// The protocol for the dispatcher to use when connecting to workers.
	Protocol string `protobuf:"bytes,2,opt,name=protocol,proto3" json:"protocol,omitempty"`
	// A work directory to use for storing dispatcher state, and for recovering
	// during restarts. The empty string indicates not to use any work directory.
	WorkDir string `protobuf:"bytes,3,opt,name=work_dir,json=workDir,proto3" json:"work_dir,omitempty"`
	// Whether to run in fault tolerant mode, where dispatcher state is saved
	// across restarts. Requires that `work_dir` is nonempty.
	FaultTolerantMode bool `protobuf:"varint,4,opt,name=fault_tolerant_mode,json=faultTolerantMode,proto3" json:"fault_tolerant_mode,omitempty"`
	// (Optional.) If the job uses auto-sharding, it needs to specify a fixed list
	// of worker addresses that will register with the dispatcher. The worker
	// addresses should be in the format "host" or "host:port", where "port" is an
	// integer, named port, or %port% to match any port.
	WorkerAddresses []string `protobuf:"bytes,7,rep,name=worker_addresses,json=workerAddresses,proto3" json:"worker_addresses,omitempty"`
	// (Optional.) tf.data service deployment mode. Supported values are "REMOTE",
	// "COLOCATED", and "HYBRID". If unspecified, it is assumed to be "REMOTE".
	DeploymentMode DeploymentMode `` /* 140-byte string literal not displayed */
	// How often the dispatcher should scan through to delete old and unused
	// jobs. A value of 0 indicates that the decision should be left up to the
	// runtime.
	JobGcCheckIntervalMs int64 `` /* 128-byte string literal not displayed */
	// How long a job needs to be unused before it becomes a candidate for garbage
	// collection. A value of -1 indicates that jobs should never be garbage
	// collected. A value of 0 indicates that the decision should be left up to
	// the runtime.
	JobGcTimeoutMs int64 `protobuf:"varint,6,opt,name=job_gc_timeout_ms,json=jobGcTimeoutMs,proto3" json:"job_gc_timeout_ms,omitempty"`
	// How long to wait before garbage-collecting a client that hasn't
	// heartbeated to the dispatcher. A value of 0 indicates that the timeout
	// should be left to the runtime.
	ClientTimeoutMs int64 `protobuf:"varint,8,opt,name=client_timeout_ms,json=clientTimeoutMs,proto3" json:"client_timeout_ms,omitempty"`
	// contains filtered or unexported fields
}

Configuration for a tf.data service DispatchServer. Next id: 10

func (*DispatcherConfig) Descriptor deprecated 

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

Deprecated: Use DispatcherConfig.ProtoReflect.Descriptor instead.

func (*DispatcherConfig) GetClientTimeoutMs 

func (x *DispatcherConfig) GetClientTimeoutMs() int64

func (*DispatcherConfig) GetDeploymentMode 

func (x *DispatcherConfig) GetDeploymentMode() DeploymentMode

func (*DispatcherConfig) GetFaultTolerantMode 

func (x *DispatcherConfig) GetFaultTolerantMode() bool

func (*DispatcherConfig) GetJobGcCheckIntervalMs 

func (x *DispatcherConfig) GetJobGcCheckIntervalMs() int64

func (*DispatcherConfig) GetJobGcTimeoutMs 

func (x *DispatcherConfig) GetJobGcTimeoutMs() int64

func (*DispatcherConfig) GetPort 

func (x *DispatcherConfig) GetPort() int64

func (*DispatcherConfig) GetProtocol 

func (x *DispatcherConfig) GetProtocol() string

func (*DispatcherConfig) GetWorkDir 

func (x *DispatcherConfig) GetWorkDir() string

func (*DispatcherConfig) GetWorkerAddresses 

func (x *DispatcherConfig) GetWorkerAddresses() []string

func (*DispatcherConfig) ProtoMessage 

func (*DispatcherConfig) ProtoMessage()

func (*DispatcherConfig) ProtoReflect 

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

func (*DispatcherConfig) Reset 

func (x *DispatcherConfig) Reset()

func (*DispatcherConfig) String 

func (x *DispatcherConfig) 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 DnnDataType 

type DnnDataType int32

Specifies the data type used by an operation. 

const (
	DataType_kFloat         DnnDataType = 0
	DataType_kDouble        DnnDataType = 1
	DataType_kHalf          DnnDataType = 2
	DataType_kInt8          DnnDataType = 3
	DataType_kInt32         DnnDataType = 4
	DataType_kComplexFloat  DnnDataType = 5
	DataType_kComplexDouble DnnDataType = 6
	DataType_kBF16          DnnDataType = 7
)

func (DnnDataType) Descriptor 

func (DnnDataType) Descriptor() protoreflect.EnumDescriptor

func (DnnDataType) Enum 

func (x DnnDataType) Enum() *DnnDataType

func (DnnDataType) EnumDescriptor deprecated 

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

Deprecated: Use DnnDataType.Descriptor instead.

func (DnnDataType) Number 

func (x DnnDataType) Number() protoreflect.EnumNumber

func (DnnDataType) String 

func (x DnnDataType) String() string

func (DnnDataType) Type 

func (DnnDataType) Type() protoreflect.EnumType

type EnqueueRequest 

type EnqueueRequest struct {
	ContextId uint64       `protobuf:"fixed64,1,opt,name=context_id,json=contextId,proto3" json:"context_id,omitempty"`
	Queue     []*QueueItem `protobuf:"bytes,3,rep,name=queue,proto3" json:"queue,omitempty"`
	// contains filtered or unexported fields
}

func (*EnqueueRequest) Descriptor deprecated 

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

Deprecated: Use EnqueueRequest.ProtoReflect.Descriptor instead.

func (*EnqueueRequest) GetContextId 

func (x *EnqueueRequest) GetContextId() uint64

func (*EnqueueRequest) GetQueue 

func (x *EnqueueRequest) GetQueue() []*QueueItem

func (*EnqueueRequest) ProtoMessage 

func (*EnqueueRequest) ProtoMessage()

func (*EnqueueRequest) ProtoReflect 

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

func (*EnqueueRequest) Reset 

func (x *EnqueueRequest) Reset()

func (*EnqueueRequest) String 

func (x *EnqueueRequest) String() string

type EnqueueResponse 

type EnqueueResponse struct {

	// A single operation response for every item in the request.
	QueueResponse []*QueueResponse `protobuf:"bytes,1,rep,name=queue_response,json=queueResponse,proto3" json:"queue_response,omitempty"`
	// contains filtered or unexported fields
}

func (*EnqueueResponse) Descriptor deprecated 

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

Deprecated: Use EnqueueResponse.ProtoReflect.Descriptor instead.

func (*EnqueueResponse) GetQueueResponse 

func (x *EnqueueResponse) GetQueueResponse() []*QueueResponse

func (*EnqueueResponse) ProtoMessage 

func (*EnqueueResponse) ProtoMessage()

func (*EnqueueResponse) ProtoReflect 

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

func (*EnqueueResponse) Reset 

func (x *EnqueueResponse) Reset()

func (*EnqueueResponse) String 

func (x *EnqueueResponse) String() string

type EnumerateDevicesRequest 

type EnumerateDevicesRequest struct {
	SessionId     uint64              `protobuf:"varint,1,opt,name=session_id,json=sessionId,proto3" json:"session_id,omitempty"`
	LocalTopology *LocalTopologyProto `protobuf:"bytes,3,opt,name=local_topology,json=localTopology,proto3" json:"local_topology,omitempty"`
	// contains filtered or unexported fields
}

func (*EnumerateDevicesRequest) Descriptor deprecated 

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

Deprecated: Use EnumerateDevicesRequest.ProtoReflect.Descriptor instead.

func (*EnumerateDevicesRequest) GetLocalTopology 

func (x *EnumerateDevicesRequest) GetLocalTopology() *LocalTopologyProto

func (*EnumerateDevicesRequest) GetSessionId 

func (x *EnumerateDevicesRequest) GetSessionId() uint64

func (*EnumerateDevicesRequest) ProtoMessage 

func (*EnumerateDevicesRequest) ProtoMessage()

func (*EnumerateDevicesRequest) ProtoReflect 

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

func (*EnumerateDevicesRequest) Reset 

func (x *EnumerateDevicesRequest) Reset()

func (*EnumerateDevicesRequest) String 

func (x *EnumerateDevicesRequest) String() string

type EnumerateDevicesResponse 

type EnumerateDevicesResponse struct {
	GlobalTopology *GlobalTopologyProto `protobuf:"bytes,1,opt,name=global_topology,json=globalTopology,proto3" json:"global_topology,omitempty"`
	// contains filtered or unexported fields
}

func (*EnumerateDevicesResponse) Descriptor deprecated 

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

Deprecated: Use EnumerateDevicesResponse.ProtoReflect.Descriptor instead.

func (*EnumerateDevicesResponse) GetGlobalTopology 

func (x *EnumerateDevicesResponse) GetGlobalTopology() *GlobalTopologyProto

func (*EnumerateDevicesResponse) ProtoMessage 

func (*EnumerateDevicesResponse) ProtoMessage()

func (*EnumerateDevicesResponse) ProtoReflect 

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

func (*EnumerateDevicesResponse) Reset 

func (x *EnumerateDevicesResponse) Reset()

func (*EnumerateDevicesResponse) String 

func (x *EnumerateDevicesResponse) String() string

type ErrorSourceProto 

type ErrorSourceProto struct {
	ErrorSource ErrorSourceProto_ErrorSource `` /* 154-byte string literal not displayed */
	// contains filtered or unexported fields
}

If included as a payload, this message contains the error source information where the error was raised. URI: "type.googleapis.com/tensorflow.core.platform.ErrorSourceProto"

func (*ErrorSourceProto) Descriptor deprecated 

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

Deprecated: Use ErrorSourceProto.ProtoReflect.Descriptor instead.

func (*ErrorSourceProto) GetErrorSource 

func (x *ErrorSourceProto) GetErrorSource() ErrorSourceProto_ErrorSource

func (*ErrorSourceProto) ProtoMessage 

func (*ErrorSourceProto) ProtoMessage()

func (*ErrorSourceProto) ProtoReflect 

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

func (*ErrorSourceProto) Reset 

func (x *ErrorSourceProto) Reset()

func (*ErrorSourceProto) String 

func (x *ErrorSourceProto) String() string

type ErrorSourceProto_ErrorSource 

type ErrorSourceProto_ErrorSource int32
const (
	ErrorSourceProto_UNKNOWN        ErrorSourceProto_ErrorSource = 0
	ErrorSourceProto_TPU_COMPILE_OP ErrorSourceProto_ErrorSource = 1
	// Old bridge.
	ErrorSourceProto_TF_XLA_BRIDGE ErrorSourceProto_ErrorSource = 2
	// TPUBridge.
	ErrorSourceProto_MLIR_BRIDGE_PHASE_1 ErrorSourceProto_ErrorSource = 3
	// LegalizeToHlo.
	ErrorSourceProto_MLIR_BRIDGE_PHASE_2 ErrorSourceProto_ErrorSource = 4
	// eager::RemoteMgr.
	ErrorSourceProto_EAGER_REMOTE_MGR ErrorSourceProto_ErrorSource = 5
)

func (ErrorSourceProto_ErrorSource) Descriptor 

func (ErrorSourceProto_ErrorSource) Descriptor() protoreflect.EnumDescriptor

func (ErrorSourceProto_ErrorSource) Enum 

func (x ErrorSourceProto_ErrorSource) Enum() *ErrorSourceProto_ErrorSource

func (ErrorSourceProto_ErrorSource) EnumDescriptor deprecated 

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

Deprecated: Use ErrorSourceProto_ErrorSource.Descriptor instead.

func (ErrorSourceProto_ErrorSource) Number 

func (x ErrorSourceProto_ErrorSource) Number() protoreflect.EnumNumber

func (ErrorSourceProto_ErrorSource) String 

func (x ErrorSourceProto_ErrorSource) String() string

func (ErrorSourceProto_ErrorSource) Type 

func (ErrorSourceProto_ErrorSource) Type() protoreflect.EnumType

type Event 

type Event struct {

	// Timestamp of the event.
	WallTime float64 `protobuf:"fixed64,1,opt,name=wall_time,json=wallTime,proto3" json:"wall_time,omitempty"`
	// Global step of the event.
	Step int64 `protobuf:"varint,2,opt,name=step,proto3" json:"step,omitempty"`
	// Types that are assignable to What:
	//	*Event_FileVersion
	//	*Event_GraphDef
	//	*Event_Summary
	//	*Event_LogMessage
	//	*Event_SessionLog
	//	*Event_TaggedRunMetadata
	//	*Event_MetaGraphDef
	What isEvent_What `protobuf_oneof:"what"`
	// contains filtered or unexported fields
}

Protocol buffer representing an event that happened during the execution of a Brain model.

func (*Event) Descriptor deprecated 

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

Deprecated: Use Event.ProtoReflect.Descriptor instead.

func (*Event) GetFileVersion 

func (x *Event) GetFileVersion() string

func (*Event) GetGraphDef 

func (x *Event) GetGraphDef() []byte

func (*Event) GetLogMessage deprecated 

func (x *Event) GetLogMessage() *LogMessage

Deprecated: Do not use.

func (*Event) GetMetaGraphDef 

func (x *Event) GetMetaGraphDef() []byte

func (*Event) GetSessionLog 

func (x *Event) GetSessionLog() *SessionLog

func (*Event) GetStep 

func (x *Event) GetStep() int64

func (*Event) GetSummary 

func (x *Event) GetSummary() *Summary

func (*Event) GetTaggedRunMetadata 

func (x *Event) GetTaggedRunMetadata() *TaggedRunMetadata

func (*Event) GetWallTime 

func (x *Event) GetWallTime() float64

func (*Event) GetWhat 

func (m *Event) GetWhat() isEvent_What

func (*Event) ProtoMessage 

func (*Event) ProtoMessage()

func (*Event) ProtoReflect 

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

func (*Event) Reset 

func (x *Event) Reset()

func (*Event) String 

func (x *Event) String() string

type Event_FileVersion 

type Event_FileVersion struct {
	// An event file was started, with the specified version.
	// This is use to identify the contents of the record IO files
	// easily.  Current version is "brain.Event:2".  All versions
	// start with "brain.Event:".
	FileVersion string `protobuf:"bytes,3,opt,name=file_version,json=fileVersion,proto3,oneof"`
}

type Event_GraphDef 

type Event_GraphDef struct {
	// An encoded version of a GraphDef.
	GraphDef []byte `protobuf:"bytes,4,opt,name=graph_def,json=graphDef,proto3,oneof"`
}

type Event_LogMessage 

type Event_LogMessage struct {
	// The user output a log message. This was theoretically used by the defunct
	// tensorboard_logging module, which has since been removed; this field is
	// now deprecated and should not be used.
	//
	// Deprecated: Do not use.
	LogMessage *LogMessage `protobuf:"bytes,6,opt,name=log_message,json=logMessage,proto3,oneof"`
}

type Event_MetaGraphDef 

type Event_MetaGraphDef struct {
	// An encoded version of a MetaGraphDef.
	MetaGraphDef []byte `protobuf:"bytes,9,opt,name=meta_graph_def,json=metaGraphDef,proto3,oneof"`
}

type Event_SessionLog 

type Event_SessionLog struct {
	// The state of the session which can be used for restarting after crashes.
	SessionLog *SessionLog `protobuf:"bytes,7,opt,name=session_log,json=sessionLog,proto3,oneof"`
}

type Event_Summary 

type Event_Summary struct {
	// A summary was generated.
	Summary *Summary `protobuf:"bytes,5,opt,name=summary,proto3,oneof"`
}

type Event_TaggedRunMetadata 

type Event_TaggedRunMetadata struct {
	// The metadata returned by running a session.run() call.
	TaggedRunMetadata *TaggedRunMetadata `protobuf:"bytes,8,opt,name=tagged_run_metadata,json=taggedRunMetadata,proto3,oneof"`
}

type Example 

type Example struct {
	Features *Features `protobuf:"bytes,1,opt,name=features,proto3" json:"features,omitempty"`
	// contains filtered or unexported fields
}

func (*Example) Descriptor deprecated 

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

Deprecated: Use Example.ProtoReflect.Descriptor instead.

func (*Example) GetFeatures 

func (x *Example) GetFeatures() *Features

func (*Example) ProtoMessage 

func (*Example) ProtoMessage()

func (*Example) ProtoReflect 

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

func (*Example) Reset 

func (x *Example) Reset()

func (*Example) String 

func (x *Example) String() string

type ExampleParserConfiguration 

type ExampleParserConfiguration struct {
	FeatureMap map[string]*FeatureConfiguration `` /* 179-byte string literal not displayed */
	// contains filtered or unexported fields
}

func (*ExampleParserConfiguration) Descriptor deprecated 

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

Deprecated: Use ExampleParserConfiguration.ProtoReflect.Descriptor instead.

func (*ExampleParserConfiguration) GetFeatureMap 

func (x *ExampleParserConfiguration) GetFeatureMap() map[string]*FeatureConfiguration

func (*ExampleParserConfiguration) ProtoMessage 

func (*ExampleParserConfiguration) ProtoMessage()

func (*ExampleParserConfiguration) ProtoReflect 

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

func (*ExampleParserConfiguration) Reset 

func (x *ExampleParserConfiguration) Reset()

func (*ExampleParserConfiguration) String 

func (x *ExampleParserConfiguration) String() string

type Execution 

type Execution struct {

	// Op type (e.g., "MatMul").
	// In the case of a Graph, this is the name of the Graph.
	OpType string `protobuf:"bytes,1,opt,name=op_type,json=opType,proto3" json:"op_type,omitempty"`
	// Number of output tensors.
	NumOutputs int32 `protobuf:"varint,2,opt,name=num_outputs,json=numOutputs,proto3" json:"num_outputs,omitempty"`
	// The graph that's executed: applicable only to the eager
	// execution of a FuncGraph.
	GraphId string `protobuf:"bytes,3,opt,name=graph_id,json=graphId,proto3" json:"graph_id,omitempty"`
	// IDs of the input tensors (if available).
	InputTensorIds []int64 `protobuf:"varint,4,rep,packed,name=input_tensor_ids,json=inputTensorIds,proto3" json:"input_tensor_ids,omitempty"`
	// IDs of the output tensors (if availbable).
	// If specified, must have the same length as tensor_protos.
	OutputTensorIds []int64 `protobuf:"varint,5,rep,packed,name=output_tensor_ids,json=outputTensorIds,proto3" json:"output_tensor_ids,omitempty"`
	// Type of the tensor value encapsulated in this proto.
	TensorDebugMode TensorDebugMode `` /* 141-byte string literal not displayed */
	// Output Tensor values in the type described by `tensor_value_type`.
	// The length of this should match `num_outputs`.
	TensorProtos []*TensorProto `protobuf:"bytes,7,rep,name=tensor_protos,json=tensorProtos,proto3" json:"tensor_protos,omitempty"`
	// Stack trace of the eager execution.
	CodeLocation *CodeLocation `protobuf:"bytes,8,opt,name=code_location,json=codeLocation,proto3" json:"code_location,omitempty"`
	// Debugged-generated IDs of the devices on which the output tensors reside.
	// To look up details about the device (e.g., name), cross-reference this
	// field with the DebuggedDevice messages.
	OutputTensorDeviceIds []int32 `` /* 136-byte string literal not displayed */
	// contains filtered or unexported fields
}

Data relating to the eager execution of an op or a Graph. For a op that generates N output tensors (N >= 0), only one Execution proto will be used to describe the execution event.

func (*Execution) Descriptor deprecated 

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

Deprecated: Use Execution.ProtoReflect.Descriptor instead.

func (*Execution) GetCodeLocation 

func (x *Execution) GetCodeLocation() *CodeLocation

func (*Execution) GetGraphId 

func (x *Execution) GetGraphId() string

func (*Execution) GetInputTensorIds 

func (x *Execution) GetInputTensorIds() []int64

func (*Execution) GetNumOutputs 

func (x *Execution) GetNumOutputs() int32

func (*Execution) GetOpType 

func (x *Execution) GetOpType() string

func (*Execution) GetOutputTensorDeviceIds 

func (x *Execution) GetOutputTensorDeviceIds() []int32

func (*Execution) GetOutputTensorIds 

func (x *Execution) GetOutputTensorIds() []int64

func (*Execution) GetTensorDebugMode 

func (x *Execution) GetTensorDebugMode() TensorDebugMode

func (*Execution) GetTensorProtos 

func (x *Execution) GetTensorProtos() []*TensorProto

func (*Execution) ProtoMessage 

func (*Execution) ProtoMessage()

func (*Execution) ProtoReflect 

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

func (*Execution) Reset 

func (x *Execution) Reset()

func (*Execution) String 

func (x *Execution) String() string

type ExecutorOpts 

type ExecutorOpts struct {
	RecordCosts                    bool `protobuf:"varint,1,opt,name=record_costs,json=recordCosts,proto3" json:"record_costs,omitempty"`
	RecordTimeline                 bool `protobuf:"varint,3,opt,name=record_timeline,json=recordTimeline,proto3" json:"record_timeline,omitempty"`
	RecordPartitionGraphs          bool `` /* 127-byte string literal not displayed */
	ReportTensorAllocationsUponOom bool `` /* 158-byte string literal not displayed */
	// contains filtered or unexported fields
}

Options specific to the execution of a single step.

func (*ExecutorOpts) Descriptor deprecated 

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

Deprecated: Use ExecutorOpts.ProtoReflect.Descriptor instead.

func (*ExecutorOpts) GetRecordCosts 

func (x *ExecutorOpts) GetRecordCosts() bool

func (*ExecutorOpts) GetRecordPartitionGraphs 

func (x *ExecutorOpts) GetRecordPartitionGraphs() bool

func (*ExecutorOpts) GetRecordTimeline 

func (x *ExecutorOpts) GetRecordTimeline() bool

func (*ExecutorOpts) GetReportTensorAllocationsUponOom 

func (x *ExecutorOpts) GetReportTensorAllocationsUponOom() bool

func (*ExecutorOpts) ProtoMessage 

func (*ExecutorOpts) ProtoMessage()

func (*ExecutorOpts) ProtoReflect 

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

func (*ExecutorOpts) Reset 

func (x *ExecutorOpts) Reset()

func (*ExecutorOpts) String 

func (x *ExecutorOpts) String() string

type ExtendSessionRequest 

type ExtendSessionRequest struct {

	// REQUIRED: session_handle must be returned by a CreateSession call
	// to the same master service.
	SessionHandle string `protobuf:"bytes,1,opt,name=session_handle,json=sessionHandle,proto3" json:"session_handle,omitempty"`
	// REQUIRED: The nodes to be added to the session's graph. If any node has
	// the same name as an existing node, the operation will fail with
	// ILLEGAL_ARGUMENT.
	GraphDef *GraphDef `protobuf:"bytes,2,opt,name=graph_def,json=graphDef,proto3" json:"graph_def,omitempty"`
	// REQUIRED: The version number of the graph to be extended. This will be
	// tested against the current server-side version number, and the operation
	// will fail with FAILED_PRECONDITION if they do not match.
	CurrentGraphVersion int64 `protobuf:"varint,3,opt,name=current_graph_version,json=currentGraphVersion,proto3" json:"current_graph_version,omitempty"`
	// contains filtered or unexported fields
}

func (*ExtendSessionRequest) Descriptor deprecated 

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

Deprecated: Use ExtendSessionRequest.ProtoReflect.Descriptor instead.

func (*ExtendSessionRequest) GetCurrentGraphVersion 

func (x *ExtendSessionRequest) GetCurrentGraphVersion() int64

func (*ExtendSessionRequest) GetGraphDef 

func (x *ExtendSessionRequest) GetGraphDef() *GraphDef

func (*ExtendSessionRequest) GetSessionHandle 

func (x *ExtendSessionRequest) GetSessionHandle() string

func (*ExtendSessionRequest) ProtoMessage 

func (*ExtendSessionRequest) ProtoMessage()

func (*ExtendSessionRequest) ProtoReflect 

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

func (*ExtendSessionRequest) Reset 

func (x *ExtendSessionRequest) Reset()

func (*ExtendSessionRequest) String 

func (x *ExtendSessionRequest) String() string

type ExtendSessionResponse 

type ExtendSessionResponse struct {

	// The new version number for the extended graph, to be used in the next call
	// to ExtendSession.
	NewGraphVersion int64 `protobuf:"varint,4,opt,name=new_graph_version,json=newGraphVersion,proto3" json:"new_graph_version,omitempty"`
	// contains filtered or unexported fields
}

func (*ExtendSessionResponse) Descriptor deprecated 

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

Deprecated: Use ExtendSessionResponse.ProtoReflect.Descriptor instead.

func (*ExtendSessionResponse) GetNewGraphVersion 

func (x *ExtendSessionResponse) GetNewGraphVersion() int64

func (*ExtendSessionResponse) ProtoMessage 

func (*ExtendSessionResponse) ProtoMessage()

func (*ExtendSessionResponse) ProtoReflect 

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

func (*ExtendSessionResponse) Reset 

func (x *ExtendSessionResponse) Reset()

func (*ExtendSessionResponse) String 

func (x *ExtendSessionResponse) String() string

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 Feature 

type Feature struct {

	// Each feature can be exactly one kind.
	//
	// Types that are assignable to Kind:
	//	*Feature_BytesList
	//	*Feature_FloatList
	//	*Feature_Int64List
	Kind isFeature_Kind `protobuf_oneof:"kind"`
	// contains filtered or unexported fields
}

Containers for non-sequential data.

func (*Feature) Descriptor deprecated 

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

Deprecated: Use Feature.ProtoReflect.Descriptor instead.

func (*Feature) GetBytesList 

func (x *Feature) GetBytesList() *BytesList

func (*Feature) GetFloatList 

func (x *Feature) GetFloatList() *FloatList

func (*Feature) GetInt64List 

func (x *Feature) GetInt64List() *Int64List

func (*Feature) GetKind 

func (m *Feature) GetKind() isFeature_Kind

func (*Feature) ProtoMessage 

func (*Feature) ProtoMessage()

func (*Feature) ProtoReflect 

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

func (*Feature) Reset 

func (x *Feature) Reset()

func (*Feature) String 

func (x *Feature) String() string

type FeatureConfiguration 

type FeatureConfiguration struct {

	// Types that are assignable to Config:
	//	*FeatureConfiguration_FixedLenFeature
	//	*FeatureConfiguration_VarLenFeature
	Config isFeatureConfiguration_Config `protobuf_oneof:"config"`
	// contains filtered or unexported fields
}

func (*FeatureConfiguration) Descriptor deprecated 

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

Deprecated: Use FeatureConfiguration.ProtoReflect.Descriptor instead.

func (*FeatureConfiguration) GetConfig 

func (m *FeatureConfiguration) GetConfig() isFeatureConfiguration_Config

func (*FeatureConfiguration) GetFixedLenFeature 

func (x *FeatureConfiguration) GetFixedLenFeature() *FixedLenFeatureProto

func (*FeatureConfiguration) GetVarLenFeature 

func (x *FeatureConfiguration) GetVarLenFeature() *VarLenFeatureProto

func (*FeatureConfiguration) ProtoMessage 

func (*FeatureConfiguration) ProtoMessage()

func (*FeatureConfiguration) ProtoReflect 

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

func (*FeatureConfiguration) Reset 

func (x *FeatureConfiguration) Reset()

func (*FeatureConfiguration) String 

func (x *FeatureConfiguration) String() string

type FeatureConfiguration_FixedLenFeature 

type FeatureConfiguration_FixedLenFeature struct {
	FixedLenFeature *FixedLenFeatureProto `protobuf:"bytes,1,opt,name=fixed_len_feature,json=fixedLenFeature,proto3,oneof"`
}

type FeatureConfiguration_VarLenFeature 

type FeatureConfiguration_VarLenFeature struct {
	VarLenFeature *VarLenFeatureProto `protobuf:"bytes,2,opt,name=var_len_feature,json=varLenFeature,proto3,oneof"`
}

type FeatureList 

type FeatureList struct {
	Feature []*Feature `protobuf:"bytes,1,rep,name=feature,proto3" json:"feature,omitempty"`
	// contains filtered or unexported fields
}

Containers for sequential data.

A FeatureList contains lists of Features. These may hold zero or more Feature values.

FeatureLists are organized into categories by name. The FeatureLists message contains the mapping from name to FeatureList.

func (*FeatureList) Descriptor deprecated 

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

Deprecated: Use FeatureList.ProtoReflect.Descriptor instead.

func (*FeatureList) GetFeature 

func (x *FeatureList) GetFeature() []*Feature

func (*FeatureList) ProtoMessage 

func (*FeatureList) ProtoMessage()

func (*FeatureList) ProtoReflect 

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

func (*FeatureList) Reset 

func (x *FeatureList) Reset()

func (*FeatureList) String 

func (x *FeatureList) String() string

type FeatureLists 

type FeatureLists struct {

	// Map from feature name to feature list.
	FeatureList map[string]*FeatureList `` /* 182-byte string literal not displayed */
	// contains filtered or unexported fields
}

func (*FeatureLists) Descriptor deprecated 

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

Deprecated: Use FeatureLists.ProtoReflect.Descriptor instead.

func (*FeatureLists) GetFeatureList 

func (x *FeatureLists) GetFeatureList() map[string]*FeatureList

func (*FeatureLists) ProtoMessage 

func (*FeatureLists) ProtoMessage()

func (*FeatureLists) ProtoReflect 

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

func (*FeatureLists) Reset 

func (x *FeatureLists) Reset()

func (*FeatureLists) String 

func (x *FeatureLists) String() string

type Feature_BytesList 

type Feature_BytesList struct {
	BytesList *BytesList `protobuf:"bytes,1,opt,name=bytes_list,json=bytesList,proto3,oneof"`
}

type Feature_FloatList 

type Feature_FloatList struct {
	FloatList *FloatList `protobuf:"bytes,2,opt,name=float_list,json=floatList,proto3,oneof"`
}

type Feature_Int64List 

type Feature_Int64List struct {
	Int64List *Int64List `protobuf:"bytes,3,opt,name=int64_list,json=int64List,proto3,oneof"`
}

type Features 

type Features struct {

	// Map from feature name to feature.
	Feature map[string]*Feature `` /* 155-byte string literal not displayed */
	// contains filtered or unexported fields
}

func (*Features) Descriptor deprecated 

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

Deprecated: Use Features.ProtoReflect.Descriptor instead.

func (*Features) GetFeature 

func (x *Features) GetFeature() map[string]*Feature

func (*Features) ProtoMessage 

func (*Features) ProtoMessage()

func (*Features) ProtoReflect 

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

func (*Features) Reset 

func (x *Features) Reset()

func (*Features) String 

func (x *Features) String() string

type FilterLayout 

type FilterLayout int32

Describes how a convolution filter is laid out in the memory. 

const (
	// Naming convention:
	// Y <-> row or height
	// X <-> column or width
	// Output <-> output feature, or N
	// Input <-> input feature, or N
	// TODO(timshen): turn them into cuDNN names, e.g. kNCHW.
	FilterLayout_kOutputInputYX   FilterLayout = 0 // cuDNN's NCHW layout
	FilterLayout_kOutputYXInput   FilterLayout = 1 // cuDNN's NHWC layout
	FilterLayout_kOutputInputYX4  FilterLayout = 2 // cuDNN's NCHW_VECT_C layout with 4-elem vectors
	FilterLayout_kOutputInputYX32 FilterLayout = 5 // cuDNN's NCHW_VECT_C layout with 32-elem vectors
	FilterLayout_kInputYXOutput   FilterLayout = 3
	FilterLayout_kYXInputOutput   FilterLayout = 4
)

func (FilterLayout) Descriptor 

func (FilterLayout) Descriptor() protoreflect.EnumDescriptor

func (FilterLayout) Enum 

func (x FilterLayout) Enum() *FilterLayout

func (FilterLayout) EnumDescriptor deprecated 

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

Deprecated: Use FilterLayout.Descriptor instead.

func (FilterLayout) Number 

func (x FilterLayout) Number() protoreflect.EnumNumber

func (FilterLayout) String 

func (x FilterLayout) String() string

func (FilterLayout) Type 

func (FilterLayout) Type() protoreflect.EnumType

type FingerprintDef 

type FingerprintDef struct {

	// Hash of the graph_def, referred to as a "checksum".
	GraphDefChecksum uint64 `protobuf:"varint,1,opt,name=graph_def_checksum,json=graphDefChecksum,proto3" json:"graph_def_checksum,omitempty"`
	// Hash of regularized graph_def.
	GraphDefProgramHash uint64 `protobuf:"varint,2,opt,name=graph_def_program_hash,json=graphDefProgramHash,proto3" json:"graph_def_program_hash,omitempty"`
	// Hash of the regularized (sorted) SignatureDefs.
	SignatureDefHash uint64 `protobuf:"varint,3,opt,name=signature_def_hash,json=signatureDefHash,proto3" json:"signature_def_hash,omitempty"`
	// Hash of the regularized SavedObjectGraph.
	SavedObjectGraphHash uint64 `` /* 126-byte string literal not displayed */
	// Hash of the checkpoint.
	CheckpointHash uint64 `protobuf:"varint,5,opt,name=checkpoint_hash,json=checkpointHash,proto3" json:"checkpoint_hash,omitempty"`
	// Version specification of the fingerprint.
	Version *VersionDef `protobuf:"bytes,6,opt,name=version,proto3" json:"version,omitempty"`
	// contains filtered or unexported fields
}

Protocol buffer representing a SavedModel Fingerprint.

If there are multiple MetaGraphDefs in the SavedModel, the FingerprintDef corresponds to the first one.

func (*FingerprintDef) Descriptor deprecated 

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

Deprecated: Use FingerprintDef.ProtoReflect.Descriptor instead.

func (*FingerprintDef) GetCheckpointHash 

func (x *FingerprintDef) GetCheckpointHash() uint64

func (*FingerprintDef) GetGraphDefChecksum 

func (x *FingerprintDef) GetGraphDefChecksum() uint64

func (*FingerprintDef) GetGraphDefProgramHash 

func (x *FingerprintDef) GetGraphDefProgramHash() uint64

func (*FingerprintDef) GetSavedObjectGraphHash 

func (x *FingerprintDef) GetSavedObjectGraphHash() uint64

func (*FingerprintDef) GetSignatureDefHash 

func (x *FingerprintDef) GetSignatureDefHash() uint64

func (*FingerprintDef) GetVersion 

func (x *FingerprintDef) GetVersion() *VersionDef

func (*FingerprintDef) ProtoMessage 

func (*FingerprintDef) ProtoMessage()

func (*FingerprintDef) ProtoReflect 

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

func (*FingerprintDef) Reset 

func (x *FingerprintDef) Reset()

func (*FingerprintDef) String 

func (x *FingerprintDef) String() string

type FixedLenFeatureProto 

type FixedLenFeatureProto 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"`
	DefaultValue           *TensorProto      `protobuf:"bytes,3,opt,name=default_value,json=defaultValue,proto3" json:"default_value,omitempty"`
	ValuesOutputTensorName string            `` /* 131-byte string literal not displayed */
	// contains filtered or unexported fields
}

func (*FixedLenFeatureProto) Descriptor deprecated 

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

Deprecated: Use FixedLenFeatureProto.ProtoReflect.Descriptor instead.

func (*FixedLenFeatureProto) GetDefaultValue 

func (x *FixedLenFeatureProto) GetDefaultValue() *TensorProto

func (*FixedLenFeatureProto) GetDtype 

func (x *FixedLenFeatureProto) GetDtype() DataType

func (*FixedLenFeatureProto) GetShape 

func (x *FixedLenFeatureProto) GetShape() *TensorShapeProto

func (*FixedLenFeatureProto) GetValuesOutputTensorName 

func (x *FixedLenFeatureProto) GetValuesOutputTensorName() string

func (*FixedLenFeatureProto) ProtoMessage 

func (*FixedLenFeatureProto) ProtoMessage()

func (*FixedLenFeatureProto) ProtoReflect 

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

func (*FixedLenFeatureProto) Reset 

func (x *FixedLenFeatureProto) Reset()

func (*FixedLenFeatureProto) String 

func (x *FixedLenFeatureProto) String() string

type FloatList 

type FloatList struct {
	Value []float32 `protobuf:"fixed32,1,rep,packed,name=value,proto3" json:"value,omitempty"`
	// contains filtered or unexported fields
}

func (*FloatList) Descriptor deprecated 

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

Deprecated: Use FloatList.ProtoReflect.Descriptor instead.

func (*FloatList) GetValue 

func (x *FloatList) GetValue() []float32

func (*FloatList) ProtoMessage 

func (*FloatList) ProtoMessage()

func (*FloatList) ProtoReflect 

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

func (*FloatList) Reset 

func (x *FloatList) Reset()

func (*FloatList) String 

func (x *FloatList) String() string

type FullTypeDef 

type FullTypeDef struct {

	// The principal type represented by this object. This may be a concrete type
	// (Tensor, Dataset) a type variable (used for dependent types) a type
	// symbol (Any, Union). See FullTypeId for details.
	TypeId FullTypeId     `protobuf:"varint,1,opt,name=type_id,json=typeId,proto3,enum=tensorflow.FullTypeId" json:"type_id,omitempty"`
	Args   []*FullTypeDef `protobuf:"bytes,2,rep,name=args,proto3" json:"args,omitempty"`
	// Literal values of this type object, if the 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 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 FunctionSpec 

type FunctionSpec struct {

	// Full arg spec from inspect.getfullargspec().
	Fullargspec *StructuredValue `protobuf:"bytes,1,opt,name=fullargspec,proto3" json:"fullargspec,omitempty"`
	// Whether this represents a class method.
	IsMethod bool `protobuf:"varint,2,opt,name=is_method,json=isMethod,proto3" json:"is_method,omitempty"`
	// The input signature, if specified.
	InputSignature *StructuredValue        `protobuf:"bytes,5,opt,name=input_signature,json=inputSignature,proto3" json:"input_signature,omitempty"`
	JitCompile     FunctionSpec_JitCompile `` /* 132-byte string literal not displayed */
	// contains filtered or unexported fields
}

Represents `FunctionSpec` used in `Function`. This represents a function that has been wrapped as a TensorFlow `Function`.

func (*FunctionSpec) Descriptor deprecated 

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

Deprecated: Use FunctionSpec.ProtoReflect.Descriptor instead.

func (*FunctionSpec) GetFullargspec 

func (x *FunctionSpec) GetFullargspec() *StructuredValue

func (*FunctionSpec) GetInputSignature 

func (x *FunctionSpec) GetInputSignature() *StructuredValue

func (*FunctionSpec) GetIsMethod 

func (x *FunctionSpec) GetIsMethod() bool

func (*FunctionSpec) GetJitCompile 

func (x *FunctionSpec) GetJitCompile() FunctionSpec_JitCompile

func (*FunctionSpec) ProtoMessage 

func (*FunctionSpec) ProtoMessage()

func (*FunctionSpec) ProtoReflect 

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

func (*FunctionSpec) Reset 

func (x *FunctionSpec) Reset()

func (*FunctionSpec) String 

func (x *FunctionSpec) String() string

type FunctionSpec_JitCompile 

type FunctionSpec_JitCompile int32

Whether the function should be compiled by XLA.

The public interface to `tf.function` uses an optional boolean to represent three distinct states for this field. Unfortunately, proto3 removes the ability to explicitly check for the presence or absence of a field, so we instead map to an enum.

See `tf.function` for details. 

const (
	FunctionSpec_DEFAULT FunctionSpec_JitCompile = 0
	FunctionSpec_ON      FunctionSpec_JitCompile = 1
	FunctionSpec_OFF     FunctionSpec_JitCompile = 2
)

func (FunctionSpec_JitCompile) Descriptor 

func (FunctionSpec_JitCompile) Descriptor() protoreflect.EnumDescriptor

func (FunctionSpec_JitCompile) Enum 

func (x FunctionSpec_JitCompile) Enum() *FunctionSpec_JitCompile

func (FunctionSpec_JitCompile) EnumDescriptor deprecated 

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

Deprecated: Use FunctionSpec_JitCompile.Descriptor instead.

func (FunctionSpec_JitCompile) Number 

func (x FunctionSpec_JitCompile) Number() protoreflect.EnumNumber

func (FunctionSpec_JitCompile) String 

func (x FunctionSpec_JitCompile) String() string

func (FunctionSpec_JitCompile) Type 

func (FunctionSpec_JitCompile) Type() protoreflect.EnumType

type GPUOptions 

type GPUOptions struct {

	// Fraction of the available GPU memory to allocate for each process.
	// 1 means to allocate all of the GPU memory, 0.5 means the process
	// allocates up to ~50% of the available GPU memory.
	//
	// GPU memory is pre-allocated unless the allow_growth option is enabled.
	//
	// If greater than 1.0, uses CUDA unified memory to potentially oversubscribe
	// the amount of memory available on the GPU device by using host memory as a
	// swap space. Accessing memory not available on the device will be
	// significantly slower as that would require memory transfer between the host
	// and the device. Options to reduce the memory requirement should be
	// considered before enabling this option as this may come with a negative
	// performance impact. Oversubscription using the unified memory requires
	// Pascal class or newer GPUs and it is currently only supported on the Linux
	// operating system. See
	// https://docs.nvidia.com/cuda/cuda-c-programming-guide/index.html#um-requirements
	// for the detailed requirements.
	PerProcessGpuMemoryFraction float64 `` /* 150-byte string literal not displayed */
	// If true, the allocator does not pre-allocate the entire specified
	// GPU memory region, instead starting small and growing as needed.
	AllowGrowth bool `protobuf:"varint,4,opt,name=allow_growth,json=allowGrowth,proto3" json:"allow_growth,omitempty"`
	// The type of GPU allocation strategy to use.
	//
	// Allowed values:
	// "": The empty string (default) uses a system-chosen default
	//     which may change over time.
	//
	// "BFC": A "Best-fit with coalescing" algorithm, simplified from a
	//        version of dlmalloc.
	AllocatorType string `protobuf:"bytes,2,opt,name=allocator_type,json=allocatorType,proto3" json:"allocator_type,omitempty"`
	// Delay deletion of up to this many bytes to reduce the number of
	// interactions with gpu driver code.  If 0, the system chooses
	// a reasonable default (several MBs).
	DeferredDeletionBytes int64 `` /* 127-byte string literal not displayed */
	// A comma-separated list of GPU ids that determines the 'visible'
	// to 'virtual' mapping of GPU devices.  For example, if TensorFlow
	// can see 8 GPU devices in the process, and one wanted to map
	// visible GPU devices 5 and 3 as "/device:GPU:0", and "/device:GPU:1",
	// then one would specify this field as "5,3".  This field is similar in
	// spirit to the CUDA_VISIBLE_DEVICES environment variable, except
	// it applies to the visible GPU devices in the process.
	//
	// NOTE:
	// 1. The GPU driver provides the process with the visible GPUs
	//    in an order which is not guaranteed to have any correlation to
	//    the *physical* GPU id in the machine.  This field is used for
	//    remapping "visible" to "virtual", which means this operates only
	//    after the process starts.  Users are required to use vendor
	//    specific mechanisms (e.g., CUDA_VISIBLE_DEVICES) to control the
	//    physical to visible device mapping prior to invoking TensorFlow.
	// 2. In the code, the ids in this list are also called "platform GPU id"s,
	//    and the 'virtual' ids of GPU devices (i.e. the ids in the device
	//    name "/device:GPU:<id>") are also called "TF GPU id"s. Please
	//    refer to third_party/tensorflow/core/common_runtime/gpu/gpu_id.h
	//    for more information.
	VisibleDeviceList string `protobuf:"bytes,5,opt,name=visible_device_list,json=visibleDeviceList,proto3" json:"visible_device_list,omitempty"`
	// In the event polling loop sleep this many microseconds between
	// PollEvents calls, when the queue is not empty.  If value is not
	// set or set to 0, gets set to a non-zero default.
	PollingActiveDelayUsecs int32 `` /* 135-byte string literal not displayed */
	// This field is deprecated and ignored.
	PollingInactiveDelayMsecs int32 `` /* 141-byte string literal not displayed */
	// Force all tensors to be gpu_compatible. On a GPU-enabled TensorFlow,
	// enabling this option forces all CPU tensors to be allocated with Cuda
	// pinned memory. Normally, TensorFlow will infer which tensors should be
	// allocated as the pinned memory. But in case where the inference is
	// incomplete, this option can significantly speed up the cross-device memory
	// copy performance as long as it fits the memory.
	// Note that this option is not something that should be
	// enabled by default for unknown or very large models, since all Cuda pinned
	// memory is unpageable, having too much pinned memory might negatively impact
	// the overall host system performance.
	ForceGpuCompatible bool `protobuf:"varint,8,opt,name=force_gpu_compatible,json=forceGpuCompatible,proto3" json:"force_gpu_compatible,omitempty"`
	// Everything inside experimental is subject to change and is not subject
	// to API stability guarantees in
	// https://www.tensorflow.org/guide/version_compat.
	Experimental *GPUOptions_Experimental `protobuf:"bytes,9,opt,name=experimental,proto3" json:"experimental,omitempty"`
	// contains filtered or unexported fields
}

func (*GPUOptions) Descriptor deprecated 

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

Deprecated: Use GPUOptions.ProtoReflect.Descriptor instead.

func (*GPUOptions) GetAllocatorType 

func (x *GPUOptions) GetAllocatorType() string

func (*GPUOptions) GetAllowGrowth 

func (x *GPUOptions) GetAllowGrowth() bool

func (*GPUOptions) GetDeferredDeletionBytes 

func (x *GPUOptions) GetDeferredDeletionBytes() int64

func (*GPUOptions) GetExperimental 

func (x *GPUOptions) GetExperimental() *GPUOptions_Experimental

func (*GPUOptions) GetForceGpuCompatible 

func (x *GPUOptions) GetForceGpuCompatible() bool

func (*GPUOptions) GetPerProcessGpuMemoryFraction 

func (x *GPUOptions) GetPerProcessGpuMemoryFraction() float64

func (*GPUOptions) GetPollingActiveDelayUsecs 

func (x *GPUOptions) GetPollingActiveDelayUsecs() int32

func (*GPUOptions) GetPollingInactiveDelayMsecs 

func (x *GPUOptions) GetPollingInactiveDelayMsecs() int32

func (*GPUOptions) GetVisibleDeviceList 

func (x *GPUOptions) GetVisibleDeviceList() string

func (*GPUOptions) ProtoMessage 

func (*GPUOptions) ProtoMessage()

func (*GPUOptions) ProtoReflect 

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

func (*GPUOptions) Reset 

func (x *GPUOptions) Reset()

func (*GPUOptions) String 

func (x *GPUOptions) String() string

type GPUOptions_Experimental 

type GPUOptions_Experimental struct {

	// The multi virtual device settings. If empty (not set), it will create
	// single virtual device on each visible GPU, according to the settings
	// in "visible_device_list" above. Otherwise, the number of elements in the
	// list must be the same as the number of visible GPUs (after
	// "visible_device_list" filtering if it is set), and the string represented
	// device names (e.g. /device:GPU:<id>) will refer to the virtual
	// devices and have the <id> field assigned sequentially starting from 0,
	// according to the order of the virtual devices determined by
	// device_ordinal and the location in the virtual device list.
	//
	// For example,
	//   visible_device_list = "1,0"
	//   virtual_devices { memory_limit: 1GB memory_limit: 2GB }
	//   virtual_devices { memory_limit: 3GB memory_limit: 4GB }
	// will create 4 virtual devices as:
	//   /device:GPU:0 -> visible GPU 1 with 1GB memory
	//   /device:GPU:1 -> visible GPU 1 with 2GB memory
	//   /device:GPU:2 -> visible GPU 0 with 3GB memory
	//   /device:GPU:3 -> visible GPU 0 with 4GB memory
	//
	// but
	//   visible_device_list = "1,0"
	//   virtual_devices { memory_limit: 1GB memory_limit: 2GB
	//                     device_ordinal: 10 device_ordinal: 20}
	//   virtual_devices { memory_limit: 3GB memory_limit: 4GB
	//                     device_ordinal: 10 device_ordinal: 20}
	// will create 4 virtual devices as:
	//   /device:GPU:0 -> visible GPU 1 with 1GB memory  (ordinal 10)
	//   /device:GPU:1 -> visible GPU 0 with 3GB memory  (ordinal 10)
	//   /device:GPU:2 -> visible GPU 1 with 2GB memory  (ordinal 20)
	//   /device:GPU:3 -> visible GPU 0 with 4GB memory  (ordinal 20)
	//
	// NOTE:
	// 1. It's invalid to set both this and "per_process_gpu_memory_fraction"
	//    at the same time.
	// 2. Currently this setting is per-process, not per-session. Using
	//    different settings in different sessions within same process will
	//    result in undefined behavior.
	VirtualDevices []*GPUOptions_Experimental_VirtualDevices `protobuf:"bytes,1,rep,name=virtual_devices,json=virtualDevices,proto3" json:"virtual_devices,omitempty"`
	// If true, uses CUDA unified memory for memory allocations. If
	// per_process_gpu_memory_fraction option is greater than 1.0, then unified
	// memory is used regardless of the value for this field. See comments for
	// per_process_gpu_memory_fraction field for more details and requirements
	// of the unified memory. This option is useful to oversubscribe memory if
	// multiple processes are sharing a single GPU while individually using less
	// than 1.0 per process memory fraction.
	UseUnifiedMemory bool `protobuf:"varint,2,opt,name=use_unified_memory,json=useUnifiedMemory,proto3" json:"use_unified_memory,omitempty"`
	// If > 1, the number of device-to-device copy streams to create
	// for each GPUDevice.  Default value is 0, which is automatically
	// converted to 1.
	NumDevToDevCopyStreams int32 `` /* 136-byte string literal not displayed */
	// If non-empty, defines a good GPU ring order on a single worker based on
	// device interconnect.  This assumes that all workers have the same GPU
	// topology.  Specify as a comma-separated string, e.g. "3,2,1,0,7,6,5,4".
	// This ring order is used by the RingReducer implementation of
	// CollectiveReduce, and serves as an override to automatic ring order
	// generation in OrderTaskDeviceMap() during CollectiveParam resolution.
	CollectiveRingOrder string `protobuf:"bytes,4,opt,name=collective_ring_order,json=collectiveRingOrder,proto3" json:"collective_ring_order,omitempty"`
	// If true then extra work is done by GPUDevice and GPUBFCAllocator to
	// keep track of when GPU memory is freed and when kernels actually
	// complete so that we can know when a nominally free memory chunk
	// is really not subject to pending use.
	TimestampedAllocator bool `protobuf:"varint,5,opt,name=timestamped_allocator,json=timestampedAllocator,proto3" json:"timestamped_allocator,omitempty"`
	// Parameters for GPUKernelTracker.  By default no kernel tracking is done.
	// Note that timestamped_allocator is only effective if some tracking is
	// specified.
	//
	// If kernel_tracker_max_interval = n > 0, then a tracking event
	// is inserted after every n kernels without an event.
	KernelTrackerMaxInterval int32 `` /* 138-byte string literal not displayed */
	// If kernel_tracker_max_bytes = n > 0, then a tracking event is
	// inserted after every series of kernels allocating a sum of
	// memory >= n.  If one kernel allocates b * n bytes, then one
	// event will be inserted after it, but it will count as b against
	// the pending limit.
	KernelTrackerMaxBytes int32 `` /* 129-byte string literal not displayed */
	// If kernel_tracker_max_pending > 0 then no more than this many
	// tracking events can be outstanding at a time.  An attempt to
	// launch an additional kernel will stall until an event
	// completes.
	KernelTrackerMaxPending int32 `` /* 135-byte string literal not displayed */
	// BFC Allocator can return an allocated chunk of memory upto 2x the
	// requested size. For virtual devices with tight memory constraints, and
	// proportionately large allocation requests, this can lead to a significant
	// reduction in available memory. The threshold below controls when a chunk
	// should be split if the chunk size exceeds requested memory size. It is
	// expressed as a fraction of total available memory for the tf device. For
	// example setting it to 0.05 would imply a chunk needs to be split if its
	// size exceeds the requested memory by 5% of the total virtual device/gpu
	// memory size.
	InternalFragmentationFraction float64 `` /* 153-byte string literal not displayed */
	// When true, use CUDA cudaMallocAsync API instead of TF gpu allocator.
	UseCudaMallocAsync bool `protobuf:"varint,11,opt,name=use_cuda_malloc_async,json=useCudaMallocAsync,proto3" json:"use_cuda_malloc_async,omitempty"`
	// By default, BFCAllocator may sleep when it runs out of memory, in the
	// hopes that another thread will free up memory in the meantime.  Setting
	// this to true disables the sleep; instead we'll OOM immediately.
	DisallowRetryOnAllocationFailure bool `` /* 165-byte string literal not displayed */
	// contains filtered or unexported fields
}

func (*GPUOptions_Experimental) Descriptor deprecated 

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

Deprecated: Use GPUOptions_Experimental.ProtoReflect.Descriptor instead.

func (*GPUOptions_Experimental) GetCollectiveRingOrder 

func (x *GPUOptions_Experimental) GetCollectiveRingOrder() string

func (*GPUOptions_Experimental) GetDisallowRetryOnAllocationFailure 

func (x *GPUOptions_Experimental) GetDisallowRetryOnAllocationFailure() bool

func (*GPUOptions_Experimental) GetInternalFragmentationFraction 

func (x *GPUOptions_Experimental) GetInternalFragmentationFraction() float64

func (*GPUOptions_Experimental) GetKernelTrackerMaxBytes 

func (x *GPUOptions_Experimental) GetKernelTrackerMaxBytes() int32

func (*GPUOptions_Experimental) GetKernelTrackerMaxInterval 

func (x *GPUOptions_Experimental) GetKernelTrackerMaxInterval() int32

func (*GPUOptions_Experimental) GetKernelTrackerMaxPending 

func (x *GPUOptions_Experimental) GetKernelTrackerMaxPending() int32

func (*GPUOptions_Experimental) GetNumDevToDevCopyStreams 

func (x *GPUOptions_Experimental) GetNumDevToDevCopyStreams() int32

func (*GPUOptions_Experimental) GetTimestampedAllocator 

func (x *GPUOptions_Experimental) GetTimestampedAllocator() bool

func (*GPUOptions_Experimental) GetUseCudaMallocAsync 

func (x *GPUOptions_Experimental) GetUseCudaMallocAsync() bool

func (*GPUOptions_Experimental) GetUseUnifiedMemory 

func (x *GPUOptions_Experimental) GetUseUnifiedMemory() bool

func (*GPUOptions_Experimental) GetVirtualDevices 

func (x *GPUOptions_Experimental) GetVirtualDevices() []*GPUOptions_Experimental_VirtualDevices

func (*GPUOptions_Experimental) ProtoMessage 

func (*GPUOptions_Experimental) ProtoMessage()

func (*GPUOptions_Experimental) ProtoReflect 

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

func (*GPUOptions_Experimental) Reset 

func (x *GPUOptions_Experimental) Reset()

func (*GPUOptions_Experimental) String 

func (x *GPUOptions_Experimental) String() string

type GPUOptions_Experimental_VirtualDevices 

type GPUOptions_Experimental_VirtualDevices struct {

	// Per "virtual" device memory limit, in MB. The number of elements in
	// the list is the number of virtual devices to create on the
	// corresponding visible GPU (see "virtual_devices" below).
	// If empty, it will create single virtual device taking all available
	// memory from the device.
	//
	// For the concept of "visible" and "virtual" GPU, see the comments for
	// "visible_device_list" above for more information.
	MemoryLimitMb []float32 `protobuf:"fixed32,1,rep,packed,name=memory_limit_mb,json=memoryLimitMb,proto3" json:"memory_limit_mb,omitempty"`
	// Priority values to use with the virtual devices. Use the cuda function
	// cudaDeviceGetStreamPriorityRange to query for valid range of values for
	// priority.
	//
	// On a P4000 GPU with cuda 10.1, the priority range reported was 0 for
	// least priority and -1 for greatest priority.
	//
	// If this field is not specified, then the virtual devices will be
	// created with the default. If this field has values set, then the size
	// of this must match with the above memory_limit_mb.
	Priority []int32 `protobuf:"varint,2,rep,packed,name=priority,proto3" json:"priority,omitempty"`
	// Virtual Device ordinal number determines the device ID of the device.
	// A Virtual device with a lower ordinal number always receives the a
	// smaller device id. The phyiscal device id and location in the
	// virtual device list is used to break ties.
	DeviceOrdinal []int32 `protobuf:"varint,3,rep,packed,name=device_ordinal,json=deviceOrdinal,proto3" json:"device_ordinal,omitempty"`
	// contains filtered or unexported fields
}

Configuration for breaking down a visible GPU into multiple "virtual" devices.

func (*GPUOptions_Experimental_VirtualDevices) Descriptor deprecated 

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

Deprecated: Use GPUOptions_Experimental_VirtualDevices.ProtoReflect.Descriptor instead.

func (*GPUOptions_Experimental_VirtualDevices) GetDeviceOrdinal 

func (x *GPUOptions_Experimental_VirtualDevices) GetDeviceOrdinal() []int32

func (*GPUOptions_Experimental_VirtualDevices) GetMemoryLimitMb 

func (x *GPUOptions_Experimental_VirtualDevices) GetMemoryLimitMb() []float32

func (*GPUOptions_Experimental_VirtualDevices) GetPriority 

func (x *GPUOptions_Experimental_VirtualDevices) GetPriority() []int32

func (*GPUOptions_Experimental_VirtualDevices) ProtoMessage 

func (*GPUOptions_Experimental_VirtualDevices) ProtoMessage()

func (*GPUOptions_Experimental_VirtualDevices) ProtoReflect 

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

func (*GPUOptions_Experimental_VirtualDevices) Reset 

func (x *GPUOptions_Experimental_VirtualDevices) Reset()

func (*GPUOptions_Experimental_VirtualDevices) String 

func (x *GPUOptions_Experimental_VirtualDevices) String() string

type GetKeyValueDirRequest 

type GetKeyValueDirRequest struct {
	DirectoryKey string `protobuf:"bytes,1,opt,name=directory_key,json=directoryKey,proto3" json:"directory_key,omitempty"`
	// contains filtered or unexported fields
}

func (*GetKeyValueDirRequest) Descriptor deprecated 

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

Deprecated: Use GetKeyValueDirRequest.ProtoReflect.Descriptor instead.

func (*GetKeyValueDirRequest) GetDirectoryKey 

func (x *GetKeyValueDirRequest) GetDirectoryKey() string

func (*GetKeyValueDirRequest) ProtoMessage 

func (*GetKeyValueDirRequest) ProtoMessage()

func (*GetKeyValueDirRequest) ProtoReflect 

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

func (*GetKeyValueDirRequest) Reset 

func (x *GetKeyValueDirRequest) Reset()

func (*GetKeyValueDirRequest) String 

func (x *GetKeyValueDirRequest) String() string

type GetKeyValueDirResponse 

type GetKeyValueDirResponse struct {
	DirectoryKey string           `protobuf:"bytes,1,opt,name=directory_key,json=directoryKey,proto3" json:"directory_key,omitempty"`
	Kv           []*KeyValueEntry `protobuf:"bytes,2,rep,name=kv,proto3" json:"kv,omitempty"`
	// contains filtered or unexported fields
}

func (*GetKeyValueDirResponse) Descriptor deprecated 

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

Deprecated: Use GetKeyValueDirResponse.ProtoReflect.Descriptor instead.

func (*GetKeyValueDirResponse) GetDirectoryKey 

func (x *GetKeyValueDirResponse) GetDirectoryKey() string

func (*GetKeyValueDirResponse) GetKv 

func (x *GetKeyValueDirResponse) GetKv() []*KeyValueEntry

func (*GetKeyValueDirResponse) ProtoMessage 

func (*GetKeyValueDirResponse) ProtoMessage()

func (*GetKeyValueDirResponse) ProtoReflect 

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

func (*GetKeyValueDirResponse) Reset 

func (x *GetKeyValueDirResponse) Reset()

func (*GetKeyValueDirResponse) String 

func (x *GetKeyValueDirResponse) String() string

type GetKeyValueRequest 

type GetKeyValueRequest struct {
	Key string `protobuf:"bytes,1,opt,name=key,proto3" json:"key,omitempty"`
	// contains filtered or unexported fields
}

Request and response messages for getting configuration key-value data.

func (*GetKeyValueRequest) Descriptor deprecated 

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

Deprecated: Use GetKeyValueRequest.ProtoReflect.Descriptor instead.

func (*GetKeyValueRequest) GetKey 

func (x *GetKeyValueRequest) GetKey() string

func (*GetKeyValueRequest) ProtoMessage 

func (*GetKeyValueRequest) ProtoMessage()

func (*GetKeyValueRequest) ProtoReflect 

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

func (*GetKeyValueRequest) Reset 

func (x *GetKeyValueRequest) Reset()

func (*GetKeyValueRequest) String 

func (x *GetKeyValueRequest) String() string

type GetKeyValueResponse 

type GetKeyValueResponse struct {
	Kv *KeyValueEntry `protobuf:"bytes,1,opt,name=kv,proto3" json:"kv,omitempty"`
	// contains filtered or unexported fields
}

func (*GetKeyValueResponse) Descriptor deprecated 

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

Deprecated: Use GetKeyValueResponse.ProtoReflect.Descriptor instead.

func (*GetKeyValueResponse) GetKv 

func (x *GetKeyValueResponse) GetKv() *KeyValueEntry

func (*GetKeyValueResponse) ProtoMessage 

func (*GetKeyValueResponse) ProtoMessage()

func (*GetKeyValueResponse) ProtoReflect 

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

func (*GetKeyValueResponse) Reset 

func (x *GetKeyValueResponse) Reset()

func (*GetKeyValueResponse) String 

func (x *GetKeyValueResponse) String() string

type GetStatusRequest 

type GetStatusRequest struct {
	// contains filtered or unexported fields
}

func (*GetStatusRequest) Descriptor deprecated 

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

Deprecated: Use GetStatusRequest.ProtoReflect.Descriptor instead.

func (*GetStatusRequest) ProtoMessage 

func (*GetStatusRequest) ProtoMessage()

func (*GetStatusRequest) ProtoReflect 

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

func (*GetStatusRequest) Reset 

func (x *GetStatusRequest) Reset()

func (*GetStatusRequest) String 

func (x *GetStatusRequest) String() string

type GetStatusResponse 

type GetStatusResponse struct {
	DeviceAttributes []*DeviceAttributes `protobuf:"bytes,1,rep,name=device_attributes,json=deviceAttributes,proto3" json:"device_attributes,omitempty"`
	// contains filtered or unexported fields
}

func (*GetStatusResponse) Descriptor deprecated 

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

Deprecated: Use GetStatusResponse.ProtoReflect.Descriptor instead.

func (*GetStatusResponse) GetDeviceAttributes 

func (x *GetStatusResponse) GetDeviceAttributes() []*DeviceAttributes

func (*GetStatusResponse) ProtoMessage 

func (*GetStatusResponse) ProtoMessage()

func (*GetStatusResponse) ProtoReflect 

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

func (*GetStatusResponse) Reset 

func (x *GetStatusResponse) Reset()

func (*GetStatusResponse) String 

func (x *GetStatusResponse) String() string

type GetStepSequenceRequest 

type GetStepSequenceRequest struct {
	GraphKey []int64 `protobuf:"varint,1,rep,packed,name=graph_key,json=graphKey,proto3" json:"graph_key,omitempty"`
	// contains filtered or unexported fields
}

Request for next agreed-upon step_id for the specified graph_keys. This is used to enable multiple graphs containing nodes from a common collective instance to coordinate using the same step_ids.

func (*GetStepSequenceRequest) Descriptor deprecated 

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

Deprecated: Use GetStepSequenceRequest.ProtoReflect.Descriptor instead.

func (*GetStepSequenceRequest) GetGraphKey 

func (x *GetStepSequenceRequest) GetGraphKey() []int64

func (*GetStepSequenceRequest) ProtoMessage 

func (*GetStepSequenceRequest) ProtoMessage()

func (*GetStepSequenceRequest) ProtoReflect 

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

func (*GetStepSequenceRequest) Reset 

func (x *GetStepSequenceRequest) Reset()

func (*GetStepSequenceRequest) String 

func (x *GetStepSequenceRequest) String() string

type GetStepSequenceResponse 

type GetStepSequenceResponse struct {
	StepSequence []*StepSequence `protobuf:"bytes,1,rep,name=step_sequence,json=stepSequence,proto3" json:"step_sequence,omitempty"`
	// contains filtered or unexported fields
}

Next valid step_ids for one or more graph_keys.

func (*GetStepSequenceResponse) Descriptor deprecated 

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

Deprecated: Use GetStepSequenceResponse.ProtoReflect.Descriptor instead.

func (*GetStepSequenceResponse) GetStepSequence 

func (x *GetStepSequenceResponse) GetStepSequence() []*StepSequence

func (*GetStepSequenceResponse) ProtoMessage 

func (*GetStepSequenceResponse) ProtoMessage()

func (*GetStepSequenceResponse) ProtoReflect 

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

func (*GetStepSequenceResponse) Reset 

func (x *GetStepSequenceResponse) Reset()

func (*GetStepSequenceResponse) String 

func (x *GetStepSequenceResponse) String() string

type GlobalTopologyProto 

type GlobalTopologyProto struct {
	Nodes []*LocalTopologyProto `protobuf:"bytes,1,rep,name=nodes,proto3" json:"nodes,omitempty"`
	// contains filtered or unexported fields
}

func (*GlobalTopologyProto) Descriptor deprecated 

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

Deprecated: Use GlobalTopologyProto.ProtoReflect.Descriptor instead.

func (*GlobalTopologyProto) GetNodes 

func (x *GlobalTopologyProto) GetNodes() []*LocalTopologyProto

func (*GlobalTopologyProto) ProtoMessage 

func (*GlobalTopologyProto) ProtoMessage()

func (*GlobalTopologyProto) ProtoReflect 

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

func (*GlobalTopologyProto) Reset 

func (x *GlobalTopologyProto) Reset()

func (*GlobalTopologyProto) String 

func (x *GlobalTopologyProto) 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 GraphDebugInfo 

type GraphDebugInfo struct {

	// This stores all the source code file names and can be indexed by the
	// `file_index`.
	Files []string `protobuf:"bytes,1,rep,name=files,proto3" json:"files,omitempty"`
	// This maps a node name to a stack trace in the source code.
	// The map key is a mangling of the containing function and op name with
	// syntax:
	//   op.name '@' func_name
	// For ops in the top-level graph, the func_name is the empty string.
	// Note that op names are restricted to a small number of characters which
	// exclude '@', making it impossible to collide keys of this form. Function
	// names accept a much wider set of characters.
	// It would be preferable to avoid mangling and use a tuple key of (op.name,
	// func_name), but this is not supported with protocol buffers.
	Traces map[string]*GraphDebugInfo_StackTrace `` /* 153-byte string literal not displayed */
	// contains filtered or unexported fields
}

func (*GraphDebugInfo) Descriptor deprecated 

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

Deprecated: Use GraphDebugInfo.ProtoReflect.Descriptor instead.

func (*GraphDebugInfo) GetFiles 

func (x *GraphDebugInfo) GetFiles() []string

func (*GraphDebugInfo) GetTraces 

func (x *GraphDebugInfo) GetTraces() map[string]*GraphDebugInfo_StackTrace

func (*GraphDebugInfo) ProtoMessage 

func (*GraphDebugInfo) ProtoMessage()

func (*GraphDebugInfo) ProtoReflect 

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

func (*GraphDebugInfo) Reset 

func (x *GraphDebugInfo) Reset()

func (*GraphDebugInfo) String 

func (x *GraphDebugInfo) String() string

type GraphDebugInfo_FileLineCol 

type GraphDebugInfo_FileLineCol struct {

	// File name index, which can be used to retrieve the file name string from
	// `files`. The value should be between 0 and (len(files)-1)
	FileIndex int32 `protobuf:"varint,1,opt,name=file_index,json=fileIndex,proto3" json:"file_index,omitempty"`
	// Line number in the file.
	Line int32 `protobuf:"varint,2,opt,name=line,proto3" json:"line,omitempty"`
	// Col number in the file line.
	Col int32 `protobuf:"varint,3,opt,name=col,proto3" json:"col,omitempty"`
	// Name of function contains the file line.
	Func string `protobuf:"bytes,4,opt,name=func,proto3" json:"func,omitempty"`
	// Source code contained in this file line.
	Code string `protobuf:"bytes,5,opt,name=code,proto3" json:"code,omitempty"`
	// contains filtered or unexported fields
}

This represents a file/line location in the source code.

func (*GraphDebugInfo_FileLineCol) Descriptor deprecated 

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

Deprecated: Use GraphDebugInfo_FileLineCol.ProtoReflect.Descriptor instead.

func (*GraphDebugInfo_FileLineCol) GetCode 

func (x *GraphDebugInfo_FileLineCol) GetCode() string

func (*GraphDebugInfo_FileLineCol) GetCol 

func (x *GraphDebugInfo_FileLineCol) GetCol() int32

func (*GraphDebugInfo_FileLineCol) GetFileIndex 

func (x *GraphDebugInfo_FileLineCol) GetFileIndex() int32

func (*GraphDebugInfo_FileLineCol) GetFunc 

func (x *GraphDebugInfo_FileLineCol) GetFunc() string

func (*GraphDebugInfo_FileLineCol) GetLine 

func (x *GraphDebugInfo_FileLineCol) GetLine() int32

func (*GraphDebugInfo_FileLineCol) ProtoMessage 

func (*GraphDebugInfo_FileLineCol) ProtoMessage()

func (*GraphDebugInfo_FileLineCol) ProtoReflect 

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

func (*GraphDebugInfo_FileLineCol) Reset 

func (x *GraphDebugInfo_FileLineCol) Reset()

func (*GraphDebugInfo_FileLineCol) String 

func (x *GraphDebugInfo_FileLineCol) String() string

type GraphDebugInfo_StackTrace 

type GraphDebugInfo_StackTrace struct {

	// Each line in the stack trace.
	FileLineCols []*GraphDebugInfo_FileLineCol `protobuf:"bytes,1,rep,name=file_line_cols,json=fileLineCols,proto3" json:"file_line_cols,omitempty"`
	// contains filtered or unexported fields
}

This represents a stack trace which is a ordered list of `FileLineCol`.

func (*GraphDebugInfo_StackTrace) Descriptor deprecated 

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

Deprecated: Use GraphDebugInfo_StackTrace.ProtoReflect.Descriptor instead.

func (*GraphDebugInfo_StackTrace) GetFileLineCols 

func (x *GraphDebugInfo_StackTrace) GetFileLineCols() []*GraphDebugInfo_FileLineCol

func (*GraphDebugInfo_StackTrace) ProtoMessage 

func (*GraphDebugInfo_StackTrace) ProtoMessage()

func (*GraphDebugInfo_StackTrace) ProtoReflect 

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

func (*GraphDebugInfo_StackTrace) Reset 

func (x *GraphDebugInfo_StackTrace) Reset()

func (*GraphDebugInfo_StackTrace) String 

func (x *GraphDebugInfo_StackTrace) String() string

type GraphDef 

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

Represents the graph of operations

func (*GraphDef) Descriptor deprecated 

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

Deprecated: Use GraphDef.ProtoReflect.Descriptor instead.

func (*GraphDef) GetLibrary 

func (x *GraphDef) GetLibrary() *FunctionDefLibrary

func (*GraphDef) GetNode 

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

func (*GraphDef) GetVersion deprecated 

func (x *GraphDef) GetVersion() int32

Deprecated: Do not use.

func (*GraphDef) GetVersions 

func (x *GraphDef) GetVersions() *VersionDef

func (*GraphDef) ProtoMessage 

func (*GraphDef) ProtoMessage()

func (*GraphDef) ProtoReflect 

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

func (*GraphDef) Reset 

func (x *GraphDef) Reset()

func (*GraphDef) String 

func (x *GraphDef) String() string

type GraphExecutionTrace 

type GraphExecutionTrace struct {

	// Unique ID of the context that the executed op(s) belong to (e.g., a
	// compiled concrete tf.function).
	TfdbgContextId string `protobuf:"bytes,1,opt,name=tfdbg_context_id,json=tfdbgContextId,proto3" json:"tfdbg_context_id,omitempty"`
	// Name of the op (applicable only in the case of the `FULL_TENSOR` trace
	// level).
	OpName string `protobuf:"bytes,2,opt,name=op_name,json=opName,proto3" json:"op_name,omitempty"`
	// Output slot of the tensor (applicable only in the case of the `FULL_TENSOR`
	// trace level).
	OutputSlot int32 `protobuf:"varint,3,opt,name=output_slot,json=outputSlot,proto3" json:"output_slot,omitempty"`
	// Type of the tensor value encapsulated in this proto.
	TensorDebugMode TensorDebugMode `` /* 141-byte string literal not displayed */
	// Tensor value in the type described by `tensor_value_type`.
	// This tensor may summarize the value of a single intermediate op of the
	// graph, or those of multiple intermediate tensors.
	TensorProto *TensorProto `protobuf:"bytes,5,opt,name=tensor_proto,json=tensorProto,proto3" json:"tensor_proto,omitempty"`
	// Name of the device that the op belongs to.
	DeviceName string `protobuf:"bytes,6,opt,name=device_name,json=deviceName,proto3" json:"device_name,omitempty"`
	// contains filtered or unexported fields
}

Data relating to an execution of a Graph (e.g., an eager execution of a FuncGraph). The values of the intermediate tensors computed in the graph are recorded in this proto. A graph execution may correspond to one or more pieces of `GraphExecutionTrace`, depending on whether the instrumented tensor values are summarized in an aggregated or separate fashion.

func (*GraphExecutionTrace) Descriptor deprecated 

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

Deprecated: Use GraphExecutionTrace.ProtoReflect.Descriptor instead.

func (*GraphExecutionTrace) GetDeviceName 

func (x *GraphExecutionTrace) GetDeviceName() string

func (*GraphExecutionTrace) GetOpName 

func (x *GraphExecutionTrace) GetOpName() string

func (*GraphExecutionTrace) GetOutputSlot 

func (x *GraphExecutionTrace) GetOutputSlot() int32

func (*GraphExecutionTrace) GetTensorDebugMode 

func (x *GraphExecutionTrace) GetTensorDebugMode() TensorDebugMode

func (*GraphExecutionTrace) GetTensorProto 

func (x *GraphExecutionTrace) GetTensorProto() *TensorProto

func (*GraphExecutionTrace) GetTfdbgContextId 

func (x *GraphExecutionTrace) GetTfdbgContextId() string

func (*GraphExecutionTrace) ProtoMessage 

func (*GraphExecutionTrace) ProtoMessage()

func (*GraphExecutionTrace) ProtoReflect 

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

func (*GraphExecutionTrace) Reset 

func (x *GraphExecutionTrace) Reset()

func (*GraphExecutionTrace) String 

func (x *GraphExecutionTrace) String() string

type GraphOpCreation 

type GraphOpCreation struct {

	// Type of the op (e.g., "MatMul").
	OpType string `protobuf:"bytes,1,opt,name=op_type,json=opType,proto3" json:"op_type,omitempty"`
	// Name of the op (e.g., "Dense/MatMul_1").
	OpName string `protobuf:"bytes,2,opt,name=op_name,json=opName,proto3" json:"op_name,omitempty"`
	// Name of the graph that the op is a part of (if available).
	GraphName string `protobuf:"bytes,3,opt,name=graph_name,json=graphName,proto3" json:"graph_name,omitempty"`
	// Unique ID of the graph (generated by debugger).
	// This is the ID of the immediately-enclosing graph.
	GraphId string `protobuf:"bytes,4,opt,name=graph_id,json=graphId,proto3" json:"graph_id,omitempty"`
	// Name of the device that the op is assigned to (if available).
	DeviceName string `protobuf:"bytes,5,opt,name=device_name,json=deviceName,proto3" json:"device_name,omitempty"`
	// Names of the input tensors to the op.
	InputNames []string `protobuf:"bytes,6,rep,name=input_names,json=inputNames,proto3" json:"input_names,omitempty"`
	// Number of output tensors emitted by the op.
	NumOutputs int32 `protobuf:"varint,7,opt,name=num_outputs,json=numOutputs,proto3" json:"num_outputs,omitempty"`
	// The unique ID for code location (stack trace) of the op's creation.
	CodeLocation *CodeLocation `protobuf:"bytes,8,opt,name=code_location,json=codeLocation,proto3" json:"code_location,omitempty"`
	// Unique IDs for the output tensors of this op.
	OutputTensorIds []int32 `protobuf:"varint,9,rep,packed,name=output_tensor_ids,json=outputTensorIds,proto3" json:"output_tensor_ids,omitempty"`
	// contains filtered or unexported fields
}

The creation of an op in a TensorFlow Graph (e.g., FuncGraph in TF2).

func (*GraphOpCreation) Descriptor deprecated 

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

Deprecated: Use GraphOpCreation.ProtoReflect.Descriptor instead.

func (*GraphOpCreation) GetCodeLocation 

func (x *GraphOpCreation) GetCodeLocation() *CodeLocation

func (*GraphOpCreation) GetDeviceName 

func (x *GraphOpCreation) GetDeviceName() string

func (*GraphOpCreation) GetGraphId 

func (x *GraphOpCreation) GetGraphId() string

func (*GraphOpCreation) GetGraphName 

func (x *GraphOpCreation) GetGraphName() string

func (*GraphOpCreation) GetInputNames 

func (x *GraphOpCreation) GetInputNames() []string

func (*GraphOpCreation) GetNumOutputs 

func (x *GraphOpCreation) GetNumOutputs() int32

func (*GraphOpCreation) GetOpName 

func (x *GraphOpCreation) GetOpName() string

func (*GraphOpCreation) GetOpType 

func (x *GraphOpCreation) GetOpType() string

func (*GraphOpCreation) GetOutputTensorIds 

func (x *GraphOpCreation) GetOutputTensorIds() []int32

func (*GraphOpCreation) ProtoMessage 

func (*GraphOpCreation) ProtoMessage()

func (*GraphOpCreation) ProtoReflect 

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

func (*GraphOpCreation) Reset 

func (x *GraphOpCreation) Reset()

func (*GraphOpCreation) String 

func (x *GraphOpCreation) String() string

type GraphOptions 

type GraphOptions struct {

	// If true, use control flow to schedule the activation of Recv nodes.
	// (Currently ignored.)
	EnableRecvScheduling bool `protobuf:"varint,2,opt,name=enable_recv_scheduling,json=enableRecvScheduling,proto3" json:"enable_recv_scheduling,omitempty"`
	// Options controlling how graph is optimized.
	OptimizerOptions *OptimizerOptions `protobuf:"bytes,3,opt,name=optimizer_options,json=optimizerOptions,proto3" json:"optimizer_options,omitempty"`
	// The number of steps to run before returning a cost model detailing
	// the memory usage and performance of each node of the graph. 0 means
	// no cost model.
	BuildCostModel int64 `protobuf:"varint,4,opt,name=build_cost_model,json=buildCostModel,proto3" json:"build_cost_model,omitempty"`
	// The number of steps to skip before collecting statistics for the
	// cost model.
	BuildCostModelAfter int64 `protobuf:"varint,9,opt,name=build_cost_model_after,json=buildCostModelAfter,proto3" json:"build_cost_model_after,omitempty"`
	// Annotate each Node with Op output shape data, to the extent it can
	// be statically inferred.
	InferShapes bool `protobuf:"varint,5,opt,name=infer_shapes,json=inferShapes,proto3" json:"infer_shapes,omitempty"`
	// Only place the subgraphs that are run, rather than the entire graph.
	//
	// This is useful for interactive graph building, where one might
	// produce graphs that cannot be placed during the debugging
	// process.  In particular, it allows the client to continue work in
	// a session after adding a node to a graph whose placement
	// constraints are unsatisfiable.
	PlacePrunedGraph bool `protobuf:"varint,6,opt,name=place_pruned_graph,json=placePrunedGraph,proto3" json:"place_pruned_graph,omitempty"`
	// If true, transfer float values between processes as bfloat16.
	EnableBfloat16Sendrecv bool `` /* 130-byte string literal not displayed */
	// If > 0, record a timeline every this many steps.
	// EXPERIMENTAL: This currently has no effect in MasterSession.
	TimelineStep int32 `protobuf:"varint,8,opt,name=timeline_step,json=timelineStep,proto3" json:"timeline_step,omitempty"`
	// Options that control the type and amount of graph rewriting.
	// Not currently configurable via the public Python API (i.e. there is no API
	// stability guarantee if you import RewriterConfig explicitly).
	RewriteOptions *RewriterConfig `protobuf:"bytes,10,opt,name=rewrite_options,json=rewriteOptions,proto3" json:"rewrite_options,omitempty"`
	// contains filtered or unexported fields
}

func (*GraphOptions) Descriptor deprecated 

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

Deprecated: Use GraphOptions.ProtoReflect.Descriptor instead.

func (*GraphOptions) GetBuildCostModel 

func (x *GraphOptions) GetBuildCostModel() int64

func (*GraphOptions) GetBuildCostModelAfter 

func (x *GraphOptions) GetBuildCostModelAfter() int64

func (*GraphOptions) GetEnableBfloat16Sendrecv 

func (x *GraphOptions) GetEnableBfloat16Sendrecv() bool

func (*GraphOptions) GetEnableRecvScheduling 

func (x *GraphOptions) GetEnableRecvScheduling() bool

func (*GraphOptions) GetInferShapes 

func (x *GraphOptions) GetInferShapes() bool

func (*GraphOptions) GetOptimizerOptions 

func (x *GraphOptions) GetOptimizerOptions() *OptimizerOptions

func (*GraphOptions) GetPlacePrunedGraph 

func (x *GraphOptions) GetPlacePrunedGraph() bool

func (*GraphOptions) GetRewriteOptions 

func (x *GraphOptions) GetRewriteOptions() *RewriterConfig

func (*GraphOptions) GetTimelineStep 

func (x *GraphOptions) GetTimelineStep() int32

func (*GraphOptions) ProtoMessage 

func (*GraphOptions) ProtoMessage()

func (*GraphOptions) ProtoReflect 

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

func (*GraphOptions) Reset 

func (x *GraphOptions) Reset()

func (*GraphOptions) String 

func (x *GraphOptions) 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 GrpcPayloadContainer 

type GrpcPayloadContainer struct {
	Payloads map[string][]byte `` /* 157-byte string literal not displayed */
	// contains filtered or unexported fields
}

Used to serialize and transmit tensorflow::Status payloads through grpc::Status `error_details` since grpc::Status lacks payload API. TODO(b/204231601): Use GRPC API once supported.

func (*GrpcPayloadContainer) Descriptor deprecated 

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

Deprecated: Use GrpcPayloadContainer.ProtoReflect.Descriptor instead.

func (*GrpcPayloadContainer) GetPayloads 

func (x *GrpcPayloadContainer) GetPayloads() map[string][]byte

func (*GrpcPayloadContainer) ProtoMessage 

func (*GrpcPayloadContainer) ProtoMessage()

func (*GrpcPayloadContainer) ProtoReflect 

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

func (*GrpcPayloadContainer) Reset 

func (x *GrpcPayloadContainer) Reset()

func (*GrpcPayloadContainer) String 

func (x *GrpcPayloadContainer) String() string

type GrpcPayloadsLost 

type GrpcPayloadsLost struct {
	// contains filtered or unexported fields
}

If included as a payload, this message flags the Status to have lost payloads during the GRPC transmission. URI: "type.googleapis.com/tensorflow.distributed_runtime.GrpcPayloadsLost"

func (*GrpcPayloadsLost) Descriptor deprecated 

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

Deprecated: Use GrpcPayloadsLost.ProtoReflect.Descriptor instead.

func (*GrpcPayloadsLost) ProtoMessage 

func (*GrpcPayloadsLost) ProtoMessage()

func (*GrpcPayloadsLost) ProtoReflect 

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

func (*GrpcPayloadsLost) Reset 

func (x *GrpcPayloadsLost) Reset()

func (*GrpcPayloadsLost) String 

func (x *GrpcPayloadsLost) String() string

type HeartbeatRequest 

type HeartbeatRequest struct {
	SessionId uint64 `protobuf:"varint,1,opt,name=session_id,json=sessionId,proto3" json:"session_id,omitempty"`
	NodeId    int32  `protobuf:"varint,2,opt,name=node_id,json=nodeId,proto3" json:"node_id,omitempty"`
	// contains filtered or unexported fields
}

func (*HeartbeatRequest) Descriptor deprecated 

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

Deprecated: Use HeartbeatRequest.ProtoReflect.Descriptor instead.

func (*HeartbeatRequest) GetNodeId 

func (x *HeartbeatRequest) GetNodeId() int32

func (*HeartbeatRequest) GetSessionId 

func (x *HeartbeatRequest) GetSessionId() uint64

func (*HeartbeatRequest) ProtoMessage 

func (*HeartbeatRequest) ProtoMessage()

func (*HeartbeatRequest) ProtoReflect 

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

func (*HeartbeatRequest) Reset 

func (x *HeartbeatRequest) Reset()

func (*HeartbeatRequest) String 

func (x *HeartbeatRequest) String() string

type HeartbeatResponse 

type HeartbeatResponse struct {
	// contains filtered or unexported fields
}

func (*HeartbeatResponse) Descriptor deprecated 

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

Deprecated: Use HeartbeatResponse.ProtoReflect.Descriptor instead.

func (*HeartbeatResponse) ProtoMessage 

func (*HeartbeatResponse) ProtoMessage()

func (*HeartbeatResponse) ProtoReflect 

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

func (*HeartbeatResponse) Reset 

func (x *HeartbeatResponse) Reset()

func (*HeartbeatResponse) String 

func (x *HeartbeatResponse) String() string

type HistogramProto 

type HistogramProto struct {
	Min        float64 `protobuf:"fixed64,1,opt,name=min,proto3" json:"min,omitempty"`
	Max        float64 `protobuf:"fixed64,2,opt,name=max,proto3" json:"max,omitempty"`
	Num        float64 `protobuf:"fixed64,3,opt,name=num,proto3" json:"num,omitempty"`
	Sum        float64 `protobuf:"fixed64,4,opt,name=sum,proto3" json:"sum,omitempty"`
	SumSquares float64 `protobuf:"fixed64,5,opt,name=sum_squares,json=sumSquares,proto3" json:"sum_squares,omitempty"`
	// Parallel arrays encoding the bucket boundaries and the bucket values.
	// bucket(i) is the count for the bucket i.  The range for
	// a bucket is:
	//   i == 0:  -DBL_MAX .. bucket_limit(0)
	//   i != 0:  bucket_limit(i-1) .. bucket_limit(i)
	BucketLimit []float64 `protobuf:"fixed64,6,rep,packed,name=bucket_limit,json=bucketLimit,proto3" json:"bucket_limit,omitempty"`
	Bucket      []float64 `protobuf:"fixed64,7,rep,packed,name=bucket,proto3" json:"bucket,omitempty"`
	// contains filtered or unexported fields
}

Serialization format for histogram module in core/lib/histogram/histogram.h

func (*HistogramProto) Descriptor deprecated 

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

Deprecated: Use HistogramProto.ProtoReflect.Descriptor instead.

func (*HistogramProto) GetBucket 

func (x *HistogramProto) GetBucket() []float64

func (*HistogramProto) GetBucketLimit 

func (x *HistogramProto) GetBucketLimit() []float64

func (*HistogramProto) GetMax 

func (x *HistogramProto) GetMax() float64

func (*HistogramProto) GetMin 

func (x *HistogramProto) GetMin() float64

func (*HistogramProto) GetNum 

func (x *HistogramProto) GetNum() float64

func (*HistogramProto) GetSum 

func (x *HistogramProto) GetSum() float64

func (*HistogramProto) GetSumSquares 

func (x *HistogramProto) GetSumSquares() float64

func (*HistogramProto) ProtoMessage 

func (*HistogramProto) ProtoMessage()

func (*HistogramProto) ProtoReflect 

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

func (*HistogramProto) Reset 

func (x *HistogramProto) Reset()

func (*HistogramProto) String 

func (x *HistogramProto) String() string

type InsertKeyValueRequest 

type InsertKeyValueRequest struct {
	Kv *KeyValueEntry `protobuf:"bytes,1,opt,name=kv,proto3" json:"kv,omitempty"`
	// contains filtered or unexported fields
}

Request and response messages for inserting configuration key-value data.

func (*InsertKeyValueRequest) Descriptor deprecated 

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

Deprecated: Use InsertKeyValueRequest.ProtoReflect.Descriptor instead.

func (*InsertKeyValueRequest) GetKv 

func (x *InsertKeyValueRequest) GetKv() *KeyValueEntry

func (*InsertKeyValueRequest) ProtoMessage 

func (*InsertKeyValueRequest) ProtoMessage()

func (*InsertKeyValueRequest) ProtoReflect 

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

func (*InsertKeyValueRequest) Reset 

func (x *InsertKeyValueRequest) Reset()

func (*InsertKeyValueRequest) String 

func (x *InsertKeyValueRequest) String() string

type InsertKeyValueResponse 

type InsertKeyValueResponse struct {
	// contains filtered or unexported fields
}

func (*InsertKeyValueResponse) Descriptor deprecated 

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

Deprecated: Use InsertKeyValueResponse.ProtoReflect.Descriptor instead.

func (*InsertKeyValueResponse) ProtoMessage 

func (*InsertKeyValueResponse) ProtoMessage()

func (*InsertKeyValueResponse) ProtoReflect 

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

func (*InsertKeyValueResponse) Reset 

func (x *InsertKeyValueResponse) Reset()

func (*InsertKeyValueResponse) String 

func (x *InsertKeyValueResponse) String() string

type Int64List 

type Int64List struct {
	Value []int64 `protobuf:"varint,1,rep,packed,name=value,proto3" json:"value,omitempty"`
	// contains filtered or unexported fields
}

func (*Int64List) Descriptor deprecated 

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

Deprecated: Use Int64List.ProtoReflect.Descriptor instead.

func (*Int64List) GetValue 

func (x *Int64List) GetValue() []int64

func (*Int64List) ProtoMessage 

func (*Int64List) ProtoMessage()

func (*Int64List) ProtoReflect 

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

func (*Int64List) Reset 

func (x *Int64List) Reset()

func (*Int64List) String 

func (x *Int64List) String() string
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 JobDef 

type JobDef struct {

	// The name of this job.
	Name string `protobuf:"bytes,1,opt,name=name,proto3" json:"name,omitempty"`
	// Mapping from task ID to "hostname:port" string.
	//
	// If the `name` field contains "worker", and the `tasks` map contains a
	// mapping from 7 to "example.org:2222", then the device prefix
	// "/job:worker/task:7" will be assigned to "example.org:2222".
	Tasks map[int32]string `` /* 152-byte string literal not displayed */
	// contains filtered or unexported fields
}

Defines a single job in a TensorFlow cluster.

func (*JobDef) Descriptor deprecated 

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

Deprecated: Use JobDef.ProtoReflect.Descriptor instead.

func (*JobDef) GetName 

func (x *JobDef) GetName() string

func (*JobDef) GetTasks 

func (x *JobDef) GetTasks() map[int32]string

func (*JobDef) ProtoMessage 

func (*JobDef) ProtoMessage()

func (*JobDef) ProtoReflect 

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

func (*JobDef) Reset 

func (x *JobDef) Reset()

func (*JobDef) String 

func (x *JobDef) String() string

type JobDeviceFilters 

type JobDeviceFilters struct {

	// The name of this job.
	Name string `protobuf:"bytes,1,opt,name=name,proto3" json:"name,omitempty"`
	// Mapping from task ID to task device filters.
	Tasks map[int32]*TaskDeviceFilters `` /* 152-byte string literal not displayed */
	// contains filtered or unexported fields
}

Defines the device filters for tasks in a job.

func (*JobDeviceFilters) Descriptor deprecated 

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

Deprecated: Use JobDeviceFilters.ProtoReflect.Descriptor instead.

func (*JobDeviceFilters) GetName 

func (x *JobDeviceFilters) GetName() string

func (*JobDeviceFilters) GetTasks 

func (x *JobDeviceFilters) GetTasks() map[int32]*TaskDeviceFilters

func (*JobDeviceFilters) ProtoMessage 

func (*JobDeviceFilters) ProtoMessage()

func (*JobDeviceFilters) ProtoReflect 

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

func (*JobDeviceFilters) Reset 

func (x *JobDeviceFilters) Reset()

func (*JobDeviceFilters) String 

func (x *JobDeviceFilters) String() string

type KeepAliveRequest 

type KeepAliveRequest struct {
	ContextId uint64 `protobuf:"fixed64,1,opt,name=context_id,json=contextId,proto3" json:"context_id,omitempty"`
	// contains filtered or unexported fields
}

func (*KeepAliveRequest) Descriptor deprecated 

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

Deprecated: Use KeepAliveRequest.ProtoReflect.Descriptor instead.

func (*KeepAliveRequest) GetContextId 

func (x *KeepAliveRequest) GetContextId() uint64

func (*KeepAliveRequest) ProtoMessage 

func (*KeepAliveRequest) ProtoMessage()

func (*KeepAliveRequest) ProtoReflect 

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

func (*KeepAliveRequest) Reset 

func (x *KeepAliveRequest) Reset()

func (*KeepAliveRequest) String 

func (x *KeepAliveRequest) String() string

type KeepAliveResponse 

type KeepAliveResponse struct {

	// If the requested context_id is on the remote host, set the context view ID.
	ContextViewId uint64 `protobuf:"fixed64,1,opt,name=context_view_id,json=contextViewId,proto3" json:"context_view_id,omitempty"`
	// contains filtered or unexported fields
}

func (*KeepAliveResponse) Descriptor deprecated 

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

Deprecated: Use KeepAliveResponse.ProtoReflect.Descriptor instead.

func (*KeepAliveResponse) GetContextViewId 

func (x *KeepAliveResponse) GetContextViewId() uint64

func (*KeepAliveResponse) ProtoMessage 

func (*KeepAliveResponse) ProtoMessage()

func (*KeepAliveResponse) ProtoReflect 

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

func (*KeepAliveResponse) Reset 

func (x *KeepAliveResponse) Reset()

func (*KeepAliveResponse) String 

func (x *KeepAliveResponse) 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 KeyValueEntry 

type KeyValueEntry struct {
	Key   string `protobuf:"bytes,1,opt,name=key,proto3" json:"key,omitempty"`
	Value []byte `protobuf:"bytes,2,opt,name=value,proto3" json:"value,omitempty"`
	// contains filtered or unexported fields
}

Message for configuration key value. Key is structured like Unix file system, with multiple levels of directory names separated by the slash ('/') characters.

func (*KeyValueEntry) Descriptor deprecated 

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

Deprecated: Use KeyValueEntry.ProtoReflect.Descriptor instead.

func (*KeyValueEntry) GetKey 

func (x *KeyValueEntry) GetKey() string

func (*KeyValueEntry) GetValue 

func (x *KeyValueEntry) GetValue() []byte

func (*KeyValueEntry) ProtoMessage 

func (*KeyValueEntry) ProtoMessage()

func (*KeyValueEntry) ProtoReflect 

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

func (*KeyValueEntry) Reset 

func (x *KeyValueEntry) Reset()

func (*KeyValueEntry) String 

func (x *KeyValueEntry) String() string

type KeyValueGetRequest 

type KeyValueGetRequest struct {
	SessionId           uint64 `protobuf:"varint,1,opt,name=session_id,json=sessionId,proto3" json:"session_id,omitempty"`
	Key                 []byte `protobuf:"bytes,2,opt,name=key,proto3" json:"key,omitempty"`
	TimeoutMilliseconds int32  `protobuf:"varint,3,opt,name=timeout_milliseconds,json=timeoutMilliseconds,proto3" json:"timeout_milliseconds,omitempty"`
	// contains filtered or unexported fields
}

func (*KeyValueGetRequest) Descriptor deprecated 

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

Deprecated: Use KeyValueGetRequest.ProtoReflect.Descriptor instead.

func (*KeyValueGetRequest) GetKey 

func (x *KeyValueGetRequest) GetKey() []byte

func (*KeyValueGetRequest) GetSessionId 

func (x *KeyValueGetRequest) GetSessionId() uint64

func (*KeyValueGetRequest) GetTimeoutMilliseconds 

func (x *KeyValueGetRequest) GetTimeoutMilliseconds() int32

func (*KeyValueGetRequest) ProtoMessage 

func (*KeyValueGetRequest) ProtoMessage()

func (*KeyValueGetRequest) ProtoReflect 

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

func (*KeyValueGetRequest) Reset 

func (x *KeyValueGetRequest) Reset()

func (*KeyValueGetRequest) String 

func (x *KeyValueGetRequest) String() string

type KeyValueGetResponse 

type KeyValueGetResponse struct {
	Found bool   `protobuf:"varint,1,opt,name=found,proto3" json:"found,omitempty"`
	Value []byte `protobuf:"bytes,2,opt,name=value,proto3" json:"value,omitempty"`
	// contains filtered or unexported fields
}

func (*KeyValueGetResponse) Descriptor deprecated 

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

Deprecated: Use KeyValueGetResponse.ProtoReflect.Descriptor instead.

func (*KeyValueGetResponse) GetFound 

func (x *KeyValueGetResponse) GetFound() bool

func (*KeyValueGetResponse) GetValue 

func (x *KeyValueGetResponse) GetValue() []byte

func (*KeyValueGetResponse) ProtoMessage 

func (*KeyValueGetResponse) ProtoMessage()

func (*KeyValueGetResponse) ProtoReflect 

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

func (*KeyValueGetResponse) Reset 

func (x *KeyValueGetResponse) Reset()

func (*KeyValueGetResponse) String 

func (x *KeyValueGetResponse) String() string

type KeyValueSetRequest 

type KeyValueSetRequest struct {
	SessionId uint64 `protobuf:"varint,1,opt,name=session_id,json=sessionId,proto3" json:"session_id,omitempty"`
	Key       []byte `protobuf:"bytes,2,opt,name=key,proto3" json:"key,omitempty"`
	Value     []byte `protobuf:"bytes,3,opt,name=value,proto3" json:"value,omitempty"`
	// contains filtered or unexported fields
}

func (*KeyValueSetRequest) Descriptor deprecated 

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

Deprecated: Use KeyValueSetRequest.ProtoReflect.Descriptor instead.

func (*KeyValueSetRequest) GetKey 

func (x *KeyValueSetRequest) GetKey() []byte

func (*KeyValueSetRequest) GetSessionId 

func (x *KeyValueSetRequest) GetSessionId() uint64

func (*KeyValueSetRequest) GetValue 

func (x *KeyValueSetRequest) GetValue() []byte

func (*KeyValueSetRequest) ProtoMessage 

func (*KeyValueSetRequest) ProtoMessage()

func (*KeyValueSetRequest) ProtoReflect 

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

func (*KeyValueSetRequest) Reset 

func (x *KeyValueSetRequest) Reset()

func (*KeyValueSetRequest) String 

func (x *KeyValueSetRequest) String() string

type KeyValueSetResponse 

type KeyValueSetResponse struct {
	// contains filtered or unexported fields
}

func (*KeyValueSetResponse) Descriptor deprecated 

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

Deprecated: Use KeyValueSetResponse.ProtoReflect.Descriptor instead.

func (*KeyValueSetResponse) ProtoMessage 

func (*KeyValueSetResponse) ProtoMessage()

func (*KeyValueSetResponse) ProtoReflect 

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

func (*KeyValueSetResponse) Reset 

func (x *KeyValueSetResponse) Reset()

func (*KeyValueSetResponse) String 

func (x *KeyValueSetResponse) String() string

type LabeledStepStats 

type LabeledStepStats struct {
	StepId    int64      `protobuf:"varint,1,opt,name=step_id,json=stepId,proto3" json:"step_id,omitempty"`
	StepStats *StepStats `protobuf:"bytes,2,opt,name=step_stats,json=stepStats,proto3" json:"step_stats,omitempty"`
	// contains filtered or unexported fields
}

func (*LabeledStepStats) Descriptor deprecated 

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

Deprecated: Use LabeledStepStats.ProtoReflect.Descriptor instead.

func (*LabeledStepStats) GetStepId 

func (x *LabeledStepStats) GetStepId() int64

func (*LabeledStepStats) GetStepStats 

func (x *LabeledStepStats) GetStepStats() *StepStats

func (*LabeledStepStats) ProtoMessage 

func (*LabeledStepStats) ProtoMessage()

func (*LabeledStepStats) ProtoReflect 

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

func (*LabeledStepStats) Reset 

func (x *LabeledStepStats) Reset()

func (*LabeledStepStats) String 

func (x *LabeledStepStats) String() string

type ListDevicesRequest 

type ListDevicesRequest struct {

	// Optional: session_handle must be returned by a CreateSession call to the
	// same master service.
	//
	// When session_handle is empty, the ClusterSpec provided when the master was
	// started is used to compute the available devices. If the session_handle is
	// provided but not recognized, an error is returned. Finally, if a valid
	// session_handle is provided, the cluster configuration for that session is
	// used when computing the response.
	SessionHandle string `protobuf:"bytes,1,opt,name=session_handle,json=sessionHandle,proto3" json:"session_handle,omitempty"`
	// contains filtered or unexported fields
}

func (*ListDevicesRequest) Descriptor deprecated 

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

Deprecated: Use ListDevicesRequest.ProtoReflect.Descriptor instead.

func (*ListDevicesRequest) GetSessionHandle 

func (x *ListDevicesRequest) GetSessionHandle() string

func (*ListDevicesRequest) ProtoMessage 

func (*ListDevicesRequest) ProtoMessage()

func (*ListDevicesRequest) ProtoReflect 

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

func (*ListDevicesRequest) Reset 

func (x *ListDevicesRequest) Reset()

func (*ListDevicesRequest) String 

func (x *ListDevicesRequest) String() string

type ListDevicesResponse 

type ListDevicesResponse struct {
	LocalDevice  []*DeviceAttributes `protobuf:"bytes,1,rep,name=local_device,json=localDevice,proto3" json:"local_device,omitempty"`
	RemoteDevice []*DeviceAttributes `protobuf:"bytes,2,rep,name=remote_device,json=remoteDevice,proto3" json:"remote_device,omitempty"`
	// contains filtered or unexported fields
}

func (*ListDevicesResponse) Descriptor deprecated 

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

Deprecated: Use ListDevicesResponse.ProtoReflect.Descriptor instead.

func (*ListDevicesResponse) GetLocalDevice 

func (x *ListDevicesResponse) GetLocalDevice() []*DeviceAttributes

func (*ListDevicesResponse) GetRemoteDevice 

func (x *ListDevicesResponse) GetRemoteDevice() []*DeviceAttributes

func (*ListDevicesResponse) ProtoMessage 

func (*ListDevicesResponse) ProtoMessage()

func (*ListDevicesResponse) ProtoReflect 

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

func (*ListDevicesResponse) Reset 

func (x *ListDevicesResponse) Reset()

func (*ListDevicesResponse) String 

func (x *ListDevicesResponse) String() string

type ListValue 

type ListValue struct {
	Values []*StructuredValue `protobuf:"bytes,1,rep,name=values,proto3" json:"values,omitempty"`
	// contains filtered or unexported fields
}

Represents a Python list.

func (*ListValue) Descriptor deprecated 

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

Deprecated: Use ListValue.ProtoReflect.Descriptor instead.

func (*ListValue) GetValues 

func (x *ListValue) GetValues() []*StructuredValue

func (*ListValue) ProtoMessage 

func (*ListValue) ProtoMessage()

func (*ListValue) ProtoReflect 

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

func (*ListValue) Reset 

func (x *ListValue) Reset()

func (*ListValue) String 

func (x *ListValue) 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 LocalTopologyProto 

type LocalTopologyProto struct {
	NodeId  int32          `protobuf:"varint,1,opt,name=node_id,json=nodeId,proto3" json:"node_id,omitempty"`
	Devices []*DeviceProto `protobuf:"bytes,2,rep,name=devices,proto3" json:"devices,omitempty"`
	// contains filtered or unexported fields
}

func (*LocalTopologyProto) Descriptor deprecated 

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

Deprecated: Use LocalTopologyProto.ProtoReflect.Descriptor instead.

func (*LocalTopologyProto) GetDevices 

func (x *LocalTopologyProto) GetDevices() []*DeviceProto

func (*LocalTopologyProto) GetNodeId 

func (x *LocalTopologyProto) GetNodeId() int32

func (*LocalTopologyProto) ProtoMessage 

func (*LocalTopologyProto) ProtoMessage()

func (*LocalTopologyProto) ProtoReflect 

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

func (*LocalTopologyProto) Reset 

func (x *LocalTopologyProto) Reset()

func (*LocalTopologyProto) String 

func (x *LocalTopologyProto) String() string

type LogMessage deprecated 

type LogMessage struct {
	Level   LogMessage_Level `protobuf:"varint,1,opt,name=level,proto3,enum=tensorflow.LogMessage_Level" json:"level,omitempty"`
	Message string           `protobuf:"bytes,2,opt,name=message,proto3" json:"message,omitempty"`
	// contains filtered or unexported fields
}

Protocol buffer used for logging messages to the events file.

This was theoretically used by the defunct tensorboard_logging module, which has been removed; this message is now deprecated and should not be used.

Deprecated: Do not use.

func (*LogMessage) Descriptor deprecated 

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

Deprecated: Use LogMessage.ProtoReflect.Descriptor instead.

func (*LogMessage) GetLevel 

func (x *LogMessage) GetLevel() LogMessage_Level

func (*LogMessage) GetMessage 

func (x *LogMessage) GetMessage() string

func (*LogMessage) ProtoMessage 

func (*LogMessage) ProtoMessage()

func (*LogMessage) ProtoReflect 

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

func (*LogMessage) Reset 

func (x *LogMessage) Reset()

func (*LogMessage) String 

func (x *LogMessage) String() string

type LogMessage_Level deprecated 

type LogMessage_Level int32

Deprecated: Do not use. 

const (
	LogMessage_UNKNOWN LogMessage_Level = 0
	// Note: The logging level 10 cannot be named DEBUG. Some software
	// projects compile their C/C++ code with -DDEBUG in debug builds. So the
	// C++ code generated from this file should not have an identifier named
	// DEBUG.
	LogMessage_DEBUGGING LogMessage_Level = 10
	LogMessage_INFO      LogMessage_Level = 20
	LogMessage_WARN      LogMessage_Level = 30
	LogMessage_ERROR     LogMessage_Level = 40
	LogMessage_FATAL     LogMessage_Level = 50
)

func (LogMessage_Level) Descriptor 

func (LogMessage_Level) Descriptor() protoreflect.EnumDescriptor

func (LogMessage_Level) Enum 

func (x LogMessage_Level) Enum() *LogMessage_Level

func (LogMessage_Level) EnumDescriptor deprecated 

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

Deprecated: Use LogMessage_Level.Descriptor instead.

func (LogMessage_Level) Number 

func (x LogMessage_Level) Number() protoreflect.EnumNumber

func (LogMessage_Level) String 

func (x LogMessage_Level) String() string

func (LogMessage_Level) Type 

func (LogMessage_Level) Type() protoreflect.EnumType

type LoggingRequest 

type LoggingRequest struct {

	// If true, RPC logging will be enabled.
	EnableRpcLogging bool `protobuf:"varint,1,opt,name=enable_rpc_logging,json=enableRpcLogging,proto3" json:"enable_rpc_logging,omitempty"`
	// If true, RPC logging will be disabled.
	DisableRpcLogging bool `protobuf:"varint,4,opt,name=disable_rpc_logging,json=disableRpcLogging,proto3" json:"disable_rpc_logging,omitempty"`
	// If true, discard any saved logging data (for all steps).
	Clear bool `protobuf:"varint,2,opt,name=clear,proto3" json:"clear,omitempty"`
	// When set, requests all saved log data pertaining to the step.
	// Any log data retrieved is eliminated from the store and cannot be
	// retrieved again.
	FetchStepId []int64 `protobuf:"varint,3,rep,packed,name=fetch_step_id,json=fetchStepId,proto3" json:"fetch_step_id,omitempty"`
	// contains filtered or unexported fields
}

Out-of-band request to begin or end logging, or to retrieve logs for particular steps.

func (*LoggingRequest) Descriptor deprecated 

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

Deprecated: Use LoggingRequest.ProtoReflect.Descriptor instead.

func (*LoggingRequest) GetClear 

func (x *LoggingRequest) GetClear() bool

func (*LoggingRequest) GetDisableRpcLogging 

func (x *LoggingRequest) GetDisableRpcLogging() bool

func (*LoggingRequest) GetEnableRpcLogging 

func (x *LoggingRequest) GetEnableRpcLogging() bool

func (*LoggingRequest) GetFetchStepId 

func (x *LoggingRequest) GetFetchStepId() []int64

func (*LoggingRequest) ProtoMessage 

func (*LoggingRequest) ProtoMessage()

func (*LoggingRequest) ProtoReflect 

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

func (*LoggingRequest) Reset 

func (x *LoggingRequest) Reset()

func (*LoggingRequest) String 

func (x *LoggingRequest) String() string

type LoggingResponse 

type LoggingResponse struct {
	Step []*LabeledStepStats `protobuf:"bytes,1,rep,name=step,proto3" json:"step,omitempty"`
	// contains filtered or unexported fields
}

func (*LoggingResponse) Descriptor deprecated 

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

Deprecated: Use LoggingResponse.ProtoReflect.Descriptor instead.

func (*LoggingResponse) GetStep 

func (x *LoggingResponse) GetStep() []*LabeledStepStats

func (*LoggingResponse) ProtoMessage 

func (*LoggingResponse) ProtoMessage()

func (*LoggingResponse) ProtoReflect 

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

func (*LoggingResponse) Reset 

func (x *LoggingResponse) Reset()

func (*LoggingResponse) String 

func (x *LoggingResponse) String() string

type MakeCallableRequest 

type MakeCallableRequest struct {

	// REQUIRED: session_handle must be returned by a CreateSession call
	// to the same master service.
	SessionHandle string `protobuf:"bytes,1,opt,name=session_handle,json=sessionHandle,proto3" json:"session_handle,omitempty"`
	// Options that define the behavior of the created callable.
	Options *CallableOptions `protobuf:"bytes,2,opt,name=options,proto3" json:"options,omitempty"`
	// Unique identifier for this request. Every MakeCallableRequest must
	// have a unique request_id, and retried MakeCallableRequest must have
	// the same request_id. If request_id is zero, retry detection is disabled.
	RequestId int64 `protobuf:"varint,3,opt,name=request_id,json=requestId,proto3" json:"request_id,omitempty"`
	// contains filtered or unexported fields
}

func (*MakeCallableRequest) Descriptor deprecated 

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

Deprecated: Use MakeCallableRequest.ProtoReflect.Descriptor instead.

func (*MakeCallableRequest) GetOptions 

func (x *MakeCallableRequest) GetOptions() *CallableOptions

func (*MakeCallableRequest) GetRequestId 

func (x *MakeCallableRequest) GetRequestId() int64

func (*MakeCallableRequest) GetSessionHandle 

func (x *MakeCallableRequest) GetSessionHandle() string

func (*MakeCallableRequest) ProtoMessage 

func (*MakeCallableRequest) ProtoMessage()

func (*MakeCallableRequest) ProtoReflect 

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

func (*MakeCallableRequest) Reset 

func (x *MakeCallableRequest) Reset()

func (*MakeCallableRequest) String 

func (x *MakeCallableRequest) String() string

type MakeCallableResponse 

type MakeCallableResponse struct {

	// A handle to the created callable.
	Handle int64 `protobuf:"varint,1,opt,name=handle,proto3" json:"handle,omitempty"`
	// contains filtered or unexported fields
}

func (*MakeCallableResponse) Descriptor deprecated 

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

Deprecated: Use MakeCallableResponse.ProtoReflect.Descriptor instead.

func (*MakeCallableResponse) GetHandle 

func (x *MakeCallableResponse) GetHandle() int64

func (*MakeCallableResponse) ProtoMessage 

func (*MakeCallableResponse) ProtoMessage()

func (*MakeCallableResponse) ProtoReflect 

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

func (*MakeCallableResponse) Reset 

func (x *MakeCallableResponse) Reset()

func (*MakeCallableResponse) String 

func (x *MakeCallableResponse) String() string

type MarkRecvFinishedRequest 

type MarkRecvFinishedRequest struct {
	RequestId int64 `protobuf:"varint,1,opt,name=request_id,json=requestId,proto3" json:"request_id,omitempty"`
	// contains filtered or unexported fields
}

Message for managing the response cache maintained on the sender side. Currently only used by the gRPC worker service.

func (*MarkRecvFinishedRequest) Descriptor deprecated 

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

Deprecated: Use MarkRecvFinishedRequest.ProtoReflect.Descriptor instead.

func (*MarkRecvFinishedRequest) GetRequestId 

func (x *MarkRecvFinishedRequest) GetRequestId() int64

func (*MarkRecvFinishedRequest) ProtoMessage 

func (*MarkRecvFinishedRequest) ProtoMessage()

func (*MarkRecvFinishedRequest) ProtoReflect 

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

func (*MarkRecvFinishedRequest) Reset 

func (x *MarkRecvFinishedRequest) Reset()

func (*MarkRecvFinishedRequest) String 

func (x *MarkRecvFinishedRequest) String() string

type MarkRecvFinishedResponse 

type MarkRecvFinishedResponse struct {
	// contains filtered or unexported fields
}

func (*MarkRecvFinishedResponse) Descriptor deprecated 

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

Deprecated: Use MarkRecvFinishedResponse.ProtoReflect.Descriptor instead.

func (*MarkRecvFinishedResponse) ProtoMessage 

func (*MarkRecvFinishedResponse) ProtoMessage()

func (*MarkRecvFinishedResponse) ProtoReflect 

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

func (*MarkRecvFinishedResponse) Reset 

func (x *MarkRecvFinishedResponse) Reset()

func (*MarkRecvFinishedResponse) String 

func (x *MarkRecvFinishedResponse) String() string

type MatmulProto 

type MatmulProto struct {
	AbDtype     DataType       `protobuf:"varint,1,opt,name=ab_dtype,json=abDtype,proto3,enum=dnn.DataType" json:"ab_dtype,omitempty"`
	CDtype      DataType       `protobuf:"varint,2,opt,name=c_dtype,json=cDtype,proto3,enum=dnn.DataType" json:"c_dtype,omitempty"`
	TransA      bool           `protobuf:"varint,3,opt,name=trans_a,json=transA,proto3" json:"trans_a,omitempty"`
	TransB      bool           `protobuf:"varint,4,opt,name=trans_b,json=transB,proto3" json:"trans_b,omitempty"`
	M           uint64         `protobuf:"varint,5,opt,name=m,proto3" json:"m,omitempty"`
	N           uint64         `protobuf:"varint,6,opt,name=n,proto3" json:"n,omitempty"`
	K           uint64         `protobuf:"varint,7,opt,name=k,proto3" json:"k,omitempty"`
	Lda         int64          `protobuf:"varint,8,opt,name=lda,proto3" json:"lda,omitempty"`
	Ldb         int64          `protobuf:"varint,9,opt,name=ldb,proto3" json:"ldb,omitempty"`
	Ldc         int64          `protobuf:"varint,10,opt,name=ldc,proto3" json:"ldc,omitempty"`
	Activation  ActivationMode `protobuf:"varint,11,opt,name=activation,proto3,enum=dnn.ActivationMode" json:"activation,omitempty"`
	AAddress    int64          `protobuf:"varint,12,opt,name=a_address,json=aAddress,proto3" json:"a_address,omitempty"`
	BAddress    int64          `protobuf:"varint,13,opt,name=b_address,json=bAddress,proto3" json:"b_address,omitempty"`
	CAddress    int64          `protobuf:"varint,14,opt,name=c_address,json=cAddress,proto3" json:"c_address,omitempty"`
	BiasAddress int64          `protobuf:"varint,15,opt,name=bias_address,json=biasAddress,proto3" json:"bias_address,omitempty"`
	// contains filtered or unexported fields
}

func (*MatmulProto) Descriptor deprecated 

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

Deprecated: Use MatmulProto.ProtoReflect.Descriptor instead.

func (*MatmulProto) GetAAddress 

func (x *MatmulProto) GetAAddress() int64

func (*MatmulProto) GetAbDtype 

func (x *MatmulProto) GetAbDtype() DataType

func (*MatmulProto) GetActivation 

func (x *MatmulProto) GetActivation() ActivationMode

func (*MatmulProto) GetBAddress 

func (x *MatmulProto) GetBAddress() int64

func (*MatmulProto) GetBiasAddress 

func (x *MatmulProto) GetBiasAddress() int64

func (*MatmulProto) GetCAddress 

func (x *MatmulProto) GetCAddress() int64

func (*MatmulProto) GetCDtype 

func (x *MatmulProto) GetCDtype() DataType

func (*MatmulProto) GetK 

func (x *MatmulProto) GetK() uint64

func (*MatmulProto) GetLda 

func (x *MatmulProto) GetLda() int64

func (*MatmulProto) GetLdb 

func (x *MatmulProto) GetLdb() int64

func (*MatmulProto) GetLdc 

func (x *MatmulProto) GetLdc() int64

func (*MatmulProto) GetM 

func (x *MatmulProto) GetM() uint64

func (*MatmulProto) GetN 

func (x *MatmulProto) GetN() uint64

func (*MatmulProto) GetTransA 

func (x *MatmulProto) GetTransA() bool

func (*MatmulProto) GetTransB 

func (x *MatmulProto) GetTransB() bool

func (*MatmulProto) ProtoMessage 

func (*MatmulProto) ProtoMessage()

func (*MatmulProto) ProtoReflect 

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

func (*MatmulProto) Reset 

func (x *MatmulProto) Reset()

func (*MatmulProto) String 

func (x *MatmulProto) String() string

type MemAllocatorStats 

type MemAllocatorStats struct {
	NumAllocs           int64   `protobuf:"varint,1,opt,name=num_allocs,json=numAllocs,proto3" json:"num_allocs,omitempty"`
	BytesInUse          int64   `protobuf:"varint,2,opt,name=bytes_in_use,json=bytesInUse,proto3" json:"bytes_in_use,omitempty"`
	PeakBytesInUse      int64   `protobuf:"varint,3,opt,name=peak_bytes_in_use,json=peakBytesInUse,proto3" json:"peak_bytes_in_use,omitempty"`
	LargestAllocSize    int64   `protobuf:"varint,4,opt,name=largest_alloc_size,json=largestAllocSize,proto3" json:"largest_alloc_size,omitempty"`
	FragmentationMetric float32 `protobuf:"fixed32,5,opt,name=fragmentation_metric,json=fragmentationMetric,proto3" json:"fragmentation_metric,omitempty"`
	// contains filtered or unexported fields
}

Some of the data from AllocatorStats

func (*MemAllocatorStats) Descriptor deprecated 

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

Deprecated: Use MemAllocatorStats.ProtoReflect.Descriptor instead.

func (*MemAllocatorStats) GetBytesInUse 

func (x *MemAllocatorStats) GetBytesInUse() int64

func (*MemAllocatorStats) GetFragmentationMetric 

func (x *MemAllocatorStats) GetFragmentationMetric() float32

func (*MemAllocatorStats) GetLargestAllocSize 

func (x *MemAllocatorStats) GetLargestAllocSize() int64

func (*MemAllocatorStats) GetNumAllocs 

func (x *MemAllocatorStats) GetNumAllocs() int64

func (*MemAllocatorStats) GetPeakBytesInUse 

func (x *MemAllocatorStats) GetPeakBytesInUse() int64

func (*MemAllocatorStats) ProtoMessage 

func (*MemAllocatorStats) ProtoMessage()

func (*MemAllocatorStats) ProtoReflect 

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

func (*MemAllocatorStats) Reset 

func (x *MemAllocatorStats) Reset()

func (*MemAllocatorStats) String 

func (x *MemAllocatorStats) String() string

type MemChunk 

type MemChunk struct {
	Address       uint64 `protobuf:"varint,1,opt,name=address,proto3" json:"address,omitempty"`
	Size          int64  `protobuf:"varint,2,opt,name=size,proto3" json:"size,omitempty"`
	RequestedSize int64  `protobuf:"varint,3,opt,name=requested_size,json=requestedSize,proto3" json:"requested_size,omitempty"`
	Bin           int32  `protobuf:"varint,4,opt,name=bin,proto3" json:"bin,omitempty"`
	OpName        string `protobuf:"bytes,5,opt,name=op_name,json=opName,proto3" json:"op_name,omitempty"`
	FreedAtCount  uint64 `protobuf:"varint,6,opt,name=freed_at_count,json=freedAtCount,proto3" json:"freed_at_count,omitempty"`
	ActionCount   uint64 `protobuf:"varint,7,opt,name=action_count,json=actionCount,proto3" json:"action_count,omitempty"`
	InUse         bool   `protobuf:"varint,8,opt,name=in_use,json=inUse,proto3" json:"in_use,omitempty"`
	StepId        uint64 `protobuf:"varint,9,opt,name=step_id,json=stepId,proto3" json:"step_id,omitempty"`
	// contains filtered or unexported fields
}

func (*MemChunk) Descriptor deprecated 

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

Deprecated: Use MemChunk.ProtoReflect.Descriptor instead.

func (*MemChunk) GetActionCount 

func (x *MemChunk) GetActionCount() uint64

func (*MemChunk) GetAddress 

func (x *MemChunk) GetAddress() uint64

func (*MemChunk) GetBin 

func (x *MemChunk) GetBin() int32

func (*MemChunk) GetFreedAtCount 

func (x *MemChunk) GetFreedAtCount() uint64

func (*MemChunk) GetInUse 

func (x *MemChunk) GetInUse() bool

func (*MemChunk) GetOpName 

func (x *MemChunk) GetOpName() string

func (*MemChunk) GetRequestedSize 

func (x *MemChunk) GetRequestedSize() int64

func (*MemChunk) GetSize 

func (x *MemChunk) GetSize() int64

func (*MemChunk) GetStepId 

func (x *MemChunk) GetStepId() uint64

func (*MemChunk) ProtoMessage 

func (*MemChunk) ProtoMessage()

func (*MemChunk) ProtoReflect 

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

func (*MemChunk) Reset 

func (x *MemChunk) Reset()

func (*MemChunk) String 

func (x *MemChunk) String() string

type MemoryDump 

type MemoryDump struct {
	AllocatorName string             `protobuf:"bytes,1,opt,name=allocator_name,json=allocatorName,proto3" json:"allocator_name,omitempty"`
	BinSummary    []*BinSummary      `protobuf:"bytes,2,rep,name=bin_summary,json=binSummary,proto3" json:"bin_summary,omitempty"`
	Chunk         []*MemChunk        `protobuf:"bytes,3,rep,name=chunk,proto3" json:"chunk,omitempty"`
	SnapShot      []*SnapShot        `protobuf:"bytes,4,rep,name=snap_shot,json=snapShot,proto3" json:"snap_shot,omitempty"`
	Stats         *MemAllocatorStats `protobuf:"bytes,5,opt,name=stats,proto3" json:"stats,omitempty"`
	// contains filtered or unexported fields
}

func (*MemoryDump) Descriptor deprecated 

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

Deprecated: Use MemoryDump.ProtoReflect.Descriptor instead.

func (*MemoryDump) GetAllocatorName 

func (x *MemoryDump) GetAllocatorName() string

func (*MemoryDump) GetBinSummary 

func (x *MemoryDump) GetBinSummary() []*BinSummary

func (*MemoryDump) GetChunk 

func (x *MemoryDump) GetChunk() []*MemChunk

func (*MemoryDump) GetSnapShot 

func (x *MemoryDump) GetSnapShot() []*SnapShot

func (*MemoryDump) GetStats 

func (x *MemoryDump) GetStats() *MemAllocatorStats

func (*MemoryDump) ProtoMessage 

func (*MemoryDump) ProtoMessage()

func (*MemoryDump) ProtoReflect 

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

func (*MemoryDump) Reset 

func (x *MemoryDump) Reset()

func (*MemoryDump) String 

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

For memory tracking.

func (*MemoryStats) Descriptor deprecated 

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

Deprecated: Use MemoryStats.ProtoReflect.Descriptor instead.

func (*MemoryStats) GetDevicePersistentMemorySize deprecated 

func (x *MemoryStats) GetDevicePersistentMemorySize() int64

Deprecated: Do not use.

func (*MemoryStats) GetDevicePersistentTensorAllocIds deprecated 

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

Deprecated: Do not use.

func (*MemoryStats) GetDeviceTempMemorySize deprecated 

func (x *MemoryStats) GetDeviceTempMemorySize() int64

Deprecated: Do not use.

func (*MemoryStats) GetPersistentMemorySize 

func (x *MemoryStats) GetPersistentMemorySize() int64

func (*MemoryStats) GetPersistentTensorAllocIds 

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

func (*MemoryStats) GetTempMemorySize 

func (x *MemoryStats) GetTempMemorySize() int64

func (*MemoryStats) ProtoMessage 

func (*MemoryStats) ProtoMessage()

func (*MemoryStats) ProtoReflect 

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

func (*MemoryStats) Reset 

func (x *MemoryStats) Reset()

func (*MemoryStats) String 

func (x *MemoryStats) String() string

type MetaGraphDef 

type MetaGraphDef struct {
	MetaInfoDef *MetaGraphDef_MetaInfoDef `protobuf:"bytes,1,opt,name=meta_info_def,json=metaInfoDef,proto3" json:"meta_info_def,omitempty"`
	// GraphDef.
	GraphDef *GraphDef `protobuf:"bytes,2,opt,name=graph_def,json=graphDef,proto3" json:"graph_def,omitempty"`
	// SaverDef.
	SaverDef *SaverDef `protobuf:"bytes,3,opt,name=saver_def,json=saverDef,proto3" json:"saver_def,omitempty"`
	// collection_def: Map from collection name to collections.
	// See CollectionDef section for details.
	CollectionDef map[string]*CollectionDef `` /* 188-byte string literal not displayed */
	// signature_def: Map from user supplied key for a signature to a single
	// SignatureDef.
	SignatureDef map[string]*SignatureDef `` /* 185-byte string literal not displayed */
	// Asset file def to be used with the defined graph.
	AssetFileDef []*AssetFileDef `protobuf:"bytes,6,rep,name=asset_file_def,json=assetFileDef,proto3" json:"asset_file_def,omitempty"`
	// Extra information about the structure of functions and stateful objects.
	ObjectGraphDef *SavedObjectGraph `protobuf:"bytes,7,opt,name=object_graph_def,json=objectGraphDef,proto3" json:"object_graph_def,omitempty"`
	// contains filtered or unexported fields
}

Protocol buffer containing the following which are necessary to restart training, run inference. It can be used to serialize/de-serialize memory objects necessary for running computation in a graph when crossing the process boundary. It can be used for long term storage of graphs, cross-language execution of graphs, etc. 

MetaInfoDef
GraphDef
SaverDef
CollectionDef
TensorInfo
SignatureDef

func (*MetaGraphDef) Descriptor deprecated 

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

Deprecated: Use MetaGraphDef.ProtoReflect.Descriptor instead.

func (*MetaGraphDef) GetAssetFileDef 

func (x *MetaGraphDef) GetAssetFileDef() []*AssetFileDef

func (*MetaGraphDef) GetCollectionDef 

func (x *MetaGraphDef) GetCollectionDef() map[string]*CollectionDef

func (*MetaGraphDef) GetGraphDef 

func (x *MetaGraphDef) GetGraphDef() *GraphDef

func (*MetaGraphDef) GetMetaInfoDef 

func (x *MetaGraphDef) GetMetaInfoDef() *MetaGraphDef_MetaInfoDef

func (*MetaGraphDef) GetObjectGraphDef 

func (x *MetaGraphDef) GetObjectGraphDef() *SavedObjectGraph

func (*MetaGraphDef) GetSaverDef 

func (x *MetaGraphDef) GetSaverDef() *SaverDef

func (*MetaGraphDef) GetSignatureDef 

func (x *MetaGraphDef) GetSignatureDef() map[string]*SignatureDef

func (*MetaGraphDef) ProtoMessage 

func (*MetaGraphDef) ProtoMessage()

func (*MetaGraphDef) ProtoReflect 

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

func (*MetaGraphDef) Reset 

func (x *MetaGraphDef) Reset()

func (*MetaGraphDef) String 

func (x *MetaGraphDef) String() string

type MetaGraphDef_MetaInfoDef 

type MetaGraphDef_MetaInfoDef struct {

	// User specified Version string. Can be the name of the model and revision,
	// steps this model has been trained to, etc.
	MetaGraphVersion string `protobuf:"bytes,1,opt,name=meta_graph_version,json=metaGraphVersion,proto3" json:"meta_graph_version,omitempty"`
	// A copy of the OpDefs used by the producer of this graph_def.
	// Descriptions and Ops not used in graph_def are stripped out.
	StrippedOpList *OpList `protobuf:"bytes,2,opt,name=stripped_op_list,json=strippedOpList,proto3" json:"stripped_op_list,omitempty"`
	// A serialized protobuf. Can be the time this meta graph is created, or
	// modified, or name of the model.
	AnyInfo *any.Any `protobuf:"bytes,3,opt,name=any_info,json=anyInfo,proto3" json:"any_info,omitempty"`
	// User supplied tag(s) on the meta_graph and included graph_def.
	//
	// MetaGraphDefs should be tagged with their capabilities or use-cases.
	// Examples: "train", "serve", "gpu", "tpu", etc.
	// These tags enable loaders to access the MetaGraph(s) appropriate for a
	// specific use-case or runtime environment.
	Tags []string `protobuf:"bytes,4,rep,name=tags,proto3" json:"tags,omitempty"`
	// The __version__ string of the tensorflow build used to write this graph.
	// This will be populated by the framework, which will overwrite any user
	// supplied value.
	TensorflowVersion string `protobuf:"bytes,5,opt,name=tensorflow_version,json=tensorflowVersion,proto3" json:"tensorflow_version,omitempty"`
	// The __git_version__ string of the tensorflow build used to write this
	// graph. This will be populated by the framework, which will overwrite any
	// user supplied value.
	TensorflowGitVersion string `protobuf:"bytes,6,opt,name=tensorflow_git_version,json=tensorflowGitVersion,proto3" json:"tensorflow_git_version,omitempty"`
	// A flag to denote whether default-valued attrs have been stripped from
	// the nodes in this graph_def.
	StrippedDefaultAttrs bool `protobuf:"varint,7,opt,name=stripped_default_attrs,json=strippedDefaultAttrs,proto3" json:"stripped_default_attrs,omitempty"`
	// FunctionDef name to aliases mapping.
	FunctionAliases map[string]string `` /* 194-byte string literal not displayed */
	// contains filtered or unexported fields
}

Meta information regarding the graph to be exported. To be used by users of this protocol buffer to encode information regarding their meta graph.

func (*MetaGraphDef_MetaInfoDef) Descriptor deprecated 

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

Deprecated: Use MetaGraphDef_MetaInfoDef.ProtoReflect.Descriptor instead.

func (*MetaGraphDef_MetaInfoDef) GetAnyInfo 

func (x *MetaGraphDef_MetaInfoDef) GetAnyInfo() *any.Any

func (*MetaGraphDef_MetaInfoDef) GetFunctionAliases 

func (x *MetaGraphDef_MetaInfoDef) GetFunctionAliases() map[string]string

func (*MetaGraphDef_MetaInfoDef) GetMetaGraphVersion 

func (x *MetaGraphDef_MetaInfoDef) GetMetaGraphVersion() string

func (*MetaGraphDef_MetaInfoDef) GetStrippedDefaultAttrs 

func (x *MetaGraphDef_MetaInfoDef) GetStrippedDefaultAttrs() bool

func (*MetaGraphDef_MetaInfoDef) GetStrippedOpList 

func (x *MetaGraphDef_MetaInfoDef) GetStrippedOpList() *OpList

func (*MetaGraphDef_MetaInfoDef) GetTags 

func (x *MetaGraphDef_MetaInfoDef) GetTags() []string

func (*MetaGraphDef_MetaInfoDef) GetTensorflowGitVersion 

func (x *MetaGraphDef_MetaInfoDef) GetTensorflowGitVersion() string

func (*MetaGraphDef_MetaInfoDef) GetTensorflowVersion 

func (x *MetaGraphDef_MetaInfoDef) GetTensorflowVersion() string

func (*MetaGraphDef_MetaInfoDef) ProtoMessage 

func (*MetaGraphDef_MetaInfoDef) ProtoMessage()

func (*MetaGraphDef_MetaInfoDef) ProtoReflect 

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

func (*MetaGraphDef_MetaInfoDef) Reset 

func (x *MetaGraphDef_MetaInfoDef) Reset()

func (*MetaGraphDef_MetaInfoDef) String 

func (x *MetaGraphDef_MetaInfoDef) 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 NamedDevice 

type NamedDevice struct {
	Name       string            `protobuf:"bytes,1,opt,name=name,proto3" json:"name,omitempty"`
	Properties *DeviceProperties `protobuf:"bytes,2,opt,name=properties,proto3" json:"properties,omitempty"`
	// contains filtered or unexported fields
}

func (*NamedDevice) Descriptor deprecated 

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

Deprecated: Use NamedDevice.ProtoReflect.Descriptor instead.

func (*NamedDevice) GetName 

func (x *NamedDevice) GetName() string

func (*NamedDevice) GetProperties 

func (x *NamedDevice) GetProperties() *DeviceProperties

func (*NamedDevice) ProtoMessage 

func (*NamedDevice) ProtoMessage()

func (*NamedDevice) ProtoReflect 

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

func (*NamedDevice) Reset 

func (x *NamedDevice) Reset()

func (*NamedDevice) String 

func (x *NamedDevice) String() string

type NamedTensorProto 

type NamedTensorProto struct {

	// Name of the tensor.
	Name string `protobuf:"bytes,1,opt,name=name,proto3" json:"name,omitempty"`
	// The client can populate a TensorProto using a tensorflow::Tensor`, or
	// directly using the protobuf field accessors.
	//
	// The client specifies whether the returned tensor values should be
	// filled tensor fields (float_val, int_val, etc.) or encoded in a
	// compact form in tensor.tensor_content.
	Tensor *TensorProto `protobuf:"bytes,2,opt,name=tensor,proto3" json:"tensor,omitempty"`
	// contains filtered or unexported fields
}

A pair of tensor name and tensor values.

func (*NamedTensorProto) Descriptor deprecated 

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

Deprecated: Use NamedTensorProto.ProtoReflect.Descriptor instead.

func (*NamedTensorProto) GetName 

func (x *NamedTensorProto) GetName() string

func (*NamedTensorProto) GetTensor 

func (x *NamedTensorProto) GetTensor() *TensorProto

func (*NamedTensorProto) ProtoMessage 

func (*NamedTensorProto) ProtoMessage()

func (*NamedTensorProto) ProtoReflect 

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

func (*NamedTensorProto) Reset 

func (x *NamedTensorProto) Reset()

func (*NamedTensorProto) String 

func (x *NamedTensorProto) String() string

type NamedTupleValue 

type NamedTupleValue struct {
	Name   string       `protobuf:"bytes,1,opt,name=name,proto3" json:"name,omitempty"`
	Values []*PairValue `protobuf:"bytes,2,rep,name=values,proto3" json:"values,omitempty"`
	// contains filtered or unexported fields
}

Represents Python's namedtuple.

func (*NamedTupleValue) Descriptor deprecated 

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

Deprecated: Use NamedTupleValue.ProtoReflect.Descriptor instead.

func (*NamedTupleValue) GetName 

func (x *NamedTupleValue) GetName() string

func (*NamedTupleValue) GetValues 

func (x *NamedTupleValue) GetValues() []*PairValue

func (*NamedTupleValue) ProtoMessage 

func (*NamedTupleValue) ProtoMessage()

func (*NamedTupleValue) ProtoReflect 

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

func (*NamedTupleValue) Reset 

func (x *NamedTupleValue) Reset()

func (*NamedTupleValue) String 

func (x *NamedTupleValue) String() string

type NewReplaySession 

type NewReplaySession struct {
	Devices       *ListDevicesResponse `protobuf:"bytes,1,opt,name=devices,proto3" json:"devices,omitempty"`
	SessionHandle string               `protobuf:"bytes,2,opt,name=session_handle,json=sessionHandle,proto3" json:"session_handle,omitempty"`
	// contains filtered or unexported fields
}

Records the creation of a new replay session. We record the device listing here to capture the state of the cluster.

func (*NewReplaySession) Descriptor deprecated 

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

Deprecated: Use NewReplaySession.ProtoReflect.Descriptor instead.

func (*NewReplaySession) GetDevices 

func (x *NewReplaySession) GetDevices() *ListDevicesResponse

func (*NewReplaySession) GetSessionHandle 

func (x *NewReplaySession) GetSessionHandle() string

func (*NewReplaySession) ProtoMessage 

func (*NewReplaySession) ProtoMessage()

func (*NewReplaySession) ProtoReflect 

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

func (*NewReplaySession) Reset 

func (x *NewReplaySession) Reset()

func (*NewReplaySession) String 

func (x *NewReplaySession) 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 NoneValue 

type NoneValue struct {
	// contains filtered or unexported fields
}

Represents None.

func (*NoneValue) Descriptor deprecated 

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

Deprecated: Use NoneValue.ProtoReflect.Descriptor instead.

func (*NoneValue) ProtoMessage 

func (*NoneValue) ProtoMessage()

func (*NoneValue) ProtoReflect 

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

func (*NoneValue) Reset 

func (x *NoneValue) Reset()

func (*NoneValue) String 

func (x *NoneValue) 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 Operation 

type Operation struct {

	// A unique identifier for the operation. Set by the client so that the client
	// can uniquely identify the outputs of the scheduled operation.
	//
	// In the initial implementation, sending duplicate IDs has undefined
	// behaviour, but additional constraints may be placed upon this in the
	// future.
	Id       int64              `protobuf:"varint,1,opt,name=id,proto3" json:"id,omitempty"`
	Name     string             `protobuf:"bytes,2,opt,name=name,proto3" json:"name,omitempty"`
	OpInputs []*Operation_Input `protobuf:"bytes,10,rep,name=op_inputs,json=opInputs,proto3" json:"op_inputs,omitempty"`
	// Control Operation IDs that will be respected when ops are re-ordered by
	// async execution. If async execution (+ op re-ordering) is not enabled, this
	// should have no effect.
	ControlOpIds []int64               `protobuf:"varint,4,rep,packed,name=control_op_ids,json=controlOpIds,proto3" json:"control_op_ids,omitempty"`
	Attrs        map[string]*AttrValue `` /* 151-byte string literal not displayed */
	Device       string                `protobuf:"bytes,6,opt,name=device,proto3" json:"device,omitempty"`
	// Indicates whether the op is a component of a multi-device function.
	IsComponentFunction bool `protobuf:"varint,7,opt,name=is_component_function,json=isComponentFunction,proto3" json:"is_component_function,omitempty"`
	// Set when is_component_function is true. It's initially generated
	// when we create an FunctionLibraryRuntime::Options (negative value) and used
	// to create Rendezvous for function execution. All components of a
	// multi-device function should use the same step id to make sure that they
	// can communicate through Send/Recv ops.
	FuncStepId int64 `protobuf:"varint,8,opt,name=func_step_id,json=funcStepId,proto3" json:"func_step_id,omitempty"`
	// Indicates whether the op is a function.
	IsFunction bool `protobuf:"varint,9,opt,name=is_function,json=isFunction,proto3" json:"is_function,omitempty"`
	// contains filtered or unexported fields
}

A proto representation of an eager operation.

func (*Operation) Descriptor deprecated 

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

Deprecated: Use Operation.ProtoReflect.Descriptor instead.

func (*Operation) GetAttrs 

func (x *Operation) GetAttrs() map[string]*AttrValue

func (*Operation) GetControlOpIds 

func (x *Operation) GetControlOpIds() []int64

func (*Operation) GetDevice 

func (x *Operation) GetDevice() string

func (*Operation) GetFuncStepId 

func (x *Operation) GetFuncStepId() int64

func (*Operation) GetId 

func (x *Operation) GetId() int64

func (*Operation) GetIsComponentFunction 

func (x *Operation) GetIsComponentFunction() bool

func (*Operation) GetIsFunction 

func (x *Operation) GetIsFunction() bool

func (*Operation) GetName 

func (x *Operation) GetName() string

func (*Operation) GetOpInputs 

func (x *Operation) GetOpInputs() []*Operation_Input

func (*Operation) ProtoMessage 

func (*Operation) ProtoMessage()

func (*Operation) ProtoReflect 

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

func (*Operation) Reset 

func (x *Operation) Reset()

func (*Operation) String 

func (x *Operation) String() string

type Operation_Input 

type Operation_Input struct {

	// Types that are assignable to Item:
	//	*Operation_Input_RemoteHandle
	//	*Operation_Input_Tensor
	Item isOperation_Input_Item `protobuf_oneof:"item"`
	// contains filtered or unexported fields
}

func (*Operation_Input) Descriptor deprecated 

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

Deprecated: Use Operation_Input.ProtoReflect.Descriptor instead.

func (*Operation_Input) GetItem 

func (m *Operation_Input) GetItem() isOperation_Input_Item

func (*Operation_Input) GetRemoteHandle 

func (x *Operation_Input) GetRemoteHandle() *RemoteTensorHandle

func (*Operation_Input) GetTensor 

func (x *Operation_Input) GetTensor() *TensorProto

func (*Operation_Input) ProtoMessage 

func (*Operation_Input) ProtoMessage()

func (*Operation_Input) ProtoReflect 

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

func (*Operation_Input) Reset 

func (x *Operation_Input) Reset()

func (*Operation_Input) String 

func (x *Operation_Input) String() string

type Operation_Input_RemoteHandle 

type Operation_Input_RemoteHandle struct {
	RemoteHandle *RemoteTensorHandle `protobuf:"bytes,1,opt,name=remote_handle,json=remoteHandle,proto3,oneof"`
}

type Operation_Input_Tensor 

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

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 OptimizerOptions 

type OptimizerOptions struct {

	// If true, optimize the graph using common subexpression elimination.
	// Note: the optimization Level L1 will override this setting to true. So in
	// order to disable common subexpression elimination the opt_level has to be
	// set to L0.
	DoCommonSubexpressionElimination bool `` /* 162-byte string literal not displayed */
	// If true, perform constant folding optimization on the graph.
	// Note: the optimization Level L1 will override this setting to true. So in
	// order to disable constant folding the opt_level has to be set to L0.
	DoConstantFolding bool `protobuf:"varint,2,opt,name=do_constant_folding,json=doConstantFolding,proto3" json:"do_constant_folding,omitempty"`
	// Constant folding optimization replaces tensors whose values can be
	// predetermined, with constant nodes. To avoid inserting too large constants,
	// the size of each constant created can be limited. If this value is zero, a
	// default limit of 10 MiB will be applied. If constant folding optimization
	// is disabled, this value is ignored.
	MaxFoldedConstantInBytes int64 `` /* 140-byte string literal not displayed */
	// If true, perform function inlining on the graph.
	DoFunctionInlining bool `protobuf:"varint,4,opt,name=do_function_inlining,json=doFunctionInlining,proto3" json:"do_function_inlining,omitempty"`
	// Overall optimization level. The actual optimizations applied will be the
	// logical OR of the flags that this level implies and any flags already set.
	OptLevel       OptimizerOptions_Level          `protobuf:"varint,3,opt,name=opt_level,json=optLevel,proto3,enum=tensorflow.OptimizerOptions_Level" json:"opt_level,omitempty"`
	GlobalJitLevel OptimizerOptions_GlobalJitLevel `` /* 154-byte string literal not displayed */
	// CPU code will be autoclustered only if global_jit_level >= ON_1 and either:
	//  - this flag is true, or
	//  - TF_XLA_FLAGS contains --tf_xla_cpu_global_jit=true.
	CpuGlobalJit bool `protobuf:"varint,7,opt,name=cpu_global_jit,json=cpuGlobalJit,proto3" json:"cpu_global_jit,omitempty"`
	// contains filtered or unexported fields
}

Options passed to the graph optimizer

func (*OptimizerOptions) Descriptor deprecated 

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

Deprecated: Use OptimizerOptions.ProtoReflect.Descriptor instead.

func (*OptimizerOptions) GetCpuGlobalJit 

func (x *OptimizerOptions) GetCpuGlobalJit() bool

func (*OptimizerOptions) GetDoCommonSubexpressionElimination 

func (x *OptimizerOptions) GetDoCommonSubexpressionElimination() bool

func (*OptimizerOptions) GetDoConstantFolding 

func (x *OptimizerOptions) GetDoConstantFolding() bool

func (*OptimizerOptions) GetDoFunctionInlining 

func (x *OptimizerOptions) GetDoFunctionInlining() bool

func (*OptimizerOptions) GetGlobalJitLevel 

func (x *OptimizerOptions) GetGlobalJitLevel() OptimizerOptions_GlobalJitLevel

func (*OptimizerOptions) GetMaxFoldedConstantInBytes 

func (x *OptimizerOptions) GetMaxFoldedConstantInBytes() int64

func (*OptimizerOptions) GetOptLevel 

func (x *OptimizerOptions) GetOptLevel() OptimizerOptions_Level

func (*OptimizerOptions) ProtoMessage 

func (*OptimizerOptions) ProtoMessage()

func (*OptimizerOptions) ProtoReflect 

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

func (*OptimizerOptions) Reset 

func (x *OptimizerOptions) Reset()

func (*OptimizerOptions) String 

func (x *OptimizerOptions) String() string

type OptimizerOptions_GlobalJitLevel 

type OptimizerOptions_GlobalJitLevel int32

Control the use of the compiler/jit. Experimental. 

const (
	OptimizerOptions_DEFAULT OptimizerOptions_GlobalJitLevel = 0 // Default setting ("off" now, but later expected to be "on")
	OptimizerOptions_OFF     OptimizerOptions_GlobalJitLevel = -1
	// The following settings turn on compilation, with higher values being
	// more aggressive.  Higher values may reduce opportunities for parallelism
	// and may use more memory.  (At present, there is no distinction, but this
	// is expected to change.)
	OptimizerOptions_ON_1 OptimizerOptions_GlobalJitLevel = 1
	OptimizerOptions_ON_2 OptimizerOptions_GlobalJitLevel = 2
)

func (OptimizerOptions_GlobalJitLevel) Descriptor 

func (OptimizerOptions_GlobalJitLevel) Descriptor() protoreflect.EnumDescriptor

func (OptimizerOptions_GlobalJitLevel) Enum 

func (x OptimizerOptions_GlobalJitLevel) Enum() *OptimizerOptions_GlobalJitLevel

func (OptimizerOptions_GlobalJitLevel) EnumDescriptor deprecated 

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

Deprecated: Use OptimizerOptions_GlobalJitLevel.Descriptor instead.

func (OptimizerOptions_GlobalJitLevel) Number 

func (x OptimizerOptions_GlobalJitLevel) Number() protoreflect.EnumNumber

func (OptimizerOptions_GlobalJitLevel) String 

func (x OptimizerOptions_GlobalJitLevel) String() string

func (OptimizerOptions_GlobalJitLevel) Type 

func (OptimizerOptions_GlobalJitLevel) Type() protoreflect.EnumType

type OptimizerOptions_Level 

type OptimizerOptions_Level int32

Optimization level 

const (
	// L1 is the default level.
	// Optimization performed at L1 :
	// 1. Common subexpression elimination
	// 2. Constant folding
	OptimizerOptions_L1 OptimizerOptions_Level = 0
	// No optimizations
	OptimizerOptions_L0 OptimizerOptions_Level = -1
)

func (OptimizerOptions_Level) Descriptor 

func (OptimizerOptions_Level) Descriptor() protoreflect.EnumDescriptor

func (OptimizerOptions_Level) Enum 

func (x OptimizerOptions_Level) Enum() *OptimizerOptions_Level

func (OptimizerOptions_Level) EnumDescriptor deprecated 

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

Deprecated: Use OptimizerOptions_Level.Descriptor instead.

func (OptimizerOptions_Level) Number 

func (x OptimizerOptions_Level) Number() protoreflect.EnumNumber

func (OptimizerOptions_Level) String 

func (x OptimizerOptions_Level) String() string

func (OptimizerOptions_Level) Type 

func (OptimizerOptions_Level) Type() protoreflect.EnumType

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

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

next: 8

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) GetSlack 

func (x *Options) GetSlack() 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 PairValue 

type PairValue struct {
	Key   string           `protobuf:"bytes,1,opt,name=key,proto3" json:"key,omitempty"`
	Value *StructuredValue `protobuf:"bytes,2,opt,name=value,proto3" json:"value,omitempty"`
	// contains filtered or unexported fields
}

Represents a (key, value) pair.

func (*PairValue) Descriptor deprecated 

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

Deprecated: Use PairValue.ProtoReflect.Descriptor instead.

func (*PairValue) GetKey 

func (x *PairValue) GetKey() string

func (*PairValue) GetValue 

func (x *PairValue) GetValue() *StructuredValue

func (*PairValue) ProtoMessage 

func (*PairValue) ProtoMessage()

func (*PairValue) ProtoReflect 

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

func (*PairValue) Reset 

func (x *PairValue) Reset()

func (*PairValue) String 

func (x *PairValue) String() string

type PartialRunSetupRequest 

type PartialRunSetupRequest struct {

	// REQUIRED: session_handle must be returned by a CreateSession call
	// to the same master service.
	SessionHandle string `protobuf:"bytes,1,opt,name=session_handle,json=sessionHandle,proto3" json:"session_handle,omitempty"`
	// Tensors to be fed in future steps.
	Feed []string `protobuf:"bytes,2,rep,name=feed,proto3" json:"feed,omitempty"`
	// Fetches. A list of tensor names. The caller expects a tensor to be returned
	// for each fetch[i] (see RunStepResponse.tensor), for corresponding partial
	// RunStepRequests. The order of specified fetches does not change the
	// execution order.
	Fetch []string `protobuf:"bytes,3,rep,name=fetch,proto3" json:"fetch,omitempty"`
	// Target Nodes. A list of node names. The named nodes will be run in future
	// steps, but their outputs will not be fetched.
	Target []string `protobuf:"bytes,4,rep,name=target,proto3" json:"target,omitempty"`
	// Unique identifier for this request. Every PartialRunSetupRequest must
	// have a unique request_id, and retried PartialRunSetupRequest must have
	// the same request_id. If request_id is zero, retry detection is disabled.
	RequestId int64 `protobuf:"varint,5,opt,name=request_id,json=requestId,proto3" json:"request_id,omitempty"`
	// contains filtered or unexported fields
}

func (*PartialRunSetupRequest) Descriptor deprecated 

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

Deprecated: Use PartialRunSetupRequest.ProtoReflect.Descriptor instead.

func (*PartialRunSetupRequest) GetFeed 

func (x *PartialRunSetupRequest) GetFeed() []string

func (*PartialRunSetupRequest) GetFetch 

func (x *PartialRunSetupRequest) GetFetch() []string

func (*PartialRunSetupRequest) GetRequestId 

func (x *PartialRunSetupRequest) GetRequestId() int64

func (*PartialRunSetupRequest) GetSessionHandle 

func (x *PartialRunSetupRequest) GetSessionHandle() string

func (*PartialRunSetupRequest) GetTarget 

func (x *PartialRunSetupRequest) GetTarget() []string

func (*PartialRunSetupRequest) ProtoMessage 

func (*PartialRunSetupRequest) ProtoMessage()

func (*PartialRunSetupRequest) ProtoReflect 

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

func (*PartialRunSetupRequest) Reset 

func (x *PartialRunSetupRequest) Reset()

func (*PartialRunSetupRequest) String 

func (x *PartialRunSetupRequest) String() string

type PartialRunSetupResponse 

type PartialRunSetupResponse struct {

	// The unique handle corresponding to the ongoing partial run call setup by
	// the invocation to PartialRunSetup. This handle may be passed to
	// RunStepRequest to send and receive tensors for this partial run.
	PartialRunHandle string `protobuf:"bytes,1,opt,name=partial_run_handle,json=partialRunHandle,proto3" json:"partial_run_handle,omitempty"`
	// contains filtered or unexported fields
}

func (*PartialRunSetupResponse) Descriptor deprecated 

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

Deprecated: Use PartialRunSetupResponse.ProtoReflect.Descriptor instead.

func (*PartialRunSetupResponse) GetPartialRunHandle 

func (x *PartialRunSetupResponse) GetPartialRunHandle() string

func (*PartialRunSetupResponse) ProtoMessage 

func (*PartialRunSetupResponse) ProtoMessage()

func (*PartialRunSetupResponse) ProtoReflect 

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

func (*PartialRunSetupResponse) Reset 

func (x *PartialRunSetupResponse) Reset()

func (*PartialRunSetupResponse) String 

func (x *PartialRunSetupResponse) String() string

type ProcessingModeDef 

type ProcessingModeDef struct {
	ShardingPolicy ProcessingModeDef_ShardingPolicy `` /* 158-byte string literal not displayed */
	// contains filtered or unexported fields
}

Next tag: 2

func (*ProcessingModeDef) Descriptor deprecated 

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

Deprecated: Use ProcessingModeDef.ProtoReflect.Descriptor instead.

func (*ProcessingModeDef) GetShardingPolicy 

func (x *ProcessingModeDef) GetShardingPolicy() ProcessingModeDef_ShardingPolicy

func (*ProcessingModeDef) ProtoMessage 

func (*ProcessingModeDef) ProtoMessage()

func (*ProcessingModeDef) ProtoReflect 

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

func (*ProcessingModeDef) Reset 

func (x *ProcessingModeDef) Reset()

func (*ProcessingModeDef) String 

func (x *ProcessingModeDef) String() string

type ProcessingModeDef_ShardingPolicy 

type ProcessingModeDef_ShardingPolicy int32

Specifies how data is sharded among tf.data service workers. 

const (
	// No sharding will be performed. Each worker produces the entire dataset
	// without any sharding. With this mode, the best practice is to shuffle the
	// dataset nondeterministically so that workers process the dataset in
	// different orders.
	ProcessingModeDef_OFF ProcessingModeDef_ShardingPolicy = 0
	// The input dataset is dynamically split among workers at runtime. Each
	// worker gets the next split when it reads data from the dispatcher. There
	// is no fixed sharding with this mode.
	ProcessingModeDef_DYNAMIC ProcessingModeDef_ShardingPolicy = 1
	// The following are static sharding policies. The semantics are similar to
	// `tf.data.experimental.AutoShardPolicy`. These policies require:
	// * The tf.data service cluster has a fixed size, and you need to specify
	//   the workers in DispatcherConfig.
	// * Each client only reads from the local tf.data service worker.
	//
	// Shards by input files (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.
	ProcessingModeDef_FILE ProcessingModeDef_ShardingPolicy = 2
	// 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.
	ProcessingModeDef_DATA ProcessingModeDef_ShardingPolicy = 3
	// Attempts FILE-based sharding, falling back to DATA-based sharding on
	// failures.
	ProcessingModeDef_FILE_OR_DATA ProcessingModeDef_ShardingPolicy = 4
	// Looks for the presence of `shard(SHARD_HINT, ...)` which is treated as a
	// placeholder to replace with `shard(num_workers, worker_index)`.
	ProcessingModeDef_HINT ProcessingModeDef_ShardingPolicy = 5
)

func (ProcessingModeDef_ShardingPolicy) Descriptor 

func (ProcessingModeDef_ShardingPolicy) Descriptor() protoreflect.EnumDescriptor

func (ProcessingModeDef_ShardingPolicy) Enum 

func (x ProcessingModeDef_ShardingPolicy) Enum() *ProcessingModeDef_ShardingPolicy

func (ProcessingModeDef_ShardingPolicy) EnumDescriptor deprecated 

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

Deprecated: Use ProcessingModeDef_ShardingPolicy.Descriptor instead.

func (ProcessingModeDef_ShardingPolicy) Number 

func (x ProcessingModeDef_ShardingPolicy) Number() protoreflect.EnumNumber

func (ProcessingModeDef_ShardingPolicy) String 

func (x ProcessingModeDef_ShardingPolicy) String() string

func (ProcessingModeDef_ShardingPolicy) Type 

func (ProcessingModeDef_ShardingPolicy) Type() protoreflect.EnumType

type QueueItem 

type QueueItem struct {

	// The remote executor should be able to handle either executing ops directly,
	// or releasing any unused tensor handles, since the tensor lifetime is
	// maintained by the client.
	//
	// Types that are assignable to Item:
	//	*QueueItem_HandleToDecref
	//	*QueueItem_Operation
	//	*QueueItem_SendTensor
	//	*QueueItem_RegisterFunction
	//	*QueueItem_CleanupFunction
	//	*QueueItem_SyncRemoteExecutorForStream
	//	*QueueItem_SendPackedHandle
	Item isQueueItem_Item `protobuf_oneof:"item"`
	// contains filtered or unexported fields
}

func (*QueueItem) Descriptor deprecated 

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

Deprecated: Use QueueItem.ProtoReflect.Descriptor instead.

func (*QueueItem) GetCleanupFunction 

func (x *QueueItem) GetCleanupFunction() *CleanupFunctionOp

func (*QueueItem) GetHandleToDecref 

func (x *QueueItem) GetHandleToDecref() *RemoteTensorHandle

func (*QueueItem) GetItem 

func (m *QueueItem) GetItem() isQueueItem_Item

func (*QueueItem) GetOperation 

func (x *QueueItem) GetOperation() *Operation

func (*QueueItem) GetRegisterFunction 

func (x *QueueItem) GetRegisterFunction() *RegisterFunctionOp

func (*QueueItem) GetSendPackedHandle 

func (x *QueueItem) GetSendPackedHandle() *SendPackedHandleOp

func (*QueueItem) GetSendTensor 

func (x *QueueItem) GetSendTensor() *SendTensorOp

func (*QueueItem) GetSyncRemoteExecutorForStream 

func (x *QueueItem) GetSyncRemoteExecutorForStream() *SyncRemoteExecutorForStream

func (*QueueItem) ProtoMessage 

func (*QueueItem) ProtoMessage()

func (*QueueItem) ProtoReflect 

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

func (*QueueItem) Reset 

func (x *QueueItem) Reset()

func (*QueueItem) String 

func (x *QueueItem) String() string

type QueueItem_CleanupFunction 

type QueueItem_CleanupFunction struct {
	CleanupFunction *CleanupFunctionOp `protobuf:"bytes,5,opt,name=cleanup_function,json=cleanupFunction,proto3,oneof"`
}

type QueueItem_HandleToDecref 

type QueueItem_HandleToDecref struct {
	HandleToDecref *RemoteTensorHandle `protobuf:"bytes,1,opt,name=handle_to_decref,json=handleToDecref,proto3,oneof"`
}

type QueueItem_Operation 

type QueueItem_Operation struct {
	Operation *Operation `protobuf:"bytes,2,opt,name=operation,proto3,oneof"`
}

type QueueItem_RegisterFunction 

type QueueItem_RegisterFunction struct {
	// Takes a FunctionDef and makes it enqueable on the remote worker.
	RegisterFunction *RegisterFunctionOp `protobuf:"bytes,4,opt,name=register_function,json=registerFunction,proto3,oneof"`
}

type QueueItem_SendPackedHandle 

type QueueItem_SendPackedHandle struct {
	SendPackedHandle *SendPackedHandleOp `protobuf:"bytes,7,opt,name=send_packed_handle,json=sendPackedHandle,proto3,oneof"`
}

type QueueItem_SendTensor 

type QueueItem_SendTensor struct {
	SendTensor *SendTensorOp `protobuf:"bytes,3,opt,name=send_tensor,json=sendTensor,proto3,oneof"`
}

type QueueItem_SyncRemoteExecutorForStream 

type QueueItem_SyncRemoteExecutorForStream struct {
	// A remote executor is created to execute ops/functions asynchronously
	// enqueued in streaming call. Request with this item type waits for pending
	// nodes to finish on the remote executor and report status.
	SyncRemoteExecutorForStream *SyncRemoteExecutorForStream `protobuf:"bytes,6,opt,name=sync_remote_executor_for_stream,json=syncRemoteExecutorForStream,proto3,oneof"`
}

type QueueResponse 

type QueueResponse struct {

	// `shape` and `tensor` cannot be set in the same response.
	// Shapes of output tensors for creating remote TensorHandles.
	Shape []*TensorShapeProto `protobuf:"bytes,1,rep,name=shape,proto3" json:"shape,omitempty"`
	// Optional. If set, represents the output devices of a function.
	Device []string `protobuf:"bytes,3,rep,name=device,proto3" json:"device,omitempty"`
	// Output tensors of a remote function. Set when Operation.id is invalid.
	Tensor []*TensorProto `protobuf:"bytes,2,rep,name=tensor,proto3" json:"tensor,omitempty"`
	// contains filtered or unexported fields
}

func (*QueueResponse) Descriptor deprecated 

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

Deprecated: Use QueueResponse.ProtoReflect.Descriptor instead.

func (*QueueResponse) GetDevice 

func (x *QueueResponse) GetDevice() []string

func (*QueueResponse) GetShape 

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

func (*QueueResponse) GetTensor 

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

func (*QueueResponse) ProtoMessage 

func (*QueueResponse) ProtoMessage()

func (*QueueResponse) ProtoReflect 

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

func (*QueueResponse) Reset 

func (x *QueueResponse) Reset()

func (*QueueResponse) String 

func (x *QueueResponse) String() string

type QueueRunnerDef 

type QueueRunnerDef struct {

	// Queue name.
	QueueName string `protobuf:"bytes,1,opt,name=queue_name,json=queueName,proto3" json:"queue_name,omitempty"`
	// A list of enqueue operations.
	EnqueueOpName []string `protobuf:"bytes,2,rep,name=enqueue_op_name,json=enqueueOpName,proto3" json:"enqueue_op_name,omitempty"`
	// The operation to run to close the queue.
	CloseOpName string `protobuf:"bytes,3,opt,name=close_op_name,json=closeOpName,proto3" json:"close_op_name,omitempty"`
	// The operation to run to cancel the queue.
	CancelOpName string `protobuf:"bytes,4,opt,name=cancel_op_name,json=cancelOpName,proto3" json:"cancel_op_name,omitempty"`
	// A list of exception types considered to signal a safely closed queue
	// if raised during enqueue operations.
	QueueClosedExceptionTypes []Code `` /* 175-byte string literal not displayed */
	// contains filtered or unexported fields
}

Protocol buffer representing a QueueRunner.

func (*QueueRunnerDef) Descriptor deprecated 

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

Deprecated: Use QueueRunnerDef.ProtoReflect.Descriptor instead.

func (*QueueRunnerDef) GetCancelOpName 

func (x *QueueRunnerDef) GetCancelOpName() string

func (*QueueRunnerDef) GetCloseOpName 

func (x *QueueRunnerDef) GetCloseOpName() string

func (*QueueRunnerDef) GetEnqueueOpName 

func (x *QueueRunnerDef) GetEnqueueOpName() []string

func (*QueueRunnerDef) GetQueueClosedExceptionTypes 

func (x *QueueRunnerDef) GetQueueClosedExceptionTypes() []Code

func (*QueueRunnerDef) GetQueueName 

func (x *QueueRunnerDef) GetQueueName() string

func (*QueueRunnerDef) ProtoMessage 

func (*QueueRunnerDef) ProtoMessage()

func (*QueueRunnerDef) ProtoReflect 

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

func (*QueueRunnerDef) Reset 

func (x *QueueRunnerDef) Reset()

func (*QueueRunnerDef) String 

func (x *QueueRunnerDef) String() string

type RPCOptions 

type RPCOptions struct {

	// If true, always use RPC to contact the session target.
	//
	// If false (the default option), TensorFlow may use an optimized
	// transport for client-master communication that avoids the RPC
	// stack. This option is primarily for used testing the RPC stack.
	UseRpcForInprocessMaster bool `` /* 140-byte string literal not displayed */
	// The compression algorithm to be used. One of "deflate", "gzip".
	CompressionAlgorithm string `protobuf:"bytes,2,opt,name=compression_algorithm,json=compressionAlgorithm,proto3" json:"compression_algorithm,omitempty"`
	// If compression_algorithm is set, the compression level to be used.
	// From 0 (no compression), up to 3.
	CompressionLevel int32 `protobuf:"varint,3,opt,name=compression_level,json=compressionLevel,proto3" json:"compression_level,omitempty"`
	// Setting cache_rpc_response to true will enable sender side caching of
	// response for RecvTensorAsync and RecvBufAsync to allow receiver to retry
	// requests . This is only necessary when the network fabric is experiencing a
	// significant error rate.  Without it we'll fail a step on an network error,
	// while with it we'll be able to complete long steps (like complex
	// initializations) in the face of some network errors during RecvTensor.
	CacheRpcResponse bool `protobuf:"varint,4,opt,name=cache_rpc_response,json=cacheRpcResponse,proto3" json:"cache_rpc_response,omitempty"`
	// Disables TCP connection sharing when opening a new RPC channel.
	DisableSessionConnectionSharing bool `` /* 159-byte string literal not displayed */
	// Setting num_channels_per_target > 0 allows uses of multiple channels to
	// communicate to the same target. This can be used to improve the aggregate
	// throughput on high speed links (e.g 100G) where single connection is not
	// sufficient to maximize link utilization. Note that a single RPC only goes
	// on a single channel, this only helps in situations where there are multiple
	// transfers to the same target overlapping in time.
	NumChannelsPerTarget int32 `` /* 126-byte string literal not displayed */
	// contains filtered or unexported fields
}

func (*RPCOptions) Descriptor deprecated 

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

Deprecated: Use RPCOptions.ProtoReflect.Descriptor instead.

func (*RPCOptions) GetCacheRpcResponse 

func (x *RPCOptions) GetCacheRpcResponse() bool

func (*RPCOptions) GetCompressionAlgorithm 

func (x *RPCOptions) GetCompressionAlgorithm() string

func (*RPCOptions) GetCompressionLevel 

func (x *RPCOptions) GetCompressionLevel() int32

func (*RPCOptions) GetDisableSessionConnectionSharing 

func (x *RPCOptions) GetDisableSessionConnectionSharing() bool

func (*RPCOptions) GetNumChannelsPerTarget 

func (x *RPCOptions) GetNumChannelsPerTarget() int32

func (*RPCOptions) GetUseRpcForInprocessMaster 

func (x *RPCOptions) GetUseRpcForInprocessMaster() bool

func (*RPCOptions) ProtoMessage 

func (*RPCOptions) ProtoMessage()

func (*RPCOptions) ProtoReflect 

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

func (*RPCOptions) Reset 

func (x *RPCOptions) Reset()

func (*RPCOptions) String 

func (x *RPCOptions) String() string

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 RecvBufRequest 

type RecvBufRequest struct {

	// Used at server side to find the correct BufRendezvous.
	StepId int64 `protobuf:"varint,1,opt,name=step_id,json=stepId,proto3" json:"step_id,omitempty"`
	// Arbitrary string identifying a BufRendezvous entry.
	BufRendezvousKey string `protobuf:"bytes,2,opt,name=buf_rendezvous_key,json=bufRendezvousKey,proto3" json:"buf_rendezvous_key,omitempty"`
	// Size of value expected, must agree with BufRendezvous entry.
	NumBytes int64 `protobuf:"varint,3,opt,name=num_bytes,json=numBytes,proto3" json:"num_bytes,omitempty"`
	// When RDMA is in use, address of destination field on client.
	BufPtr uint64 `protobuf:"fixed64,4,opt,name=buf_ptr,json=bufPtr,proto3" json:"buf_ptr,omitempty"`
	// Optional information on client-side device locality.
	ClientLocality *DeviceLocality `protobuf:"bytes,5,opt,name=client_locality,json=clientLocality,proto3" json:"client_locality,omitempty"`
	// Optional information on server-side device locality.
	ServerLocality *DeviceLocality `protobuf:"bytes,6,opt,name=server_locality,json=serverLocality,proto3" json:"server_locality,omitempty"`
	// Optional, implementation-specific data.
	TransportOptions *any.Any `protobuf:"bytes,7,opt,name=transport_options,json=transportOptions,proto3" json:"transport_options,omitempty"`
	// For annotating timeline and device incarnation check.
	SrcDevice string `protobuf:"bytes,8,opt,name=src_device,json=srcDevice,proto3" json:"src_device,omitempty"`
	// Optional, for annotating the timeline.
	DstDevice string `protobuf:"bytes,9,opt,name=dst_device,json=dstDevice,proto3" json:"dst_device,omitempty"`
	// Depending on the RPC system in use, it may be necessary to set this
	// id to detect resends of RPCs where the server is not aware that
	// the prior RPC failed.
	RequestId int64 `protobuf:"varint,10,opt,name=request_id,json=requestId,proto3" json:"request_id,omitempty"`
	// Incarnation number of the source device, used to detect worker failures.
	SrcIncarnation uint64 `protobuf:"varint,11,opt,name=src_incarnation,json=srcIncarnation,proto3" json:"src_incarnation,omitempty"`
	// contains filtered or unexported fields
}

func (*RecvBufRequest) Descriptor deprecated 

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

Deprecated: Use RecvBufRequest.ProtoReflect.Descriptor instead.

func (*RecvBufRequest) GetBufPtr 

func (x *RecvBufRequest) GetBufPtr() uint64

func (*RecvBufRequest) GetBufRendezvousKey 

func (x *RecvBufRequest) GetBufRendezvousKey() string

func (*RecvBufRequest) GetClientLocality 

func (x *RecvBufRequest) GetClientLocality() *DeviceLocality

func (*RecvBufRequest) GetDstDevice 

func (x *RecvBufRequest) GetDstDevice() string

func (*RecvBufRequest) GetNumBytes 

func (x *RecvBufRequest) GetNumBytes() int64

func (*RecvBufRequest) GetRequestId 

func (x *RecvBufRequest) GetRequestId() int64

func (*RecvBufRequest) GetServerLocality 

func (x *RecvBufRequest) GetServerLocality() *DeviceLocality

func (*RecvBufRequest) GetSrcDevice 

func (x *RecvBufRequest) GetSrcDevice() string

func (*RecvBufRequest) GetSrcIncarnation 

func (x *RecvBufRequest) GetSrcIncarnation() uint64

func (*RecvBufRequest) GetStepId 

func (x *RecvBufRequest) GetStepId() int64

func (*RecvBufRequest) GetTransportOptions 

func (x *RecvBufRequest) GetTransportOptions() *any.Any

func (*RecvBufRequest) ProtoMessage 

func (*RecvBufRequest) ProtoMessage()

func (*RecvBufRequest) ProtoReflect 

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

func (*RecvBufRequest) Reset 

func (x *RecvBufRequest) Reset()

func (*RecvBufRequest) String 

func (x *RecvBufRequest) String() string

type RecvBufRespExtra 

type RecvBufRespExtra struct {
	TensorContent [][]byte `protobuf:"bytes,1,rep,name=tensor_content,json=tensorContent,proto3" json:"tensor_content,omitempty"`
	// contains filtered or unexported fields
}

Extra data needed on a non-RDMA RecvBufResponse.

func (*RecvBufRespExtra) Descriptor deprecated 

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

Deprecated: Use RecvBufRespExtra.ProtoReflect.Descriptor instead.

func (*RecvBufRespExtra) GetTensorContent 

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

func (*RecvBufRespExtra) ProtoMessage 

func (*RecvBufRespExtra) ProtoMessage()

func (*RecvBufRespExtra) ProtoReflect 

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

func (*RecvBufRespExtra) Reset 

func (x *RecvBufRespExtra) Reset()

func (*RecvBufRespExtra) String 

func (x *RecvBufRespExtra) String() string

type RecvBufResponse 

type RecvBufResponse struct {
	BufPtr   uint64 `protobuf:"fixed64,1,opt,name=buf_ptr,json=bufPtr,proto3" json:"buf_ptr,omitempty"`      // Address of source field on server.
	NumBytes int64  `protobuf:"varint,2,opt,name=num_bytes,json=numBytes,proto3" json:"num_bytes,omitempty"` // Byte length of buf_ptr field, if set.
	IsDead   bool   `protobuf:"varint,3,opt,name=is_dead,json=isDead,proto3" json:"is_dead,omitempty"`       // True if value is 'dead' like a tensor.
	// Optional, implementation-specific data.
	TransportOptions *any.Any `protobuf:"bytes,4,opt,name=transport_options,json=transportOptions,proto3" json:"transport_options,omitempty"`
	// Optional, for timeline.
	SendStartMicros int64 `protobuf:"varint,5,opt,name=send_start_micros,json=sendStartMicros,proto3" json:"send_start_micros,omitempty"`
	// Whether the receiver should send a MarkRecvFinishedRequest to the sender
	// to ack the message.
	RequireAck bool `protobuf:"varint,6,opt,name=require_ack,json=requireAck,proto3" json:"require_ack,omitempty"`
	// contains filtered or unexported fields
}

func (*RecvBufResponse) Descriptor deprecated 

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

Deprecated: Use RecvBufResponse.ProtoReflect.Descriptor instead.

func (*RecvBufResponse) GetBufPtr 

func (x *RecvBufResponse) GetBufPtr() uint64

func (*RecvBufResponse) GetIsDead 

func (x *RecvBufResponse) GetIsDead() bool

func (*RecvBufResponse) GetNumBytes 

func (x *RecvBufResponse) GetNumBytes() int64

func (*RecvBufResponse) GetRequireAck 

func (x *RecvBufResponse) GetRequireAck() bool

func (*RecvBufResponse) GetSendStartMicros 

func (x *RecvBufResponse) GetSendStartMicros() int64

func (*RecvBufResponse) GetTransportOptions 

func (x *RecvBufResponse) GetTransportOptions() *any.Any

func (*RecvBufResponse) ProtoMessage 

func (*RecvBufResponse) ProtoMessage()

func (*RecvBufResponse) ProtoReflect 

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

func (*RecvBufResponse) Reset 

func (x *RecvBufResponse) Reset()

func (*RecvBufResponse) String 

func (x *RecvBufResponse) String() string

type RecvTensorRequest 

type RecvTensorRequest struct {

	// The step in which the tensor will be produced.
	//
	// REQUIRED: This must eventually correspond to the `step_id` passed
	// into a RunGraph call on the same WorkerService.
	StepId int64 `protobuf:"varint,1,opt,name=step_id,json=stepId,proto3" json:"step_id,omitempty"`
	// A key identifying the channel to receive tensors from. A RecvTensor request
	// retrieves one tensor from the channel, but multiple tensors can be sent and
	// received over the same channel with multiple RecvTensor requests. See
	// rendezvous.h for details.
	RendezvousKey string `protobuf:"bytes,2,opt,name=rendezvous_key,json=rendezvousKey,proto3" json:"rendezvous_key,omitempty"`
	// If true, use an out-of-band DMA mechanism to transfer the
	// received tensor.
	DmaOk bool `protobuf:"varint,3,opt,name=dma_ok,json=dmaOk,proto3" json:"dma_ok,omitempty"`
	// Optional information on client-side device locality.
	ClientLocality *DeviceLocality `protobuf:"bytes,4,opt,name=client_locality,json=clientLocality,proto3" json:"client_locality,omitempty"`
	// Optional information on server-side device locality.
	ServerLocality *DeviceLocality `protobuf:"bytes,5,opt,name=server_locality,json=serverLocality,proto3" json:"server_locality,omitempty"`
	// Optional information needed by the RPC subsystem.
	TransportOptions *any.Any `protobuf:"bytes,6,opt,name=transport_options,json=transportOptions,proto3" json:"transport_options,omitempty"`
	// Unique identifier for this request. Every RecvTensorRequest must have a
	// unique request_id, and retried RecvTensorRequests must have the same
	// request_id. If request_id is zero, retry detection and response cache
	// are disabled.
	//
	// Retried RecvTensorRequests are problematic because a RecvTensor with no
	// corresponding sender will wait forever, and the tensor may have been
	// delivered to a previous retry. Workers use request_ids to reject retried
	// RecvTensor requests instead of waiting forever.
	RequestId int64 `protobuf:"varint,7,opt,name=request_id,json=requestId,proto3" json:"request_id,omitempty"`
	// contains filtered or unexported fields
}

func (*RecvTensorRequest) Descriptor deprecated 

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

Deprecated: Use RecvTensorRequest.ProtoReflect.Descriptor instead.

func (*RecvTensorRequest) GetClientLocality 

func (x *RecvTensorRequest) GetClientLocality() *DeviceLocality

func (*RecvTensorRequest) GetDmaOk 

func (x *RecvTensorRequest) GetDmaOk() bool

func (*RecvTensorRequest) GetRendezvousKey 

func (x *RecvTensorRequest) GetRendezvousKey() string

func (*RecvTensorRequest) GetRequestId 

func (x *RecvTensorRequest) GetRequestId() int64

func (*RecvTensorRequest) GetServerLocality 

func (x *RecvTensorRequest) GetServerLocality() *DeviceLocality

func (*RecvTensorRequest) GetStepId 

func (x *RecvTensorRequest) GetStepId() int64

func (*RecvTensorRequest) GetTransportOptions 

func (x *RecvTensorRequest) GetTransportOptions() *any.Any

func (*RecvTensorRequest) ProtoMessage 

func (*RecvTensorRequest) ProtoMessage()

func (*RecvTensorRequest) ProtoReflect 

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

func (*RecvTensorRequest) Reset 

func (x *RecvTensorRequest) Reset()

func (*RecvTensorRequest) String 

func (x *RecvTensorRequest) String() string

type RecvTensorResponse 

type RecvTensorResponse struct {

	// The tensor as a proto.
	Tensor *TensorProto `protobuf:"bytes,1,opt,name=tensor,proto3" json:"tensor,omitempty"`
	// If true, this tensor was the output of a dead node, and the
	// content is invalid.
	IsDead bool `protobuf:"varint,2,opt,name=is_dead,json=isDead,proto3" json:"is_dead,omitempty"`
	// The time at which tensor was available and started to be returned.
	SendStartMicros int64 `protobuf:"varint,3,opt,name=send_start_micros,json=sendStartMicros,proto3" json:"send_start_micros,omitempty"`
	// Optional additional information about how to receive the tensor,
	// e.g. in the event that `RecvTensorRequest.dma_ok` was true.
	TransportOptions *any.Any `protobuf:"bytes,4,opt,name=transport_options,json=transportOptions,proto3" json:"transport_options,omitempty"`
	// Whether the receiver should send a MarkRecvFinishedRequest to the sender
	// to ack the message.
	RequireAck bool `protobuf:"varint,5,opt,name=require_ack,json=requireAck,proto3" json:"require_ack,omitempty"`
	// contains filtered or unexported fields
}

func (*RecvTensorResponse) Descriptor deprecated 

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

Deprecated: Use RecvTensorResponse.ProtoReflect.Descriptor instead.

func (*RecvTensorResponse) GetIsDead 

func (x *RecvTensorResponse) GetIsDead() bool

func (*RecvTensorResponse) GetRequireAck 

func (x *RecvTensorResponse) GetRequireAck() bool

func (*RecvTensorResponse) GetSendStartMicros 

func (x *RecvTensorResponse) GetSendStartMicros() int64

func (*RecvTensorResponse) GetTensor 

func (x *RecvTensorResponse) GetTensor() *TensorProto

func (*RecvTensorResponse) GetTransportOptions 

func (x *RecvTensorResponse) GetTransportOptions() *any.Any

func (*RecvTensorResponse) ProtoMessage 

func (*RecvTensorResponse) ProtoMessage()

func (*RecvTensorResponse) ProtoReflect 

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

func (*RecvTensorResponse) Reset 

func (x *RecvTensorResponse) Reset()

func (*RecvTensorResponse) String 

func (x *RecvTensorResponse) String() string

type RegisterFunctionOp 

type RegisterFunctionOp struct {
	FunctionDef *FunctionDef `protobuf:"bytes,1,opt,name=function_def,json=functionDef,proto3" json:"function_def,omitempty"`
	// If true, it means that function_def is produced by graph partition during
	// multi-device function instantiation.
	IsComponentFunction bool `protobuf:"varint,2,opt,name=is_component_function,json=isComponentFunction,proto3" json:"is_component_function,omitempty"`
	// All necessary FunctionDefs and GradientDefs to expand `function_def`.
	// When is_component_function is true, `function_def` could be a nested
	// function, since some nodes in its parent's function body could be
	// replaced with a new function by the graph optimization passes. No need to
	// add FunctionDefs here to the function cache in EagerContext since they
	// won't be executed as KernelAndDevices.
	Library *FunctionDefLibrary `protobuf:"bytes,3,opt,name=library,proto3" json:"library,omitempty"`
	// contains filtered or unexported fields
}

func (*RegisterFunctionOp) Descriptor deprecated 

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

Deprecated: Use RegisterFunctionOp.ProtoReflect.Descriptor instead.

func (*RegisterFunctionOp) GetFunctionDef 

func (x *RegisterFunctionOp) GetFunctionDef() *FunctionDef

func (*RegisterFunctionOp) GetIsComponentFunction 

func (x *RegisterFunctionOp) GetIsComponentFunction() bool

func (*RegisterFunctionOp) GetLibrary 

func (x *RegisterFunctionOp) GetLibrary() *FunctionDefLibrary

func (*RegisterFunctionOp) ProtoMessage 

func (*RegisterFunctionOp) ProtoMessage()

func (*RegisterFunctionOp) ProtoReflect 

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

func (*RegisterFunctionOp) Reset 

func (x *RegisterFunctionOp) Reset()

func (*RegisterFunctionOp) String 

func (x *RegisterFunctionOp) String() string

type RegisterGraphRequest 

type RegisterGraphRequest struct {

	// Subgraphs are scoped within one session.
	SessionHandle string `protobuf:"bytes,1,opt,name=session_handle,json=sessionHandle,proto3" json:"session_handle,omitempty"`
	// Set to true if `CreateWorkerSession` was called for `session_handle`.
	CreateWorkerSessionCalled bool `` /* 141-byte string literal not displayed */
	// "graph_def" has the subgraph of nodes for this worker, with each node
	// having its device_name filled in.
	GraphDef *GraphDef `protobuf:"bytes,2,opt,name=graph_def,json=graphDef,proto3" json:"graph_def,omitempty"`
	// True iff the graph (before partitioning) contains control flow nodes.
	//
	// As of 01/11/2015, this is no longer set by clients.
	//
	// Deprecated: Do not use.
	HasControlFlow bool `protobuf:"varint,3,opt,name=has_control_flow,json=hasControlFlow,proto3" json:"has_control_flow,omitempty"`
	// Configuration options for the session in which this graph was created.
	GraphOptions *GraphOptions `protobuf:"bytes,4,opt,name=graph_options,json=graphOptions,proto3" json:"graph_options,omitempty"`
	// Field(s) used by TensorFlow Debugger (tfdbg).
	DebugOptions *DebugOptions `protobuf:"bytes,5,opt,name=debug_options,json=debugOptions,proto3" json:"debug_options,omitempty"`
	// If graph_def contains any collective ops this must be a positive
	// integer used to coordinate execution with other graphs.  All
	// graphs in a distributed execution with the same
	// collective_graph_key will coordinate to use the same step_id
	// concurrently so that BufRendezvous entries will make the correct
	// values accessible.
	CollectiveGraphKey int64 `protobuf:"varint,7,opt,name=collective_graph_key,json=collectiveGraphKey,proto3" json:"collective_graph_key,omitempty"`
	// ConfigProto from the session in which this graph was created.
	// Contains additional parameters beyond graph_options, including
	// the name of the requested executor.
	ConfigProto *ConfigProto `protobuf:"bytes,8,opt,name=config_proto,json=configProto,proto3" json:"config_proto,omitempty"`
	// contains filtered or unexported fields
}

func (*RegisterGraphRequest) Descriptor deprecated 

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

Deprecated: Use RegisterGraphRequest.ProtoReflect.Descriptor instead.

func (*RegisterGraphRequest) GetCollectiveGraphKey 

func (x *RegisterGraphRequest) GetCollectiveGraphKey() int64

func (*RegisterGraphRequest) GetConfigProto 

func (x *RegisterGraphRequest) GetConfigProto() *ConfigProto

func (*RegisterGraphRequest) GetCreateWorkerSessionCalled 

func (x *RegisterGraphRequest) GetCreateWorkerSessionCalled() bool

func (*RegisterGraphRequest) GetDebugOptions 

func (x *RegisterGraphRequest) GetDebugOptions() *DebugOptions

func (*RegisterGraphRequest) GetGraphDef 

func (x *RegisterGraphRequest) GetGraphDef() *GraphDef

func (*RegisterGraphRequest) GetGraphOptions 

func (x *RegisterGraphRequest) GetGraphOptions() *GraphOptions

func (*RegisterGraphRequest) GetHasControlFlow deprecated 

func (x *RegisterGraphRequest) GetHasControlFlow() bool

Deprecated: Do not use.

func (*RegisterGraphRequest) GetSessionHandle 

func (x *RegisterGraphRequest) GetSessionHandle() string

func (*RegisterGraphRequest) ProtoMessage 

func (*RegisterGraphRequest) ProtoMessage()

func (*RegisterGraphRequest) ProtoReflect 

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

func (*RegisterGraphRequest) Reset 

func (x *RegisterGraphRequest) Reset()

func (*RegisterGraphRequest) String 

func (x *RegisterGraphRequest) String() string

type RegisterGraphResponse 

type RegisterGraphResponse struct {

	// If the registration succeeds, returns an opaque graph_handle to
	// the master. The master calls RunGraph with graph_handle to
	// compute different steps.
	GraphHandle string `protobuf:"bytes,1,opt,name=graph_handle,json=graphHandle,proto3" json:"graph_handle,omitempty"`
	// contains filtered or unexported fields
}

func (*RegisterGraphResponse) Descriptor deprecated 

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

Deprecated: Use RegisterGraphResponse.ProtoReflect.Descriptor instead.

func (*RegisterGraphResponse) GetGraphHandle 

func (x *RegisterGraphResponse) GetGraphHandle() string

func (*RegisterGraphResponse) ProtoMessage 

func (*RegisterGraphResponse) ProtoMessage()

func (*RegisterGraphResponse) ProtoReflect 

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

func (*RegisterGraphResponse) Reset 

func (x *RegisterGraphResponse) Reset()

func (*RegisterGraphResponse) String 

func (x *RegisterGraphResponse) String() string

type RegisterTaskRequest 

type RegisterTaskRequest struct {
	Incarnation uint64           `protobuf:"fixed64,3,opt,name=incarnation,proto3" json:"incarnation,omitempty"`
	SourceTask  *CoordinatedTask `protobuf:"bytes,5,opt,name=source_task,json=sourceTask,proto3" json:"source_task,omitempty"`
	// contains filtered or unexported fields
}

Request and response messages for registering a task to the cluster leader. A task is uniquely represented by its `job_name`, `task_id` and `incarnation`. Leader responds with its `incarnation` to identify a leader process.

func (*RegisterTaskRequest) Descriptor deprecated 

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

Deprecated: Use RegisterTaskRequest.ProtoReflect.Descriptor instead.

func (*RegisterTaskRequest) GetIncarnation 

func (x *RegisterTaskRequest) GetIncarnation() uint64

func (*RegisterTaskRequest) GetSourceTask 

func (x *RegisterTaskRequest) GetSourceTask() *CoordinatedTask

func (*RegisterTaskRequest) ProtoMessage 

func (*RegisterTaskRequest) ProtoMessage()

func (*RegisterTaskRequest) ProtoReflect 

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

func (*RegisterTaskRequest) Reset 

func (x *RegisterTaskRequest) Reset()

func (*RegisterTaskRequest) String 

func (x *RegisterTaskRequest) String() string

type RegisterTaskResponse 

type RegisterTaskResponse struct {
	LeaderIncarnation uint64 `protobuf:"fixed64,1,opt,name=leader_incarnation,json=leaderIncarnation,proto3" json:"leader_incarnation,omitempty"`
	// contains filtered or unexported fields
}

func (*RegisterTaskResponse) Descriptor deprecated 

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

Deprecated: Use RegisterTaskResponse.ProtoReflect.Descriptor instead.

func (*RegisterTaskResponse) GetLeaderIncarnation 

func (x *RegisterTaskResponse) GetLeaderIncarnation() uint64

func (*RegisterTaskResponse) ProtoMessage 

func (*RegisterTaskResponse) ProtoMessage()

func (*RegisterTaskResponse) ProtoReflect 

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

func (*RegisterTaskResponse) Reset 

func (x *RegisterTaskResponse) Reset()

func (*RegisterTaskResponse) String 

func (x *RegisterTaskResponse) 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 RegisteredSaver 

type RegisteredSaver struct {

	// The name of the registered saver/restore function.
	Name string `protobuf:"bytes,1,opt,name=name,proto3" json:"name,omitempty"`
	// Unique auto-generated name of the object.
	ObjectName string `protobuf:"bytes,2,opt,name=object_name,json=objectName,proto3" json:"object_name,omitempty"`
	// contains filtered or unexported fields
}

func (*RegisteredSaver) Descriptor deprecated 

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

Deprecated: Use RegisteredSaver.ProtoReflect.Descriptor instead.

func (*RegisteredSaver) GetName 

func (x *RegisteredSaver) GetName() string

func (*RegisteredSaver) GetObjectName 

func (x *RegisteredSaver) GetObjectName() string

func (*RegisteredSaver) ProtoMessage 

func (*RegisteredSaver) ProtoMessage()

func (*RegisteredSaver) ProtoReflect 

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

func (*RegisteredSaver) Reset 

func (x *RegisteredSaver) Reset()

func (*RegisteredSaver) String 

func (x *RegisteredSaver) String() string

type ReleaseCallableRequest 

type ReleaseCallableRequest struct {

	// REQUIRED: session_handle must be returned by a CreateSession call
	// to the same master service.
	SessionHandle string `protobuf:"bytes,1,opt,name=session_handle,json=sessionHandle,proto3" json:"session_handle,omitempty"`
	// REQUIRED: handle must be returned by a MakeCallable call to the same
	// master service.
	Handle int64 `protobuf:"varint,2,opt,name=handle,proto3" json:"handle,omitempty"`
	// contains filtered or unexported fields
}

func (*ReleaseCallableRequest) Descriptor deprecated 

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

Deprecated: Use ReleaseCallableRequest.ProtoReflect.Descriptor instead.

func (*ReleaseCallableRequest) GetHandle 

func (x *ReleaseCallableRequest) GetHandle() int64

func (*ReleaseCallableRequest) GetSessionHandle 

func (x *ReleaseCallableRequest) GetSessionHandle() string

func (*ReleaseCallableRequest) ProtoMessage 

func (*ReleaseCallableRequest) ProtoMessage()

func (*ReleaseCallableRequest) ProtoReflect 

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

func (*ReleaseCallableRequest) Reset 

func (x *ReleaseCallableRequest) Reset()

func (*ReleaseCallableRequest) String 

func (x *ReleaseCallableRequest) String() string

type ReleaseCallableResponse 

type ReleaseCallableResponse struct {
	// contains filtered or unexported fields
}

func (*ReleaseCallableResponse) Descriptor deprecated 

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

Deprecated: Use ReleaseCallableResponse.ProtoReflect.Descriptor instead.

func (*ReleaseCallableResponse) ProtoMessage 

func (*ReleaseCallableResponse) ProtoMessage()

func (*ReleaseCallableResponse) ProtoReflect 

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

func (*ReleaseCallableResponse) Reset 

func (x *ReleaseCallableResponse) Reset()

func (*ReleaseCallableResponse) String 

func (x *ReleaseCallableResponse) String() string

type RemoteTensorHandle 

type RemoteTensorHandle struct {

	// The ID of the operation that produced this tensor.
	OpId int64 `protobuf:"varint,1,opt,name=op_id,json=opId,proto3" json:"op_id,omitempty"`
	// The index into the outputs of the operation that produced this tensor.
	OutputNum int32 `protobuf:"varint,2,opt,name=output_num,json=outputNum,proto3" json:"output_num,omitempty"`
	// Device where the tensor is located. Cannot be empty.
	// For multi-device functions, it's the default device passed to placer.
	Device string `protobuf:"bytes,3,opt,name=device,proto3" json:"device,omitempty"`
	// Device of the operation producing this tensor. Can be empty if the
	// operation producing this tensor is a multi-device function.
	OpDevice string `protobuf:"bytes,4,opt,name=op_device,json=opDevice,proto3" json:"op_device,omitempty"`
	// Tensor type.
	Dtype DataType `protobuf:"varint,5,opt,name=dtype,proto3,enum=tensorflow.DataType" json:"dtype,omitempty"`
	// Optional data types and shapes of a remote resource variable.
	ResourceDtypesAndShapes []*ResourceDtypeAndShape `` /* 134-byte string literal not displayed */
	// contains filtered or unexported fields
}

func (*RemoteTensorHandle) Descriptor deprecated 

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

Deprecated: Use RemoteTensorHandle.ProtoReflect.Descriptor instead.

func (*RemoteTensorHandle) GetDevice 

func (x *RemoteTensorHandle) GetDevice() string

func (*RemoteTensorHandle) GetDtype 

func (x *RemoteTensorHandle) GetDtype() DataType

func (*RemoteTensorHandle) GetOpDevice 

func (x *RemoteTensorHandle) GetOpDevice() string

func (*RemoteTensorHandle) GetOpId 

func (x *RemoteTensorHandle) GetOpId() int64

func (*RemoteTensorHandle) GetOutputNum 

func (x *RemoteTensorHandle) GetOutputNum() int32

func (*RemoteTensorHandle) GetResourceDtypesAndShapes 

func (x *RemoteTensorHandle) GetResourceDtypesAndShapes() []*ResourceDtypeAndShape

func (*RemoteTensorHandle) ProtoMessage 

func (*RemoteTensorHandle) ProtoMessage()

func (*RemoteTensorHandle) ProtoReflect 

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

func (*RemoteTensorHandle) Reset 

func (x *RemoteTensorHandle) Reset()

func (*RemoteTensorHandle) String 

func (x *RemoteTensorHandle) String() string

type ReplayOp 

type ReplayOp struct {
	StartTimeUs float64 `protobuf:"fixed64,31,opt,name=start_time_us,json=startTimeUs,proto3" json:"start_time_us,omitempty"`
	EndTimeUs   float64 `protobuf:"fixed64,32,opt,name=end_time_us,json=endTimeUs,proto3" json:"end_time_us,omitempty"`
	// Types that are assignable to Op:
	//	*ReplayOp_CreateSession
	//	*ReplayOp_ExtendSession
	//	*ReplayOp_PartialRunSetup
	//	*ReplayOp_RunStep
	//	*ReplayOp_CloseSession
	//	*ReplayOp_ListDevices
	//	*ReplayOp_ResetRequest
	//	*ReplayOp_MakeCallable
	//	*ReplayOp_RunCallable
	//	*ReplayOp_ReleaseCallable
	//	*ReplayOp_NewReplaySession
	Op isReplayOp_Op `protobuf_oneof:"op"`
	// Types that are assignable to Response:
	//	*ReplayOp_CreateSessionResponse
	//	*ReplayOp_ExtendSessionResponse
	//	*ReplayOp_PartialRunSetupResponse
	//	*ReplayOp_RunStepResponse
	//	*ReplayOp_CloseSessionResponse
	//	*ReplayOp_ListDevicesResponse
	//	*ReplayOp_ResetRequestResponse
	//	*ReplayOp_MakeCallableResponse
	//	*ReplayOp_RunCallableResponse
	//	*ReplayOp_ReleaseCallableResponse
	Response isReplayOp_Response `protobuf_oneof:"response"`
	// contains filtered or unexported fields
}

func (*ReplayOp) Descriptor deprecated 

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

Deprecated: Use ReplayOp.ProtoReflect.Descriptor instead.

func (*ReplayOp) GetCloseSession 

func (x *ReplayOp) GetCloseSession() *CloseSessionRequest

func (*ReplayOp) GetCloseSessionResponse 

func (x *ReplayOp) GetCloseSessionResponse() *CloseSessionResponse

func (*ReplayOp) GetCreateSession 

func (x *ReplayOp) GetCreateSession() *CreateSessionRequest

func (*ReplayOp) GetCreateSessionResponse 

func (x *ReplayOp) GetCreateSessionResponse() *CreateSessionResponse

func (*ReplayOp) GetEndTimeUs 

func (x *ReplayOp) GetEndTimeUs() float64

func (*ReplayOp) GetExtendSession 

func (x *ReplayOp) GetExtendSession() *ExtendSessionRequest

func (*ReplayOp) GetExtendSessionResponse 

func (x *ReplayOp) GetExtendSessionResponse() *ExtendSessionResponse

func (*ReplayOp) GetListDevices 

func (x *ReplayOp) GetListDevices() *ListDevicesRequest

func (*ReplayOp) GetListDevicesResponse 

func (x *ReplayOp) GetListDevicesResponse() *ListDevicesResponse

func (*ReplayOp) GetMakeCallable 

func (x *ReplayOp) GetMakeCallable() *MakeCallableRequest

func (*ReplayOp) GetMakeCallableResponse 

func (x *ReplayOp) GetMakeCallableResponse() *MakeCallableResponse

func (*ReplayOp) GetNewReplaySession 

func (x *ReplayOp) GetNewReplaySession() *NewReplaySession

func (*ReplayOp) GetOp 

func (m *ReplayOp) GetOp() isReplayOp_Op

func (*ReplayOp) GetPartialRunSetup 

func (x *ReplayOp) GetPartialRunSetup() *PartialRunSetupRequest

func (*ReplayOp) GetPartialRunSetupResponse 

func (x *ReplayOp) GetPartialRunSetupResponse() *PartialRunSetupResponse

func (*ReplayOp) GetReleaseCallable 

func (x *ReplayOp) GetReleaseCallable() *ReleaseCallableRequest

func (*ReplayOp) GetReleaseCallableResponse 

func (x *ReplayOp) GetReleaseCallableResponse() *ReleaseCallableResponse

func (*ReplayOp) GetResetRequest 

func (x *ReplayOp) GetResetRequest() *ResetRequest

func (*ReplayOp) GetResetRequestResponse 

func (x *ReplayOp) GetResetRequestResponse() *ResetResponse

func (*ReplayOp) GetResponse 

func (m *ReplayOp) GetResponse() isReplayOp_Response

func (*ReplayOp) GetRunCallable 

func (x *ReplayOp) GetRunCallable() *RunCallableRequest

func (*ReplayOp) GetRunCallableResponse 

func (x *ReplayOp) GetRunCallableResponse() *RunCallableResponse

func (*ReplayOp) GetRunStep 

func (x *ReplayOp) GetRunStep() *RunStepRequest

func (*ReplayOp) GetRunStepResponse 

func (x *ReplayOp) GetRunStepResponse() *RunStepResponse

func (*ReplayOp) GetStartTimeUs 

func (x *ReplayOp) GetStartTimeUs() float64

func (*ReplayOp) ProtoMessage 

func (*ReplayOp) ProtoMessage()

func (*ReplayOp) ProtoReflect 

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

func (*ReplayOp) Reset 

func (x *ReplayOp) Reset()

func (*ReplayOp) String 

func (x *ReplayOp) String() string

type ReplayOp_CloseSession 

type ReplayOp_CloseSession struct {
	CloseSession *CloseSessionRequest `protobuf:"bytes,5,opt,name=close_session,json=closeSession,proto3,oneof"`
}

type ReplayOp_CloseSessionResponse 

type ReplayOp_CloseSessionResponse struct {
	CloseSessionResponse *CloseSessionResponse `protobuf:"bytes,25,opt,name=close_session_response,json=closeSessionResponse,proto3,oneof"`
}

type ReplayOp_CreateSession 

type ReplayOp_CreateSession struct {
	CreateSession *CreateSessionRequest `protobuf:"bytes,1,opt,name=create_session,json=createSession,proto3,oneof"`
}

type ReplayOp_CreateSessionResponse 

type ReplayOp_CreateSessionResponse struct {
	CreateSessionResponse *CreateSessionResponse `protobuf:"bytes,21,opt,name=create_session_response,json=createSessionResponse,proto3,oneof"`
}

type ReplayOp_ExtendSession 

type ReplayOp_ExtendSession struct {
	ExtendSession *ExtendSessionRequest `protobuf:"bytes,2,opt,name=extend_session,json=extendSession,proto3,oneof"`
}

type ReplayOp_ExtendSessionResponse 

type ReplayOp_ExtendSessionResponse struct {
	ExtendSessionResponse *ExtendSessionResponse `protobuf:"bytes,22,opt,name=extend_session_response,json=extendSessionResponse,proto3,oneof"`
}

type ReplayOp_ListDevices 

type ReplayOp_ListDevices struct {
	ListDevices *ListDevicesRequest `protobuf:"bytes,6,opt,name=list_devices,json=listDevices,proto3,oneof"`
}

type ReplayOp_ListDevicesResponse 

type ReplayOp_ListDevicesResponse struct {
	ListDevicesResponse *ListDevicesResponse `protobuf:"bytes,26,opt,name=list_devices_response,json=listDevicesResponse,proto3,oneof"`
}

type ReplayOp_MakeCallable 

type ReplayOp_MakeCallable struct {
	MakeCallable *MakeCallableRequest `protobuf:"bytes,8,opt,name=make_callable,json=makeCallable,proto3,oneof"`
}

type ReplayOp_MakeCallableResponse 

type ReplayOp_MakeCallableResponse struct {
	MakeCallableResponse *MakeCallableResponse `protobuf:"bytes,28,opt,name=make_callable_response,json=makeCallableResponse,proto3,oneof"`
}

type ReplayOp_NewReplaySession 

type ReplayOp_NewReplaySession struct {
	NewReplaySession *NewReplaySession `protobuf:"bytes,11,opt,name=new_replay_session,json=newReplaySession,proto3,oneof"`
}

type ReplayOp_PartialRunSetup 

type ReplayOp_PartialRunSetup struct {
	PartialRunSetup *PartialRunSetupRequest `protobuf:"bytes,3,opt,name=partial_run_setup,json=partialRunSetup,proto3,oneof"`
}

type ReplayOp_PartialRunSetupResponse 

type ReplayOp_PartialRunSetupResponse struct {
	PartialRunSetupResponse *PartialRunSetupResponse `protobuf:"bytes,23,opt,name=partial_run_setup_response,json=partialRunSetupResponse,proto3,oneof"`
}

type ReplayOp_ReleaseCallable 

type ReplayOp_ReleaseCallable struct {
	ReleaseCallable *ReleaseCallableRequest `protobuf:"bytes,10,opt,name=release_callable,json=releaseCallable,proto3,oneof"`
}

type ReplayOp_ReleaseCallableResponse 

type ReplayOp_ReleaseCallableResponse struct {
	ReleaseCallableResponse *ReleaseCallableResponse `protobuf:"bytes,30,opt,name=release_callable_response,json=releaseCallableResponse,proto3,oneof"`
}

type ReplayOp_ResetRequest 

type ReplayOp_ResetRequest struct {
	ResetRequest *ResetRequest `protobuf:"bytes,7,opt,name=reset_request,json=resetRequest,proto3,oneof"`
}

type ReplayOp_ResetRequestResponse 

type ReplayOp_ResetRequestResponse struct {
	ResetRequestResponse *ResetResponse `protobuf:"bytes,27,opt,name=reset_request_response,json=resetRequestResponse,proto3,oneof"`
}

type ReplayOp_RunCallable 

type ReplayOp_RunCallable struct {
	RunCallable *RunCallableRequest `protobuf:"bytes,9,opt,name=run_callable,json=runCallable,proto3,oneof"`
}

type ReplayOp_RunCallableResponse 

type ReplayOp_RunCallableResponse struct {
	RunCallableResponse *RunCallableResponse `protobuf:"bytes,29,opt,name=run_callable_response,json=runCallableResponse,proto3,oneof"`
}

type ReplayOp_RunStep 

type ReplayOp_RunStep struct {
	RunStep *RunStepRequest `protobuf:"bytes,4,opt,name=run_step,json=runStep,proto3,oneof"`
}

type ReplayOp_RunStepResponse 

type ReplayOp_RunStepResponse struct {
	RunStepResponse *RunStepResponse `protobuf:"bytes,24,opt,name=run_step_response,json=runStepResponse,proto3,oneof"`
}

type ReportErrorToServiceRequest 

type ReportErrorToServiceRequest struct {
	ErrorCode    int32            `protobuf:"varint,1,opt,name=error_code,json=errorCode,proto3" json:"error_code,omitempty"`
	ErrorMessage string           `protobuf:"bytes,2,opt,name=error_message,json=errorMessage,proto3" json:"error_message,omitempty"`
	ErrorOrigin  *CoordinatedTask `protobuf:"bytes,5,opt,name=error_origin,json=errorOrigin,proto3" json:"error_origin,omitempty"`
	// contains filtered or unexported fields
}

Request and response messages for reporting errors to service instance.

func (*ReportErrorToServiceRequest) Descriptor deprecated 

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

Deprecated: Use ReportErrorToServiceRequest.ProtoReflect.Descriptor instead.

func (*ReportErrorToServiceRequest) GetErrorCode 

func (x *ReportErrorToServiceRequest) GetErrorCode() int32

func (*ReportErrorToServiceRequest) GetErrorMessage 

func (x *ReportErrorToServiceRequest) GetErrorMessage() string

func (*ReportErrorToServiceRequest) GetErrorOrigin 

func (x *ReportErrorToServiceRequest) GetErrorOrigin() *CoordinatedTask

func (*ReportErrorToServiceRequest) ProtoMessage 

func (*ReportErrorToServiceRequest) ProtoMessage()

func (*ReportErrorToServiceRequest) ProtoReflect 

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

func (*ReportErrorToServiceRequest) Reset 

func (x *ReportErrorToServiceRequest) Reset()

func (*ReportErrorToServiceRequest) String 

func (x *ReportErrorToServiceRequest) String() string

type ReportErrorToServiceResponse 

type ReportErrorToServiceResponse struct {
	// contains filtered or unexported fields
}

func (*ReportErrorToServiceResponse) Descriptor deprecated 

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

Deprecated: Use ReportErrorToServiceResponse.ProtoReflect.Descriptor instead.

func (*ReportErrorToServiceResponse) ProtoMessage 

func (*ReportErrorToServiceResponse) ProtoMessage()

func (*ReportErrorToServiceResponse) ProtoReflect 

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

func (*ReportErrorToServiceResponse) Reset 

func (x *ReportErrorToServiceResponse) Reset()

func (*ReportErrorToServiceResponse) String 

func (x *ReportErrorToServiceResponse) String() string

type ReportErrorToTaskRequest 

type ReportErrorToTaskRequest struct {
	ErrorCode    int32                     `protobuf:"varint,1,opt,name=error_code,json=errorCode,proto3" json:"error_code,omitempty"`
	ErrorMessage string                    `protobuf:"bytes,2,opt,name=error_message,json=errorMessage,proto3" json:"error_message,omitempty"`
	ErrorPayload *CoordinationServiceError `protobuf:"bytes,5,opt,name=error_payload,json=errorPayload,proto3" json:"error_payload,omitempty"`
	// contains filtered or unexported fields
}

Request and response messages for reporting errors to task.

func (*ReportErrorToTaskRequest) Descriptor deprecated 

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

Deprecated: Use ReportErrorToTaskRequest.ProtoReflect.Descriptor instead.

func (*ReportErrorToTaskRequest) GetErrorCode 

func (x *ReportErrorToTaskRequest) GetErrorCode() int32

func (*ReportErrorToTaskRequest) GetErrorMessage 

func (x *ReportErrorToTaskRequest) GetErrorMessage() string

func (*ReportErrorToTaskRequest) GetErrorPayload 

func (x *ReportErrorToTaskRequest) GetErrorPayload() *CoordinationServiceError

func (*ReportErrorToTaskRequest) ProtoMessage 

func (*ReportErrorToTaskRequest) ProtoMessage()

func (*ReportErrorToTaskRequest) ProtoReflect 

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

func (*ReportErrorToTaskRequest) Reset 

func (x *ReportErrorToTaskRequest) Reset()

func (*ReportErrorToTaskRequest) String 

func (x *ReportErrorToTaskRequest) String() string

type ReportErrorToTaskResponse 

type ReportErrorToTaskResponse struct {
	// contains filtered or unexported fields
}

func (*ReportErrorToTaskResponse) Descriptor deprecated 

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

Deprecated: Use ReportErrorToTaskResponse.ProtoReflect.Descriptor instead.

func (*ReportErrorToTaskResponse) ProtoMessage 

func (*ReportErrorToTaskResponse) ProtoMessage()

func (*ReportErrorToTaskResponse) ProtoReflect 

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

func (*ReportErrorToTaskResponse) Reset 

func (x *ReportErrorToTaskResponse) Reset()

func (*ReportErrorToTaskResponse) String 

func (x *ReportErrorToTaskResponse) String() string

type RequestedExitCode 

type RequestedExitCode struct {
	ExitCode int32 `protobuf:"varint,1,opt,name=exit_code,json=exitCode,proto3" json:"exit_code,omitempty"`
	// contains filtered or unexported fields
}

func (*RequestedExitCode) Descriptor deprecated 

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

Deprecated: Use RequestedExitCode.ProtoReflect.Descriptor instead.

func (*RequestedExitCode) GetExitCode 

func (x *RequestedExitCode) GetExitCode() int32

func (*RequestedExitCode) ProtoMessage 

func (*RequestedExitCode) ProtoMessage()

func (*RequestedExitCode) ProtoReflect 

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

func (*RequestedExitCode) Reset 

func (x *RequestedExitCode) Reset()

func (*RequestedExitCode) String 

func (x *RequestedExitCode) String() string

type ResetRequest 

type ResetRequest struct {

	// A list of container names, which may be empty.
	//
	// If 'container' is not empty, releases resources in the given
	// containers in all devices.
	//
	// If 'container' is empty, releases resources in the default
	// container in all devices.
	Container []string `protobuf:"bytes,1,rep,name=container,proto3" json:"container,omitempty"`
	// When any filters are present, only devices that match the filters
	// will be reset. Each filter can be partially specified,
	// e.g. "/job:ps" "/job:worker/replica:3", etc.
	DeviceFilters []string `protobuf:"bytes,2,rep,name=device_filters,json=deviceFilters,proto3" json:"device_filters,omitempty"`
	// contains filtered or unexported fields
}

Reset() allows misbehaving or slow sessions to be aborted and closed, and causes their resources eventually to be released. Reset() does not wait for the computations in old sessions to cease; it merely starts the process of tearing them down. However, if a new session is started after a Reset(), the new session is isolated from changes that old sessions (started prior to the Reset()) may continue to make to resources, provided all those resources are in containers listed in "containers".

Old sessions may continue to have side-effects on resources not in containers listed in "containers", and thus may affect future sessions' results in ways that are hard to predict. Thus, if well-defined behavior is desired, is it recommended that all containers be listed in "containers". Similarly, if a device_filter is specified, results may be hard to predict.

func (*ResetRequest) Descriptor deprecated 

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

Deprecated: Use ResetRequest.ProtoReflect.Descriptor instead.

func (*ResetRequest) GetContainer 

func (x *ResetRequest) GetContainer() []string

func (*ResetRequest) GetDeviceFilters 

func (x *ResetRequest) GetDeviceFilters() []string

func (*ResetRequest) ProtoMessage 

func (*ResetRequest) ProtoMessage()

func (*ResetRequest) ProtoReflect 

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

func (*ResetRequest) Reset 

func (x *ResetRequest) Reset()

func (*ResetRequest) String 

func (x *ResetRequest) String() string

type ResetResponse 

type ResetResponse struct {
	// contains filtered or unexported fields
}

func (*ResetResponse) Descriptor deprecated 

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

Deprecated: Use ResetResponse.ProtoReflect.Descriptor instead.

func (*ResetResponse) ProtoMessage 

func (*ResetResponse) ProtoMessage()

func (*ResetResponse) ProtoReflect 

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

func (*ResetResponse) Reset 

func (x *ResetResponse) Reset()

func (*ResetResponse) String 

func (x *ResetResponse) String() string

type ResetTaskRequest 

type ResetTaskRequest struct {
	SourceTask *CoordinatedTask `protobuf:"bytes,1,opt,name=source_task,json=sourceTask,proto3" json:"source_task,omitempty"`
	// contains filtered or unexported fields
}

Request and response messages for resetting a task state in the service.

func (*ResetTaskRequest) Descriptor deprecated 

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

Deprecated: Use ResetTaskRequest.ProtoReflect.Descriptor instead.

func (*ResetTaskRequest) GetSourceTask 

func (x *ResetTaskRequest) GetSourceTask() *CoordinatedTask

func (*ResetTaskRequest) ProtoMessage 

func (*ResetTaskRequest) ProtoMessage()

func (*ResetTaskRequest) ProtoReflect 

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

func (*ResetTaskRequest) Reset 

func (x *ResetTaskRequest) Reset()

func (*ResetTaskRequest) String 

func (x *ResetTaskRequest) String() string

type ResetTaskResponse 

type ResetTaskResponse struct {
	// contains filtered or unexported fields
}

func (*ResetTaskResponse) Descriptor deprecated 

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

Deprecated: Use ResetTaskResponse.ProtoReflect.Descriptor instead.

func (*ResetTaskResponse) ProtoMessage 

func (*ResetTaskResponse) ProtoMessage()

func (*ResetTaskResponse) ProtoReflect 

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

func (*ResetTaskResponse) Reset 

func (x *ResetTaskResponse) Reset()

func (*ResetTaskResponse) String 

func (x *ResetTaskResponse) String() string

type ResourceDtypeAndShape 

type ResourceDtypeAndShape 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
}

func (*ResourceDtypeAndShape) Descriptor deprecated 

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

Deprecated: Use ResourceDtypeAndShape.ProtoReflect.Descriptor instead.

func (*ResourceDtypeAndShape) GetDtype 

func (x *ResourceDtypeAndShape) GetDtype() DataType

func (*ResourceDtypeAndShape) GetShape 

func (x *ResourceDtypeAndShape) GetShape() *TensorShapeProto

func (*ResourceDtypeAndShape) ProtoMessage 

func (*ResourceDtypeAndShape) ProtoMessage()

func (*ResourceDtypeAndShape) ProtoReflect 

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

func (*ResourceDtypeAndShape) Reset 

func (x *ResourceDtypeAndShape) Reset()

func (*ResourceDtypeAndShape) String 

func (x *ResourceDtypeAndShape) 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 RewriterConfig 

type RewriterConfig struct {

	// CPU Conversion settings between NHCW and NCHW.
	CpuLayoutConversion RewriterConfig_CpuLayout `` /* 163-byte string literal not displayed */
	// Optimize tensor layouts (default is ON)
	// e.g. This will try to use NCHW layout on GPU which is faster.
	LayoutOptimizer RewriterConfig_Toggle `` /* 145-byte string literal not displayed */
	// Fold constants (default is ON)
	// Statically infer the value of tensors when possible, and materialize the
	// result using constants.
	ConstantFolding RewriterConfig_Toggle `` /* 145-byte string literal not displayed */
	// Shape optimizations (default is ON)
	// Simplify computations made on shapes.
	ShapeOptimization RewriterConfig_Toggle `` /* 152-byte string literal not displayed */
	// Remapping (default is ON)
	// Remap subgraphs onto more efficient implementations.
	Remapping RewriterConfig_Toggle `protobuf:"varint,14,opt,name=remapping,proto3,enum=tensorflow.RewriterConfig_Toggle" json:"remapping,omitempty"`
	// Common subgraph elimination (default is ON)
	// e.g. Simplify arithmetic ops; merge ops with same value (like constants).
	CommonSubgraphElimination RewriterConfig_Toggle `` /* 178-byte string literal not displayed */
	// Arithmetic optimizations (default is ON)
	// e.g. Simplify arithmetic ops; merge ops with same value (like constants).
	ArithmeticOptimization RewriterConfig_Toggle `` /* 166-byte string literal not displayed */
	// Control dependency optimizations (default is ON).
	// Remove redundant control dependencies, which may enable other optimization.
	DependencyOptimization RewriterConfig_Toggle `` /* 166-byte string literal not displayed */
	// Loop optimizations (default is ON).
	LoopOptimization RewriterConfig_Toggle `` /* 148-byte string literal not displayed */
	// Function optimizations (default is ON).
	FunctionOptimization RewriterConfig_Toggle `` /* 161-byte string literal not displayed */
	// Strips debug-related nodes from the graph (off by default).
	DebugStripper RewriterConfig_Toggle `` /* 140-byte string literal not displayed */
	// If true, don't remove unnecessary ops from the graph
	DisableModelPruning bool `protobuf:"varint,2,opt,name=disable_model_pruning,json=disableModelPruning,proto3" json:"disable_model_pruning,omitempty"`
	// Try to allocate some independent Op outputs contiguously in order to
	// merge or eliminate downstream Ops (off by default).
	ScopedAllocatorOptimization RewriterConfig_Toggle `` /* 184-byte string literal not displayed */
	// Force small ops onto the CPU (default is OFF).
	PinToHostOptimization RewriterConfig_Toggle `` /* 168-byte string literal not displayed */
	// Enable the swap of kernel implementations based on the device placement
	// (default is ON).
	ImplementationSelector RewriterConfig_Toggle `` /* 167-byte string literal not displayed */
	// Optimize data types for CUDA (default is OFF).
	// This will try to use float16 on GPU which is faster.
	// Note that this can change the numerical stability of the graph and may
	// require the use of loss scaling to maintain model convergence.
	AutoMixedPrecision RewriterConfig_Toggle `` /* 157-byte string literal not displayed */
	// Optimize data types for oneDNN (default is OFF).
	// This will try to use bfloat16 on CPUs, which is faster.
	// Note that this can change the numerical stability of the graph.
	// Note: this is deprecated.
	// It is replaced by auto_mixed_precision_onednn_bfloat16
	AutoMixedPrecisionMkl RewriterConfig_Toggle `` /* 168-byte string literal not displayed */
	// Optimize data types for oneDNN (default is OFF).
	// This will try to use bfloat16 on CPUs, which is faster.
	// Note that this can change the numerical stability of the graph.
	// Note: this is equivalent to the deprecated option auto_mixed_precision_mkl
	AutoMixedPrecisionOnednnBfloat16 RewriterConfig_Toggle `` /* 203-byte string literal not displayed */
	// Emulate a model using data type float16 on CPU (default is OFF).
	// This will try to emulate the float16 inputs and outputs of an operator
	// on CPU to have better correlation with float16 on GPU; however the
	// computation in the operator is based on float32.
	// Note that this can change the numerical stability of the graph.
	AutoMixedPrecisionCpu RewriterConfig_Toggle `` /* 168-byte string literal not displayed */
	// Disable the entire meta optimizer (off by default).
	DisableMetaOptimizer bool `protobuf:"varint,19,opt,name=disable_meta_optimizer,json=disableMetaOptimizer,proto3" json:"disable_meta_optimizer,omitempty"`
	// Optimizers registered by plugin (default is ON)
	UsePluginOptimizers RewriterConfig_Toggle `` /* 160-byte string literal not displayed */
	// Conditional code motion (default is ON).
	ExperimentalConditionalCodeMotion RewriterConfig_Toggle `` /* 204-byte string literal not displayed */
	// Controls how many times we run the optimizers in meta optimizer (default
	// is once).
	MetaOptimizerIterations RewriterConfig_NumIterationsType `` /* 183-byte string literal not displayed */
	// The minimum number of nodes in a graph to optimizer. For smaller graphs,
	// optimization is skipped.
	// 0 means the system picks an appropriate number.
	// < 0 means do not skip optimization.
	MinGraphNodes int32 `protobuf:"varint,17,opt,name=min_graph_nodes,json=minGraphNodes,proto3" json:"min_graph_nodes,omitempty"`
	// Disable optimizations that assume compressed tensors. Note that this flag
	// is experimental and may be removed in the future.
	ExperimentalDisableCompressedTensorOptimization bool `` /* 210-byte string literal not displayed */
	// Disable folding quantization emulation ops such as FakeQuantWithMinMax* and
	// QuantizeAndDequantize*. Some compilers (e.g. the TF-to-tflite converter)
	// have to extract quantization configs (e.g. min/max range, number of bits,
	// and per-channel) from the quantization emulation ops. Note that this flag
	// is experimental and may be removed in the future. See b/174138564 for more
	// details.
	ExperimentalDisableFoldingQuantizationEmulation bool `` /* 210-byte string literal not displayed */
	// Configures memory optimization passes through the meta-optimizer. Has no
	// effect on manually requested memory optimization passes in the optimizers
	// field.
	MemoryOptimization RewriterConfig_MemOptType `` /* 158-byte string literal not displayed */
	// A node name scope for node names which are valid outputs of recomputations.
	// Inputs to nodes that match this scope may be recomputed (subject either to
	// manual annotation of those input nodes or to manual annotation and
	// heuristics depending on memory_optimization), but the nodes themselves will
	// not be recomputed. This matches any sub-scopes as well, meaning the scope
	// can appear not just as a top-level scope. For example, if the value is
	// "gradients/", the default, it will match node name "gradients/foo",
	// "foo/gradients/bar", but not "foo_gradients/"
	MemoryOptimizerTargetNodeNameScope string `` /* 171-byte string literal not displayed */
	// Maximum number of milliseconds to spend optimizing a single graph before
	// timing out. If less than or equal to 0 (default value) the optimizer will
	// never time out.
	MetaOptimizerTimeoutMs int64 `` /* 133-byte string literal not displayed */
	// Configures AutoParallel optimization passes either through the
	// meta-optimizer or when manually specified through the optimizers field.
	AutoParallel *AutoParallelOptions `protobuf:"bytes,5,opt,name=auto_parallel,json=autoParallel,proto3" json:"auto_parallel,omitempty"`
	// If true, any optimization pass failing will cause the MetaOptimizer to
	// stop with an error. By default - or when set to false, failing passes are
	// skipped silently.
	FailOnOptimizerErrors bool                    `` /* 130-byte string literal not displayed */
	ScopedAllocatorOpts   *ScopedAllocatorOptions `protobuf:"bytes,16,opt,name=scoped_allocator_opts,json=scopedAllocatorOpts,proto3" json:"scoped_allocator_opts,omitempty"`
	// If non-empty, will use this as an alternative way to specify a list of
	// optimizations to turn on and the order of the optimizations (replacing the
	// meta-optimizer).
	//
	// Of the RewriterConfig options, only the AutoParallel configuration options
	// (the auto_parallel field) apply to manually requested optimization passes
	// ("autoparallel"). Memory optimization passes ("memory") invoked here are
	// not configurable (in contrast to memory optimization passes through the
	// meta-optimizer) and act only on manual op annotations.
	//
	// Custom optimizers (see custom_optimizers) that are not part of this
	// schedule will be run after - in the order that they were specified.
	Optimizers []string `protobuf:"bytes,100,rep,name=optimizers,proto3" json:"optimizers,omitempty"`
	// list of CustomGraphOptimizers to apply.
	CustomOptimizers []*RewriterConfig_CustomGraphOptimizer `protobuf:"bytes,200,rep,name=custom_optimizers,json=customOptimizers,proto3" json:"custom_optimizers,omitempty"`
	// VerifierConfig specifying the verifiers to be run after every optimizer.
	InterOptimizerVerifierConfig *VerifierConfig `` /* 151-byte string literal not displayed */
	// VerifierConfig specifying the verifiers to be run at the end, after all
	// optimizers have run.
	PostOptimizationVerifierConfig *VerifierConfig `` /* 157-byte string literal not displayed */
	// contains filtered or unexported fields
}

func (*RewriterConfig) Descriptor deprecated 

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

Deprecated: Use RewriterConfig.ProtoReflect.Descriptor instead.

func (*RewriterConfig) GetArithmeticOptimization 

func (x *RewriterConfig) GetArithmeticOptimization() RewriterConfig_Toggle

func (*RewriterConfig) GetAutoMixedPrecision 

func (x *RewriterConfig) GetAutoMixedPrecision() RewriterConfig_Toggle

func (*RewriterConfig) GetAutoMixedPrecisionCpu 

func (x *RewriterConfig) GetAutoMixedPrecisionCpu() RewriterConfig_Toggle

func (*RewriterConfig) GetAutoMixedPrecisionMkl 

func (x *RewriterConfig) GetAutoMixedPrecisionMkl() RewriterConfig_Toggle

func (*RewriterConfig) GetAutoMixedPrecisionOnednnBfloat16 

func (x *RewriterConfig) GetAutoMixedPrecisionOnednnBfloat16() RewriterConfig_Toggle

func (*RewriterConfig) GetAutoParallel 

func (x *RewriterConfig) GetAutoParallel() *AutoParallelOptions

func (*RewriterConfig) GetCommonSubgraphElimination 

func (x *RewriterConfig) GetCommonSubgraphElimination() RewriterConfig_Toggle

func (*RewriterConfig) GetConstantFolding 

func (x *RewriterConfig) GetConstantFolding() RewriterConfig_Toggle

func (*RewriterConfig) GetCpuLayoutConversion 

func (x *RewriterConfig) GetCpuLayoutConversion() RewriterConfig_CpuLayout

func (*RewriterConfig) GetCustomOptimizers 

func (x *RewriterConfig) GetCustomOptimizers() []*RewriterConfig_CustomGraphOptimizer

func (*RewriterConfig) GetDebugStripper 

func (x *RewriterConfig) GetDebugStripper() RewriterConfig_Toggle

func (*RewriterConfig) GetDependencyOptimization 

func (x *RewriterConfig) GetDependencyOptimization() RewriterConfig_Toggle

func (*RewriterConfig) GetDisableMetaOptimizer 

func (x *RewriterConfig) GetDisableMetaOptimizer() bool

func (*RewriterConfig) GetDisableModelPruning 

func (x *RewriterConfig) GetDisableModelPruning() bool

func (*RewriterConfig) GetExperimentalConditionalCodeMotion 

func (x *RewriterConfig) GetExperimentalConditionalCodeMotion() RewriterConfig_Toggle

func (*RewriterConfig) GetExperimentalDisableCompressedTensorOptimization 

func (x *RewriterConfig) GetExperimentalDisableCompressedTensorOptimization() bool

func (*RewriterConfig) GetExperimentalDisableFoldingQuantizationEmulation 

func (x *RewriterConfig) GetExperimentalDisableFoldingQuantizationEmulation() bool

func (*RewriterConfig) GetFailOnOptimizerErrors 

func (x *RewriterConfig) GetFailOnOptimizerErrors() bool

func (*RewriterConfig) GetFunctionOptimization 

func (x *RewriterConfig) GetFunctionOptimization() RewriterConfig_Toggle

func (*RewriterConfig) GetImplementationSelector 

func (x *RewriterConfig) GetImplementationSelector() RewriterConfig_Toggle

func (*RewriterConfig) GetInterOptimizerVerifierConfig 

func (x *RewriterConfig) GetInterOptimizerVerifierConfig() *VerifierConfig

func (*RewriterConfig) GetLayoutOptimizer 

func (x *RewriterConfig) GetLayoutOptimizer() RewriterConfig_Toggle

func (*RewriterConfig) GetLoopOptimization 

func (x *RewriterConfig) GetLoopOptimization() RewriterConfig_Toggle

func (*RewriterConfig) GetMemoryOptimization 

func (x *RewriterConfig) GetMemoryOptimization() RewriterConfig_MemOptType

func (*RewriterConfig) GetMemoryOptimizerTargetNodeNameScope 

func (x *RewriterConfig) GetMemoryOptimizerTargetNodeNameScope() string

func (*RewriterConfig) GetMetaOptimizerIterations 

func (x *RewriterConfig) GetMetaOptimizerIterations() RewriterConfig_NumIterationsType

func (*RewriterConfig) GetMetaOptimizerTimeoutMs 

func (x *RewriterConfig) GetMetaOptimizerTimeoutMs() int64

func (*RewriterConfig) GetMinGraphNodes 

func (x *RewriterConfig) GetMinGraphNodes() int32

func (*RewriterConfig) GetOptimizers 

func (x *RewriterConfig) GetOptimizers() []string

func (*RewriterConfig) GetPinToHostOptimization 

func (x *RewriterConfig) GetPinToHostOptimization() RewriterConfig_Toggle

func (*RewriterConfig) GetPostOptimizationVerifierConfig 

func (x *RewriterConfig) GetPostOptimizationVerifierConfig() *VerifierConfig

func (*RewriterConfig) GetRemapping 

func (x *RewriterConfig) GetRemapping() RewriterConfig_Toggle

func (*RewriterConfig) GetScopedAllocatorOptimization 

func (x *RewriterConfig) GetScopedAllocatorOptimization() RewriterConfig_Toggle

func (*RewriterConfig) GetScopedAllocatorOpts 

func (x *RewriterConfig) GetScopedAllocatorOpts() *ScopedAllocatorOptions

func (*RewriterConfig) GetShapeOptimization 

func (x *RewriterConfig) GetShapeOptimization() RewriterConfig_Toggle

func (*RewriterConfig) GetUsePluginOptimizers 

func (x *RewriterConfig) GetUsePluginOptimizers() RewriterConfig_Toggle

func (*RewriterConfig) ProtoMessage 

func (*RewriterConfig) ProtoMessage()

func (*RewriterConfig) ProtoReflect 

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

func (*RewriterConfig) Reset 

func (x *RewriterConfig) Reset()

func (*RewriterConfig) String 

func (x *RewriterConfig) String() string

type RewriterConfig_CpuLayout 

type RewriterConfig_CpuLayout int32

Enum for layout conversion between NCHW and NHWC on CPU. Default is OFF. 

const (
	RewriterConfig_NO_CONVERSION_ON_CPU RewriterConfig_CpuLayout = 0
	RewriterConfig_NCHW_TO_NHWC         RewriterConfig_CpuLayout = 1
	RewriterConfig_NHWC_TO_NCHW         RewriterConfig_CpuLayout = 2
)

func (RewriterConfig_CpuLayout) Descriptor 

func (RewriterConfig_CpuLayout) Descriptor() protoreflect.EnumDescriptor

func (RewriterConfig_CpuLayout) Enum 

func (x RewriterConfig_CpuLayout) Enum() *RewriterConfig_CpuLayout

func (RewriterConfig_CpuLayout) EnumDescriptor deprecated 

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

Deprecated: Use RewriterConfig_CpuLayout.Descriptor instead.

func (RewriterConfig_CpuLayout) Number 

func (x RewriterConfig_CpuLayout) Number() protoreflect.EnumNumber

func (RewriterConfig_CpuLayout) String 

func (x RewriterConfig_CpuLayout) String() string

func (RewriterConfig_CpuLayout) Type 

func (RewriterConfig_CpuLayout) Type() protoreflect.EnumType

type RewriterConfig_CustomGraphOptimizer 

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

Message to describe custom graph optimizer and its parameters

func (*RewriterConfig_CustomGraphOptimizer) Descriptor deprecated 

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

Deprecated: Use RewriterConfig_CustomGraphOptimizer.ProtoReflect.Descriptor instead.

func (*RewriterConfig_CustomGraphOptimizer) GetName 

func (x *RewriterConfig_CustomGraphOptimizer) GetName() string

func (*RewriterConfig_CustomGraphOptimizer) GetParameterMap 

func (x *RewriterConfig_CustomGraphOptimizer) GetParameterMap() map[string]*AttrValue

func (*RewriterConfig_CustomGraphOptimizer) ProtoMessage 

func (*RewriterConfig_CustomGraphOptimizer) ProtoMessage()

func (*RewriterConfig_CustomGraphOptimizer) ProtoReflect 

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

func (*RewriterConfig_CustomGraphOptimizer) Reset 

func (x *RewriterConfig_CustomGraphOptimizer) Reset()

func (*RewriterConfig_CustomGraphOptimizer) String 

func (x *RewriterConfig_CustomGraphOptimizer) String() string

type RewriterConfig_MemOptType 

type RewriterConfig_MemOptType int32
const (
	// The default setting (SCHEDULING and SWAPPING HEURISTICS only)
	RewriterConfig_DEFAULT_MEM_OPT RewriterConfig_MemOptType = 0
	// Disabled in the meta-optimizer.
	RewriterConfig_NO_MEM_OPT RewriterConfig_MemOptType = 1
	// Driven by manual op-level annotations.
	RewriterConfig_MANUAL RewriterConfig_MemOptType = 2
	// Swapping heuristic will move a tensor from the GPU to the CPU and move
	// it back when needed to reduce peak memory usage.
	RewriterConfig_SWAPPING_HEURISTICS RewriterConfig_MemOptType = 4
	// Recomputation heuristics will recompute ops (such as Relu activation)
	// during backprop instead of storing them, reducing peak memory usage.
	RewriterConfig_RECOMPUTATION_HEURISTICS RewriterConfig_MemOptType = 5
	// Scheduling will split big ops such as AddN and try to enforce a schedule
	// of the new computations that decreases peak memory usage.
	RewriterConfig_SCHEDULING_HEURISTICS RewriterConfig_MemOptType = 6
	// Use any combination of swapping and recomputation heuristics.
	RewriterConfig_HEURISTICS RewriterConfig_MemOptType = 3
)

func (RewriterConfig_MemOptType) Descriptor 

func (RewriterConfig_MemOptType) Descriptor() protoreflect.EnumDescriptor

func (RewriterConfig_MemOptType) Enum 

func (x RewriterConfig_MemOptType) Enum() *RewriterConfig_MemOptType

func (RewriterConfig_MemOptType) EnumDescriptor deprecated 

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

Deprecated: Use RewriterConfig_MemOptType.Descriptor instead.

func (RewriterConfig_MemOptType) Number 

func (x RewriterConfig_MemOptType) Number() protoreflect.EnumNumber

func (RewriterConfig_MemOptType) String 

func (x RewriterConfig_MemOptType) String() string

func (RewriterConfig_MemOptType) Type 

func (RewriterConfig_MemOptType) Type() protoreflect.EnumType

type RewriterConfig_NumIterationsType 

type RewriterConfig_NumIterationsType int32

Enum controlling the number of times to run optimizers. The default is to run them twice. 

const (
	RewriterConfig_DEFAULT_NUM_ITERS RewriterConfig_NumIterationsType = 0
	RewriterConfig_ONE               RewriterConfig_NumIterationsType = 1
	RewriterConfig_TWO               RewriterConfig_NumIterationsType = 2
)

func (RewriterConfig_NumIterationsType) Descriptor 

func (RewriterConfig_NumIterationsType) Descriptor() protoreflect.EnumDescriptor

func (RewriterConfig_NumIterationsType) Enum 

func (x RewriterConfig_NumIterationsType) Enum() *RewriterConfig_NumIterationsType

func (RewriterConfig_NumIterationsType) EnumDescriptor deprecated 

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

Deprecated: Use RewriterConfig_NumIterationsType.Descriptor instead.

func (RewriterConfig_NumIterationsType) Number 

func (x RewriterConfig_NumIterationsType) Number() protoreflect.EnumNumber

func (RewriterConfig_NumIterationsType) String 

func (x RewriterConfig_NumIterationsType) String() string

func (RewriterConfig_NumIterationsType) Type 

func (RewriterConfig_NumIterationsType) Type() protoreflect.EnumType

type RewriterConfig_Toggle 

type RewriterConfig_Toggle int32
const (
	RewriterConfig_DEFAULT RewriterConfig_Toggle = 0
	RewriterConfig_ON      RewriterConfig_Toggle = 1
	RewriterConfig_OFF     RewriterConfig_Toggle = 2
	// Enable some aggressive optimizations that use assumptions that TF graphs
	// may break. For example, assume the shape of a placeholder matches its
	// actual feed.
	RewriterConfig_AGGRESSIVE RewriterConfig_Toggle = 3
	// Run MLIR pass if there's one implemented in TFG, do nothing otherwise.
	// I.e., if there's no corresponding TFG pass, it's an OFF. This is supposed
	// to be mapped with `ON` and there's no `AGGRESSIVE` in MLIR pass now.
	RewriterConfig_EXPERIMENTAL_MLIR RewriterConfig_Toggle = 4
	// Run both MLIR and Grappler passes consecutively and MLIR pass will come
	// first.
	RewriterConfig_EXPERIMENTAL_BOTH RewriterConfig_Toggle = 5
)

func (RewriterConfig_Toggle) Descriptor 

func (RewriterConfig_Toggle) Descriptor() protoreflect.EnumDescriptor

func (RewriterConfig_Toggle) Enum 

func (x RewriterConfig_Toggle) Enum() *RewriterConfig_Toggle

func (RewriterConfig_Toggle) EnumDescriptor deprecated 

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

Deprecated: Use RewriterConfig_Toggle.Descriptor instead.

func (RewriterConfig_Toggle) Number 

func (x RewriterConfig_Toggle) Number() protoreflect.EnumNumber

func (RewriterConfig_Toggle) String 

func (x RewriterConfig_Toggle) String() string

func (RewriterConfig_Toggle) Type 

func (RewriterConfig_Toggle) Type() protoreflect.EnumType

type RunCallableRequest 

type RunCallableRequest struct {

	// REQUIRED: session_handle must be returned by a CreateSession call
	// to the same master service.
	SessionHandle string `protobuf:"bytes,1,opt,name=session_handle,json=sessionHandle,proto3" json:"session_handle,omitempty"`
	// REQUIRED: handle must be returned by a MakeCallable call to the same
	// master service.
	Handle int64 `protobuf:"varint,2,opt,name=handle,proto3" json:"handle,omitempty"`
	// Values of the tensors passed as arguments to the callable, in the order
	// defined in the CallableOptions.feed field passed to MakeCallable.
	Feed []*TensorProto `protobuf:"bytes,3,rep,name=feed,proto3" json:"feed,omitempty"`
	// Unique identifier for this request. Every RunCallableRequest must
	// have a unique request_id, and retried RunCallableRequest must have
	// the same request_id. If request_id is zero, retry detection is disabled.
	RequestId int64 `protobuf:"varint,4,opt,name=request_id,json=requestId,proto3" json:"request_id,omitempty"`
	// contains filtered or unexported fields
}

func (*RunCallableRequest) Descriptor deprecated 

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

Deprecated: Use RunCallableRequest.ProtoReflect.Descriptor instead.

func (*RunCallableRequest) GetFeed 

func (x *RunCallableRequest) GetFeed() []*TensorProto

func (*RunCallableRequest) GetHandle 

func (x *RunCallableRequest) GetHandle() int64

func (*RunCallableRequest) GetRequestId 

func (x *RunCallableRequest) GetRequestId() int64

func (*RunCallableRequest) GetSessionHandle 

func (x *RunCallableRequest) GetSessionHandle() string

func (*RunCallableRequest) ProtoMessage 

func (*RunCallableRequest) ProtoMessage()

func (*RunCallableRequest) ProtoReflect 

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

func (*RunCallableRequest) Reset 

func (x *RunCallableRequest) Reset()

func (*RunCallableRequest) String 

func (x *RunCallableRequest) String() string

type RunCallableResponse 

type RunCallableResponse struct {

	// Values of the tensors returned by the callable, in the order defined in the
	// CallableOptions.fetch field passed to MakeCallable.
	Fetch []*TensorProto `protobuf:"bytes,1,rep,name=fetch,proto3" json:"fetch,omitempty"`
	// Returned metadata if requested in the options.
	Metadata *RunMetadata `protobuf:"bytes,2,opt,name=metadata,proto3" json:"metadata,omitempty"`
	// contains filtered or unexported fields
}

func (*RunCallableResponse) Descriptor deprecated 

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

Deprecated: Use RunCallableResponse.ProtoReflect.Descriptor instead.

func (*RunCallableResponse) GetFetch 

func (x *RunCallableResponse) GetFetch() []*TensorProto

func (*RunCallableResponse) GetMetadata 

func (x *RunCallableResponse) GetMetadata() *RunMetadata

func (*RunCallableResponse) ProtoMessage 

func (*RunCallableResponse) ProtoMessage()

func (*RunCallableResponse) ProtoReflect 

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

func (*RunCallableResponse) Reset 

func (x *RunCallableResponse) Reset()

func (*RunCallableResponse) String 

func (x *RunCallableResponse) String() string

type RunComponentFunctionRequest 

type RunComponentFunctionRequest struct {
	ContextId uint64     `protobuf:"fixed64,1,opt,name=context_id,json=contextId,proto3" json:"context_id,omitempty"`
	Operation *Operation `protobuf:"bytes,2,opt,name=operation,proto3" json:"operation,omitempty"`
	// The output indices of its parent function.
	OutputNum []int32 `protobuf:"varint,3,rep,packed,name=output_num,json=outputNum,proto3" json:"output_num,omitempty"`
	// contains filtered or unexported fields
}

func (*RunComponentFunctionRequest) Descriptor deprecated 

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

Deprecated: Use RunComponentFunctionRequest.ProtoReflect.Descriptor instead.

func (*RunComponentFunctionRequest) GetContextId 

func (x *RunComponentFunctionRequest) GetContextId() uint64

func (*RunComponentFunctionRequest) GetOperation 

func (x *RunComponentFunctionRequest) GetOperation() *Operation

func (*RunComponentFunctionRequest) GetOutputNum 

func (x *RunComponentFunctionRequest) GetOutputNum() []int32

func (*RunComponentFunctionRequest) ProtoMessage 

func (*RunComponentFunctionRequest) ProtoMessage()

func (*RunComponentFunctionRequest) ProtoReflect 

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

func (*RunComponentFunctionRequest) Reset 

func (x *RunComponentFunctionRequest) Reset()

func (*RunComponentFunctionRequest) String 

func (x *RunComponentFunctionRequest) String() string

type RunComponentFunctionResponse 

type RunComponentFunctionResponse struct {
	Shape  []*TensorShapeProto `protobuf:"bytes,1,rep,name=shape,proto3" json:"shape,omitempty"`
	Tensor []*TensorProto      `protobuf:"bytes,2,rep,name=tensor,proto3" json:"tensor,omitempty"`
	// contains filtered or unexported fields
}

func (*RunComponentFunctionResponse) Descriptor deprecated 

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

Deprecated: Use RunComponentFunctionResponse.ProtoReflect.Descriptor instead.

func (*RunComponentFunctionResponse) GetShape 

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

func (*RunComponentFunctionResponse) GetTensor 

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

func (*RunComponentFunctionResponse) ProtoMessage 

func (*RunComponentFunctionResponse) ProtoMessage()

func (*RunComponentFunctionResponse) ProtoReflect 

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

func (*RunComponentFunctionResponse) Reset 

func (x *RunComponentFunctionResponse) Reset()

func (*RunComponentFunctionResponse) String 

func (x *RunComponentFunctionResponse) String() string

type RunGraphRequest 

type RunGraphRequest struct {

	// session_handle is the master-generated unique id for this session.
	// If session_handle is non-empty, it must be the same as used when
	// registering the graph. If it is empty, a single global namespace is used to
	// search for the graph_handle.
	SessionHandle string `protobuf:"bytes,8,opt,name=session_handle,json=sessionHandle,proto3" json:"session_handle,omitempty"`
	// Set to true if `CreateWorkerSession` was called for `session_handle`.
	CreateWorkerSessionCalled bool `` /* 142-byte string literal not displayed */
	// REQUIRED: graph_handle must be returned by a RegisterGraph call
	// to the same WorkerService.
	GraphHandle string `protobuf:"bytes,1,opt,name=graph_handle,json=graphHandle,proto3" json:"graph_handle,omitempty"`
	// A unique ID to distinguish different runs of the same graph.
	//
	// The master generates a global unique `step_id` to distinguish
	// different runs of the graph computation. Subgraphs communicate
	// (e.g., send/recv ops) with each other using `step_id` to
	// distinguish tensors generated by different runs.
	StepId int64 `protobuf:"varint,2,opt,name=step_id,json=stepId,proto3" json:"step_id,omitempty"`
	// Options for this step.
	ExecOpts *ExecutorOpts `protobuf:"bytes,5,opt,name=exec_opts,json=execOpts,proto3" json:"exec_opts,omitempty"`
	// Runs the graph.
	//
	// Sends the tensors in "send" into the graph before the run and
	// fetches the keys into `RunGraphResponse.recv` after the run.
	Send    []*NamedTensorProto `protobuf:"bytes,3,rep,name=send,proto3" json:"send,omitempty"`
	RecvKey []string            `protobuf:"bytes,4,rep,name=recv_key,json=recvKey,proto3" json:"recv_key,omitempty"`
	// True if the RunGraphRequest is a partial run request.
	IsPartial bool `protobuf:"varint,6,opt,name=is_partial,json=isPartial,proto3" json:"is_partial,omitempty"`
	// True if this is the last partial run request in a sequence of requests.
	IsLastPartialRun bool `protobuf:"varint,7,opt,name=is_last_partial_run,json=isLastPartialRun,proto3" json:"is_last_partial_run,omitempty"`
	// If true then some errors, e.g., execution errors that have long
	// error messages, may return an OK RunGraphResponse with the actual
	// error saved in the status_code/status_error_message fields of the
	// response body. This is a workaround since the RPC subsystem may
	// truncate long metadata messages.
	StoreErrorsInResponseBody bool `` /* 143-byte string literal not displayed */
	// Unique identifier for this request. Every RunGraphRequest must have a
	// unique request_id, and retried RunGraphRequests must have the same
	// request_id. If request_id is zero, retry detection is disabled.
	//
	// Retried RunGraphRequests are problematic because they may issue a
	// RecvTensor that will have no corresponding sender and will wait forever.
	// Workers use request_ids to reject retried RunGraph requests instead of
	// waiting forever.
	RequestId int64 `protobuf:"varint,11,opt,name=request_id,json=requestId,proto3" json:"request_id,omitempty"`
	// contains filtered or unexported fields
}

func (*RunGraphRequest) Descriptor deprecated 

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

Deprecated: Use RunGraphRequest.ProtoReflect.Descriptor instead.

func (*RunGraphRequest) GetCreateWorkerSessionCalled 

func (x *RunGraphRequest) GetCreateWorkerSessionCalled() bool

func (*RunGraphRequest) GetExecOpts 

func (x *RunGraphRequest) GetExecOpts() *ExecutorOpts

func (*RunGraphRequest) GetGraphHandle 

func (x *RunGraphRequest) GetGraphHandle() string

func (*RunGraphRequest) GetIsLastPartialRun 

func (x *RunGraphRequest) GetIsLastPartialRun() bool

func (*RunGraphRequest) GetIsPartial 

func (x *RunGraphRequest) GetIsPartial() bool

func (*RunGraphRequest) GetRecvKey 

func (x *RunGraphRequest) GetRecvKey() []string

func (*RunGraphRequest) GetRequestId 

func (x *RunGraphRequest) GetRequestId() int64

func (*RunGraphRequest) GetSend 

func (x *RunGraphRequest) GetSend() []*NamedTensorProto

func (*RunGraphRequest) GetSessionHandle 

func (x *RunGraphRequest) GetSessionHandle() string

func (*RunGraphRequest) GetStepId 

func (x *RunGraphRequest) GetStepId() int64

func (*RunGraphRequest) GetStoreErrorsInResponseBody 

func (x *RunGraphRequest) GetStoreErrorsInResponseBody() bool

func (*RunGraphRequest) ProtoMessage 

func (*RunGraphRequest) ProtoMessage()

func (*RunGraphRequest) ProtoReflect 

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

func (*RunGraphRequest) Reset 

func (x *RunGraphRequest) Reset()

func (*RunGraphRequest) String 

func (x *RunGraphRequest) String() string

type RunGraphResponse 

type RunGraphResponse struct {

	// A list of tensors corresponding to those requested by
	// `RunGraphRequest.recv_key`.
	Recv []*NamedTensorProto `protobuf:"bytes,1,rep,name=recv,proto3" json:"recv,omitempty"`
	// If the request asked for execution stats, the cost graph, or the partition
	// graphs, these are returned here.
	// TODO(suharshs): Package these in a RunMetadata instead.
	StepStats      *StepStats    `protobuf:"bytes,2,opt,name=step_stats,json=stepStats,proto3" json:"step_stats,omitempty"`
	CostGraph      *CostGraphDef `protobuf:"bytes,3,opt,name=cost_graph,json=costGraph,proto3" json:"cost_graph,omitempty"`
	PartitionGraph []*GraphDef   `protobuf:"bytes,4,rep,name=partition_graph,json=partitionGraph,proto3" json:"partition_graph,omitempty"`
	// If store_errors_in_response_body is true in the request, then
	// optionally the server may return an OK status for the RPC and
	// fill the true status into the fields below, to allow for messages
	// that are too long to fit in metadata.
	StatusCode         Code   `protobuf:"varint,5,opt,name=status_code,json=statusCode,proto3,enum=tensorflow.error.Code" json:"status_code,omitempty"`
	StatusErrorMessage string `protobuf:"bytes,6,opt,name=status_error_message,json=statusErrorMessage,proto3" json:"status_error_message,omitempty"`
	// contains filtered or unexported fields
}

func (*RunGraphResponse) Descriptor deprecated 

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

Deprecated: Use RunGraphResponse.ProtoReflect.Descriptor instead.

func (*RunGraphResponse) GetCostGraph 

func (x *RunGraphResponse) GetCostGraph() *CostGraphDef

func (*RunGraphResponse) GetPartitionGraph 

func (x *RunGraphResponse) GetPartitionGraph() []*GraphDef

func (*RunGraphResponse) GetRecv 

func (x *RunGraphResponse) GetRecv() []*NamedTensorProto

func (*RunGraphResponse) GetStatusCode 

func (x *RunGraphResponse) GetStatusCode() Code

func (*RunGraphResponse) GetStatusErrorMessage 

func (x *RunGraphResponse) GetStatusErrorMessage() string

func (*RunGraphResponse) GetStepStats 

func (x *RunGraphResponse) GetStepStats() *StepStats

func (*RunGraphResponse) ProtoMessage 

func (*RunGraphResponse) ProtoMessage()

func (*RunGraphResponse) ProtoReflect 

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

func (*RunGraphResponse) Reset 

func (x *RunGraphResponse) Reset()

func (*RunGraphResponse) String 

func (x *RunGraphResponse) String() string

type RunMetadata 

type RunMetadata struct {

	// Statistics traced for this step. Populated if tracing is turned on via the
	// "RunOptions" proto.
	// EXPERIMENTAL: The format and set of events may change in future versions.
	StepStats *StepStats `protobuf:"bytes,1,opt,name=step_stats,json=stepStats,proto3" json:"step_stats,omitempty"`
	// The cost graph for the computation defined by the run call.
	CostGraph *CostGraphDef `protobuf:"bytes,2,opt,name=cost_graph,json=costGraph,proto3" json:"cost_graph,omitempty"`
	// Graphs of the partitions executed by executors.
	PartitionGraphs []*GraphDef `protobuf:"bytes,3,rep,name=partition_graphs,json=partitionGraphs,proto3" json:"partition_graphs,omitempty"`
	// This is only populated for graphs that are run as functions in TensorFlow
	// V2. There will be an entry below for each function that is traced.
	// The main use cases of the post_optimization_graph and the partition_graphs
	// is to give the caller insight into the graphs that were actually run by the
	// runtime. Additional information (such as those in step_stats) will match
	// these graphs.
	// We also include the pre_optimization_graph since it is usually easier to
	// read, and is helpful in situations where the caller wants to get a high
	// level idea of what the built graph looks like (since the various graph
	// optimization passes might change the structure of the graph significantly).
	FunctionGraphs []*RunMetadata_FunctionGraphs `protobuf:"bytes,4,rep,name=function_graphs,json=functionGraphs,proto3" json:"function_graphs,omitempty"`
	// Metadata about the session.
	SessionMetadata *SessionMetadata `protobuf:"bytes,5,opt,name=session_metadata,json=sessionMetadata,proto3" json:"session_metadata,omitempty"`
	// contains filtered or unexported fields
}

Metadata output (i.e., non-Tensor) for a single Run() call.

func (*RunMetadata) Descriptor deprecated 

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

Deprecated: Use RunMetadata.ProtoReflect.Descriptor instead.

func (*RunMetadata) GetCostGraph 

func (x *RunMetadata) GetCostGraph() *CostGraphDef

func (*RunMetadata) GetFunctionGraphs 

func (x *RunMetadata) GetFunctionGraphs() []*RunMetadata_FunctionGraphs

func (*RunMetadata) GetPartitionGraphs 

func (x *RunMetadata) GetPartitionGraphs() []*GraphDef

func (*RunMetadata) GetSessionMetadata 

func (x *RunMetadata) GetSessionMetadata() *SessionMetadata

func (*RunMetadata) GetStepStats 

func (x *RunMetadata) GetStepStats() *StepStats

func (*RunMetadata) ProtoMessage 

func (*RunMetadata) ProtoMessage()

func (*RunMetadata) ProtoReflect 

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

func (*RunMetadata) Reset 

func (x *RunMetadata) Reset()

func (*RunMetadata) String 

func (x *RunMetadata) String() string

type RunMetadata_FunctionGraphs 

type RunMetadata_FunctionGraphs struct {

	// TODO(nareshmodi): Include some sort of function/cache-key identifier?
	PartitionGraphs       []*GraphDef `protobuf:"bytes,1,rep,name=partition_graphs,json=partitionGraphs,proto3" json:"partition_graphs,omitempty"`
	PreOptimizationGraph  *GraphDef   `protobuf:"bytes,2,opt,name=pre_optimization_graph,json=preOptimizationGraph,proto3" json:"pre_optimization_graph,omitempty"`
	PostOptimizationGraph *GraphDef   `` /* 126-byte string literal not displayed */
	// contains filtered or unexported fields
}

func (*RunMetadata_FunctionGraphs) Descriptor deprecated 

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

Deprecated: Use RunMetadata_FunctionGraphs.ProtoReflect.Descriptor instead.

func (*RunMetadata_FunctionGraphs) GetPartitionGraphs 

func (x *RunMetadata_FunctionGraphs) GetPartitionGraphs() []*GraphDef

func (*RunMetadata_FunctionGraphs) GetPostOptimizationGraph 

func (x *RunMetadata_FunctionGraphs) GetPostOptimizationGraph() *GraphDef

func (*RunMetadata_FunctionGraphs) GetPreOptimizationGraph 

func (x *RunMetadata_FunctionGraphs) GetPreOptimizationGraph() *GraphDef

func (*RunMetadata_FunctionGraphs) ProtoMessage 

func (*RunMetadata_FunctionGraphs) ProtoMessage()

func (*RunMetadata_FunctionGraphs) ProtoReflect 

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

func (*RunMetadata_FunctionGraphs) Reset 

func (x *RunMetadata_FunctionGraphs) Reset()

func (*RunMetadata_FunctionGraphs) String 

func (x *RunMetadata_FunctionGraphs) String() string

type RunOptions 

type RunOptions struct {
	TraceLevel RunOptions_TraceLevel `` /* 130-byte string literal not displayed */
	// Time to wait for operation to complete in milliseconds.
	TimeoutInMs int64 `protobuf:"varint,2,opt,name=timeout_in_ms,json=timeoutInMs,proto3" json:"timeout_in_ms,omitempty"`
	// The thread pool to use, if session_inter_op_thread_pool is configured.
	// To use the caller thread set this to -1 - this uses the caller thread
	// to execute Session::Run() and thus avoids a context switch. Using the
	// caller thread to execute Session::Run() should be done ONLY for simple
	// graphs, where the overhead of an additional context switch is
	// comparable with the overhead of Session::Run().
	InterOpThreadPool int32 `protobuf:"varint,3,opt,name=inter_op_thread_pool,json=interOpThreadPool,proto3" json:"inter_op_thread_pool,omitempty"`
	// Whether the partition graph(s) executed by the executor(s) should be
	// outputted via RunMetadata.
	OutputPartitionGraphs bool `` /* 127-byte string literal not displayed */
	// EXPERIMENTAL.  Options used to initialize DebuggerState, if enabled.
	DebugOptions *DebugOptions `protobuf:"bytes,6,opt,name=debug_options,json=debugOptions,proto3" json:"debug_options,omitempty"`
	// When enabled, causes tensor allocation information to be included in
	// the error message when the Run() call fails because the allocator ran
	// out of memory (OOM).
	//
	// Enabling this option can slow down the Run() call.
	ReportTensorAllocationsUponOom bool                     `` /* 158-byte string literal not displayed */
	Experimental                   *RunOptions_Experimental `protobuf:"bytes,8,opt,name=experimental,proto3" json:"experimental,omitempty"`
	// contains filtered or unexported fields
}

Options for a single Run() call.

func (*RunOptions) Descriptor deprecated 

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

Deprecated: Use RunOptions.ProtoReflect.Descriptor instead.

func (*RunOptions) GetDebugOptions 

func (x *RunOptions) GetDebugOptions() *DebugOptions

func (*RunOptions) GetExperimental 

func (x *RunOptions) GetExperimental() *RunOptions_Experimental

func (*RunOptions) GetInterOpThreadPool 

func (x *RunOptions) GetInterOpThreadPool() int32

func (*RunOptions) GetOutputPartitionGraphs 

func (x *RunOptions) GetOutputPartitionGraphs() bool

func (*RunOptions) GetReportTensorAllocationsUponOom 

func (x *RunOptions) GetReportTensorAllocationsUponOom() bool

func (*RunOptions) GetTimeoutInMs 

func (x *RunOptions) GetTimeoutInMs() int64

func (*RunOptions) GetTraceLevel 

func (x *RunOptions) GetTraceLevel() RunOptions_TraceLevel

func (*RunOptions) ProtoMessage 

func (*RunOptions) ProtoMessage()

func (*RunOptions) ProtoReflect 

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

func (*RunOptions) Reset 

func (x *RunOptions) Reset()

func (*RunOptions) String 

func (x *RunOptions) String() string

type RunOptions_Experimental 

type RunOptions_Experimental struct {

	// If non-zero, declares that this graph is going to use collective
	// ops and must synchronize step_ids with any other graph with this
	// same group_key value (in a distributed computation where tasks
	// run disjoint graphs).
	CollectiveGraphKey int64 `protobuf:"varint,1,opt,name=collective_graph_key,json=collectiveGraphKey,proto3" json:"collective_graph_key,omitempty"`
	// If true, then operations (using the inter-op pool) across all
	// session::run() calls will be centrally scheduled, optimizing for (median
	// and tail) latency.
	// Consider using this option for CPU-bound workloads like inference.
	UseRunHandlerPool     bool                                           `protobuf:"varint,2,opt,name=use_run_handler_pool,json=useRunHandlerPool,proto3" json:"use_run_handler_pool,omitempty"`
	RunHandlerPoolOptions *RunOptions_Experimental_RunHandlerPoolOptions `` /* 128-byte string literal not displayed */
	// contains filtered or unexported fields
}

Everything inside Experimental is subject to change and is not subject to API stability guarantees in https://www.tensorflow.org/guide/version_compat.

func (*RunOptions_Experimental) Descriptor deprecated 

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

Deprecated: Use RunOptions_Experimental.ProtoReflect.Descriptor instead.

func (*RunOptions_Experimental) GetCollectiveGraphKey 

func (x *RunOptions_Experimental) GetCollectiveGraphKey() int64

func (*RunOptions_Experimental) GetRunHandlerPoolOptions 

func (x *RunOptions_Experimental) GetRunHandlerPoolOptions() *RunOptions_Experimental_RunHandlerPoolOptions

func (*RunOptions_Experimental) GetUseRunHandlerPool 

func (x *RunOptions_Experimental) GetUseRunHandlerPool() bool

func (*RunOptions_Experimental) ProtoMessage 

func (*RunOptions_Experimental) ProtoMessage()

func (*RunOptions_Experimental) ProtoReflect 

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

func (*RunOptions_Experimental) Reset 

func (x *RunOptions_Experimental) Reset()

func (*RunOptions_Experimental) String 

func (x *RunOptions_Experimental) String() string

type RunOptions_Experimental_RunHandlerPoolOptions 

type RunOptions_Experimental_RunHandlerPoolOptions struct {

	// Priority of the request. The run handler thread pool will schedule ops
	// based on the priority number. The larger number means higher priority.
	Priority int64 `protobuf:"varint,1,opt,name=priority,proto3" json:"priority,omitempty"`
	// contains filtered or unexported fields
}

Options for run handler thread pool.

func (*RunOptions_Experimental_RunHandlerPoolOptions) Descriptor deprecated 

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

Deprecated: Use RunOptions_Experimental_RunHandlerPoolOptions.ProtoReflect.Descriptor instead.

func (*RunOptions_Experimental_RunHandlerPoolOptions) GetPriority 

func (x *RunOptions_Experimental_RunHandlerPoolOptions) GetPriority() int64

func (*RunOptions_Experimental_RunHandlerPoolOptions) ProtoMessage 

func (*RunOptions_Experimental_RunHandlerPoolOptions) ProtoMessage()

func (*RunOptions_Experimental_RunHandlerPoolOptions) ProtoReflect 

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

func (*RunOptions_Experimental_RunHandlerPoolOptions) Reset 

func (x *RunOptions_Experimental_RunHandlerPoolOptions) Reset()

func (*RunOptions_Experimental_RunHandlerPoolOptions) String 

func (x *RunOptions_Experimental_RunHandlerPoolOptions) String() string

type RunOptions_TraceLevel 

type RunOptions_TraceLevel int32

TODO(pbar) Turn this into a TraceOptions proto which allows tracing to be controlled in a more orthogonal manner? 

const (
	RunOptions_NO_TRACE       RunOptions_TraceLevel = 0
	RunOptions_SOFTWARE_TRACE RunOptions_TraceLevel = 1
	RunOptions_HARDWARE_TRACE RunOptions_TraceLevel = 2
	RunOptions_FULL_TRACE     RunOptions_TraceLevel = 3
)

func (RunOptions_TraceLevel) Descriptor 

func (RunOptions_TraceLevel) Descriptor() protoreflect.EnumDescriptor

func (RunOptions_TraceLevel) Enum 

func (x RunOptions_TraceLevel) Enum() *RunOptions_TraceLevel

func (RunOptions_TraceLevel) EnumDescriptor deprecated 

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

Deprecated: Use RunOptions_TraceLevel.Descriptor instead.

func (RunOptions_TraceLevel) Number 

func (x RunOptions_TraceLevel) Number() protoreflect.EnumNumber

func (RunOptions_TraceLevel) String 

func (x RunOptions_TraceLevel) String() string

func (RunOptions_TraceLevel) Type 

func (RunOptions_TraceLevel) Type() protoreflect.EnumType

type RunStepRequest 

type RunStepRequest struct {

	// REQUIRED: session_handle must be returned by a CreateSession call
	// to the same master service.
	SessionHandle string `protobuf:"bytes,1,opt,name=session_handle,json=sessionHandle,proto3" json:"session_handle,omitempty"`
	// Tensors to be fed in the step. Each feed is a named tensor.
	Feed []*NamedTensorProto `protobuf:"bytes,2,rep,name=feed,proto3" json:"feed,omitempty"`
	// Fetches. A list of tensor names. The caller expects a tensor to
	// be returned for each fetch[i] (see RunStepResponse.tensor). The
	// order of specified fetches does not change the execution order.
	Fetch []string `protobuf:"bytes,3,rep,name=fetch,proto3" json:"fetch,omitempty"`
	// Target Nodes. A list of node names. The named nodes will be run
	// to but their outputs will not be fetched.
	Target []string `protobuf:"bytes,4,rep,name=target,proto3" json:"target,omitempty"`
	// Options for the run call.
	Options *RunOptions `protobuf:"bytes,5,opt,name=options,proto3" json:"options,omitempty"`
	// Partial run handle (optional). If specified, this will be a partial run
	// execution, run up to the specified fetches.
	PartialRunHandle string `protobuf:"bytes,6,opt,name=partial_run_handle,json=partialRunHandle,proto3" json:"partial_run_handle,omitempty"`
	// If true then some errors, e.g., execution errors that have long
	// error messages, may return an OK RunStepResponse with the actual
	// error saved in the status_code/status_error_message fields of the
	// response body. This is a workaround since the RPC subsystem may
	// truncate long metadata messages.
	StoreErrorsInResponseBody bool `` /* 143-byte string literal not displayed */
	// Unique identifier for this request. Every RunStepRequest must
	// have a unique request_id, and retried RunStepRequest must have
	// the same request_id. If request_id is zero, retry detection is disabled.
	RequestId int64 `protobuf:"varint,8,opt,name=request_id,json=requestId,proto3" json:"request_id,omitempty"`
	// contains filtered or unexported fields
}

func (*RunStepRequest) Descriptor deprecated 

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

Deprecated: Use RunStepRequest.ProtoReflect.Descriptor instead.

func (*RunStepRequest) GetFeed 

func (x *RunStepRequest) GetFeed() []*NamedTensorProto

func (*RunStepRequest) GetFetch 

func (x *RunStepRequest) GetFetch() []string

func (*RunStepRequest) GetOptions 

func (x *RunStepRequest) GetOptions() *RunOptions

func (*RunStepRequest) GetPartialRunHandle 

func (x *RunStepRequest) GetPartialRunHandle() string

func (*RunStepRequest) GetRequestId 

func (x *RunStepRequest) GetRequestId() int64

func (*RunStepRequest) GetSessionHandle 

func (x *RunStepRequest) GetSessionHandle() string

func (*RunStepRequest) GetStoreErrorsInResponseBody 

func (x *RunStepRequest) GetStoreErrorsInResponseBody() bool

func (*RunStepRequest) GetTarget 

func (x *RunStepRequest) GetTarget() []string

func (*RunStepRequest) ProtoMessage 

func (*RunStepRequest) ProtoMessage()

func (*RunStepRequest) ProtoReflect 

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

func (*RunStepRequest) Reset 

func (x *RunStepRequest) Reset()

func (*RunStepRequest) String 

func (x *RunStepRequest) String() string

type RunStepResponse 

type RunStepResponse struct {

	// NOTE: The order of the returned tensors may or may not match
	// the fetch order specified in RunStepRequest.
	Tensor []*NamedTensorProto `protobuf:"bytes,1,rep,name=tensor,proto3" json:"tensor,omitempty"`
	// Returned metadata if requested in the options.
	Metadata *RunMetadata `protobuf:"bytes,2,opt,name=metadata,proto3" json:"metadata,omitempty"`
	// If store_errors_in_response_body is true in the request, then
	// optionally the server may return an OK status for the RPC and
	// fill the true status into the fields below, to allow for messages
	// that are too long to fit in metadata.
	StatusCode         Code   `protobuf:"varint,3,opt,name=status_code,json=statusCode,proto3,enum=tensorflow.error.Code" json:"status_code,omitempty"`
	StatusErrorMessage string `protobuf:"bytes,4,opt,name=status_error_message,json=statusErrorMessage,proto3" json:"status_error_message,omitempty"`
	// contains filtered or unexported fields
}

func (*RunStepResponse) Descriptor deprecated 

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

Deprecated: Use RunStepResponse.ProtoReflect.Descriptor instead.

func (*RunStepResponse) GetMetadata 

func (x *RunStepResponse) GetMetadata() *RunMetadata

func (*RunStepResponse) GetStatusCode 

func (x *RunStepResponse) GetStatusCode() Code

func (*RunStepResponse) GetStatusErrorMessage 

func (x *RunStepResponse) GetStatusErrorMessage() string

func (*RunStepResponse) GetTensor 

func (x *RunStepResponse) GetTensor() []*NamedTensorProto

func (*RunStepResponse) ProtoMessage 

func (*RunStepResponse) ProtoMessage()

func (*RunStepResponse) ProtoReflect 

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

func (*RunStepResponse) Reset 

func (x *RunStepResponse) Reset()

func (*RunStepResponse) String 

func (x *RunStepResponse) 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 SaveableObject 

type SaveableObject struct {

	// Node ids of concrete functions for saving and loading from a checkpoint.
	// These functions save and restore directly from tensors.
	SaveFunction    int32 `protobuf:"varint,2,opt,name=save_function,json=saveFunction,proto3" json:"save_function,omitempty"`
	RestoreFunction int32 `protobuf:"varint,3,opt,name=restore_function,json=restoreFunction,proto3" json:"restore_function,omitempty"`
	// contains filtered or unexported fields
}

func (*SaveableObject) Descriptor deprecated 

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

Deprecated: Use SaveableObject.ProtoReflect.Descriptor instead.

func (*SaveableObject) GetRestoreFunction 

func (x *SaveableObject) GetRestoreFunction() int32

func (*SaveableObject) GetSaveFunction 

func (x *SaveableObject) GetSaveFunction() int32

func (*SaveableObject) ProtoMessage 

func (*SaveableObject) ProtoMessage()

func (*SaveableObject) ProtoReflect 

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

func (*SaveableObject) Reset 

func (x *SaveableObject) Reset()

func (*SaveableObject) String 

func (x *SaveableObject) String() string

type SavedAsset 

type SavedAsset struct {

	// Index into `MetaGraphDef.asset_file_def[]` that describes the Asset.
	//
	// Only the field `AssetFileDef.filename` is used. Other fields, such as
	// `AssetFileDef.tensor_info`, MUST be ignored.
	AssetFileDefIndex int32 `protobuf:"varint,1,opt,name=asset_file_def_index,json=assetFileDefIndex,proto3" json:"asset_file_def_index,omitempty"`
	// contains filtered or unexported fields
}

A SavedAsset points to an asset in the MetaGraph.

When bound to a function this object evaluates to a tensor with the absolute filename. Users should not depend on a particular part of the filename to remain stable (e.g. basename could be changed).

func (*SavedAsset) Descriptor deprecated 

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

Deprecated: Use SavedAsset.ProtoReflect.Descriptor instead.

func (*SavedAsset) GetAssetFileDefIndex 

func (x *SavedAsset) GetAssetFileDefIndex() int32

func (*SavedAsset) ProtoMessage 

func (*SavedAsset) ProtoMessage()

func (*SavedAsset) ProtoReflect 

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

func (*SavedAsset) Reset 

func (x *SavedAsset) Reset()

func (*SavedAsset) String 

func (x *SavedAsset) String() string

type SavedBareConcreteFunction 

type SavedBareConcreteFunction struct {

	// Identifies a SavedConcreteFunction.
	ConcreteFunctionName string `protobuf:"bytes,1,opt,name=concrete_function_name,json=concreteFunctionName,proto3" json:"concrete_function_name,omitempty"`
	// A sequence of unique strings, one per Tensor argument.
	ArgumentKeywords []string `protobuf:"bytes,2,rep,name=argument_keywords,json=argumentKeywords,proto3" json:"argument_keywords,omitempty"`
	// The prefix of `argument_keywords` which may be identified by position.
	AllowedPositionalArguments int64 `` /* 142-byte string literal not displayed */
	// The spec of the function that this ConcreteFunction is traced from. This
	// allows the ConcreteFunction to be called with nest structure inputs. This
	// field may not be populated. If this field is absent, the concrete function
	// can only be called with flat inputs.
	// TODO(b/169361281): support calling saved ConcreteFunction with structured
	// inputs in C++ SavedModel API.
	FunctionSpec *FunctionSpec `protobuf:"bytes,4,opt,name=function_spec,json=functionSpec,proto3" json:"function_spec,omitempty"`
	// contains filtered or unexported fields
}

func (*SavedBareConcreteFunction) Descriptor deprecated 

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

Deprecated: Use SavedBareConcreteFunction.ProtoReflect.Descriptor instead.

func (*SavedBareConcreteFunction) GetAllowedPositionalArguments 

func (x *SavedBareConcreteFunction) GetAllowedPositionalArguments() int64

func (*SavedBareConcreteFunction) GetArgumentKeywords 

func (x *SavedBareConcreteFunction) GetArgumentKeywords() []string

func (*SavedBareConcreteFunction) GetConcreteFunctionName 

func (x *SavedBareConcreteFunction) GetConcreteFunctionName() string

func (*SavedBareConcreteFunction) GetFunctionSpec 

func (x *SavedBareConcreteFunction) GetFunctionSpec() *FunctionSpec

func (*SavedBareConcreteFunction) ProtoMessage 

func (*SavedBareConcreteFunction) ProtoMessage()

func (*SavedBareConcreteFunction) ProtoReflect 

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

func (*SavedBareConcreteFunction) Reset 

func (x *SavedBareConcreteFunction) Reset()

func (*SavedBareConcreteFunction) String 

func (x *SavedBareConcreteFunction) String() string

type SavedConcreteFunction 

type SavedConcreteFunction struct {
	BoundInputs []int32 `protobuf:"varint,2,rep,packed,name=bound_inputs,json=boundInputs,proto3" json:"bound_inputs,omitempty"`
	// Input in canonicalized form that was received to create this concrete
	// function.
	CanonicalizedInputSignature *StructuredValue `` /* 144-byte string literal not displayed */
	// Output that was the return value of this function after replacing all
	// Tensors with TensorSpecs. This can be an arbitrary nested function and will
	// be used to reconstruct the full structure from pure tensors.
	OutputSignature *StructuredValue `protobuf:"bytes,4,opt,name=output_signature,json=outputSignature,proto3" json:"output_signature,omitempty"`
	// contains filtered or unexported fields
}

Stores low-level information about a concrete function. Referenced in either a SavedFunction or a SavedBareConcreteFunction.

func (*SavedConcreteFunction) Descriptor deprecated 

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

Deprecated: Use SavedConcreteFunction.ProtoReflect.Descriptor instead.

func (*SavedConcreteFunction) GetBoundInputs 

func (x *SavedConcreteFunction) GetBoundInputs() []int32

func (*SavedConcreteFunction) GetCanonicalizedInputSignature 

func (x *SavedConcreteFunction) GetCanonicalizedInputSignature() *StructuredValue

func (*SavedConcreteFunction) GetOutputSignature 

func (x *SavedConcreteFunction) GetOutputSignature() *StructuredValue

func (*SavedConcreteFunction) ProtoMessage 

func (*SavedConcreteFunction) ProtoMessage()

func (*SavedConcreteFunction) ProtoReflect 

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

func (*SavedConcreteFunction) Reset 

func (x *SavedConcreteFunction) Reset()

func (*SavedConcreteFunction) String 

func (x *SavedConcreteFunction) String() string

type SavedConstant 

type SavedConstant struct {

	// An Operation name for a ConstantOp in this SavedObjectGraph's MetaGraph.
	Operation string `protobuf:"bytes,1,opt,name=operation,proto3" json:"operation,omitempty"`
	// contains filtered or unexported fields
}

func (*SavedConstant) Descriptor deprecated 

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

Deprecated: Use SavedConstant.ProtoReflect.Descriptor instead.

func (*SavedConstant) GetOperation 

func (x *SavedConstant) GetOperation() string

func (*SavedConstant) ProtoMessage 

func (*SavedConstant) ProtoMessage()

func (*SavedConstant) ProtoReflect 

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

func (*SavedConstant) Reset 

func (x *SavedConstant) Reset()

func (*SavedConstant) String 

func (x *SavedConstant) String() string

type SavedFunction 

type SavedFunction struct {
	ConcreteFunctions []string      `protobuf:"bytes,1,rep,name=concrete_functions,json=concreteFunctions,proto3" json:"concrete_functions,omitempty"`
	FunctionSpec      *FunctionSpec `protobuf:"bytes,2,opt,name=function_spec,json=functionSpec,proto3" json:"function_spec,omitempty"`
	// contains filtered or unexported fields
}

A function with multiple signatures, possibly with non-Tensor arguments.

func (*SavedFunction) Descriptor deprecated 

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

Deprecated: Use SavedFunction.ProtoReflect.Descriptor instead.

func (*SavedFunction) GetConcreteFunctions 

func (x *SavedFunction) GetConcreteFunctions() []string

func (*SavedFunction) GetFunctionSpec 

func (x *SavedFunction) GetFunctionSpec() *FunctionSpec

func (*SavedFunction) ProtoMessage 

func (*SavedFunction) ProtoMessage()

func (*SavedFunction) ProtoReflect 

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

func (*SavedFunction) Reset 

func (x *SavedFunction) Reset()

func (*SavedFunction) String 

func (x *SavedFunction) String() string

type SavedModel 

type SavedModel struct {

	// The schema version of the SavedModel instance. Used for versioning when
	// making future changes to the specification/implementation. Initial value
	// at release will be 1.
	SavedModelSchemaVersion int64 `` /* 135-byte string literal not displayed */
	// One or more MetaGraphs.
	MetaGraphs []*MetaGraphDef `protobuf:"bytes,2,rep,name=meta_graphs,json=metaGraphs,proto3" json:"meta_graphs,omitempty"`
	// contains filtered or unexported fields
}

SavedModel is the high level serialization format for TensorFlow Models. See [todo: doc links, similar to session_bundle] for more information.

func (*SavedModel) Descriptor deprecated 

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

Deprecated: Use SavedModel.ProtoReflect.Descriptor instead.

func (*SavedModel) GetMetaGraphs 

func (x *SavedModel) GetMetaGraphs() []*MetaGraphDef

func (*SavedModel) GetSavedModelSchemaVersion 

func (x *SavedModel) GetSavedModelSchemaVersion() int64

func (*SavedModel) ProtoMessage 

func (*SavedModel) ProtoMessage()

func (*SavedModel) ProtoReflect 

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

func (*SavedModel) Reset 

func (x *SavedModel) Reset()

func (*SavedModel) String 

func (x *SavedModel) String() string

type SavedObject 

type SavedObject struct {

	// Objects which this object depends on: named edges in the dependency
	// graph.
	//
	// Note: All kinds of SavedObject may have children, except
	// "constant" and "captured_tensor".
	Children []*TrackableObjectGraph_TrackableObject_ObjectReference `protobuf:"bytes,1,rep,name=children,proto3" json:"children,omitempty"`
	// Ordered list of dependencies that must be loaded before this object.
	// SavedModel loads with the bottom-up approach, by first creating all objects
	// (in the order defined by the dependencies), then connecting the edges.
	Dependencies []*TrackableObjectGraph_TrackableObject_ObjectReference `protobuf:"bytes,15,rep,name=dependencies,proto3" json:"dependencies,omitempty"`
	// Slot variables owned by this object. This describes the three-way
	// (optimizer, variable, slot variable) relationship; none of the three
	// depend on the others directly.
	//
	// Note: currently only valid if kind == "user_object".
	SlotVariables []*TrackableObjectGraph_TrackableObject_SlotVariableReference `protobuf:"bytes,3,rep,name=slot_variables,json=slotVariables,proto3" json:"slot_variables,omitempty"`
	// Types that are assignable to Kind:
	//	*SavedObject_UserObject
	//	*SavedObject_Asset
	//	*SavedObject_Function
	//	*SavedObject_Variable
	//	*SavedObject_BareConcreteFunction
	//	*SavedObject_Constant
	//	*SavedObject_Resource
	//	*SavedObject_CapturedTensor
	Kind isSavedObject_Kind `protobuf_oneof:"kind"`
	// Stores the functions used to save and restore this object. At most one of
	// `saveable_objects` or `registered_saver` is defined for each SavedObject.
	// See the comment below for the difference between SaveableObject and
	// registered savers.
	SaveableObjects map[string]*SaveableObject `` /* 195-byte string literal not displayed */
	// The name of the registered class of the form "{package}.{class_name}".
	// This field is used to search for the registered class at loading time.
	RegisteredName string `protobuf:"bytes,13,opt,name=registered_name,json=registeredName,proto3" json:"registered_name,omitempty"`
	// The user-generated proto storing metadata for this object, to be passed to
	// the registered classes's _deserialize_from_proto method when this object is
	// loaded from the SavedModel.
	SerializedUserProto *any.Any `protobuf:"bytes,14,opt,name=serialized_user_proto,json=serializedUserProto,proto3" json:"serialized_user_proto,omitempty"`
	// String name of the registered saver. At most one of `saveable_objects` or
	// `registered_saver` is defined for each SavedObject.
	RegisteredSaver string `protobuf:"bytes,16,opt,name=registered_saver,json=registeredSaver,proto3" json:"registered_saver,omitempty"`
	// contains filtered or unexported fields
}

func (*SavedObject) Descriptor deprecated 

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

Deprecated: Use SavedObject.ProtoReflect.Descriptor instead.

func (*SavedObject) GetAsset 

func (x *SavedObject) GetAsset() *SavedAsset

func (*SavedObject) GetBareConcreteFunction 

func (x *SavedObject) GetBareConcreteFunction() *SavedBareConcreteFunction

func (*SavedObject) GetCapturedTensor 

func (x *SavedObject) GetCapturedTensor() *CapturedTensor

func (*SavedObject) GetChildren 

func (x *SavedObject) GetChildren() []*TrackableObjectGraph_TrackableObject_ObjectReference

func (*SavedObject) GetConstant 

func (x *SavedObject) GetConstant() *SavedConstant

func (*SavedObject) GetDependencies 

func (x *SavedObject) GetDependencies() []*TrackableObjectGraph_TrackableObject_ObjectReference

func (*SavedObject) GetFunction 

func (x *SavedObject) GetFunction() *SavedFunction

func (*SavedObject) GetKind 

func (m *SavedObject) GetKind() isSavedObject_Kind

func (*SavedObject) GetRegisteredName 

func (x *SavedObject) GetRegisteredName() string

func (*SavedObject) GetRegisteredSaver 

func (x *SavedObject) GetRegisteredSaver() string

func (*SavedObject) GetResource 

func (x *SavedObject) GetResource() *SavedResource

func (*SavedObject) GetSaveableObjects 

func (x *SavedObject) GetSaveableObjects() map[string]*SaveableObject

func (*SavedObject) GetSerializedUserProto 

func (x *SavedObject) GetSerializedUserProto() *any.Any

func (*SavedObject) GetSlotVariables 

func (x *SavedObject) GetSlotVariables() []*TrackableObjectGraph_TrackableObject_SlotVariableReference

func (*SavedObject) GetUserObject 

func (x *SavedObject) GetUserObject() *SavedUserObject

func (*SavedObject) GetVariable 

func (x *SavedObject) GetVariable() *SavedVariable

func (*SavedObject) ProtoMessage 

func (*SavedObject) ProtoMessage()

func (*SavedObject) ProtoReflect 

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

func (*SavedObject) Reset 

func (x *SavedObject) Reset()

func (*SavedObject) String 

func (x *SavedObject) String() string

type SavedObjectGraph 

type SavedObjectGraph struct {

	// Flattened list of objects in the object graph.
	//
	// The position of the object in this list indicates its id.
	// Nodes[0] is considered the root node.
	Nodes []*SavedObject `protobuf:"bytes,1,rep,name=nodes,proto3" json:"nodes,omitempty"`
	// Information about captures and output structures in concrete functions.
	// Referenced from SavedBareConcreteFunction and SavedFunction.
	ConcreteFunctions map[string]*SavedConcreteFunction `` /* 200-byte string literal not displayed */
	// contains filtered or unexported fields
}

func (*SavedObjectGraph) Descriptor deprecated 

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

Deprecated: Use SavedObjectGraph.ProtoReflect.Descriptor instead.

func (*SavedObjectGraph) GetConcreteFunctions 

func (x *SavedObjectGraph) GetConcreteFunctions() map[string]*SavedConcreteFunction

func (*SavedObjectGraph) GetNodes 

func (x *SavedObjectGraph) GetNodes() []*SavedObject

func (*SavedObjectGraph) ProtoMessage 

func (*SavedObjectGraph) ProtoMessage()

func (*SavedObjectGraph) ProtoReflect 

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

func (*SavedObjectGraph) Reset 

func (x *SavedObjectGraph) Reset()

func (*SavedObjectGraph) String 

func (x *SavedObjectGraph) String() string

type SavedObject_Asset 

type SavedObject_Asset struct {
	Asset *SavedAsset `protobuf:"bytes,5,opt,name=asset,proto3,oneof"`
}

type SavedObject_BareConcreteFunction 

type SavedObject_BareConcreteFunction struct {
	BareConcreteFunction *SavedBareConcreteFunction `protobuf:"bytes,8,opt,name=bare_concrete_function,json=bareConcreteFunction,proto3,oneof"`
}

type SavedObject_CapturedTensor 

type SavedObject_CapturedTensor struct {
	CapturedTensor *CapturedTensor `protobuf:"bytes,12,opt,name=captured_tensor,json=capturedTensor,proto3,oneof"`
}

type SavedObject_Constant 

type SavedObject_Constant struct {
	Constant *SavedConstant `protobuf:"bytes,9,opt,name=constant,proto3,oneof"`
}

type SavedObject_Function 

type SavedObject_Function struct {
	Function *SavedFunction `protobuf:"bytes,6,opt,name=function,proto3,oneof"`
}

type SavedObject_Resource 

type SavedObject_Resource struct {
	Resource *SavedResource `protobuf:"bytes,10,opt,name=resource,proto3,oneof"`
}

type SavedObject_UserObject 

type SavedObject_UserObject struct {
	UserObject *SavedUserObject `protobuf:"bytes,4,opt,name=user_object,json=userObject,proto3,oneof"`
}

type SavedObject_Variable 

type SavedObject_Variable struct {
	Variable *SavedVariable `protobuf:"bytes,7,opt,name=variable,proto3,oneof"`
}

type SavedResource 

type SavedResource struct {

	// A device specification indicating a required placement for the resource
	// creation function, e.g. "CPU". An empty string allows the user to select a
	// device.
	Device string `protobuf:"bytes,1,opt,name=device,proto3" json:"device,omitempty"`
	// contains filtered or unexported fields
}

A SavedResource represents a TF object that holds state during its lifetime. An object of this type can have a reference to a: create_resource() and an initialize() function.

func (*SavedResource) Descriptor deprecated 

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

Deprecated: Use SavedResource.ProtoReflect.Descriptor instead.

func (*SavedResource) GetDevice 

func (x *SavedResource) GetDevice() string

func (*SavedResource) ProtoMessage 

func (*SavedResource) ProtoMessage()

func (*SavedResource) ProtoReflect 

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

func (*SavedResource) Reset 

func (x *SavedResource) Reset()

func (*SavedResource) String 

func (x *SavedResource) String() string

type SavedUserObject 

type SavedUserObject struct {

	// Corresponds to a registration of the type to use in the loading program.
	Identifier string `protobuf:"bytes,1,opt,name=identifier,proto3" json:"identifier,omitempty"`
	// Version information from the producer of this SavedUserObject.
	Version *VersionDef `protobuf:"bytes,2,opt,name=version,proto3" json:"version,omitempty"`
	// Metadata for deserializing this object.
	//
	// Deprecated! At the time of deprecation, Keras was the only user of this
	// field, and its saving and loading code will be updated shortly.
	// Please save your application-specific metadata to a separate file.
	//
	// Deprecated: Do not use.
	Metadata string `protobuf:"bytes,3,opt,name=metadata,proto3" json:"metadata,omitempty"`
	// contains filtered or unexported fields
}

A SavedUserObject is an object (in the object-oriented language of the TensorFlow program) of some user- or framework-defined class other than those handled specifically by the other kinds of SavedObjects.

This object cannot be evaluated as a tensor, and therefore cannot be bound to an input of a function.

func (*SavedUserObject) Descriptor deprecated 

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

Deprecated: Use SavedUserObject.ProtoReflect.Descriptor instead.

func (*SavedUserObject) GetIdentifier 

func (x *SavedUserObject) GetIdentifier() string

func (*SavedUserObject) GetMetadata deprecated 

func (x *SavedUserObject) GetMetadata() string

Deprecated: Do not use.

func (*SavedUserObject) GetVersion 

func (x *SavedUserObject) GetVersion() *VersionDef

func (*SavedUserObject) ProtoMessage 

func (*SavedUserObject) ProtoMessage()

func (*SavedUserObject) ProtoReflect 

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

func (*SavedUserObject) Reset 

func (x *SavedUserObject) Reset()

func (*SavedUserObject) String 

func (x *SavedUserObject) String() string

type SavedVariable 

type SavedVariable 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"`
	Trainable       bool                    `protobuf:"varint,3,opt,name=trainable,proto3" json:"trainable,omitempty"`
	Synchronization VariableSynchronization `protobuf:"varint,4,opt,name=synchronization,proto3,enum=tensorflow.VariableSynchronization" json:"synchronization,omitempty"`
	Aggregation     VariableAggregation     `protobuf:"varint,5,opt,name=aggregation,proto3,enum=tensorflow.VariableAggregation" json:"aggregation,omitempty"`
	Name            string                  `protobuf:"bytes,6,opt,name=name,proto3" json:"name,omitempty"`
	Device          string                  `protobuf:"bytes,7,opt,name=device,proto3" json:"device,omitempty"`
	// List of component variables for a distributed variable.
	//
	// When this field is non-empty, the SavedVariable will be assumed
	// to be a distributed variable defined by the components listed here.
	//
	// This is only supported by experimental loaders at the moment.
	ExperimentalDistributedVariableComponents []*SavedVariable `` /* 188-byte string literal not displayed */
	// contains filtered or unexported fields
}

Represents a Variable that is initialized by loading the contents from the checkpoint.

func (*SavedVariable) Descriptor deprecated 

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

Deprecated: Use SavedVariable.ProtoReflect.Descriptor instead.

func (*SavedVariable) GetAggregation 

func (x *SavedVariable) GetAggregation() VariableAggregation

func (*SavedVariable) GetDevice 

func (x *SavedVariable) GetDevice() string

func (*SavedVariable) GetDtype 

func (x *SavedVariable) GetDtype() DataType

func (*SavedVariable) GetExperimentalDistributedVariableComponents 

func (x *SavedVariable) GetExperimentalDistributedVariableComponents() []*SavedVariable

func (*SavedVariable) GetName 

func (x *SavedVariable) GetName() string

func (*SavedVariable) GetShape 

func (x *SavedVariable) GetShape() *TensorShapeProto

func (*SavedVariable) GetSynchronization 

func (x *SavedVariable) GetSynchronization() VariableSynchronization

func (*SavedVariable) GetTrainable 

func (x *SavedVariable) GetTrainable() bool

func (*SavedVariable) ProtoMessage 

func (*SavedVariable) ProtoMessage()

func (*SavedVariable) ProtoReflect 

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

func (*SavedVariable) Reset 

func (x *SavedVariable) Reset()

func (*SavedVariable) String 

func (x *SavedVariable) String() string

type SaverDef 

type SaverDef struct {

	// The name of the tensor in which to specify the filename when saving or
	// restoring a model checkpoint.
	FilenameTensorName string `protobuf:"bytes,1,opt,name=filename_tensor_name,json=filenameTensorName,proto3" json:"filename_tensor_name,omitempty"`
	// The operation to run when saving a model checkpoint.
	SaveTensorName string `protobuf:"bytes,2,opt,name=save_tensor_name,json=saveTensorName,proto3" json:"save_tensor_name,omitempty"`
	// The operation to run when restoring a model checkpoint.
	RestoreOpName string `protobuf:"bytes,3,opt,name=restore_op_name,json=restoreOpName,proto3" json:"restore_op_name,omitempty"`
	// Maximum number of checkpoints to keep.  If 0, no checkpoints are deleted.
	MaxToKeep int32 `protobuf:"varint,4,opt,name=max_to_keep,json=maxToKeep,proto3" json:"max_to_keep,omitempty"`
	// Shard the save files, one per device that has Variable nodes.
	Sharded bool `protobuf:"varint,5,opt,name=sharded,proto3" json:"sharded,omitempty"`
	// How often to keep an additional checkpoint. If not specified, only the last
	// "max_to_keep" checkpoints are kept; if specified, in addition to keeping
	// the last "max_to_keep" checkpoints, an additional checkpoint will be kept
	// for every n hours of training.
	KeepCheckpointEveryNHours float32                          `` /* 144-byte string literal not displayed */
	Version                   SaverDef_CheckpointFormatVersion `protobuf:"varint,7,opt,name=version,proto3,enum=tensorflow.SaverDef_CheckpointFormatVersion" json:"version,omitempty"`
	// contains filtered or unexported fields
}

Protocol buffer representing the configuration of a Saver.

func (*SaverDef) Descriptor deprecated 

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

Deprecated: Use SaverDef.ProtoReflect.Descriptor instead.

func (*SaverDef) GetFilenameTensorName 

func (x *SaverDef) GetFilenameTensorName() string

func (*SaverDef) GetKeepCheckpointEveryNHours 

func (x *SaverDef) GetKeepCheckpointEveryNHours() float32

func (*SaverDef) GetMaxToKeep 

func (x *SaverDef) GetMaxToKeep() int32

func (*SaverDef) GetRestoreOpName 

func (x *SaverDef) GetRestoreOpName() string

func (*SaverDef) GetSaveTensorName 

func (x *SaverDef) GetSaveTensorName() string

func (*SaverDef) GetSharded 

func (x *SaverDef) GetSharded() bool

func (*SaverDef) GetVersion 

func (x *SaverDef) GetVersion() SaverDef_CheckpointFormatVersion

func (*SaverDef) ProtoMessage 

func (*SaverDef) ProtoMessage()

func (*SaverDef) ProtoReflect 

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

func (*SaverDef) Reset 

func (x *SaverDef) Reset()

func (*SaverDef) String 

func (x *SaverDef) String() string

type SaverDef_CheckpointFormatVersion 

type SaverDef_CheckpointFormatVersion int32

A version number that identifies a different on-disk checkpoint format. Usually, each subclass of BaseSaverBuilder works with a particular version/format. However, it is possible that the same builder may be upgraded to support a newer checkpoint format in the future. 

const (
	// Internal legacy format.
	SaverDef_LEGACY SaverDef_CheckpointFormatVersion = 0
	// Deprecated format: tf.Saver() which works with tensorflow::table::Table.
	SaverDef_V1 SaverDef_CheckpointFormatVersion = 1
	// Current format: more efficient.
	SaverDef_V2 SaverDef_CheckpointFormatVersion = 2
)

func (SaverDef_CheckpointFormatVersion) Descriptor 

func (SaverDef_CheckpointFormatVersion) Descriptor() protoreflect.EnumDescriptor

func (SaverDef_CheckpointFormatVersion) Enum 

func (x SaverDef_CheckpointFormatVersion) Enum() *SaverDef_CheckpointFormatVersion

func (SaverDef_CheckpointFormatVersion) EnumDescriptor deprecated 

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

Deprecated: Use SaverDef_CheckpointFormatVersion.Descriptor instead.

func (SaverDef_CheckpointFormatVersion) Number 

func (x SaverDef_CheckpointFormatVersion) Number() protoreflect.EnumNumber

func (SaverDef_CheckpointFormatVersion) String 

func (x SaverDef_CheckpointFormatVersion) String() string

func (SaverDef_CheckpointFormatVersion) Type 

func (SaverDef_CheckpointFormatVersion) Type() protoreflect.EnumType

type ScopedAllocatorOptions 

type ScopedAllocatorOptions struct {

	// If present, only perform optimization for these ops.
	EnableOp []string `protobuf:"bytes,1,rep,name=enable_op,json=enableOp,proto3" json:"enable_op,omitempty"`
	// contains filtered or unexported fields
}

func (*ScopedAllocatorOptions) Descriptor deprecated 

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

Deprecated: Use ScopedAllocatorOptions.ProtoReflect.Descriptor instead.

func (*ScopedAllocatorOptions) GetEnableOp 

func (x *ScopedAllocatorOptions) GetEnableOp() []string

func (*ScopedAllocatorOptions) ProtoMessage 

func (*ScopedAllocatorOptions) ProtoMessage()

func (*ScopedAllocatorOptions) ProtoReflect 

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

func (*ScopedAllocatorOptions) Reset 

func (x *ScopedAllocatorOptions) Reset()

func (*ScopedAllocatorOptions) String 

func (x *ScopedAllocatorOptions) String() string

type SendPackedHandleOp 

type SendPackedHandleOp struct {

	// Op id of the remote packed TensorHandle.
	OpId       int64                        `protobuf:"varint,1,opt,name=op_id,json=opId,proto3" json:"op_id,omitempty"`
	Handles    []*SendPackedHandleOp_Handle `protobuf:"bytes,2,rep,name=handles,proto3" json:"handles,omitempty"`
	DeviceName string                       `protobuf:"bytes,3,opt,name=device_name,json=deviceName,proto3" json:"device_name,omitempty"`
	// contains filtered or unexported fields
}

Send a packed TensorHandle to a remote worker.

func (*SendPackedHandleOp) Descriptor deprecated 

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

Deprecated: Use SendPackedHandleOp.ProtoReflect.Descriptor instead.

func (*SendPackedHandleOp) GetDeviceName 

func (x *SendPackedHandleOp) GetDeviceName() string

func (*SendPackedHandleOp) GetHandles 

func (x *SendPackedHandleOp) GetHandles() []*SendPackedHandleOp_Handle

func (*SendPackedHandleOp) GetOpId 

func (x *SendPackedHandleOp) GetOpId() int64

func (*SendPackedHandleOp) ProtoMessage 

func (*SendPackedHandleOp) ProtoMessage()

func (*SendPackedHandleOp) ProtoReflect 

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

func (*SendPackedHandleOp) Reset 

func (x *SendPackedHandleOp) Reset()

func (*SendPackedHandleOp) String 

func (x *SendPackedHandleOp) String() string

type SendPackedHandleOp_Handle 

type SendPackedHandleOp_Handle struct {

	// Types that are assignable to Item:
	//	*SendPackedHandleOp_Handle_LocalHandle
	//	*SendPackedHandleOp_Handle_RemoteHandle
	Item isSendPackedHandleOp_Handle_Item `protobuf_oneof:"item"`
	// contains filtered or unexported fields
}

func (*SendPackedHandleOp_Handle) Descriptor deprecated 

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

Deprecated: Use SendPackedHandleOp_Handle.ProtoReflect.Descriptor instead.

func (*SendPackedHandleOp_Handle) GetItem 

func (m *SendPackedHandleOp_Handle) GetItem() isSendPackedHandleOp_Handle_Item

func (*SendPackedHandleOp_Handle) GetLocalHandle 

func (x *SendPackedHandleOp_Handle) GetLocalHandle() *SendPackedHandleOp_LocalTensorHandle

func (*SendPackedHandleOp_Handle) GetRemoteHandle 

func (x *SendPackedHandleOp_Handle) GetRemoteHandle() *RemoteTensorHandle

func (*SendPackedHandleOp_Handle) ProtoMessage 

func (*SendPackedHandleOp_Handle) ProtoMessage()

func (*SendPackedHandleOp_Handle) ProtoReflect 

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

func (*SendPackedHandleOp_Handle) Reset 

func (x *SendPackedHandleOp_Handle) Reset()

func (*SendPackedHandleOp_Handle) String 

func (x *SendPackedHandleOp_Handle) String() string

type SendPackedHandleOp_Handle_LocalHandle 

type SendPackedHandleOp_Handle_LocalHandle struct {
	LocalHandle *SendPackedHandleOp_LocalTensorHandle `protobuf:"bytes,1,opt,name=local_handle,json=localHandle,proto3,oneof"`
}

type SendPackedHandleOp_Handle_RemoteHandle 

type SendPackedHandleOp_Handle_RemoteHandle struct {
	RemoteHandle *RemoteTensorHandle `protobuf:"bytes,2,opt,name=remote_handle,json=remoteHandle,proto3,oneof"`
}

type SendPackedHandleOp_LocalTensorHandle 

type SendPackedHandleOp_LocalTensorHandle struct {
	Tensor *TensorProto `protobuf:"bytes,1,opt,name=tensor,proto3" json:"tensor,omitempty"`
	// Device where the tensor is produced.
	Device string `protobuf:"bytes,2,opt,name=device,proto3" json:"device,omitempty"`
	// contains filtered or unexported fields
}

func (*SendPackedHandleOp_LocalTensorHandle) Descriptor deprecated 

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

Deprecated: Use SendPackedHandleOp_LocalTensorHandle.ProtoReflect.Descriptor instead.

func (*SendPackedHandleOp_LocalTensorHandle) GetDevice 

func (x *SendPackedHandleOp_LocalTensorHandle) GetDevice() string

func (*SendPackedHandleOp_LocalTensorHandle) GetTensor 

func (x *SendPackedHandleOp_LocalTensorHandle) GetTensor() *TensorProto

func (*SendPackedHandleOp_LocalTensorHandle) ProtoMessage 

func (*SendPackedHandleOp_LocalTensorHandle) ProtoMessage()

func (*SendPackedHandleOp_LocalTensorHandle) ProtoReflect 

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

func (*SendPackedHandleOp_LocalTensorHandle) Reset 

func (x *SendPackedHandleOp_LocalTensorHandle) Reset()

func (*SendPackedHandleOp_LocalTensorHandle) String 

func (x *SendPackedHandleOp_LocalTensorHandle) String() string

type SendTensorOp 

type SendTensorOp struct {

	// All remote tensors are identified by <Op ID, Output num>. To mimic this
	// situation when directly sending tensors, we include an "artificial" op ID
	// (which would have corresponded to the _Recv op when not using SendTensor).
	OpId int64 `protobuf:"varint,1,opt,name=op_id,json=opId,proto3" json:"op_id,omitempty"`
	// The index within the repeated field is the output number that will help
	// uniquely identify (along with the above op_id) the particular tensor.
	Tensors []*TensorProto `protobuf:"bytes,2,rep,name=tensors,proto3" json:"tensors,omitempty"`
	// The device on which the tensors should be resident.
	DeviceName string `protobuf:"bytes,3,opt,name=device_name,json=deviceName,proto3" json:"device_name,omitempty"`
	// contains filtered or unexported fields
}

func (*SendTensorOp) Descriptor deprecated 

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

Deprecated: Use SendTensorOp.ProtoReflect.Descriptor instead.

func (*SendTensorOp) GetDeviceName 

func (x *SendTensorOp) GetDeviceName() string

func (*SendTensorOp) GetOpId 

func (x *SendTensorOp) GetOpId() int64

func (*SendTensorOp) GetTensors 

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

func (*SendTensorOp) ProtoMessage 

func (*SendTensorOp) ProtoMessage()

func (*SendTensorOp) ProtoReflect 

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

func (*SendTensorOp) Reset 

func (x *SendTensorOp) Reset()

func (*SendTensorOp) String 

func (x *SendTensorOp) String() string

type SequenceExample 

type SequenceExample struct {
	Context      *Features     `protobuf:"bytes,1,opt,name=context,proto3" json:"context,omitempty"`
	FeatureLists *FeatureLists `protobuf:"bytes,2,opt,name=feature_lists,json=featureLists,proto3" json:"feature_lists,omitempty"`
	// contains filtered or unexported fields
}

func (*SequenceExample) Descriptor deprecated 

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

Deprecated: Use SequenceExample.ProtoReflect.Descriptor instead.

func (*SequenceExample) GetContext 

func (x *SequenceExample) GetContext() *Features

func (*SequenceExample) GetFeatureLists 

func (x *SequenceExample) GetFeatureLists() *FeatureLists

func (*SequenceExample) ProtoMessage 

func (*SequenceExample) ProtoMessage()

func (*SequenceExample) ProtoReflect 

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

func (*SequenceExample) Reset 

func (x *SequenceExample) Reset()

func (*SequenceExample) String 

func (x *SequenceExample) 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 ServerDef 

type ServerDef struct {

	// The cluster of which this server is a member.
	Cluster *ClusterDef `protobuf:"bytes,1,opt,name=cluster,proto3" json:"cluster,omitempty"`
	// The name of the job of which this server is a member.
	//
	// NOTE(mrry): The `cluster` field must contain a `JobDef` with a `name` field
	// that matches this name.
	JobName string `protobuf:"bytes,2,opt,name=job_name,json=jobName,proto3" json:"job_name,omitempty"`
	// The task index of this server in its job.
	//
	// NOTE: The `cluster` field must contain a `JobDef` with a matching `name`
	// and a mapping in its `tasks` field for this index.
	TaskIndex int32 `protobuf:"varint,3,opt,name=task_index,json=taskIndex,proto3" json:"task_index,omitempty"`
	// The default configuration for sessions that run on this server.
	DefaultSessionConfig *ConfigProto `protobuf:"bytes,4,opt,name=default_session_config,json=defaultSessionConfig,proto3" json:"default_session_config,omitempty"`
	// The protocol to be used by this server.
	//
	// Acceptable values include: "grpc", "grpc+verbs".
	Protocol string `protobuf:"bytes,5,opt,name=protocol,proto3" json:"protocol,omitempty"`
	// The server port. If not set, then we identify the port from the job_name.
	Port int32 `protobuf:"varint,6,opt,name=port,proto3" json:"port,omitempty"`
	// Device filters for remote tasks in the cluster.
	// NOTE: This is an experimental feature and only effective in TensorFlow 2.x.
	ClusterDeviceFilters *ClusterDeviceFilters `protobuf:"bytes,7,opt,name=cluster_device_filters,json=clusterDeviceFilters,proto3" json:"cluster_device_filters,omitempty"`
	// contains filtered or unexported fields
}

Defines the configuration of a single TensorFlow server.

func (*ServerDef) Descriptor deprecated 

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

Deprecated: Use ServerDef.ProtoReflect.Descriptor instead.

func (*ServerDef) GetCluster 

func (x *ServerDef) GetCluster() *ClusterDef

func (*ServerDef) GetClusterDeviceFilters 

func (x *ServerDef) GetClusterDeviceFilters() *ClusterDeviceFilters

func (*ServerDef) GetDefaultSessionConfig 

func (x *ServerDef) GetDefaultSessionConfig() *ConfigProto

func (*ServerDef) GetJobName 

func (x *ServerDef) GetJobName() string

func (*ServerDef) GetPort 

func (x *ServerDef) GetPort() int32

func (*ServerDef) GetProtocol 

func (x *ServerDef) GetProtocol() string

func (*ServerDef) GetTaskIndex 

func (x *ServerDef) GetTaskIndex() int32

func (*ServerDef) ProtoMessage 

func (*ServerDef) ProtoMessage()

func (*ServerDef) ProtoReflect 

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

func (*ServerDef) Reset 

func (x *ServerDef) Reset()

func (*ServerDef) String 

func (x *ServerDef) String() string

type SessionLog 

type SessionLog struct {
	Status SessionLog_SessionStatus `protobuf:"varint,1,opt,name=status,proto3,enum=tensorflow.SessionLog_SessionStatus" json:"status,omitempty"`
	// This checkpoint_path contains both the path and filename.
	CheckpointPath string `protobuf:"bytes,2,opt,name=checkpoint_path,json=checkpointPath,proto3" json:"checkpoint_path,omitempty"`
	Msg            string `protobuf:"bytes,3,opt,name=msg,proto3" json:"msg,omitempty"`
	// contains filtered or unexported fields
}

Protocol buffer used for logging session state.

func (*SessionLog) Descriptor deprecated 

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

Deprecated: Use SessionLog.ProtoReflect.Descriptor instead.

func (*SessionLog) GetCheckpointPath 

func (x *SessionLog) GetCheckpointPath() string

func (*SessionLog) GetMsg 

func (x *SessionLog) GetMsg() string

func (*SessionLog) GetStatus 

func (x *SessionLog) GetStatus() SessionLog_SessionStatus

func (*SessionLog) ProtoMessage 

func (*SessionLog) ProtoMessage()

func (*SessionLog) ProtoReflect 

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

func (*SessionLog) Reset 

func (x *SessionLog) Reset()

func (*SessionLog) String 

func (x *SessionLog) String() string

type SessionLog_SessionStatus 

type SessionLog_SessionStatus int32
const (
	SessionLog_STATUS_UNSPECIFIED SessionLog_SessionStatus = 0
	SessionLog_START              SessionLog_SessionStatus = 1
	SessionLog_STOP               SessionLog_SessionStatus = 2
	SessionLog_CHECKPOINT         SessionLog_SessionStatus = 3
)

func (SessionLog_SessionStatus) Descriptor 

func (SessionLog_SessionStatus) Descriptor() protoreflect.EnumDescriptor

func (SessionLog_SessionStatus) Enum 

func (x SessionLog_SessionStatus) Enum() *SessionLog_SessionStatus

func (SessionLog_SessionStatus) EnumDescriptor deprecated 

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

Deprecated: Use SessionLog_SessionStatus.Descriptor instead.

func (SessionLog_SessionStatus) Number 

func (x SessionLog_SessionStatus) Number() protoreflect.EnumNumber

func (SessionLog_SessionStatus) String 

func (x SessionLog_SessionStatus) String() string

func (SessionLog_SessionStatus) Type 

func (SessionLog_SessionStatus) Type() protoreflect.EnumType

type SessionMetadata 

type SessionMetadata struct {
	Name string `protobuf:"bytes,1,opt,name=name,proto3" json:"name,omitempty"`
	// The version is optional. If set, needs to be >= 0.
	Version int64 `protobuf:"varint,2,opt,name=version,proto3" json:"version,omitempty"`
	// contains filtered or unexported fields
}

Metadata about the session.

This can be used by the runtime and the Ops for debugging, monitoring, etc.

The (name, version) tuple is expected to be a unique identifier for sessions within the same process.

NOTE: This is currently used and propagated only by the direct session.

func (*SessionMetadata) Descriptor deprecated 

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

Deprecated: Use SessionMetadata.ProtoReflect.Descriptor instead.

func (*SessionMetadata) GetName 

func (x *SessionMetadata) GetName() string

func (*SessionMetadata) GetVersion 

func (x *SessionMetadata) GetVersion() int64

func (*SessionMetadata) ProtoMessage 

func (*SessionMetadata) ProtoMessage()

func (*SessionMetadata) ProtoReflect 

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

func (*SessionMetadata) Reset 

func (x *SessionMetadata) Reset()

func (*SessionMetadata) String 

func (x *SessionMetadata) String() string

type ShutdownRequest 

type ShutdownRequest struct {
	SessionId uint64 `protobuf:"varint,1,opt,name=session_id,json=sessionId,proto3" json:"session_id,omitempty"`
	NodeId    int32  `protobuf:"varint,2,opt,name=node_id,json=nodeId,proto3" json:"node_id,omitempty"`
	// contains filtered or unexported fields
}

func (*ShutdownRequest) Descriptor deprecated 

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

Deprecated: Use ShutdownRequest.ProtoReflect.Descriptor instead.

func (*ShutdownRequest) GetNodeId 

func (x *ShutdownRequest) GetNodeId() int32

func (*ShutdownRequest) GetSessionId 

func (x *ShutdownRequest) GetSessionId() uint64

func (*ShutdownRequest) ProtoMessage 

func (*ShutdownRequest) ProtoMessage()

func (*ShutdownRequest) ProtoReflect 

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

func (*ShutdownRequest) Reset 

func (x *ShutdownRequest) Reset()

func (*ShutdownRequest) String 

func (x *ShutdownRequest) String() string

type ShutdownResponse 

type ShutdownResponse struct {
	// contains filtered or unexported fields
}

func (*ShutdownResponse) Descriptor deprecated 

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

Deprecated: Use ShutdownResponse.ProtoReflect.Descriptor instead.

func (*ShutdownResponse) ProtoMessage 

func (*ShutdownResponse) ProtoMessage()

func (*ShutdownResponse) ProtoReflect 

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

func (*ShutdownResponse) Reset 

func (x *ShutdownResponse) Reset()

func (*ShutdownResponse) String 

func (x *ShutdownResponse) String() string

type ShutdownTaskRequest 

type ShutdownTaskRequest struct {
	SourceTask *CoordinatedTask `protobuf:"bytes,1,opt,name=source_task,json=sourceTask,proto3" json:"source_task,omitempty"`
	// contains filtered or unexported fields
}

Request and response messages for disconnecting a task from the service.

func (*ShutdownTaskRequest) Descriptor deprecated 

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

Deprecated: Use ShutdownTaskRequest.ProtoReflect.Descriptor instead.

func (*ShutdownTaskRequest) GetSourceTask 

func (x *ShutdownTaskRequest) GetSourceTask() *CoordinatedTask

func (*ShutdownTaskRequest) ProtoMessage 

func (*ShutdownTaskRequest) ProtoMessage()

func (*ShutdownTaskRequest) ProtoReflect 

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

func (*ShutdownTaskRequest) Reset 

func (x *ShutdownTaskRequest) Reset()

func (*ShutdownTaskRequest) String 

func (x *ShutdownTaskRequest) String() string

type ShutdownTaskResponse 

type ShutdownTaskResponse struct {
	// contains filtered or unexported fields
}

func (*ShutdownTaskResponse) Descriptor deprecated 

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

Deprecated: Use ShutdownTaskResponse.ProtoReflect.Descriptor instead.

func (*ShutdownTaskResponse) ProtoMessage 

func (*ShutdownTaskResponse) ProtoMessage()

func (*ShutdownTaskResponse) ProtoReflect 

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

func (*ShutdownTaskResponse) Reset 

func (x *ShutdownTaskResponse) Reset()

func (*ShutdownTaskResponse) String 

func (x *ShutdownTaskResponse) String() string

type SignatureDef 

type SignatureDef struct {

	// Named input parameters.
	Inputs map[string]*TensorInfo `` /* 153-byte string literal not displayed */
	// Named output parameters.
	Outputs map[string]*TensorInfo `` /* 155-byte string literal not displayed */
	// Extensible method_name information enabling third-party users to mark a
	// SignatureDef as supporting a particular method. This enables producers and
	// consumers of SignatureDefs, e.g. a model definition library and a serving
	// library to have a clear hand-off regarding the semantics of a computation.
	//
	// Note that multiple SignatureDefs in a single MetaGraphDef may have the same
	// method_name. This is commonly used to support multi-headed computation,
	// where a single graph computation may return multiple results.
	MethodName string `protobuf:"bytes,3,opt,name=method_name,json=methodName,proto3" json:"method_name,omitempty"`
	// contains filtered or unexported fields
}

SignatureDef defines the signature of a computation supported by a TensorFlow graph.

For example, a model with two loss computations, sharing a single input, might have the following signature_def map, in a MetaGraphDef message.

Note that across the two SignatureDefs "loss_A" and "loss_B", the input key, output key, and method_name are identical, and will be used by system(s) that implement or rely upon this particular loss method. The output tensor names differ, demonstrating how different outputs can exist for the same method. 

signature_def {
  key: "loss_A"
  value {
    inputs {
      key: "input"
      value {
        name: "input:0"
        dtype: DT_STRING
        tensor_shape: ...
      }
    }
    outputs {
      key: "loss_output"
      value {
        name: "loss_output_A:0"
        dtype: DT_FLOAT
        tensor_shape: ...
      }
    }
    method_name: "some/package/compute_loss"
  }
  ...
}

signature_def {
  key: "loss_B"
  value {
    inputs {
      key: "input"
      value {
        name: "input:0"
        dtype: DT_STRING
        tensor_shape: ...
      }
    }
    outputs {
      key: "loss_output"
      value {
        name: "loss_output_B:0"
        dtype: DT_FLOAT
        tensor_shape: ...
      }
    }
    method_name: "some/package/compute_loss"
  }
  ...
}

func (*SignatureDef) Descriptor deprecated 

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

Deprecated: Use SignatureDef.ProtoReflect.Descriptor instead.

func (*SignatureDef) GetInputs 

func (x *SignatureDef) GetInputs() map[string]*TensorInfo

func (*SignatureDef) GetMethodName 

func (x *SignatureDef) GetMethodName() string

func (*SignatureDef) GetOutputs 

func (x *SignatureDef) GetOutputs() map[string]*TensorInfo

func (*SignatureDef) ProtoMessage 

func (*SignatureDef) ProtoMessage()

func (*SignatureDef) ProtoReflect 

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

func (*SignatureDef) Reset 

func (x *SignatureDef) Reset()

func (*SignatureDef) String 

func (x *SignatureDef) String() string

type SnapShot 

type SnapShot struct {
	ActionCount uint64 `protobuf:"varint,1,opt,name=action_count,json=actionCount,proto3" json:"action_count,omitempty"`
	Size        int64  `protobuf:"varint,2,opt,name=size,proto3" json:"size,omitempty"`
	// contains filtered or unexported fields
}

func (*SnapShot) Descriptor deprecated 

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

Deprecated: Use SnapShot.ProtoReflect.Descriptor instead.

func (*SnapShot) GetActionCount 

func (x *SnapShot) GetActionCount() uint64

func (*SnapShot) GetSize 

func (x *SnapShot) GetSize() int64

func (*SnapShot) ProtoMessage 

func (*SnapShot) ProtoMessage()

func (*SnapShot) ProtoReflect 

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

func (*SnapShot) Reset 

func (x *SnapShot) Reset()

func (*SnapShot) String 

func (x *SnapShot) String() string

type SnapshotMetadataRecord 

type SnapshotMetadataRecord struct {

	// Stores the fingerprint of the graph that describes the dataset that is
	// snapshotted.
	GraphHash string `protobuf:"bytes,1,opt,name=graph_hash,json=graphHash,proto3" json:"graph_hash,omitempty"`
	// Run ID that this snapshot corresponds to.
	RunId string `protobuf:"bytes,2,opt,name=run_id,json=runId,proto3" json:"run_id,omitempty"`
	// Time when we started creating this snapshot.
	CreationTimestamp int64 `protobuf:"varint,3,opt,name=creation_timestamp,json=creationTimestamp,proto3" json:"creation_timestamp,omitempty"`
	// Version of the snapshot data file format.
	Version int64 `protobuf:"varint,4,opt,name=version,proto3" json:"version,omitempty"`
	// A list of tensor dtype corresponding to each element of the snapshot.
	Dtype []DataType `protobuf:"varint,5,rep,packed,name=dtype,proto3,enum=tensorflow.DataType" json:"dtype,omitempty"`
	// The number of elements in the snapshot.
	NumElements int64 `protobuf:"varint,6,opt,name=num_elements,json=numElements,proto3" json:"num_elements,omitempty"`
	Finalized   bool  `protobuf:"varint,1000,opt,name=finalized,proto3" json:"finalized,omitempty"`
	// contains filtered or unexported fields
}

This stores the metadata information present in each snapshot record.

func (*SnapshotMetadataRecord) Descriptor deprecated 

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

Deprecated: Use SnapshotMetadataRecord.ProtoReflect.Descriptor instead.

func (*SnapshotMetadataRecord) GetCreationTimestamp 

func (x *SnapshotMetadataRecord) GetCreationTimestamp() int64

func (*SnapshotMetadataRecord) GetDtype 

func (x *SnapshotMetadataRecord) GetDtype() []DataType

func (*SnapshotMetadataRecord) GetFinalized 

func (x *SnapshotMetadataRecord) GetFinalized() bool

func (*SnapshotMetadataRecord) GetGraphHash 

func (x *SnapshotMetadataRecord) GetGraphHash() string

func (*SnapshotMetadataRecord) GetNumElements 

func (x *SnapshotMetadataRecord) GetNumElements() int64

func (*SnapshotMetadataRecord) GetRunId 

func (x *SnapshotMetadataRecord) GetRunId() string

func (*SnapshotMetadataRecord) GetVersion 

func (x *SnapshotMetadataRecord) GetVersion() int64

func (*SnapshotMetadataRecord) ProtoMessage 

func (*SnapshotMetadataRecord) ProtoMessage()

func (*SnapshotMetadataRecord) ProtoReflect 

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

func (*SnapshotMetadataRecord) Reset 

func (x *SnapshotMetadataRecord) Reset()

func (*SnapshotMetadataRecord) String 

func (x *SnapshotMetadataRecord) String() string

type SnapshotRecord 

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

Each SnapshotRecord represents one batch of pre-processed input data. A batch consists of a list of tensors that we encode as TensorProtos. This message doesn't store the structure of the batch.

func (*SnapshotRecord) Descriptor deprecated 

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

Deprecated: Use SnapshotRecord.ProtoReflect.Descriptor instead.

func (*SnapshotRecord) GetTensor 

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

func (*SnapshotRecord) ProtoMessage 

func (*SnapshotRecord) ProtoMessage()

func (*SnapshotRecord) ProtoReflect 

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

func (*SnapshotRecord) Reset 

func (x *SnapshotRecord) Reset()

func (*SnapshotRecord) String 

func (x *SnapshotRecord) String() string

type SnapshotTensorMetadata 

type SnapshotTensorMetadata struct {
	TensorMetadata []*TensorMetadata `protobuf:"bytes,1,rep,name=tensor_metadata,json=tensorMetadata,proto3" json:"tensor_metadata,omitempty"`
	// contains filtered or unexported fields
}

Metadata for all the tensors in a Snapshot Record.

func (*SnapshotTensorMetadata) Descriptor deprecated 

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

Deprecated: Use SnapshotTensorMetadata.ProtoReflect.Descriptor instead.

func (*SnapshotTensorMetadata) GetTensorMetadata 

func (x *SnapshotTensorMetadata) GetTensorMetadata() []*TensorMetadata

func (*SnapshotTensorMetadata) ProtoMessage 

func (*SnapshotTensorMetadata) ProtoMessage()

func (*SnapshotTensorMetadata) ProtoReflect 

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

func (*SnapshotTensorMetadata) Reset 

func (x *SnapshotTensorMetadata) Reset()

func (*SnapshotTensorMetadata) String 

func (x *SnapshotTensorMetadata) String() string

type SourceFile 

type SourceFile struct {

	// Path to the file.
	FilePath string `protobuf:"bytes,1,opt,name=file_path,json=filePath,proto3" json:"file_path,omitempty"`
	// Name of the host on which the file is located.
	HostName string `protobuf:"bytes,2,opt,name=host_name,json=hostName,proto3" json:"host_name,omitempty"`
	// Line-by-line content of the file.
	Lines []string `protobuf:"bytes,3,rep,name=lines,proto3" json:"lines,omitempty"`
	// contains filtered or unexported fields
}

Content of a source file involved in the execution of the debugged TensorFlow program.

func (*SourceFile) Descriptor deprecated 

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

Deprecated: Use SourceFile.ProtoReflect.Descriptor instead.

func (*SourceFile) GetFilePath 

func (x *SourceFile) GetFilePath() string

func (*SourceFile) GetHostName 

func (x *SourceFile) GetHostName() string

func (*SourceFile) GetLines 

func (x *SourceFile) GetLines() []string

func (*SourceFile) ProtoMessage 

func (*SourceFile) ProtoMessage()

func (*SourceFile) ProtoReflect 

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

func (*SourceFile) Reset 

func (x *SourceFile) Reset()

func (*SourceFile) String 

func (x *SourceFile) String() string

type StackFrameWithId 

type StackFrameWithId struct {

	// A unique ID for the stack frame: A UUID-like string.
	Id string `protobuf:"bytes,1,opt,name=id,proto3" json:"id,omitempty"`
	// Stack frame, i.e., a frame of a stack trace, containing information
	// regarding the file name, line number, function name, code content
	// of the line, and column number (if available).
	FileLineCol *GraphDebugInfo_FileLineCol `protobuf:"bytes,2,opt,name=file_line_col,json=fileLineCol,proto3" json:"file_line_col,omitempty"`
	// contains filtered or unexported fields
}

A stack frame with ID.

func (*StackFrameWithId) Descriptor deprecated 

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

Deprecated: Use StackFrameWithId.ProtoReflect.Descriptor instead.

func (*StackFrameWithId) GetFileLineCol 

func (x *StackFrameWithId) GetFileLineCol() *GraphDebugInfo_FileLineCol

func (*StackFrameWithId) GetId 

func (x *StackFrameWithId) GetId() string

func (*StackFrameWithId) ProtoMessage 

func (*StackFrameWithId) ProtoMessage()

func (*StackFrameWithId) ProtoReflect 

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

func (*StackFrameWithId) Reset 

func (x *StackFrameWithId) Reset()

func (*StackFrameWithId) String 

func (x *StackFrameWithId) String() string

type StackTracePayload 

type StackTracePayload struct {
	StackFrames []*StackTracePayload_StackFrame `protobuf:"bytes,1,rep,name=stack_frames,json=stackFrames,proto3" json:"stack_frames,omitempty"`
	// contains filtered or unexported fields
}

If included as a payload, this message contains associated source location for the error Status.

func (*StackTracePayload) Descriptor deprecated 

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

Deprecated: Use StackTracePayload.ProtoReflect.Descriptor instead.

func (*StackTracePayload) GetStackFrames 

func (x *StackTracePayload) GetStackFrames() []*StackTracePayload_StackFrame

func (*StackTracePayload) ProtoMessage 

func (*StackTracePayload) ProtoMessage()

func (*StackTracePayload) ProtoReflect 

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

func (*StackTracePayload) Reset 

func (x *StackTracePayload) Reset()

func (*StackTracePayload) String 

func (x *StackTracePayload) String() string

type StackTracePayload_StackFrame 

type StackTracePayload_StackFrame struct {
	FileName     string `protobuf:"bytes,1,opt,name=file_name,json=fileName,proto3" json:"file_name,omitempty"`
	LineNumber   int32  `protobuf:"varint,2,opt,name=line_number,json=lineNumber,proto3" json:"line_number,omitempty"`
	FunctionName string `protobuf:"bytes,3,opt,name=function_name,json=functionName,proto3" json:"function_name,omitempty"`
	// contains filtered or unexported fields
}

func (*StackTracePayload_StackFrame) Descriptor deprecated 

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

Deprecated: Use StackTracePayload_StackFrame.ProtoReflect.Descriptor instead.

func (*StackTracePayload_StackFrame) GetFileName 

func (x *StackTracePayload_StackFrame) GetFileName() string

func (*StackTracePayload_StackFrame) GetFunctionName 

func (x *StackTracePayload_StackFrame) GetFunctionName() string

func (*StackTracePayload_StackFrame) GetLineNumber 

func (x *StackTracePayload_StackFrame) GetLineNumber() int32

func (*StackTracePayload_StackFrame) ProtoMessage 

func (*StackTracePayload_StackFrame) ProtoMessage()

func (*StackTracePayload_StackFrame) ProtoReflect 

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

func (*StackTracePayload_StackFrame) Reset 

func (x *StackTracePayload_StackFrame) Reset()

func (*StackTracePayload_StackFrame) String 

func (x *StackTracePayload_StackFrame) String() string

type StepSequence 

type StepSequence struct {
	GraphKey   int64 `protobuf:"varint,1,opt,name=graph_key,json=graphKey,proto3" json:"graph_key,omitempty"`
	NextStepId int64 `protobuf:"varint,2,opt,name=next_step_id,json=nextStepId,proto3" json:"next_step_id,omitempty"`
	// contains filtered or unexported fields
}

func (*StepSequence) Descriptor deprecated 

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

Deprecated: Use StepSequence.ProtoReflect.Descriptor instead.

func (*StepSequence) GetGraphKey 

func (x *StepSequence) GetGraphKey() int64

func (*StepSequence) GetNextStepId 

func (x *StepSequence) GetNextStepId() int64

func (*StepSequence) ProtoMessage 

func (*StepSequence) ProtoMessage()

func (*StepSequence) ProtoReflect 

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

func (*StepSequence) Reset 

func (x *StepSequence) Reset()

func (*StepSequence) String 

func (x *StepSequence) 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 StructuredValue 

type StructuredValue struct {

	// The kind of value.
	//
	// Types that are assignable to Kind:
	//	*StructuredValue_NoneValue
	//	*StructuredValue_Float64Value
	//	*StructuredValue_Int64Value
	//	*StructuredValue_StringValue
	//	*StructuredValue_BoolValue
	//	*StructuredValue_TensorShapeValue
	//	*StructuredValue_TensorDtypeValue
	//	*StructuredValue_TensorSpecValue
	//	*StructuredValue_TypeSpecValue
	//	*StructuredValue_BoundedTensorSpecValue
	//	*StructuredValue_ListValue
	//	*StructuredValue_TupleValue
	//	*StructuredValue_DictValue
	//	*StructuredValue_NamedTupleValue
	Kind isStructuredValue_Kind `protobuf_oneof:"kind"`
	// contains filtered or unexported fields
}

`StructuredValue` represents a dynamically typed value representing various data structures that are inspired by Python data structures typically used in TensorFlow functions as inputs and outputs.

For example when saving a Layer there may be a `training` argument. If the user passes a boolean True/False, that switches between two concrete TensorFlow functions. In order to switch between them in the same way after loading the SavedModel, we need to represent "True" and "False".

A more advanced example might be a function which takes a list of dictionaries mapping from strings to Tensors. In order to map from user-specified arguments `[{"a": tf.constant(1.)}, {"q": tf.constant(3.)}]` after load to the right saved TensorFlow function, we need to represent the nested structure and the strings, recording that we have a trace for anything matching `[{"a": tf.TensorSpec(None, tf.float32)}, {"q": tf.TensorSpec([], tf.float64)}]` as an example.

Likewise functions may return nested structures of Tensors, for example returning a dictionary mapping from strings to Tensors. In order for the loaded function to return the same structure we need to serialize it.

This is an ergonomic aid for working with loaded SavedModels, not a promise to serialize all possible function signatures. For example we do not expect to pickle generic Python objects, and ideally we'd stay language-agnostic.

func (*StructuredValue) Descriptor deprecated 

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

Deprecated: Use StructuredValue.ProtoReflect.Descriptor instead.

func (*StructuredValue) GetBoolValue 

func (x *StructuredValue) GetBoolValue() bool

func (*StructuredValue) GetBoundedTensorSpecValue 

func (x *StructuredValue) GetBoundedTensorSpecValue() *BoundedTensorSpecProto

func (*StructuredValue) GetDictValue 

func (x *StructuredValue) GetDictValue() *DictValue

func (*StructuredValue) GetFloat64Value 

func (x *StructuredValue) GetFloat64Value() float64

func (*StructuredValue) GetInt64Value 

func (x *StructuredValue) GetInt64Value() int64

func (*StructuredValue) GetKind 

func (m *StructuredValue) GetKind() isStructuredValue_Kind

func (*StructuredValue) GetListValue 

func (x *StructuredValue) GetListValue() *ListValue

func (*StructuredValue) GetNamedTupleValue 

func (x *StructuredValue) GetNamedTupleValue() *NamedTupleValue

func (*StructuredValue) GetNoneValue 

func (x *StructuredValue) GetNoneValue() *NoneValue

func (*StructuredValue) GetStringValue 

func (x *StructuredValue) GetStringValue() string

func (*StructuredValue) GetTensorDtypeValue 

func (x *StructuredValue) GetTensorDtypeValue() DataType

func (*StructuredValue) GetTensorShapeValue 

func (x *StructuredValue) GetTensorShapeValue() *TensorShapeProto

func (*StructuredValue) GetTensorSpecValue 

func (x *StructuredValue) GetTensorSpecValue() *TensorSpecProto

func (*StructuredValue) GetTupleValue 

func (x *StructuredValue) GetTupleValue() *TupleValue

func (*StructuredValue) GetTypeSpecValue 

func (x *StructuredValue) GetTypeSpecValue() *TypeSpecProto

func (*StructuredValue) ProtoMessage 

func (*StructuredValue) ProtoMessage()

func (*StructuredValue) ProtoReflect 

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

func (*StructuredValue) Reset 

func (x *StructuredValue) Reset()

func (*StructuredValue) String 

func (x *StructuredValue) String() string

type StructuredValue_BoolValue 

type StructuredValue_BoolValue struct {
	// Represents a boolean value.
	BoolValue bool `protobuf:"varint,14,opt,name=bool_value,json=boolValue,proto3,oneof"`
}

type StructuredValue_BoundedTensorSpecValue 

type StructuredValue_BoundedTensorSpecValue struct {
	// Represents a value for tf.BoundedTensorSpec.
	BoundedTensorSpecValue *BoundedTensorSpecProto `protobuf:"bytes,35,opt,name=bounded_tensor_spec_value,json=boundedTensorSpecValue,proto3,oneof"`
}

type StructuredValue_DictValue 

type StructuredValue_DictValue struct {
	// Represents a dict `Value`.
	DictValue *DictValue `protobuf:"bytes,53,opt,name=dict_value,json=dictValue,proto3,oneof"`
}

type StructuredValue_Float64Value 

type StructuredValue_Float64Value struct {
	// Represents a double-precision floating-point value (a Python `float`).
	Float64Value float64 `protobuf:"fixed64,11,opt,name=float64_value,json=float64Value,proto3,oneof"`
}

type StructuredValue_Int64Value 

type StructuredValue_Int64Value struct {
	// Represents a signed integer value, limited to 64 bits.
	// Larger values from Python's arbitrary-precision integers are unsupported.
	Int64Value int64 `protobuf:"zigzag64,12,opt,name=int64_value,json=int64Value,proto3,oneof"`
}

type StructuredValue_ListValue 

type StructuredValue_ListValue struct {
	// Represents a list of `Value`.
	ListValue *ListValue `protobuf:"bytes,51,opt,name=list_value,json=listValue,proto3,oneof"`
}

type StructuredValue_NamedTupleValue 

type StructuredValue_NamedTupleValue struct {
	// Represents Python's namedtuple.
	NamedTupleValue *NamedTupleValue `protobuf:"bytes,54,opt,name=named_tuple_value,json=namedTupleValue,proto3,oneof"`
}

type StructuredValue_NoneValue 

type StructuredValue_NoneValue struct {
	// Represents None.
	NoneValue *NoneValue `protobuf:"bytes,1,opt,name=none_value,json=noneValue,proto3,oneof"`
}

type StructuredValue_StringValue 

type StructuredValue_StringValue struct {
	// Represents a string of Unicode characters stored in a Python `str`.
	// In Python 3, this is exactly what type `str` is.
	// In Python 2, this is the UTF-8 encoding of the characters.
	// For strings with ASCII characters only (as often used in TensorFlow code)
	// there is effectively no difference between the language versions.
	// The obsolescent `unicode` type of Python 2 is not supported here.
	StringValue string `protobuf:"bytes,13,opt,name=string_value,json=stringValue,proto3,oneof"`
}

type StructuredValue_TensorDtypeValue 

type StructuredValue_TensorDtypeValue struct {
	// Represents an enum value for dtype.
	TensorDtypeValue DataType `protobuf:"varint,32,opt,name=tensor_dtype_value,json=tensorDtypeValue,proto3,enum=tensorflow.DataType,oneof"`
}

type StructuredValue_TensorShapeValue 

type StructuredValue_TensorShapeValue struct {
	// Represents a TensorShape.
	TensorShapeValue *TensorShapeProto `protobuf:"bytes,31,opt,name=tensor_shape_value,json=tensorShapeValue,proto3,oneof"`
}

type StructuredValue_TensorSpecValue 

type StructuredValue_TensorSpecValue struct {
	// Represents a value for tf.TensorSpec.
	TensorSpecValue *TensorSpecProto `protobuf:"bytes,33,opt,name=tensor_spec_value,json=tensorSpecValue,proto3,oneof"`
}

type StructuredValue_TupleValue 

type StructuredValue_TupleValue struct {
	// Represents a tuple of `Value`.
	TupleValue *TupleValue `protobuf:"bytes,52,opt,name=tuple_value,json=tupleValue,proto3,oneof"`
}

type StructuredValue_TypeSpecValue 

type StructuredValue_TypeSpecValue struct {
	// Represents a value for tf.TypeSpec.
	TypeSpecValue *TypeSpecProto `protobuf:"bytes,34,opt,name=type_spec_value,json=typeSpecValue,proto3,oneof"`
}

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() *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 *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 SyncRemoteExecutorForStream 

type SyncRemoteExecutorForStream struct {
	// contains filtered or unexported fields
}

func (*SyncRemoteExecutorForStream) Descriptor deprecated 

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

Deprecated: Use SyncRemoteExecutorForStream.ProtoReflect.Descriptor instead.

func (*SyncRemoteExecutorForStream) ProtoMessage 

func (*SyncRemoteExecutorForStream) ProtoMessage()

func (*SyncRemoteExecutorForStream) ProtoReflect 

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

func (*SyncRemoteExecutorForStream) Reset 

func (x *SyncRemoteExecutorForStream) Reset()

func (*SyncRemoteExecutorForStream) String 

func (x *SyncRemoteExecutorForStream) String() string

type TaggedRunMetadata 

type TaggedRunMetadata struct {

	// Tag name associated with this metadata.
	Tag string `protobuf:"bytes,1,opt,name=tag,proto3" json:"tag,omitempty"`
	// Byte-encoded version of the `RunMetadata` proto in order to allow lazy
	// deserialization.
	RunMetadata []byte `protobuf:"bytes,2,opt,name=run_metadata,json=runMetadata,proto3" json:"run_metadata,omitempty"`
	// contains filtered or unexported fields
}

For logging the metadata output for a single session.run() call.

func (*TaggedRunMetadata) Descriptor deprecated 

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

Deprecated: Use TaggedRunMetadata.ProtoReflect.Descriptor instead.

func (*TaggedRunMetadata) GetRunMetadata 

func (x *TaggedRunMetadata) GetRunMetadata() []byte

func (*TaggedRunMetadata) GetTag 

func (x *TaggedRunMetadata) GetTag() string

func (*TaggedRunMetadata) ProtoMessage 

func (*TaggedRunMetadata) ProtoMessage()

func (*TaggedRunMetadata) ProtoReflect 

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

func (*TaggedRunMetadata) Reset 

func (x *TaggedRunMetadata) Reset()

func (*TaggedRunMetadata) String 

func (x *TaggedRunMetadata) String() string

type TaskDeviceFilters 

type TaskDeviceFilters struct {
	DeviceFilters []string `protobuf:"bytes,1,rep,name=device_filters,json=deviceFilters,proto3" json:"device_filters,omitempty"`
	// contains filtered or unexported fields
}

Defines the device filters for a remote task.

func (*TaskDeviceFilters) Descriptor deprecated 

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

Deprecated: Use TaskDeviceFilters.ProtoReflect.Descriptor instead.

func (*TaskDeviceFilters) GetDeviceFilters 

func (x *TaskDeviceFilters) GetDeviceFilters() []string

func (*TaskDeviceFilters) ProtoMessage 

func (*TaskDeviceFilters) ProtoMessage()

func (*TaskDeviceFilters) ProtoReflect 

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

func (*TaskDeviceFilters) Reset 

func (x *TaskDeviceFilters) Reset()

func (*TaskDeviceFilters) String 

func (x *TaskDeviceFilters) String() string

type TensorConnection 

type TensorConnection struct {

	// A tensor name. The value of this tensor will be substituted for
	// the tensor named in `to_tensor`.
	FromTensor string `protobuf:"bytes,1,opt,name=from_tensor,json=fromTensor,proto3" json:"from_tensor,omitempty"`
	// A tensor name. The value of this tensor will be bound to the
	// value of the tensor named in `from_tensor`.
	ToTensor string `protobuf:"bytes,2,opt,name=to_tensor,json=toTensor,proto3" json:"to_tensor,omitempty"`
	// contains filtered or unexported fields
}

Defines a connection between two tensors in a `GraphDef`.

func (*TensorConnection) Descriptor deprecated 

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

Deprecated: Use TensorConnection.ProtoReflect.Descriptor instead.

func (*TensorConnection) GetFromTensor 

func (x *TensorConnection) GetFromTensor() string

func (*TensorConnection) GetToTensor 

func (x *TensorConnection) GetToTensor() string

func (*TensorConnection) ProtoMessage 

func (*TensorConnection) ProtoMessage()

func (*TensorConnection) ProtoReflect 

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

func (*TensorConnection) Reset 

func (x *TensorConnection) Reset()

func (*TensorConnection) String 

func (x *TensorConnection) String() string

type TensorDebugMode 

type TensorDebugMode int32

Available modes for extracting debugging information from a Tensor. TODO(cais): Document the detailed column names and semantics in a separate markdown file once the implementation settles. 

const (
	TensorDebugMode_UNSPECIFIED TensorDebugMode = 0
	// Only records what tensors are computed, eagerly or in graphs.
	// No information regarding the value of the tensor is available.
	TensorDebugMode_NO_TENSOR TensorDebugMode = 1
	// A minimalist health summary for float-type tensors.
	// Contains information only about the presence/absence of pathological
	// values including Infinity and NaN.
	// Applicable only to float dtypes.
	TensorDebugMode_CURT_HEALTH TensorDebugMode = 2
	// A concise health summary for float-type tensors.
	// Contains more information that CURT_HEALTH.
	// Infinity and NaN are treated differently.
	// Applicable only to float and integer dtypes.
	TensorDebugMode_CONCISE_HEALTH TensorDebugMode = 3
	// A detailed health summary.
	// Contains further detailed information than `CONCISE_HEALTH`.
	// Information about device, dtype and shape are included.
	// Counts for various types of values (Infinity, NaN, negative, zero,
	// positive) are included.
	// Applicable to float, integer and boolean dtypes.
	TensorDebugMode_FULL_HEALTH TensorDebugMode = 4
	// Provides full runtime shape information, up to a maximum rank, beyond
	// which the dimension sizes are truncated.
	TensorDebugMode_SHAPE TensorDebugMode = 5
	// Full numeric summary.
	// Including device, dtype, shape, counts of various types of values
	// (Infinity, NaN, negative, zero, positive), and summary statistics
	// (minimum, maximum, mean and variance).
	// Applicable to float, integer and boolean dtypes.
	TensorDebugMode_FULL_NUMERICS TensorDebugMode = 6
	// Full tensor value.
	TensorDebugMode_FULL_TENSOR TensorDebugMode = 7
	// Reduce the elements of a tensor to a rank-1 tensor of shape [3], in which
	// - the 1st element is -inf if any element of the tensor is -inf,
	//   or zero otherwise.
	// - the 2nd element is +inf if any element of the tensor is +inf,
	//   or zero otherwise.
	// - the 3rd element is nan if any element of the tensor is nan, or zero
	//   otherwise.
	TensorDebugMode_REDUCE_INF_NAN_THREE_SLOTS TensorDebugMode = 8
)

func (TensorDebugMode) Descriptor 

func (TensorDebugMode) Descriptor() protoreflect.EnumDescriptor

func (TensorDebugMode) Enum 

func (x TensorDebugMode) Enum() *TensorDebugMode

func (TensorDebugMode) EnumDescriptor deprecated 

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

Deprecated: Use TensorDebugMode.Descriptor instead.

func (TensorDebugMode) Number 

func (x TensorDebugMode) Number() protoreflect.EnumNumber

func (TensorDebugMode) String 

func (x TensorDebugMode) String() string

func (TensorDebugMode) Type 

func (TensorDebugMode) Type() protoreflect.EnumType

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 TensorDescriptorProto 

type TensorDescriptorProto struct {
	Dimensions []int64     `protobuf:"varint,1,rep,packed,name=dimensions,proto3" json:"dimensions,omitempty"`
	DataType   DnnDataType `protobuf:"varint,2,opt,name=data_type,json=dataType,proto3,enum=stream_executor.dnn.DataType" json:"data_type,omitempty"`
	// Types that are assignable to LayoutOneof:
	//	*TensorDescriptorProto_DataLayout
	//	*TensorDescriptorProto_FilterLayout
	LayoutOneof isTensorDescriptorProto_LayoutOneof `protobuf_oneof:"layout_oneof"`
	// contains filtered or unexported fields
}

Generic tensor representation.

func (*TensorDescriptorProto) Descriptor deprecated 

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

Deprecated: Use TensorDescriptorProto.ProtoReflect.Descriptor instead.

func (*TensorDescriptorProto) GetDataLayout 

func (x *TensorDescriptorProto) GetDataLayout() DataLayout

func (*TensorDescriptorProto) GetDataType 

func (x *TensorDescriptorProto) GetDataType() DnnDataType

func (*TensorDescriptorProto) GetDimensions 

func (x *TensorDescriptorProto) GetDimensions() []int64

func (*TensorDescriptorProto) GetFilterLayout 

func (x *TensorDescriptorProto) GetFilterLayout() FilterLayout

func (*TensorDescriptorProto) GetLayoutOneof 

func (m *TensorDescriptorProto) GetLayoutOneof() isTensorDescriptorProto_LayoutOneof

func (*TensorDescriptorProto) ProtoMessage 

func (*TensorDescriptorProto) ProtoMessage()

func (*TensorDescriptorProto) ProtoReflect 

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

func (*TensorDescriptorProto) Reset 

func (x *TensorDescriptorProto) Reset()

func (*TensorDescriptorProto) String 

func (x *TensorDescriptorProto) String() string

type TensorDescriptorProto_DataLayout 

type TensorDescriptorProto_DataLayout struct {
	DataLayout DataLayout `protobuf:"varint,3,opt,name=data_layout,json=dataLayout,proto3,enum=stream_executor.dnn.DataLayout,oneof"`
}

type TensorDescriptorProto_FilterLayout 

type TensorDescriptorProto_FilterLayout struct {
	FilterLayout FilterLayout `protobuf:"varint,4,opt,name=filter_layout,json=filterLayout,proto3,enum=stream_executor.dnn.FilterLayout,oneof"`
}

type TensorInfo 

type TensorInfo struct {

	// Types that are assignable to Encoding:
	//	*TensorInfo_Name
	//	*TensorInfo_CooSparse_
	//	*TensorInfo_CompositeTensor_
	Encoding isTensorInfo_Encoding `protobuf_oneof:"encoding"`
	Dtype    DataType              `protobuf:"varint,2,opt,name=dtype,proto3,enum=tensorflow.DataType" json:"dtype,omitempty"`
	// The static shape should be recorded here, to the extent that it can
	// be known in advance.  In the case of a SparseTensor, this field describes
	// the logical shape of the represented tensor (aka dense_shape).
	TensorShape *TensorShapeProto `protobuf:"bytes,3,opt,name=tensor_shape,json=tensorShape,proto3" json:"tensor_shape,omitempty"`
	// contains filtered or unexported fields
}

Information about a Tensor necessary for feeding or retrieval.

func (*TensorInfo) Descriptor deprecated 

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

Deprecated: Use TensorInfo.ProtoReflect.Descriptor instead.

func (*TensorInfo) GetCompositeTensor 

func (x *TensorInfo) GetCompositeTensor() *TensorInfo_CompositeTensor

func (*TensorInfo) GetCooSparse 

func (x *TensorInfo) GetCooSparse() *TensorInfo_CooSparse

func (*TensorInfo) GetDtype 

func (x *TensorInfo) GetDtype() DataType

func (*TensorInfo) GetEncoding 

func (m *TensorInfo) GetEncoding() isTensorInfo_Encoding

func (*TensorInfo) GetName 

func (x *TensorInfo) GetName() string

func (*TensorInfo) GetTensorShape 

func (x *TensorInfo) GetTensorShape() *TensorShapeProto

func (*TensorInfo) ProtoMessage 

func (*TensorInfo) ProtoMessage()

func (*TensorInfo) ProtoReflect 

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

func (*TensorInfo) Reset 

func (x *TensorInfo) Reset()

func (*TensorInfo) String 

func (x *TensorInfo) String() string

type TensorInfo_CompositeTensor 

type TensorInfo_CompositeTensor struct {

	// The serialized TypeSpec for the composite tensor.
	TypeSpec *TypeSpecProto `protobuf:"bytes,1,opt,name=type_spec,json=typeSpec,proto3" json:"type_spec,omitempty"`
	// A TensorInfo for each flattened component tensor.
	Components []*TensorInfo `protobuf:"bytes,2,rep,name=components,proto3" json:"components,omitempty"`
	// contains filtered or unexported fields
}

Generic encoding for composite tensors.

func (*TensorInfo_CompositeTensor) Descriptor deprecated 

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

Deprecated: Use TensorInfo_CompositeTensor.ProtoReflect.Descriptor instead.

func (*TensorInfo_CompositeTensor) GetComponents 

func (x *TensorInfo_CompositeTensor) GetComponents() []*TensorInfo

func (*TensorInfo_CompositeTensor) GetTypeSpec 

func (x *TensorInfo_CompositeTensor) GetTypeSpec() *TypeSpecProto

func (*TensorInfo_CompositeTensor) ProtoMessage 

func (*TensorInfo_CompositeTensor) ProtoMessage()

func (*TensorInfo_CompositeTensor) ProtoReflect 

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

func (*TensorInfo_CompositeTensor) Reset 

func (x *TensorInfo_CompositeTensor) Reset()

func (*TensorInfo_CompositeTensor) String 

func (x *TensorInfo_CompositeTensor) String() string

type TensorInfo_CompositeTensor_ 

type TensorInfo_CompositeTensor_ struct {
	// Generic encoding for CompositeTensors.
	CompositeTensor *TensorInfo_CompositeTensor `protobuf:"bytes,5,opt,name=composite_tensor,json=compositeTensor,proto3,oneof"`
}

type TensorInfo_CooSparse 

type TensorInfo_CooSparse struct {

	// The shape of the values Tensor is [?].  Its dtype must be the dtype of
	// the SparseTensor as a whole, given in the enclosing TensorInfo.
	ValuesTensorName string `protobuf:"bytes,1,opt,name=values_tensor_name,json=valuesTensorName,proto3" json:"values_tensor_name,omitempty"`
	// The indices Tensor must have dtype int64 and shape [?, ?].
	IndicesTensorName string `protobuf:"bytes,2,opt,name=indices_tensor_name,json=indicesTensorName,proto3" json:"indices_tensor_name,omitempty"`
	// The dynamic logical shape represented by the SparseTensor is recorded in
	// the Tensor referenced here.  It must have dtype int64 and shape [?].
	DenseShapeTensorName string `protobuf:"bytes,3,opt,name=dense_shape_tensor_name,json=denseShapeTensorName,proto3" json:"dense_shape_tensor_name,omitempty"`
	// contains filtered or unexported fields
}

For sparse tensors, The COO encoding stores a triple of values, indices, and shape.

func (*TensorInfo_CooSparse) Descriptor deprecated 

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

Deprecated: Use TensorInfo_CooSparse.ProtoReflect.Descriptor instead.

func (*TensorInfo_CooSparse) GetDenseShapeTensorName 

func (x *TensorInfo_CooSparse) GetDenseShapeTensorName() string

func (*TensorInfo_CooSparse) GetIndicesTensorName 

func (x *TensorInfo_CooSparse) GetIndicesTensorName() string

func (*TensorInfo_CooSparse) GetValuesTensorName 

func (x *TensorInfo_CooSparse) GetValuesTensorName() string

func (*TensorInfo_CooSparse) ProtoMessage 

func (*TensorInfo_CooSparse) ProtoMessage()

func (*TensorInfo_CooSparse) ProtoReflect 

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

func (*TensorInfo_CooSparse) Reset 

func (x *TensorInfo_CooSparse) Reset()

func (*TensorInfo_CooSparse) String 

func (x *TensorInfo_CooSparse) String() string

type TensorInfo_CooSparse_ 

type TensorInfo_CooSparse_ struct {
	// There are many possible encodings of sparse matrices
	// (https://en.wikipedia.org/wiki/Sparse_matrix).  Currently, TensorFlow
	// uses only the COO encoding.  This is supported and documented in the
	// SparseTensor Python class.
	CooSparse *TensorInfo_CooSparse `protobuf:"bytes,4,opt,name=coo_sparse,json=cooSparse,proto3,oneof"`
}

type TensorInfo_Name 

type TensorInfo_Name struct {
	// For dense `Tensor`s, the name of the tensor in the graph.
	Name string `protobuf:"bytes,1,opt,name=name,proto3,oneof"`
}

type TensorMetadata 

type TensorMetadata struct {
	TensorShape *TensorShapeProto `protobuf:"bytes,2,opt,name=tensor_shape,json=tensorShape,proto3" json:"tensor_shape,omitempty"`
	// Number of uncompressed bytes used to store the tensor representation.
	TensorSizeBytes int64 `protobuf:"varint,3,opt,name=tensor_size_bytes,json=tensorSizeBytes,proto3" json:"tensor_size_bytes,omitempty"`
	// contains filtered or unexported fields
}

Metadata for a single tensor in the Snapshot Record.

func (*TensorMetadata) Descriptor deprecated 

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

Deprecated: Use TensorMetadata.ProtoReflect.Descriptor instead.

func (*TensorMetadata) GetTensorShape 

func (x *TensorMetadata) GetTensorShape() *TensorShapeProto

func (*TensorMetadata) GetTensorSizeBytes 

func (x *TensorMetadata) GetTensorSizeBytes() int64

func (*TensorMetadata) ProtoMessage 

func (*TensorMetadata) ProtoMessage()

func (*TensorMetadata) ProtoReflect 

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

func (*TensorMetadata) Reset 

func (x *TensorMetadata) Reset()

func (*TensorMetadata) String 

func (x *TensorMetadata) String() string

type TensorProto 

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

Protocol buffer representing a tensor.

func (*TensorProto) Descriptor deprecated 

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

Deprecated: Use TensorProto.ProtoReflect.Descriptor instead.

func (*TensorProto) GetBoolVal 

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

func (*TensorProto) GetDcomplexVal 

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

func (*TensorProto) GetDoubleVal 

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

func (*TensorProto) GetDtype 

func (x *TensorProto) GetDtype() DataType

func (*TensorProto) GetFloatVal 

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

func (*TensorProto) GetHalfVal 

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

func (*TensorProto) GetInt64Val 

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

func (*TensorProto) GetIntVal 

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

func (*TensorProto) GetResourceHandleVal 

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

func (*TensorProto) GetScomplexVal 

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

func (*TensorProto) GetStringVal 

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

func (*TensorProto) GetTensorContent 

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

func (*TensorProto) GetTensorShape 

func (x *TensorProto) GetTensorShape() *TensorShapeProto

func (*TensorProto) GetUint32Val 

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

func (*TensorProto) GetUint64Val 

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

func (*TensorProto) GetVariantVal 

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

func (*TensorProto) GetVersionNumber 

func (x *TensorProto) GetVersionNumber() int32

func (*TensorProto) ProtoMessage 

func (*TensorProto) ProtoMessage()

func (*TensorProto) ProtoReflect 

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

func (*TensorProto) Reset 

func (x *TensorProto) Reset()

func (*TensorProto) String 

func (x *TensorProto) String() string

type TensorShapeProto 

type TensorShapeProto struct {

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

Dimensions of a tensor.

func (*TensorShapeProto) Descriptor deprecated 

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

Deprecated: Use TensorShapeProto.ProtoReflect.Descriptor instead.

func (*TensorShapeProto) GetDim 

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

func (*TensorShapeProto) GetUnknownRank 

func (x *TensorShapeProto) GetUnknownRank() bool

func (*TensorShapeProto) ProtoMessage 

func (*TensorShapeProto) ProtoMessage()

func (*TensorShapeProto) ProtoReflect 

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

func (*TensorShapeProto) Reset 

func (x *TensorShapeProto) Reset()

func (*TensorShapeProto) String 

func (x *TensorShapeProto) String() string

type TensorShapeProto_Dim 

type TensorShapeProto_Dim struct {

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

One dimension of the tensor.

func (*TensorShapeProto_Dim) Descriptor deprecated 

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

Deprecated: Use TensorShapeProto_Dim.ProtoReflect.Descriptor instead.

func (*TensorShapeProto_Dim) GetName 

func (x *TensorShapeProto_Dim) GetName() string

func (*TensorShapeProto_Dim) GetSize 

func (x *TensorShapeProto_Dim) GetSize() int64

func (*TensorShapeProto_Dim) ProtoMessage 

func (*TensorShapeProto_Dim) ProtoMessage()

func (*TensorShapeProto_Dim) ProtoReflect 

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

func (*TensorShapeProto_Dim) Reset 

func (x *TensorShapeProto_Dim) Reset()

func (*TensorShapeProto_Dim) String 

func (x *TensorShapeProto_Dim) String() string

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

type TensorSpecProto struct {
	Name  string            `protobuf:"bytes,1,opt,name=name,proto3" json:"name,omitempty"`
	Shape *TensorShapeProto `protobuf:"bytes,2,opt,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
}

A protobuf to represent tf.TensorSpec.

func (*TensorSpecProto) Descriptor deprecated 

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

Deprecated: Use TensorSpecProto.ProtoReflect.Descriptor instead.

func (*TensorSpecProto) GetDtype 

func (x *TensorSpecProto) GetDtype() DataType

func (*TensorSpecProto) GetName 

func (x *TensorSpecProto) GetName() string

func (*TensorSpecProto) GetShape 

func (x *TensorSpecProto) GetShape() *TensorShapeProto

func (*TensorSpecProto) ProtoMessage 

func (*TensorSpecProto) ProtoMessage()

func (*TensorSpecProto) ProtoReflect 

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

func (*TensorSpecProto) Reset 

func (x *TensorSpecProto) Reset()

func (*TensorSpecProto) String 

func (x *TensorSpecProto) String() string

type TfDeviceList 

type TfDeviceList struct {
	Devices []*DeviceAttributes `protobuf:"bytes,1,rep,name=devices,proto3" json:"devices,omitempty"`
	// contains filtered or unexported fields
}

Represent device information from different runtimes.

func (*TfDeviceList) Descriptor deprecated 

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

Deprecated: Use TfDeviceList.ProtoReflect.Descriptor instead.

func (*TfDeviceList) GetDevices 

func (x *TfDeviceList) GetDevices() []*DeviceAttributes

func (*TfDeviceList) ProtoMessage 

func (*TfDeviceList) ProtoMessage()

func (*TfDeviceList) ProtoReflect 

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

func (*TfDeviceList) Reset 

func (x *TfDeviceList) Reset()

func (*TfDeviceList) String 

func (x *TfDeviceList) String() string

type ThreadPoolOptionProto 

type ThreadPoolOptionProto struct {

	// The number of threads in the pool.
	//
	// 0 means the system picks a value based on where this option proto is used
	// (see the declaration of the specific field for more info).
	NumThreads int32 `protobuf:"varint,1,opt,name=num_threads,json=numThreads,proto3" json:"num_threads,omitempty"`
	// The global name of the threadpool.
	//
	// If empty, then the threadpool is made and used according to the scope it's
	// in - e.g., for a session threadpool, it is used by that session only.
	//
	// If non-empty, then:
	// - a global threadpool associated with this name is looked
	//   up or created. This allows, for example, sharing one threadpool across
	//   many sessions (e.g., like the default behavior, if
	//   inter_op_parallelism_threads is not configured), but still partitioning
	//   into a large and small pool.
	// - if the threadpool for this global_name already exists, then it is an
	//   error if the existing pool was created using a different num_threads
	//   value as is specified on this call.
	// - threadpools created this way are never garbage collected.
	GlobalName string `protobuf:"bytes,2,opt,name=global_name,json=globalName,proto3" json:"global_name,omitempty"`
	// contains filtered or unexported fields
}

func (*ThreadPoolOptionProto) Descriptor deprecated 

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

Deprecated: Use ThreadPoolOptionProto.ProtoReflect.Descriptor instead.

func (*ThreadPoolOptionProto) GetGlobalName 

func (x *ThreadPoolOptionProto) GetGlobalName() string

func (*ThreadPoolOptionProto) GetNumThreads 

func (x *ThreadPoolOptionProto) GetNumThreads() int32

func (*ThreadPoolOptionProto) ProtoMessage 

func (*ThreadPoolOptionProto) ProtoMessage()

func (*ThreadPoolOptionProto) ProtoReflect 

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

func (*ThreadPoolOptionProto) Reset 

func (x *ThreadPoolOptionProto) Reset()

func (*ThreadPoolOptionProto) String 

func (x *ThreadPoolOptionProto) String() string

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 TraceOpts 

type TraceOpts struct {

	// Length of the trace to be taken, in seconds.
	Duration float64 `protobuf:"fixed64,1,opt,name=duration,proto3" json:"duration,omitempty"`
	// If true, capture step profile locally in each worker. Currently
	// unimplemented.
	UseStepProfiler bool `protobuf:"varint,2,opt,name=use_step_profiler,json=useStepProfiler,proto3" json:"use_step_profiler,omitempty"`
	// If true, capture kernel events from each worker.
	UseKernelProfiler bool `protobuf:"varint,3,opt,name=use_kernel_profiler,json=useKernelProfiler,proto3" json:"use_kernel_profiler,omitempty"`
	// If true, capture extended profiling events from TensorFlow process.
	UseExtendedProfiler bool `protobuf:"varint,4,opt,name=use_extended_profiler,json=useExtendedProfiler,proto3" json:"use_extended_profiler,omitempty"`
	// If true, capture GPU profiling events locally on each
	// machine. Currently unimplemented.
	UseGpuProfiler bool `protobuf:"varint,5,opt,name=use_gpu_profiler,json=useGpuProfiler,proto3" json:"use_gpu_profiler,omitempty"`
	// If true, collect sampled profile events. Currently unimplemented.
	UseSampleProfiler bool `protobuf:"varint,6,opt,name=use_sample_profiler,json=useSampleProfiler,proto3" json:"use_sample_profiler,omitempty"`
	// contains filtered or unexported fields
}

func (*TraceOpts) Descriptor deprecated 

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

Deprecated: Use TraceOpts.ProtoReflect.Descriptor instead.

func (*TraceOpts) GetDuration 

func (x *TraceOpts) GetDuration() float64

func (*TraceOpts) GetUseExtendedProfiler 

func (x *TraceOpts) GetUseExtendedProfiler() bool

func (*TraceOpts) GetUseGpuProfiler 

func (x *TraceOpts) GetUseGpuProfiler() bool

func (*TraceOpts) GetUseKernelProfiler 

func (x *TraceOpts) GetUseKernelProfiler() bool

func (*TraceOpts) GetUseSampleProfiler 

func (x *TraceOpts) GetUseSampleProfiler() bool

func (*TraceOpts) GetUseStepProfiler 

func (x *TraceOpts) GetUseStepProfiler() bool

func (*TraceOpts) ProtoMessage 

func (*TraceOpts) ProtoMessage()

func (*TraceOpts) ProtoReflect 

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

func (*TraceOpts) Reset 

func (x *TraceOpts) Reset()

func (*TraceOpts) String 

func (x *TraceOpts) String() string

type TracingRequest 

type TracingRequest struct {
	Options *TraceOpts `protobuf:"bytes,1,opt,name=options,proto3" json:"options,omitempty"`
	// contains filtered or unexported fields
}

Out-of-band request to configure distributed tracing.

func (*TracingRequest) Descriptor deprecated 

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

Deprecated: Use TracingRequest.ProtoReflect.Descriptor instead.

func (*TracingRequest) GetOptions 

func (x *TracingRequest) GetOptions() *TraceOpts

func (*TracingRequest) ProtoMessage 

func (*TracingRequest) ProtoMessage()

func (*TracingRequest) ProtoReflect 

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

func (*TracingRequest) Reset 

func (x *TracingRequest) Reset()

func (*TracingRequest) String 

func (x *TracingRequest) String() string

type TracingResponse 

type TracingResponse struct {
	// contains filtered or unexported fields
}

func (*TracingResponse) Descriptor deprecated 

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

Deprecated: Use TracingResponse.ProtoReflect.Descriptor instead.

func (*TracingResponse) ProtoMessage 

func (*TracingResponse) ProtoMessage()

func (*TracingResponse) ProtoReflect 

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

func (*TracingResponse) Reset 

func (x *TracingResponse) Reset()

func (*TracingResponse) String 

func (x *TracingResponse) String() string

type TrackableObjectGraph 

type TrackableObjectGraph struct {
	Nodes []*TrackableObjectGraph_TrackableObject `protobuf:"bytes,1,rep,name=nodes,proto3" json:"nodes,omitempty"`
	// contains filtered or unexported fields
}

func (*TrackableObjectGraph) Descriptor deprecated 

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

Deprecated: Use TrackableObjectGraph.ProtoReflect.Descriptor instead.

func (*TrackableObjectGraph) GetNodes 

func (x *TrackableObjectGraph) GetNodes() []*TrackableObjectGraph_TrackableObject

func (*TrackableObjectGraph) ProtoMessage 

func (*TrackableObjectGraph) ProtoMessage()

func (*TrackableObjectGraph) ProtoReflect 

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

func (*TrackableObjectGraph) Reset 

func (x *TrackableObjectGraph) Reset()

func (*TrackableObjectGraph) String 

func (x *TrackableObjectGraph) String() string

type TrackableObjectGraph_TrackableObject 

type TrackableObjectGraph_TrackableObject struct {

	// Objects which this object depends on.
	Children []*TrackableObjectGraph_TrackableObject_ObjectReference `protobuf:"bytes,1,rep,name=children,proto3" json:"children,omitempty"`
	// Serialized data specific to this object.
	Attributes []*TrackableObjectGraph_TrackableObject_SerializedTensor `protobuf:"bytes,2,rep,name=attributes,proto3" json:"attributes,omitempty"`
	// Slot variables owned by this object.
	SlotVariables []*TrackableObjectGraph_TrackableObject_SlotVariableReference `protobuf:"bytes,3,rep,name=slot_variables,json=slotVariables,proto3" json:"slot_variables,omitempty"`
	// The registered saver used to save this object. If this saver is not
	// present when loading the checkpoint, then loading will fail.
	RegisteredSaver *RegisteredSaver `protobuf:"bytes,4,opt,name=registered_saver,json=registeredSaver,proto3" json:"registered_saver,omitempty"`
	// Whether this object has checkpoint values or descendants with checkpoint
	// values. This is computed at save time to avoid traversing the entire
	// object graph proto when restoring (which also has to traverse the live
	// object graph).
	HasCheckpointValues *wrappers.BoolValue `protobuf:"bytes,5,opt,name=has_checkpoint_values,json=hasCheckpointValues,proto3" json:"has_checkpoint_values,omitempty"`
	// contains filtered or unexported fields
}

func (*TrackableObjectGraph_TrackableObject) Descriptor deprecated 

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

Deprecated: Use TrackableObjectGraph_TrackableObject.ProtoReflect.Descriptor instead.

func (*TrackableObjectGraph_TrackableObject) GetAttributes 

func (x *TrackableObjectGraph_TrackableObject) GetAttributes() []*TrackableObjectGraph_TrackableObject_SerializedTensor

func (*TrackableObjectGraph_TrackableObject) GetChildren 

func (x *TrackableObjectGraph_TrackableObject) GetChildren() []*TrackableObjectGraph_TrackableObject_ObjectReference

func (*TrackableObjectGraph_TrackableObject) GetHasCheckpointValues 

func (x *TrackableObjectGraph_TrackableObject) GetHasCheckpointValues() *wrappers.BoolValue

func (*TrackableObjectGraph_TrackableObject) GetRegisteredSaver 

func (x *TrackableObjectGraph_TrackableObject) GetRegisteredSaver() *RegisteredSaver

func (*TrackableObjectGraph_TrackableObject) GetSlotVariables 

func (x *TrackableObjectGraph_TrackableObject) GetSlotVariables() []*TrackableObjectGraph_TrackableObject_SlotVariableReference

func (*TrackableObjectGraph_TrackableObject) ProtoMessage 

func (*TrackableObjectGraph_TrackableObject) ProtoMessage()

func (*TrackableObjectGraph_TrackableObject) ProtoReflect 

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

func (*TrackableObjectGraph_TrackableObject) Reset 

func (x *TrackableObjectGraph_TrackableObject) Reset()

func (*TrackableObjectGraph_TrackableObject) String 

func (x *TrackableObjectGraph_TrackableObject) String() string

type TrackableObjectGraph_TrackableObject_ObjectReference 

type TrackableObjectGraph_TrackableObject_ObjectReference struct {

	// An index into `TrackableObjectGraph.nodes`, indicating the object
	// being referenced.
	NodeId int32 `protobuf:"varint,1,opt,name=node_id,json=nodeId,proto3" json:"node_id,omitempty"`
	// A user-provided name for the edge.
	LocalName string `protobuf:"bytes,2,opt,name=local_name,json=localName,proto3" json:"local_name,omitempty"`
	// contains filtered or unexported fields
}

func (*TrackableObjectGraph_TrackableObject_ObjectReference) Descriptor deprecated 

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

Deprecated: Use TrackableObjectGraph_TrackableObject_ObjectReference.ProtoReflect.Descriptor instead.

func (*TrackableObjectGraph_TrackableObject_ObjectReference) GetLocalName 

func (x *TrackableObjectGraph_TrackableObject_ObjectReference) GetLocalName() string

func (*TrackableObjectGraph_TrackableObject_ObjectReference) GetNodeId 

func (x *TrackableObjectGraph_TrackableObject_ObjectReference) GetNodeId() int32

func (*TrackableObjectGraph_TrackableObject_ObjectReference) ProtoMessage 

func (*TrackableObjectGraph_TrackableObject_ObjectReference) ProtoMessage()

func (*TrackableObjectGraph_TrackableObject_ObjectReference) ProtoReflect 

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

func (*TrackableObjectGraph_TrackableObject_ObjectReference) Reset 

func (x *TrackableObjectGraph_TrackableObject_ObjectReference) Reset()

func (*TrackableObjectGraph_TrackableObject_ObjectReference) String 

func (x *TrackableObjectGraph_TrackableObject_ObjectReference) String() string

type TrackableObjectGraph_TrackableObject_SerializedTensor 

type TrackableObjectGraph_TrackableObject_SerializedTensor struct {

	// A name for the Tensor. Simple variables have only one
	// `SerializedTensor` named "VARIABLE_VALUE" by convention. This value may
	// be restored on object creation as an optimization.
	Name string `protobuf:"bytes,1,opt,name=name,proto3" json:"name,omitempty"`
	// The full name of the variable/tensor, if applicable. Used to allow
	// name-based loading of checkpoints which were saved using an
	// object-based API. Should match the checkpoint key which would have been
	// assigned by tf.train.Saver.
	FullName string `protobuf:"bytes,2,opt,name=full_name,json=fullName,proto3" json:"full_name,omitempty"`
	// The generated name of the Tensor in the checkpoint.
	CheckpointKey string `protobuf:"bytes,3,opt,name=checkpoint_key,json=checkpointKey,proto3" json:"checkpoint_key,omitempty"`
	// contains filtered or unexported fields
}

func (*TrackableObjectGraph_TrackableObject_SerializedTensor) Descriptor deprecated 

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

Deprecated: Use TrackableObjectGraph_TrackableObject_SerializedTensor.ProtoReflect.Descriptor instead.

func (*TrackableObjectGraph_TrackableObject_SerializedTensor) GetCheckpointKey 

func (x *TrackableObjectGraph_TrackableObject_SerializedTensor) GetCheckpointKey() string

func (*TrackableObjectGraph_TrackableObject_SerializedTensor) GetFullName 

func (x *TrackableObjectGraph_TrackableObject_SerializedTensor) GetFullName() string

func (*TrackableObjectGraph_TrackableObject_SerializedTensor) GetName 

func (x *TrackableObjectGraph_TrackableObject_SerializedTensor) GetName() string

func (*TrackableObjectGraph_TrackableObject_SerializedTensor) ProtoMessage 

func (*TrackableObjectGraph_TrackableObject_SerializedTensor) ProtoMessage()

func (*TrackableObjectGraph_TrackableObject_SerializedTensor) ProtoReflect 

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

func (*TrackableObjectGraph_TrackableObject_SerializedTensor) Reset 

func (x *TrackableObjectGraph_TrackableObject_SerializedTensor) Reset()

func (*TrackableObjectGraph_TrackableObject_SerializedTensor) String 

func (x *TrackableObjectGraph_TrackableObject_SerializedTensor) String() string

type TrackableObjectGraph_TrackableObject_SlotVariableReference 

type TrackableObjectGraph_TrackableObject_SlotVariableReference struct {

	// An index into `TrackableObjectGraph.nodes`, indicating the
	// variable object this slot was created for.
	OriginalVariableNodeId int32 `` /* 132-byte string literal not displayed */
	// The name of the slot (e.g. "m"/"v").
	SlotName string `protobuf:"bytes,2,opt,name=slot_name,json=slotName,proto3" json:"slot_name,omitempty"`
	// An index into `TrackableObjectGraph.nodes`, indicating the
	// `Object` with the value of the slot variable.
	SlotVariableNodeId int32 `protobuf:"varint,3,opt,name=slot_variable_node_id,json=slotVariableNodeId,proto3" json:"slot_variable_node_id,omitempty"`
	// contains filtered or unexported fields
}

func (*TrackableObjectGraph_TrackableObject_SlotVariableReference) Descriptor deprecated 

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

Deprecated: Use TrackableObjectGraph_TrackableObject_SlotVariableReference.ProtoReflect.Descriptor instead.

func (*TrackableObjectGraph_TrackableObject_SlotVariableReference) GetOriginalVariableNodeId 

func (x *TrackableObjectGraph_TrackableObject_SlotVariableReference) GetOriginalVariableNodeId() int32

func (*TrackableObjectGraph_TrackableObject_SlotVariableReference) GetSlotName 

func (x *TrackableObjectGraph_TrackableObject_SlotVariableReference) GetSlotName() string

func (*TrackableObjectGraph_TrackableObject_SlotVariableReference) GetSlotVariableNodeId 

func (x *TrackableObjectGraph_TrackableObject_SlotVariableReference) GetSlotVariableNodeId() int32

func (*TrackableObjectGraph_TrackableObject_SlotVariableReference) ProtoMessage 

func (*TrackableObjectGraph_TrackableObject_SlotVariableReference) ProtoMessage()

func (*TrackableObjectGraph_TrackableObject_SlotVariableReference) ProtoReflect 

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

func (*TrackableObjectGraph_TrackableObject_SlotVariableReference) Reset 

func (x *TrackableObjectGraph_TrackableObject_SlotVariableReference) Reset()

func (*TrackableObjectGraph_TrackableObject_SlotVariableReference) String 

func (x *TrackableObjectGraph_TrackableObject_SlotVariableReference) String() string

type TryGetKeyValueRequest 

type TryGetKeyValueRequest struct {
	Key string `protobuf:"bytes,1,opt,name=key,proto3" json:"key,omitempty"`
	// contains filtered or unexported fields
}

func (*TryGetKeyValueRequest) Descriptor deprecated 

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

Deprecated: Use TryGetKeyValueRequest.ProtoReflect.Descriptor instead.

func (*TryGetKeyValueRequest) GetKey 

func (x *TryGetKeyValueRequest) GetKey() string

func (*TryGetKeyValueRequest) ProtoMessage 

func (*TryGetKeyValueRequest) ProtoMessage()

func (*TryGetKeyValueRequest) ProtoReflect 

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

func (*TryGetKeyValueRequest) Reset 

func (x *TryGetKeyValueRequest) Reset()

func (*TryGetKeyValueRequest) String 

func (x *TryGetKeyValueRequest) String() string

type TryGetKeyValueResponse 

type TryGetKeyValueResponse struct {
	Kv *KeyValueEntry `protobuf:"bytes,1,opt,name=kv,proto3" json:"kv,omitempty"`
	// contains filtered or unexported fields
}

func (*TryGetKeyValueResponse) Descriptor deprecated 

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

Deprecated: Use TryGetKeyValueResponse.ProtoReflect.Descriptor instead.

func (*TryGetKeyValueResponse) GetKv 

func (x *TryGetKeyValueResponse) GetKv() *KeyValueEntry

func (*TryGetKeyValueResponse) ProtoMessage 

func (*TryGetKeyValueResponse) ProtoMessage()

func (*TryGetKeyValueResponse) ProtoReflect 

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

func (*TryGetKeyValueResponse) Reset 

func (x *TryGetKeyValueResponse) Reset()

func (*TryGetKeyValueResponse) String 

func (x *TryGetKeyValueResponse) String() string

type TupleValue 

type TupleValue struct {
	Values []*StructuredValue `protobuf:"bytes,1,rep,name=values,proto3" json:"values,omitempty"`
	// contains filtered or unexported fields
}

Represents a Python tuple.

func (*TupleValue) Descriptor deprecated 

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

Deprecated: Use TupleValue.ProtoReflect.Descriptor instead.

func (*TupleValue) GetValues 

func (x *TupleValue) GetValues() []*StructuredValue

func (*TupleValue) ProtoMessage 

func (*TupleValue) ProtoMessage()

func (*TupleValue) ProtoReflect 

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

func (*TupleValue) Reset 

func (x *TupleValue) Reset()

func (*TupleValue) String 

func (x *TupleValue) String() string

type TypeSpecProto 

type TypeSpecProto struct {
	TypeSpecClass TypeSpecProto_TypeSpecClass `` /* 147-byte string literal not displayed */
	// The value returned by TypeSpec._serialize().
	TypeState *StructuredValue `protobuf:"bytes,2,opt,name=type_state,json=typeState,proto3" json:"type_state,omitempty"`
	// The name of the TypeSpec class.
	//  * If type_spec_class == REGISTERED_TYPE_SPEC, the TypeSpec class is
	//    the one registered under this name. For types registered outside
	//    core TensorFlow by an add-on library, that library must be loaded
	//    before this value can be deserialized by nested_structure_coder.
	//  * If type_spec_class specifies a particular TypeSpec class, this field is
	//    redundant with the type_spec_class enum, and is only used for error
	//    reporting in older binaries that do not know the tupe_spec_class enum.
	TypeSpecClassName string `protobuf:"bytes,3,opt,name=type_spec_class_name,json=typeSpecClassName,proto3" json:"type_spec_class_name,omitempty"`
	// The number of flat tensor components required by this TypeSpec.
	NumFlatComponents int32 `protobuf:"varint,4,opt,name=num_flat_components,json=numFlatComponents,proto3" json:"num_flat_components,omitempty"`
	// contains filtered or unexported fields
}

Represents a tf.TypeSpec

func (*TypeSpecProto) Descriptor deprecated 

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

Deprecated: Use TypeSpecProto.ProtoReflect.Descriptor instead.

func (*TypeSpecProto) GetNumFlatComponents 

func (x *TypeSpecProto) GetNumFlatComponents() int32

func (*TypeSpecProto) GetTypeSpecClass 

func (x *TypeSpecProto) GetTypeSpecClass() TypeSpecProto_TypeSpecClass

func (*TypeSpecProto) GetTypeSpecClassName 

func (x *TypeSpecProto) GetTypeSpecClassName() string

func (*TypeSpecProto) GetTypeState 

func (x *TypeSpecProto) GetTypeState() *StructuredValue

func (*TypeSpecProto) ProtoMessage 

func (*TypeSpecProto) ProtoMessage()

func (*TypeSpecProto) ProtoReflect 

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

func (*TypeSpecProto) Reset 

func (x *TypeSpecProto) Reset()

func (*TypeSpecProto) String 

func (x *TypeSpecProto) String() string

type TypeSpecProto_TypeSpecClass 

type TypeSpecProto_TypeSpecClass int32
const (
	TypeSpecProto_UNKNOWN              TypeSpecProto_TypeSpecClass = 0
	TypeSpecProto_SPARSE_TENSOR_SPEC   TypeSpecProto_TypeSpecClass = 1  // tf.SparseTensorSpec
	TypeSpecProto_INDEXED_SLICES_SPEC  TypeSpecProto_TypeSpecClass = 2  // tf.IndexedSlicesSpec
	TypeSpecProto_RAGGED_TENSOR_SPEC   TypeSpecProto_TypeSpecClass = 3  // tf.RaggedTensorSpec
	TypeSpecProto_TENSOR_ARRAY_SPEC    TypeSpecProto_TypeSpecClass = 4  // tf.TensorArraySpec
	TypeSpecProto_DATA_DATASET_SPEC    TypeSpecProto_TypeSpecClass = 5  // tf.data.DatasetSpec
	TypeSpecProto_DATA_ITERATOR_SPEC   TypeSpecProto_TypeSpecClass = 6  // IteratorSpec from data/ops/iterator_ops.py
	TypeSpecProto_OPTIONAL_SPEC        TypeSpecProto_TypeSpecClass = 7  // tf.OptionalSpec
	TypeSpecProto_PER_REPLICA_SPEC     TypeSpecProto_TypeSpecClass = 8  // PerReplicaSpec from distribute/values.py
	TypeSpecProto_VARIABLE_SPEC        TypeSpecProto_TypeSpecClass = 9  // tf.VariableSpec
	TypeSpecProto_ROW_PARTITION_SPEC   TypeSpecProto_TypeSpecClass = 10 // RowPartitionSpec from ragged/row_partition.py
	TypeSpecProto_REGISTERED_TYPE_SPEC TypeSpecProto_TypeSpecClass = 12 // The type registered as type_spec_class_name.
	TypeSpecProto_EXTENSION_TYPE_SPEC  TypeSpecProto_TypeSpecClass = 13 // Subclasses of tf.ExtensionType
)

func (TypeSpecProto_TypeSpecClass) Descriptor 

func (TypeSpecProto_TypeSpecClass) Descriptor() protoreflect.EnumDescriptor

func (TypeSpecProto_TypeSpecClass) Enum 

func (x TypeSpecProto_TypeSpecClass) Enum() *TypeSpecProto_TypeSpecClass

func (TypeSpecProto_TypeSpecClass) EnumDescriptor deprecated 

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

Deprecated: Use TypeSpecProto_TypeSpecClass.Descriptor instead.

func (TypeSpecProto_TypeSpecClass) Number 

func (x TypeSpecProto_TypeSpecClass) Number() protoreflect.EnumNumber

func (TypeSpecProto_TypeSpecClass) String 

func (x TypeSpecProto_TypeSpecClass) String() string

func (TypeSpecProto_TypeSpecClass) Type 

func (TypeSpecProto_TypeSpecClass) Type() protoreflect.EnumType

type UpdateContextRequest 

type UpdateContextRequest struct {

	// Identifies the full cluster, and this particular worker's position within.
	ServerDef *ServerDef `protobuf:"bytes,1,opt,name=server_def,json=serverDef,proto3" json:"server_def,omitempty"`
	// Device attributes in the cluster.
	// If this field is empty, it indicates that this is a simple update request
	// that only increments the cluster view ID and does not require changes to
	// the workers it connects to.
	ClusterDeviceAttributes []*DeviceAttributes `` /* 132-byte string literal not displayed */
	// The ID of the context to be updated. A context with the specified ID must
	// already exist on the recepient server of this request.
	ContextId uint64 `protobuf:"fixed64,3,opt,name=context_id,json=contextId,proto3" json:"context_id,omitempty"`
	// The view ID of the context, which should be contiguously incremented when
	// updating the same context.
	ContextViewId uint64 `protobuf:"fixed64,4,opt,name=context_view_id,json=contextViewId,proto3" json:"context_view_id,omitempty"`
	// contains filtered or unexported fields
}

func (*UpdateContextRequest) Descriptor deprecated 

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

Deprecated: Use UpdateContextRequest.ProtoReflect.Descriptor instead.

func (*UpdateContextRequest) GetClusterDeviceAttributes 

func (x *UpdateContextRequest) GetClusterDeviceAttributes() []*DeviceAttributes

func (*UpdateContextRequest) GetContextId 

func (x *UpdateContextRequest) GetContextId() uint64

func (*UpdateContextRequest) GetContextViewId 

func (x *UpdateContextRequest) GetContextViewId() uint64

func (*UpdateContextRequest) GetServerDef 

func (x *UpdateContextRequest) GetServerDef() *ServerDef

func (*UpdateContextRequest) ProtoMessage 

func (*UpdateContextRequest) ProtoMessage()

func (*UpdateContextRequest) ProtoReflect 

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

func (*UpdateContextRequest) Reset 

func (x *UpdateContextRequest) Reset()

func (*UpdateContextRequest) String 

func (x *UpdateContextRequest) String() string

type UpdateContextResponse 

type UpdateContextResponse struct {

	// List of devices that are locally accessible to the worker.
	DeviceAttributes []*DeviceAttributes `protobuf:"bytes,1,rep,name=device_attributes,json=deviceAttributes,proto3" json:"device_attributes,omitempty"`
	// contains filtered or unexported fields
}

func (*UpdateContextResponse) Descriptor deprecated 

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

Deprecated: Use UpdateContextResponse.ProtoReflect.Descriptor instead.

func (*UpdateContextResponse) GetDeviceAttributes 

func (x *UpdateContextResponse) GetDeviceAttributes() []*DeviceAttributes

func (*UpdateContextResponse) ProtoMessage 

func (*UpdateContextResponse) ProtoMessage()

func (*UpdateContextResponse) ProtoReflect 

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

func (*UpdateContextResponse) Reset 

func (x *UpdateContextResponse) Reset()

func (*UpdateContextResponse) String 

func (x *UpdateContextResponse) String() string

type ValuesDef 

type ValuesDef struct {

	// Value names that have been seen in this context.
	Values []string `protobuf:"bytes,1,rep,name=values,proto3" json:"values,omitempty"`
	// Value names referenced by but external to this context.
	ExternalValues map[string]string `` /* 191-byte string literal not displayed */
	// contains filtered or unexported fields
}

Protocol buffer representing the values in ControlFlowContext.

func (*ValuesDef) Descriptor deprecated 

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

Deprecated: Use ValuesDef.ProtoReflect.Descriptor instead.

func (*ValuesDef) GetExternalValues 

func (x *ValuesDef) GetExternalValues() map[string]string

func (*ValuesDef) GetValues 

func (x *ValuesDef) GetValues() []string

func (*ValuesDef) ProtoMessage 

func (*ValuesDef) ProtoMessage()

func (*ValuesDef) ProtoReflect 

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

func (*ValuesDef) Reset 

func (x *ValuesDef) Reset()

func (*ValuesDef) String 

func (x *ValuesDef) String() string

type VarLenFeatureProto 

type VarLenFeatureProto struct {
	Dtype                   DataType `protobuf:"varint,1,opt,name=dtype,proto3,enum=tensorflow.DataType" json:"dtype,omitempty"`
	ValuesOutputTensorName  string   `` /* 131-byte string literal not displayed */
	IndicesOutputTensorName string   `` /* 134-byte string literal not displayed */
	ShapesOutputTensorName  string   `` /* 131-byte string literal not displayed */
	// contains filtered or unexported fields
}

func (*VarLenFeatureProto) Descriptor deprecated 

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

Deprecated: Use VarLenFeatureProto.ProtoReflect.Descriptor instead.

func (*VarLenFeatureProto) GetDtype 

func (x *VarLenFeatureProto) GetDtype() DataType

func (*VarLenFeatureProto) GetIndicesOutputTensorName 

func (x *VarLenFeatureProto) GetIndicesOutputTensorName() string

func (*VarLenFeatureProto) GetShapesOutputTensorName 

func (x *VarLenFeatureProto) GetShapesOutputTensorName() string

func (*VarLenFeatureProto) GetValuesOutputTensorName 

func (x *VarLenFeatureProto) GetValuesOutputTensorName() string

func (*VarLenFeatureProto) ProtoMessage 

func (*VarLenFeatureProto) ProtoMessage()

func (*VarLenFeatureProto) ProtoReflect 

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

func (*VarLenFeatureProto) Reset 

func (x *VarLenFeatureProto) Reset()

func (*VarLenFeatureProto) String 

func (x *VarLenFeatureProto) 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 VerifierConfig 

type VerifierConfig struct {

	// Deadline for completion of all verification i.e. all the Toggle ON
	// verifiers must complete execution within this time.
	VerificationTimeoutInMs int64 `` /* 135-byte string literal not displayed */
	// Perform structural validation on a tensorflow graph. Default is OFF.
	StructureVerifier VerifierConfig_Toggle `` /* 151-byte string literal not displayed */
	// contains filtered or unexported fields
}

The config for graph verifiers.

func (*VerifierConfig) Descriptor deprecated 

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

Deprecated: Use VerifierConfig.ProtoReflect.Descriptor instead.

func (*VerifierConfig) GetStructureVerifier 

func (x *VerifierConfig) GetStructureVerifier() VerifierConfig_Toggle

func (*VerifierConfig) GetVerificationTimeoutInMs 

func (x *VerifierConfig) GetVerificationTimeoutInMs() int64

func (*VerifierConfig) ProtoMessage 

func (*VerifierConfig) ProtoMessage()

func (*VerifierConfig) ProtoReflect 

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

func (*VerifierConfig) Reset 

func (x *VerifierConfig) Reset()

func (*VerifierConfig) String 

func (x *VerifierConfig) String() string

type VerifierConfig_Toggle 

type VerifierConfig_Toggle int32
const (
	VerifierConfig_DEFAULT VerifierConfig_Toggle = 0
	VerifierConfig_ON      VerifierConfig_Toggle = 1
	VerifierConfig_OFF     VerifierConfig_Toggle = 2
)

func (VerifierConfig_Toggle) Descriptor 

func (VerifierConfig_Toggle) Descriptor() protoreflect.EnumDescriptor

func (VerifierConfig_Toggle) Enum 

func (x VerifierConfig_Toggle) Enum() *VerifierConfig_Toggle

func (VerifierConfig_Toggle) EnumDescriptor deprecated 

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

Deprecated: Use VerifierConfig_Toggle.Descriptor instead.

func (VerifierConfig_Toggle) Number 

func (x VerifierConfig_Toggle) Number() protoreflect.EnumNumber

func (VerifierConfig_Toggle) String 

func (x VerifierConfig_Toggle) String() string

func (VerifierConfig_Toggle) Type 

func (VerifierConfig_Toggle) Type() protoreflect.EnumType

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

type WaitAtBarrierRequest 

type WaitAtBarrierRequest struct {
	SessionId           uint64 `protobuf:"varint,1,opt,name=session_id,json=sessionId,proto3" json:"session_id,omitempty"`
	BarrierId           []byte `protobuf:"bytes,2,opt,name=barrier_id,json=barrierId,proto3" json:"barrier_id,omitempty"`
	NodeId              int32  `protobuf:"varint,3,opt,name=node_id,json=nodeId,proto3" json:"node_id,omitempty"`
	TimeoutMilliseconds int32  `protobuf:"varint,4,opt,name=timeout_milliseconds,json=timeoutMilliseconds,proto3" json:"timeout_milliseconds,omitempty"`
	// contains filtered or unexported fields
}

func (*WaitAtBarrierRequest) Descriptor deprecated 

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

Deprecated: Use WaitAtBarrierRequest.ProtoReflect.Descriptor instead.

func (*WaitAtBarrierRequest) GetBarrierId 

func (x *WaitAtBarrierRequest) GetBarrierId() []byte

func (*WaitAtBarrierRequest) GetNodeId 

func (x *WaitAtBarrierRequest) GetNodeId() int32

func (*WaitAtBarrierRequest) GetSessionId 

func (x *WaitAtBarrierRequest) GetSessionId() uint64

func (*WaitAtBarrierRequest) GetTimeoutMilliseconds 

func (x *WaitAtBarrierRequest) GetTimeoutMilliseconds() int32

func (*WaitAtBarrierRequest) ProtoMessage 

func (*WaitAtBarrierRequest) ProtoMessage()

func (*WaitAtBarrierRequest) ProtoReflect 

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

func (*WaitAtBarrierRequest) Reset 

func (x *WaitAtBarrierRequest) Reset()

func (*WaitAtBarrierRequest) String 

func (x *WaitAtBarrierRequest) String() string

type WaitAtBarrierResponse 

type WaitAtBarrierResponse struct {
	// contains filtered or unexported fields
}

func (*WaitAtBarrierResponse) Descriptor deprecated 

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

Deprecated: Use WaitAtBarrierResponse.ProtoReflect.Descriptor instead.

func (*WaitAtBarrierResponse) ProtoMessage 

func (*WaitAtBarrierResponse) ProtoMessage()

func (*WaitAtBarrierResponse) ProtoReflect 

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

func (*WaitAtBarrierResponse) Reset 

func (x *WaitAtBarrierResponse) Reset()

func (*WaitAtBarrierResponse) String 

func (x *WaitAtBarrierResponse) String() string

type WaitForAllTasksRequest 

type WaitForAllTasksRequest struct {

	// All local device attributes on the request sender.
	LocalDeviceInfo *CoordinationServiceDeviceInfo `protobuf:"bytes,4,opt,name=local_device_info,json=localDeviceInfo,proto3" json:"local_device_info,omitempty"`
	SourceTask      *CoordinatedTask               `protobuf:"bytes,5,opt,name=source_task,json=sourceTask,proto3" json:"source_task,omitempty"`
	// contains filtered or unexported fields
}

Request and response messages for waiting for all tasks.

func (*WaitForAllTasksRequest) Descriptor deprecated 

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

Deprecated: Use WaitForAllTasksRequest.ProtoReflect.Descriptor instead.

func (*WaitForAllTasksRequest) GetLocalDeviceInfo 

func (x *WaitForAllTasksRequest) GetLocalDeviceInfo() *CoordinationServiceDeviceInfo

func (*WaitForAllTasksRequest) GetSourceTask 

func (x *WaitForAllTasksRequest) GetSourceTask() *CoordinatedTask

func (*WaitForAllTasksRequest) ProtoMessage 

func (*WaitForAllTasksRequest) ProtoMessage()

func (*WaitForAllTasksRequest) ProtoReflect 

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

func (*WaitForAllTasksRequest) Reset 

func (x *WaitForAllTasksRequest) Reset()

func (*WaitForAllTasksRequest) String 

func (x *WaitForAllTasksRequest) String() string

type WaitForAllTasksResponse 

type WaitForAllTasksResponse struct {
	LeaderIncarnation uint64 `protobuf:"fixed64,1,opt,name=leader_incarnation,json=leaderIncarnation,proto3" json:"leader_incarnation,omitempty"`
	// All devices in the cluster.
	ClusterDeviceInfo *CoordinationServiceDeviceInfo `protobuf:"bytes,3,opt,name=cluster_device_info,json=clusterDeviceInfo,proto3" json:"cluster_device_info,omitempty"`
	// contains filtered or unexported fields
}

func (*WaitForAllTasksResponse) Descriptor deprecated 

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

Deprecated: Use WaitForAllTasksResponse.ProtoReflect.Descriptor instead.

func (*WaitForAllTasksResponse) GetClusterDeviceInfo 

func (x *WaitForAllTasksResponse) GetClusterDeviceInfo() *CoordinationServiceDeviceInfo

func (*WaitForAllTasksResponse) GetLeaderIncarnation 

func (x *WaitForAllTasksResponse) GetLeaderIncarnation() uint64

func (*WaitForAllTasksResponse) ProtoMessage 

func (*WaitForAllTasksResponse) ProtoMessage()

func (*WaitForAllTasksResponse) ProtoReflect 

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

func (*WaitForAllTasksResponse) Reset 

func (x *WaitForAllTasksResponse) Reset()

func (*WaitForAllTasksResponse) String 

func (x *WaitForAllTasksResponse) String() string

type WaitQueueDoneRequest 

type WaitQueueDoneRequest struct {
	ContextId uint64 `protobuf:"fixed64,1,opt,name=context_id,json=contextId,proto3" json:"context_id,omitempty"`
	// Ids to wait on. If empty, wait on everything currently pending.
	OpId []int64 `protobuf:"varint,2,rep,packed,name=op_id,json=opId,proto3" json:"op_id,omitempty"`
	// contains filtered or unexported fields
}

func (*WaitQueueDoneRequest) Descriptor deprecated 

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

Deprecated: Use WaitQueueDoneRequest.ProtoReflect.Descriptor instead.

func (*WaitQueueDoneRequest) GetContextId 

func (x *WaitQueueDoneRequest) GetContextId() uint64

func (*WaitQueueDoneRequest) GetOpId 

func (x *WaitQueueDoneRequest) GetOpId() []int64

func (*WaitQueueDoneRequest) ProtoMessage 

func (*WaitQueueDoneRequest) ProtoMessage()

func (*WaitQueueDoneRequest) ProtoReflect 

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

func (*WaitQueueDoneRequest) Reset 

func (x *WaitQueueDoneRequest) Reset()

func (*WaitQueueDoneRequest) String 

func (x *WaitQueueDoneRequest) String() string

type WaitQueueDoneResponse 

type WaitQueueDoneResponse struct {
	// contains filtered or unexported fields
}

func (*WaitQueueDoneResponse) Descriptor deprecated 

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

Deprecated: Use WaitQueueDoneResponse.ProtoReflect.Descriptor instead.

func (*WaitQueueDoneResponse) ProtoMessage 

func (*WaitQueueDoneResponse) ProtoMessage()

func (*WaitQueueDoneResponse) ProtoReflect 

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

func (*WaitQueueDoneResponse) Reset 

func (x *WaitQueueDoneResponse) Reset()

func (*WaitQueueDoneResponse) String 

func (x *WaitQueueDoneResponse) String() string

type WatchdogConfig 

type WatchdogConfig struct {
	TimeoutMs int64 `protobuf:"varint,1,opt,name=timeout_ms,json=timeoutMs,proto3" json:"timeout_ms,omitempty"`
	// contains filtered or unexported fields
}

func (*WatchdogConfig) Descriptor deprecated 

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

Deprecated: Use WatchdogConfig.ProtoReflect.Descriptor instead.

func (*WatchdogConfig) GetTimeoutMs 

func (x *WatchdogConfig) GetTimeoutMs() int64

func (*WatchdogConfig) ProtoMessage 

func (*WatchdogConfig) ProtoMessage()

func (*WatchdogConfig) ProtoReflect 

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

func (*WatchdogConfig) Reset 

func (x *WatchdogConfig) Reset()

func (*WatchdogConfig) String 

func (x *WatchdogConfig) String() string

type WhileContextDef 

type WhileContextDef struct {

	// Name of the context.
	ContextName string `protobuf:"bytes,1,opt,name=context_name,json=contextName,proto3" json:"context_name,omitempty"`
	// The number of iterations allowed to run in parallel.
	ParallelIterations int32 `protobuf:"varint,2,opt,name=parallel_iterations,json=parallelIterations,proto3" json:"parallel_iterations,omitempty"`
	// Whether backprop is enabled for this while loop.
	BackProp bool `protobuf:"varint,3,opt,name=back_prop,json=backProp,proto3" json:"back_prop,omitempty"`
	// Whether GPU-CPU memory swap is enabled for this loop.
	SwapMemory bool `protobuf:"varint,4,opt,name=swap_memory,json=swapMemory,proto3" json:"swap_memory,omitempty"`
	// Name of the pivot tensor.
	PivotName string `protobuf:"bytes,5,opt,name=pivot_name,json=pivotName,proto3" json:"pivot_name,omitempty"`
	// Name of the pivot_for_pred tensor.
	PivotForPredName string `protobuf:"bytes,6,opt,name=pivot_for_pred_name,json=pivotForPredName,proto3" json:"pivot_for_pred_name,omitempty"`
	// Name of the pivot_for_body tensor.
	PivotForBodyName string `protobuf:"bytes,7,opt,name=pivot_for_body_name,json=pivotForBodyName,proto3" json:"pivot_for_body_name,omitempty"`
	// List of names for exit tensors.
	LoopExitNames []string `protobuf:"bytes,8,rep,name=loop_exit_names,json=loopExitNames,proto3" json:"loop_exit_names,omitempty"`
	// List of names for enter tensors.
	LoopEnterNames []string `protobuf:"bytes,10,rep,name=loop_enter_names,json=loopEnterNames,proto3" json:"loop_enter_names,omitempty"`
	// Values and external values in control flow context.
	ValuesDef *ValuesDef `protobuf:"bytes,9,opt,name=values_def,json=valuesDef,proto3" json:"values_def,omitempty"`
	// Optional name of the maximum_iterations tensor.
	MaximumIterationsName string `` /* 127-byte string literal not displayed */
	// Contexts contained inside this context (e.g. nested whiles).
	NestedContexts []*ControlFlowContextDef `protobuf:"bytes,12,rep,name=nested_contexts,json=nestedContexts,proto3" json:"nested_contexts,omitempty"`
	// contains filtered or unexported fields
}

Protocol buffer representing a WhileContext object.

func (*WhileContextDef) Descriptor deprecated 

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

Deprecated: Use WhileContextDef.ProtoReflect.Descriptor instead.

func (*WhileContextDef) GetBackProp 

func (x *WhileContextDef) GetBackProp() bool

func (*WhileContextDef) GetContextName 

func (x *WhileContextDef) GetContextName() string

func (*WhileContextDef) GetLoopEnterNames 

func (x *WhileContextDef) GetLoopEnterNames() []string

func (*WhileContextDef) GetLoopExitNames 

func (x *WhileContextDef) GetLoopExitNames() []string

func (*WhileContextDef) GetMaximumIterationsName 

func (x *WhileContextDef) GetMaximumIterationsName() string

func (*WhileContextDef) GetNestedContexts 

func (x *WhileContextDef) GetNestedContexts() []*ControlFlowContextDef

func (*WhileContextDef) GetParallelIterations 

func (x *WhileContextDef) GetParallelIterations() int32

func (*WhileContextDef) GetPivotForBodyName 

func (x *WhileContextDef) GetPivotForBodyName() string

func (*WhileContextDef) GetPivotForPredName 

func (x *WhileContextDef) GetPivotForPredName() string

func (*WhileContextDef) GetPivotName 

func (x *WhileContextDef) GetPivotName() string

func (*WhileContextDef) GetSwapMemory 

func (x *WhileContextDef) GetSwapMemory() bool

func (*WhileContextDef) GetValuesDef 

func (x *WhileContextDef) GetValuesDef() *ValuesDef

func (*WhileContextDef) ProtoMessage 

func (*WhileContextDef) ProtoMessage()

func (*WhileContextDef) ProtoReflect 

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

func (*WhileContextDef) Reset 

func (x *WhileContextDef) Reset()

func (*WhileContextDef) String 

func (x *WhileContextDef) String() string

type WorkerConfig 

type WorkerConfig struct {

	// The port for the worker to bind to. A value of 0 indicates that the
	// worker may bind to any available port.
	Port int64 `protobuf:"varint,1,opt,name=port,proto3" json:"port,omitempty"`
	// The protocol for the worker to use when connecting to the dispatcher.
	Protocol string `protobuf:"bytes,2,opt,name=protocol,proto3" json:"protocol,omitempty"`
	// The address of the dispatcher to register with.
	DispatcherAddress string `protobuf:"bytes,3,opt,name=dispatcher_address,json=dispatcherAddress,proto3" json:"dispatcher_address,omitempty"`
	// The address of the worker server. The substring "%port%", if specified,
	// will be replaced with the worker's bound port. This is useful when the port
	// is set to `0`.
	WorkerAddress string `protobuf:"bytes,4,opt,name=worker_address,json=workerAddress,proto3" json:"worker_address,omitempty"`
	// Tags attached to the worker. This allows reading from selected workers.
	// For example, by applying a "COLOCATED" tag, tf.data service is able to read
	// from the local tf.data worker if one exists, then from off-TF-host workers,
	// to avoid cross-TF-host reads.
	WorkerTags []string `protobuf:"bytes,10,rep,name=worker_tags,json=workerTags,proto3" json:"worker_tags,omitempty"`
	// How often the worker should heartbeat to the master. A value of 0 indicates
	// that the decision should be left up to the runtime.
	HeartbeatIntervalMs int64 `protobuf:"varint,5,opt,name=heartbeat_interval_ms,json=heartbeatIntervalMs,proto3" json:"heartbeat_interval_ms,omitempty"`
	// How long to retry requests to the dispatcher before giving up and reporting
	// an error. A value of 0 indicates that the decision should be left up to the
	// runtime.
	DispatcherTimeoutMs int64 `protobuf:"varint,6,opt,name=dispatcher_timeout_ms,json=dispatcherTimeoutMs,proto3" json:"dispatcher_timeout_ms,omitempty"`
	// The protocol for the worker to use when transferring data to clients.
	DataTransferProtocol string `protobuf:"bytes,7,opt,name=data_transfer_protocol,json=dataTransferProtocol,proto3" json:"data_transfer_protocol,omitempty"`
	// The data transfer address of the worker server. The substring "%port%", if
	// specified, will be replaced with the worker's bound port. This is useful
	// when the port is set to `0`.
	DataTransferAddress string `protobuf:"bytes,8,opt,name=data_transfer_address,json=dataTransferAddress,proto3" json:"data_transfer_address,omitempty"`
	// Maximum size of the cross-trainer cache in bytes. If enabled, make sure
	// your training job provides sufficient memory resources.
	CrossTrainerCacheSizeBytes int64 `` /* 147-byte string literal not displayed */
	// When shutting down a worker, how long to wait for the gRPC server to
	// process the final requests. This is used to achieve clean shutdown in unit
	// tests.
	ShutdownQuietPeriodMs int64 `` /* 129-byte string literal not displayed */
	// contains filtered or unexported fields
}

Configuration for a tf.data service WorkerServer. Next id: 12

func (*WorkerConfig) Descriptor deprecated 

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

Deprecated: Use WorkerConfig.ProtoReflect.Descriptor instead.

func (*WorkerConfig) GetCrossTrainerCacheSizeBytes 

func (x *WorkerConfig) GetCrossTrainerCacheSizeBytes() int64

func (*WorkerConfig) GetDataTransferAddress 

func (x *WorkerConfig) GetDataTransferAddress() string

func (*WorkerConfig) GetDataTransferProtocol 

func (x *WorkerConfig) GetDataTransferProtocol() string

func (*WorkerConfig) GetDispatcherAddress 

func (x *WorkerConfig) GetDispatcherAddress() string

func (*WorkerConfig) GetDispatcherTimeoutMs 

func (x *WorkerConfig) GetDispatcherTimeoutMs() int64

func (*WorkerConfig) GetHeartbeatIntervalMs 

func (x *WorkerConfig) GetHeartbeatIntervalMs() int64

func (*WorkerConfig) GetPort 

func (x *WorkerConfig) GetPort() int64

func (*WorkerConfig) GetProtocol 

func (x *WorkerConfig) GetProtocol() string

func (*WorkerConfig) GetShutdownQuietPeriodMs 

func (x *WorkerConfig) GetShutdownQuietPeriodMs() int64

func (*WorkerConfig) GetWorkerAddress 

func (x *WorkerConfig) GetWorkerAddress() string

func (*WorkerConfig) GetWorkerTags 

func (x *WorkerConfig) GetWorkerTags() []string

func (*WorkerConfig) ProtoMessage 

func (*WorkerConfig) ProtoMessage()

func (*WorkerConfig) ProtoReflect 

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

func (*WorkerConfig) Reset 

func (x *WorkerConfig) Reset()

func (*WorkerConfig) String 

func (x *WorkerConfig) String() string

type WorkerHealth 

type WorkerHealth int32

Current health status of a worker. 

const (
	WorkerHealth_OK                       WorkerHealth = 0 // By default a worker is healthy.
	WorkerHealth_RECEIVED_SHUTDOWN_SIGNAL WorkerHealth = 1
	WorkerHealth_INTERNAL_ERROR           WorkerHealth = 2
	WorkerHealth_SHUTTING_DOWN            WorkerHealth = 3 // Worker has been instructed to shutdown after a timeout.
)

func (WorkerHealth) Descriptor 

func (WorkerHealth) Descriptor() protoreflect.EnumDescriptor

func (WorkerHealth) Enum 

func (x WorkerHealth) Enum() *WorkerHealth

func (WorkerHealth) EnumDescriptor deprecated 

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

Deprecated: Use WorkerHealth.Descriptor instead.

func (WorkerHealth) Number 

func (x WorkerHealth) Number() protoreflect.EnumNumber

func (WorkerHealth) String 

func (x WorkerHealth) String() string

func (WorkerHealth) Type 

func (WorkerHealth) Type() protoreflect.EnumType

type WorkerHeartbeatRequest 

type WorkerHeartbeatRequest struct {
	ShutdownMode   WorkerShutdownMode `` /* 133-byte string literal not displayed */
	WatchdogConfig *WatchdogConfig    `protobuf:"bytes,2,opt,name=watchdog_config,json=watchdogConfig,proto3" json:"watchdog_config,omitempty"`
	ExitCode       *RequestedExitCode `protobuf:"bytes,3,opt,name=exit_code,json=exitCode,proto3" json:"exit_code,omitempty"`
	// contains filtered or unexported fields
}

func (*WorkerHeartbeatRequest) Descriptor deprecated 

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

Deprecated: Use WorkerHeartbeatRequest.ProtoReflect.Descriptor instead.

func (*WorkerHeartbeatRequest) GetExitCode 

func (x *WorkerHeartbeatRequest) GetExitCode() *RequestedExitCode

func (*WorkerHeartbeatRequest) GetShutdownMode 

func (x *WorkerHeartbeatRequest) GetShutdownMode() WorkerShutdownMode

func (*WorkerHeartbeatRequest) GetWatchdogConfig 

func (x *WorkerHeartbeatRequest) GetWatchdogConfig() *WatchdogConfig

func (*WorkerHeartbeatRequest) ProtoMessage 

func (*WorkerHeartbeatRequest) ProtoMessage()

func (*WorkerHeartbeatRequest) ProtoReflect 

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

func (*WorkerHeartbeatRequest) Reset 

func (x *WorkerHeartbeatRequest) Reset()

func (*WorkerHeartbeatRequest) String 

func (x *WorkerHeartbeatRequest) String() string

type WorkerHeartbeatResponse 

type WorkerHeartbeatResponse struct {
	HealthStatus WorkerHealth `` /* 127-byte string literal not displayed */
	WorkerLog    []*Event     `protobuf:"bytes,2,rep,name=worker_log,json=workerLog,proto3" json:"worker_log,omitempty"`
	Hostname     string       `protobuf:"bytes,3,opt,name=hostname,proto3" json:"hostname,omitempty"`
	// contains filtered or unexported fields
}

func (*WorkerHeartbeatResponse) Descriptor deprecated 

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

Deprecated: Use WorkerHeartbeatResponse.ProtoReflect.Descriptor instead.

func (*WorkerHeartbeatResponse) GetHealthStatus 

func (x *WorkerHeartbeatResponse) GetHealthStatus() WorkerHealth

func (*WorkerHeartbeatResponse) GetHostname 

func (x *WorkerHeartbeatResponse) GetHostname() string

func (*WorkerHeartbeatResponse) GetWorkerLog 

func (x *WorkerHeartbeatResponse) GetWorkerLog() []*Event

func (*WorkerHeartbeatResponse) ProtoMessage 

func (*WorkerHeartbeatResponse) ProtoMessage()

func (*WorkerHeartbeatResponse) ProtoReflect 

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

func (*WorkerHeartbeatResponse) Reset 

func (x *WorkerHeartbeatResponse) Reset()

func (*WorkerHeartbeatResponse) String 

func (x *WorkerHeartbeatResponse) String() string

type WorkerPossiblyRestarted 

type WorkerPossiblyRestarted struct {
	// contains filtered or unexported fields
}

If included as a payload, this message flags the Status to be a possible outcome of a worker restart. URI: "type.googleapis.com/tensorflow.distributed_runtime.WorkerPossiblyRestarted"

func (*WorkerPossiblyRestarted) Descriptor deprecated 

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

Deprecated: Use WorkerPossiblyRestarted.ProtoReflect.Descriptor instead.

func (*WorkerPossiblyRestarted) ProtoMessage 

func (*WorkerPossiblyRestarted) ProtoMessage()

func (*WorkerPossiblyRestarted) ProtoReflect 

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

func (*WorkerPossiblyRestarted) Reset 

func (x *WorkerPossiblyRestarted) Reset()

func (*WorkerPossiblyRestarted) String 

func (x *WorkerPossiblyRestarted) String() string

type WorkerShutdownMode 

type WorkerShutdownMode int32

Indicates the behavior of the worker when an internal error or shutdown signal is received. 

const (
	WorkerShutdownMode_DEFAULT                WorkerShutdownMode = 0
	WorkerShutdownMode_NOT_CONFIGURED         WorkerShutdownMode = 1
	WorkerShutdownMode_WAIT_FOR_COORDINATOR   WorkerShutdownMode = 2
	WorkerShutdownMode_SHUTDOWN_AFTER_TIMEOUT WorkerShutdownMode = 3
)

func (WorkerShutdownMode) Descriptor 

func (WorkerShutdownMode) Descriptor() protoreflect.EnumDescriptor

func (WorkerShutdownMode) Enum 

func (x WorkerShutdownMode) Enum() *WorkerShutdownMode

func (WorkerShutdownMode) EnumDescriptor deprecated 

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

Deprecated: Use WorkerShutdownMode.Descriptor instead.

func (WorkerShutdownMode) Number 

func (x WorkerShutdownMode) Number() protoreflect.EnumNumber

func (WorkerShutdownMode) String 

func (x WorkerShutdownMode) String() string

func (WorkerShutdownMode) Type 

func (WorkerShutdownMode) Type() protoreflect.EnumType

type XlaDeviceList 

type XlaDeviceList struct {
	Devices *GlobalTopologyProto `protobuf:"bytes,1,opt,name=devices,proto3" json:"devices,omitempty"`
	// contains filtered or unexported fields
}

func (*XlaDeviceList) Descriptor deprecated 

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

Deprecated: Use XlaDeviceList.ProtoReflect.Descriptor instead.

func (*XlaDeviceList) GetDevices 

func (x *XlaDeviceList) GetDevices() *GlobalTopologyProto

func (*XlaDeviceList) ProtoMessage 

func (*XlaDeviceList) ProtoMessage()

func (*XlaDeviceList) ProtoReflect 

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

func (*XlaDeviceList) Reset 

func (x *XlaDeviceList) Reset()

func (*XlaDeviceList) String 

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