protobuf

package
v2.14.1 Latest Latest
Warning

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

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

Documentation

Index

Constants

View Source 
const (
	CoordinationService_RegisterTask_FullMethodName         = "/tensorflow.CoordinationService/RegisterTask"
	CoordinationService_Heartbeat_FullMethodName            = "/tensorflow.CoordinationService/Heartbeat"
	CoordinationService_WaitForAllTasks_FullMethodName      = "/tensorflow.CoordinationService/WaitForAllTasks"
	CoordinationService_ShutdownTask_FullMethodName         = "/tensorflow.CoordinationService/ShutdownTask"
	CoordinationService_ResetTask_FullMethodName            = "/tensorflow.CoordinationService/ResetTask"
	CoordinationService_ReportErrorToTask_FullMethodName    = "/tensorflow.CoordinationService/ReportErrorToTask"
	CoordinationService_ReportErrorToService_FullMethodName = "/tensorflow.CoordinationService/ReportErrorToService"
	CoordinationService_GetTaskState_FullMethodName         = "/tensorflow.CoordinationService/GetTaskState"
	CoordinationService_InsertKeyValue_FullMethodName       = "/tensorflow.CoordinationService/InsertKeyValue"
	CoordinationService_GetKeyValue_FullMethodName          = "/tensorflow.CoordinationService/GetKeyValue"
	CoordinationService_TryGetKeyValue_FullMethodName       = "/tensorflow.CoordinationService/TryGetKeyValue"
	CoordinationService_GetKeyValueDir_FullMethodName       = "/tensorflow.CoordinationService/GetKeyValueDir"
	CoordinationService_DeleteKeyValue_FullMethodName       = "/tensorflow.CoordinationService/DeleteKeyValue"
	CoordinationService_Barrier_FullMethodName              = "/tensorflow.CoordinationService/Barrier"
	CoordinationService_CancelBarrier_FullMethodName        = "/tensorflow.CoordinationService/CancelBarrier"
)

Variables

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 (
	DataType_name = map[int32]string{
		0:  "kFloat",
		1:  "kDouble",
		2:  "kHalf",
		3:  "kInt8",
		4:  "kInt32",
		5:  "kComplexFloat",
		6:  "kComplexDouble",
		7:  "kBF16",
		8:  "kF8E5M2",
		9:  "kF8E4M3FN",
		10: "kF8E5M2FNUZ",
		11: "kF8E4M3FNUZ",
		12: "kInt64",
	}
	DataType_value = map[string]int32{
		"kFloat":         0,
		"kDouble":        1,
		"kHalf":          2,
		"kInt8":          3,
		"kInt32":         4,
		"kComplexFloat":  5,
		"kComplexDouble": 6,
		"kBF16":          7,
		"kF8E5M2":        8,
		"kF8E4M3FN":      9,
		"kF8E5M2FNUZ":    10,
		"kF8E4M3FNUZ":    11,
		"kInt64":         12,
	}
)

Enum value maps for DataType.

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",
		6: "kOutputInputYX32_CudnnReordered",
		3: "kInputYXOutput",
		4: "kYXInputOutput",
	}
	FilterLayout_value = map[string]int32{
		"kOutputInputYX":                  0,
		"kOutputYXInput":                  1,
		"kOutputInputYX4":                 2,
		"kOutputInputYX32":                5,
		"kOutputInputYX32_CudnnReordered": 6,
		"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",
		7: "kElu",
		8: "kLeakyRelu",
		9: "kGeluExact",
	}
	ActivationMode_value = map[string]int32{
		"kNone":      0,
		"kSigmoid":   1,
		"kRelu":      2,
		"kRelu6":     3,
		"kReluX":     4,
		"kTanh":      5,
		"kBandPass":  6,
		"kElu":       7,
		"kLeakyRelu": 8,
		"kGeluExact": 9,
	}
)

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",
		5: "FORWARD_GRAPH",
	}
	ConvolutionKind_value = map[string]int32{
		"INVALID":                 0,
		"FORWARD":                 1,
		"BACKWARD_FILTER":         2,
		"BACKWARD_DATA":           3,
		"FORWARD_BIAS_ACTIVATION": 4,
		"FORWARD_GRAPH":           5,
	}
)

Enum value maps for ConvolutionKind.

View Source 
var (
	FusedMHAKind_name = map[int32]string{
		0: "BMM1_OUTPUT_UNKNOWN",
		1: "BMM1_OUTPUT_INPUT_TYPE",
		2: "BMM1_OUTPUT_FLOAT",
	}
	FusedMHAKind_value = map[string]int32{
		"BMM1_OUTPUT_UNKNOWN":    0,
		"BMM1_OUTPUT_INPUT_TYPE": 1,
		"BMM1_OUTPUT_FLOAT":      2,
	}
)

Enum value maps for FusedMHAKind.

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 (
	TestResults_BenchmarkType_name = map[int32]string{
		0: "UNKNOWN",
		1: "CPP_MICROBENCHMARK",
		2: "PYTHON_BENCHMARK",
		3: "ANDROID_BENCHMARK",
		4: "EDGE_BENCHMARK",
		5: "IOS_BENCHMARK",
	}
	TestResults_BenchmarkType_value = map[string]int32{
		"UNKNOWN":            0,
		"CPP_MICROBENCHMARK": 1,
		"PYTHON_BENCHMARK":   2,
		"ANDROID_BENCHMARK":  3,
		"EDGE_BENCHMARK":     4,
		"IOS_BENCHMARK":      5,
	}
)

Enum value maps for TestResults_BenchmarkType.

View Source 
var CoordinationService_ServiceDesc = grpc.ServiceDesc{
	ServiceName: "tensorflow.CoordinationService",
	HandlerType: (*CoordinationServiceServer)(nil),
	Methods: []grpc.MethodDesc{
		{
			MethodName: "RegisterTask",
			Handler:    _CoordinationService_RegisterTask_Handler,
		},
		{
			MethodName: "Heartbeat",
			Handler:    _CoordinationService_Heartbeat_Handler,
		},
		{
			MethodName: "WaitForAllTasks",
			Handler:    _CoordinationService_WaitForAllTasks_Handler,
		},
		{
			MethodName: "ShutdownTask",
			Handler:    _CoordinationService_ShutdownTask_Handler,
		},
		{
			MethodName: "ResetTask",
			Handler:    _CoordinationService_ResetTask_Handler,
		},
		{
			MethodName: "ReportErrorToTask",
			Handler:    _CoordinationService_ReportErrorToTask_Handler,
		},
		{
			MethodName: "ReportErrorToService",
			Handler:    _CoordinationService_ReportErrorToService_Handler,
		},
		{
			MethodName: "GetTaskState",
			Handler:    _CoordinationService_GetTaskState_Handler,
		},
		{
			MethodName: "InsertKeyValue",
			Handler:    _CoordinationService_InsertKeyValue_Handler,
		},
		{
			MethodName: "GetKeyValue",
			Handler:    _CoordinationService_GetKeyValue_Handler,
		},
		{
			MethodName: "TryGetKeyValue",
			Handler:    _CoordinationService_TryGetKeyValue_Handler,
		},
		{
			MethodName: "GetKeyValueDir",
			Handler:    _CoordinationService_GetKeyValueDir_Handler,
		},
		{
			MethodName: "DeleteKeyValue",
			Handler:    _CoordinationService_DeleteKeyValue_Handler,
		},
		{
			MethodName: "Barrier",
			Handler:    _CoordinationService_Barrier_Handler,
		},
		{
			MethodName: "CancelBarrier",
			Handler:    _CoordinationService_CancelBarrier_Handler,
		},
	},
	Streams:  []grpc.StreamDesc{},
	Metadata: "tensorflow/tsl/protobuf/coordination_service.proto",
}

CoordinationService_ServiceDesc is the grpc.ServiceDesc for CoordinationService service. It's only intended for direct use with grpc.RegisterService, and not to be introspected or modified (even as a copy)

View Source 
var File_tensorflow_tsl_protobuf_bfc_memory_map_proto protoreflect.FileDescriptor
View Source 
var File_tensorflow_tsl_protobuf_coordination_config_proto protoreflect.FileDescriptor
View Source 
var File_tensorflow_tsl_protobuf_coordination_service_proto protoreflect.FileDescriptor
View Source 
var File_tensorflow_tsl_protobuf_distributed_runtime_payloads_proto protoreflect.FileDescriptor
View Source 
var File_tensorflow_tsl_protobuf_dnn_proto protoreflect.FileDescriptor
View Source 
var File_tensorflow_tsl_protobuf_error_codes_proto protoreflect.FileDescriptor
View Source 
var File_tensorflow_tsl_protobuf_histogram_proto protoreflect.FileDescriptor
View Source 
var File_tensorflow_tsl_protobuf_rpc_options_proto protoreflect.FileDescriptor
View Source 
var File_tensorflow_tsl_protobuf_status_proto protoreflect.FileDescriptor
View Source 
var File_tensorflow_tsl_protobuf_test_log_proto protoreflect.FileDescriptor

Functions

func RegisterCoordinationServiceServer added in v2.12.0 

func RegisterCoordinationServiceServer(s grpc.ServiceRegistrar, srv CoordinationServiceServer)

Types

type ActivationMode added in v2.12.0 

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
	// Exponential linear activation: f(x) = x < 0 ? e^x - 1 : x
	ActivationMode_kElu ActivationMode = 7
	// Leaky Rectified linear activation: f(x) = x < 0 ? alpha * x : x
	ActivationMode_kLeakyRelu ActivationMode = 8
	// Gaussian Error linear unit activation:
	//
	//	x * P(X <= x) = 0.5 * x * (1 + erf(x / sqrt(2))), where P(X) ~ N(0, 1).
	ActivationMode_kGeluExact ActivationMode = 9
)

func (ActivationMode) Descriptor added in v2.12.0 

func (ActivationMode) Descriptor() protoreflect.EnumDescriptor

func (ActivationMode) Enum added in v2.12.0 

func (x ActivationMode) Enum() *ActivationMode

func (ActivationMode) EnumDescriptor deprecated added in v2.12.0 

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

Deprecated: Use ActivationMode.Descriptor instead.

func (ActivationMode) Number added in v2.12.0 

func (x ActivationMode) Number() protoreflect.EnumNumber

func (ActivationMode) String added in v2.12.0 

func (x ActivationMode) String() string

func (ActivationMode) Type added in v2.12.0 

func (ActivationMode) Type() protoreflect.EnumType

type AlgorithmConfigProto added in v2.12.0 

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 added in v2.12.0 

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

Deprecated: Use AlgorithmConfigProto.ProtoReflect.Descriptor instead.

func (*AlgorithmConfigProto) GetAlgorithm added in v2.12.0 

func (x *AlgorithmConfigProto) GetAlgorithm() *AlgorithmProto

func (*AlgorithmConfigProto) GetAlgorithmNoScratch added in v2.12.0 

func (x *AlgorithmConfigProto) GetAlgorithmNoScratch() *AlgorithmProto

func (*AlgorithmConfigProto) GetOptionalAlgorithm added in v2.12.0 

func (m *AlgorithmConfigProto) GetOptionalAlgorithm() isAlgorithmConfigProto_OptionalAlgorithm

func (*AlgorithmConfigProto) GetOptionalAlgorithmNoScratch added in v2.12.0 

func (m *AlgorithmConfigProto) GetOptionalAlgorithmNoScratch() isAlgorithmConfigProto_OptionalAlgorithmNoScratch

func (*AlgorithmConfigProto) GetOptionalScratchSize added in v2.12.0 

func (m *AlgorithmConfigProto) GetOptionalScratchSize() isAlgorithmConfigProto_OptionalScratchSize

func (*AlgorithmConfigProto) GetScratchSize added in v2.12.0 

func (x *AlgorithmConfigProto) GetScratchSize() int64

func (*AlgorithmConfigProto) ProtoMessage added in v2.12.0 

func (*AlgorithmConfigProto) ProtoMessage()

func (*AlgorithmConfigProto) ProtoReflect added in v2.12.0 

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

func (*AlgorithmConfigProto) Reset added in v2.12.0 

func (x *AlgorithmConfigProto) Reset()

func (*AlgorithmConfigProto) String added in v2.12.0 

func (x *AlgorithmConfigProto) String() string

type AlgorithmConfigProto_Algorithm added in v2.12.0 

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

type AlgorithmConfigProto_AlgorithmNoScratch added in v2.12.0 

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

type AlgorithmConfigProto_ScratchSize added in v2.12.0 

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

type AlgorithmProto added in v2.12.0 

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 *wrapperspb.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 added in v2.12.0 

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

Deprecated: Use AlgorithmProto.ProtoReflect.Descriptor instead.

func (*AlgorithmProto) GetAlgoId added in v2.12.0 

func (x *AlgorithmProto) GetAlgoId() int64

func (*AlgorithmProto) GetIsCudnnFrontend added in v2.12.0 

func (x *AlgorithmProto) GetIsCudnnFrontend() bool

func (*AlgorithmProto) GetMathType added in v2.12.0 

func (x *AlgorithmProto) GetMathType() AlgorithmProto_MathType

func (*AlgorithmProto) GetTuningKnobs added in v2.12.0 

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

func (*AlgorithmProto) GetWorkspaceSize added in v2.12.0 

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

func (*AlgorithmProto) ProtoMessage added in v2.12.0 

func (*AlgorithmProto) ProtoMessage()

func (*AlgorithmProto) ProtoReflect added in v2.12.0 

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

func (*AlgorithmProto) Reset added in v2.12.0 

func (x *AlgorithmProto) Reset()

func (*AlgorithmProto) String added in v2.12.0 

func (x *AlgorithmProto) String() string

type AlgorithmProto_MathType added in v2.12.0 

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 added in v2.12.0 

func (AlgorithmProto_MathType) Descriptor() protoreflect.EnumDescriptor

func (AlgorithmProto_MathType) Enum added in v2.12.0 

func (x AlgorithmProto_MathType) Enum() *AlgorithmProto_MathType

func (AlgorithmProto_MathType) EnumDescriptor deprecated added in v2.12.0 

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

Deprecated: Use AlgorithmProto_MathType.Descriptor instead.

func (AlgorithmProto_MathType) Number added in v2.12.0 

func (x AlgorithmProto_MathType) Number() protoreflect.EnumNumber

func (AlgorithmProto_MathType) String added in v2.12.0 

func (x AlgorithmProto_MathType) String() string

func (AlgorithmProto_MathType) Type added in v2.12.0 

func (AlgorithmProto_MathType) Type() protoreflect.EnumType

type AvailableDeviceInfo added in v2.12.0 

type AvailableDeviceInfo struct {
	Name                string `protobuf:"bytes,1,opt,name=name,proto3" json:"name,omitempty"`                                                          // Device name.
	Type                string `protobuf:"bytes,2,opt,name=type,proto3" json:"type,omitempty"`                                                          // Device type, e.g. 'CPU' or 'GPU'.
	MemoryLimit         int64  `protobuf:"varint,3,opt,name=memory_limit,json=memoryLimit,proto3" json:"memory_limit,omitempty"`                        // Memory capacity in bytes.
	PhysicalDescription string `protobuf:"bytes,4,opt,name=physical_description,json=physicalDescription,proto3" json:"physical_description,omitempty"` // The physical description of this device.
	// contains filtered or unexported fields
}

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

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

Deprecated: Use AvailableDeviceInfo.ProtoReflect.Descriptor instead.

func (*AvailableDeviceInfo) GetMemoryLimit added in v2.12.0 

func (x *AvailableDeviceInfo) GetMemoryLimit() int64

func (*AvailableDeviceInfo) GetName added in v2.12.0 

func (x *AvailableDeviceInfo) GetName() string

func (*AvailableDeviceInfo) GetPhysicalDescription added in v2.12.0 

func (x *AvailableDeviceInfo) GetPhysicalDescription() string

func (*AvailableDeviceInfo) GetType added in v2.12.0 

func (x *AvailableDeviceInfo) GetType() string

func (*AvailableDeviceInfo) ProtoMessage added in v2.12.0 

func (*AvailableDeviceInfo) ProtoMessage()

func (*AvailableDeviceInfo) ProtoReflect added in v2.12.0 

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

func (*AvailableDeviceInfo) Reset added in v2.12.0 

func (x *AvailableDeviceInfo) Reset()

func (*AvailableDeviceInfo) String added in v2.12.0 

func (x *AvailableDeviceInfo) String() string

type BarrierRequest added in v2.12.0 

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 added in v2.12.0 

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

Deprecated: Use BarrierRequest.ProtoReflect.Descriptor instead.

func (*BarrierRequest) GetBarrierId added in v2.12.0 

func (x *BarrierRequest) GetBarrierId() string

func (*BarrierRequest) GetBarrierTimeoutInMs added in v2.12.0 

func (x *BarrierRequest) GetBarrierTimeoutInMs() int64

func (*BarrierRequest) GetSourceTask added in v2.12.0 

func (x *BarrierRequest) GetSourceTask() *CoordinatedTask

func (*BarrierRequest) GetTasks added in v2.12.0 

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

func (*BarrierRequest) ProtoMessage added in v2.12.0 

func (*BarrierRequest) ProtoMessage()

func (*BarrierRequest) ProtoReflect added in v2.12.0 

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

func (*BarrierRequest) Reset added in v2.12.0 

func (x *BarrierRequest) Reset()

func (*BarrierRequest) String added in v2.12.0 

func (x *BarrierRequest) String() string

type BarrierResponse added in v2.12.0 

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

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

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

Deprecated: Use BarrierResponse.ProtoReflect.Descriptor instead.

func (*BarrierResponse) ProtoMessage added in v2.12.0 

func (*BarrierResponse) ProtoMessage()

func (*BarrierResponse) ProtoReflect added in v2.12.0 

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

func (*BarrierResponse) Reset added in v2.12.0 

func (x *BarrierResponse) Reset()

func (*BarrierResponse) String added in v2.12.0 

func (x *BarrierResponse) String() string

type BenchmarkEntries added in v2.12.0 

type BenchmarkEntries struct {
	Entry []*BenchmarkEntry `protobuf:"bytes,1,rep,name=entry,proto3" json:"entry,omitempty"`
	// contains filtered or unexported fields
}

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

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

Deprecated: Use BenchmarkEntries.ProtoReflect.Descriptor instead.

func (*BenchmarkEntries) GetEntry added in v2.12.0 

func (x *BenchmarkEntries) GetEntry() []*BenchmarkEntry

func (*BenchmarkEntries) ProtoMessage added in v2.12.0 

func (*BenchmarkEntries) ProtoMessage()

func (*BenchmarkEntries) ProtoReflect added in v2.12.0 

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

func (*BenchmarkEntries) Reset added in v2.12.0 

func (x *BenchmarkEntries) Reset()

func (*BenchmarkEntries) String added in v2.12.0 

func (x *BenchmarkEntries) String() string

type BenchmarkEntry added in v2.12.0 

type BenchmarkEntry struct {

	// The name of the specific benchmark or test
	// (e.g. BM_AdjustContrast_gpu_B_W_H)
	Name string `protobuf:"bytes,1,opt,name=name,proto3" json:"name,omitempty"`
	// If a benchmark, how many iterations it was run for
	Iters int64 `protobuf:"varint,2,opt,name=iters,proto3" json:"iters,omitempty"`
	// Total cpu time used for all iterations (in seconds)
	CpuTime float64 `protobuf:"fixed64,3,opt,name=cpu_time,json=cpuTime,proto3" json:"cpu_time,omitempty"`
	// Total wall time used for all iterations (in seconds)
	WallTime float64 `protobuf:"fixed64,4,opt,name=wall_time,json=wallTime,proto3" json:"wall_time,omitempty"`
	// Throughput (in MB/s)
	Throughput float64 `protobuf:"fixed64,5,opt,name=throughput,proto3" json:"throughput,omitempty"`
	// Generic map from result key to value.
	Extras map[string]*EntryValue `` /* 153-byte string literal not displayed */
	// Metric name, value and expected range. This can include accuracy metrics
	// typically used to determine whether the accuracy test has passed
	Metrics []*MetricEntry `protobuf:"bytes,7,rep,name=metrics,proto3" json:"metrics,omitempty"`
	// contains filtered or unexported fields
}

Each unit test or benchmark in a test or benchmark run provides some set of information. Here we provide some reasonable keys one would expect to see, with optional key/value pairs for things we haven't considered.

This BenchmarkEntry should be emitted by each unit test or benchmark reporter.

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

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

Deprecated: Use BenchmarkEntry.ProtoReflect.Descriptor instead.

func (*BenchmarkEntry) GetCpuTime added in v2.12.0 

func (x *BenchmarkEntry) GetCpuTime() float64

func (*BenchmarkEntry) GetExtras added in v2.12.0 

func (x *BenchmarkEntry) GetExtras() map[string]*EntryValue

func (*BenchmarkEntry) GetIters added in v2.12.0 

func (x *BenchmarkEntry) GetIters() int64

func (*BenchmarkEntry) GetMetrics added in v2.12.0 

func (x *BenchmarkEntry) GetMetrics() []*MetricEntry

func (*BenchmarkEntry) GetName added in v2.12.0 

func (x *BenchmarkEntry) GetName() string

func (*BenchmarkEntry) GetThroughput added in v2.12.0 

func (x *BenchmarkEntry) GetThroughput() float64

func (*BenchmarkEntry) GetWallTime added in v2.12.0 

func (x *BenchmarkEntry) GetWallTime() float64

func (*BenchmarkEntry) ProtoMessage added in v2.12.0 

func (*BenchmarkEntry) ProtoMessage()

func (*BenchmarkEntry) ProtoReflect added in v2.12.0 

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

func (*BenchmarkEntry) Reset added in v2.12.0 

func (x *BenchmarkEntry) Reset()

func (*BenchmarkEntry) String added in v2.12.0 

func (x *BenchmarkEntry) String() string

type BinSummary added in v2.12.0 

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 added in v2.12.0 

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

Deprecated: Use BinSummary.ProtoReflect.Descriptor instead.

func (*BinSummary) GetBin added in v2.12.0 

func (x *BinSummary) GetBin() int32

func (*BinSummary) GetTotalBytesInBin added in v2.12.0 

func (x *BinSummary) GetTotalBytesInBin() int64

func (*BinSummary) GetTotalBytesInUse added in v2.12.0 

func (x *BinSummary) GetTotalBytesInUse() int64

func (*BinSummary) GetTotalChunksInBin added in v2.12.0 

func (x *BinSummary) GetTotalChunksInBin() int64

func (*BinSummary) GetTotalChunksInUse added in v2.12.0 

func (x *BinSummary) GetTotalChunksInUse() int64

func (*BinSummary) ProtoMessage added in v2.12.0 

func (*BinSummary) ProtoMessage()

func (*BinSummary) ProtoReflect added in v2.12.0 

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

func (*BinSummary) Reset added in v2.12.0 

func (x *BinSummary) Reset()

func (*BinSummary) String added in v2.12.0 

func (x *BinSummary) String() string

type BuildConfiguration added in v2.12.0 

type BuildConfiguration struct {
	Mode    string   `protobuf:"bytes,1,opt,name=mode,proto3" json:"mode,omitempty"`                      // opt, dbg, etc
	CcFlags []string `protobuf:"bytes,2,rep,name=cc_flags,json=ccFlags,proto3" json:"cc_flags,omitempty"` // CC compiler flags, if known
	Opts    []string `protobuf:"bytes,3,rep,name=opts,proto3" json:"opts,omitempty"`                      // Bazel compilation options, if known
	// contains filtered or unexported fields
}

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

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

Deprecated: Use BuildConfiguration.ProtoReflect.Descriptor instead.

func (*BuildConfiguration) GetCcFlags added in v2.12.0 

func (x *BuildConfiguration) GetCcFlags() []string

func (*BuildConfiguration) GetMode added in v2.12.0 

func (x *BuildConfiguration) GetMode() string

func (*BuildConfiguration) GetOpts added in v2.12.0 

func (x *BuildConfiguration) GetOpts() []string

func (*BuildConfiguration) ProtoMessage added in v2.12.0 

func (*BuildConfiguration) ProtoMessage()

func (*BuildConfiguration) ProtoReflect added in v2.12.0 

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

func (*BuildConfiguration) Reset added in v2.12.0 

func (x *BuildConfiguration) Reset()

func (*BuildConfiguration) String added in v2.12.0 

func (x *BuildConfiguration) String() string

type CPUInfo added in v2.12.0 

type CPUInfo struct {
	NumCores        int64 `protobuf:"varint,1,opt,name=num_cores,json=numCores,proto3" json:"num_cores,omitempty"`
	NumCoresAllowed int64 `protobuf:"varint,2,opt,name=num_cores_allowed,json=numCoresAllowed,proto3" json:"num_cores_allowed,omitempty"`
	// How fast are these cpus?
	MhzPerCpu float64 `protobuf:"fixed64,3,opt,name=mhz_per_cpu,json=mhzPerCpu,proto3" json:"mhz_per_cpu,omitempty"`
	// Additional cpu information. For example,
	// Intel Ivybridge with HyperThreading (24 cores) dL1:32KB dL2:256KB dL3:30MB
	CpuInfo string `protobuf:"bytes,4,opt,name=cpu_info,json=cpuInfo,proto3" json:"cpu_info,omitempty"`
	// What kind of cpu scaling is enabled on the host.
	// Examples include "performance", "ondemand", "conservative", "mixed".
	CpuGovernor string `protobuf:"bytes,5,opt,name=cpu_governor,json=cpuGovernor,proto3" json:"cpu_governor,omitempty"`
	// Cache sizes (in bytes), e.g. "L2": 262144 (for 256KB)
	CacheSize map[string]int64 `` /* 177-byte string literal not displayed */
	// contains filtered or unexported fields
}

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

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

Deprecated: Use CPUInfo.ProtoReflect.Descriptor instead.

func (*CPUInfo) GetCacheSize added in v2.12.0 

func (x *CPUInfo) GetCacheSize() map[string]int64

func (*CPUInfo) GetCpuGovernor added in v2.12.0 

func (x *CPUInfo) GetCpuGovernor() string

func (*CPUInfo) GetCpuInfo added in v2.12.0 

func (x *CPUInfo) GetCpuInfo() string

func (*CPUInfo) GetMhzPerCpu added in v2.12.0 

func (x *CPUInfo) GetMhzPerCpu() float64

func (*CPUInfo) GetNumCores added in v2.12.0 

func (x *CPUInfo) GetNumCores() int64

func (*CPUInfo) GetNumCoresAllowed added in v2.12.0 

func (x *CPUInfo) GetNumCoresAllowed() int64

func (*CPUInfo) ProtoMessage added in v2.12.0 

func (*CPUInfo) ProtoMessage()

func (*CPUInfo) ProtoReflect added in v2.12.0 

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

func (*CPUInfo) Reset added in v2.12.0 

func (x *CPUInfo) Reset()

func (*CPUInfo) String added in v2.12.0 

func (x *CPUInfo) String() string

type CancelBarrierRequest added in v2.12.0 

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 added in v2.12.0 

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

Deprecated: Use CancelBarrierRequest.ProtoReflect.Descriptor instead.

func (*CancelBarrierRequest) GetBarrierId added in v2.12.0 

func (x *CancelBarrierRequest) GetBarrierId() string

func (*CancelBarrierRequest) GetSourceTask added in v2.12.0 

func (x *CancelBarrierRequest) GetSourceTask() *CoordinatedTask

func (*CancelBarrierRequest) ProtoMessage added in v2.12.0 

func (*CancelBarrierRequest) ProtoMessage()

func (*CancelBarrierRequest) ProtoReflect added in v2.12.0 

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

func (*CancelBarrierRequest) Reset added in v2.12.0 

func (x *CancelBarrierRequest) Reset()

func (*CancelBarrierRequest) String added in v2.12.0 

func (x *CancelBarrierRequest) String() string

type CancelBarrierResponse added in v2.12.0 

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

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

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

Deprecated: Use CancelBarrierResponse.ProtoReflect.Descriptor instead.

func (*CancelBarrierResponse) ProtoMessage added in v2.12.0 

func (*CancelBarrierResponse) ProtoMessage()

func (*CancelBarrierResponse) ProtoReflect added in v2.12.0 

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

func (*CancelBarrierResponse) Reset added in v2.12.0 

func (x *CancelBarrierResponse) Reset()

func (*CancelBarrierResponse) String added in v2.12.0 

func (x *CancelBarrierResponse) 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 CommitId added in v2.12.0 

type CommitId struct {

	// Types that are assignable to Kind:
	//
	//	*CommitId_Changelist
	//	*CommitId_Hash
	Kind isCommitId_Kind `protobuf_oneof:"kind"`
	// Hash of intermediate change between hash/changelist and what was tested.
	// Not used if the build is from a commit without modifications.
	Snapshot string `protobuf:"bytes,3,opt,name=snapshot,proto3" json:"snapshot,omitempty"`
	// Changelist tested if the change list is not already submitted.
	PendingChangelist int64 `protobuf:"varint,4,opt,name=pending_changelist,json=pendingChangelist,proto3" json:"pending_changelist,omitempty"`
	// contains filtered or unexported fields
}

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

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

Deprecated: Use CommitId.ProtoReflect.Descriptor instead.

func (*CommitId) GetChangelist added in v2.12.0 

func (x *CommitId) GetChangelist() int64

func (*CommitId) GetHash added in v2.12.0 

func (x *CommitId) GetHash() string

func (*CommitId) GetKind added in v2.12.0 

func (m *CommitId) GetKind() isCommitId_Kind

func (*CommitId) GetPendingChangelist added in v2.12.0 

func (x *CommitId) GetPendingChangelist() int64

func (*CommitId) GetSnapshot added in v2.12.0 

func (x *CommitId) GetSnapshot() string

func (*CommitId) ProtoMessage added in v2.12.0 

func (*CommitId) ProtoMessage()

func (*CommitId) ProtoReflect added in v2.12.0 

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

func (*CommitId) Reset added in v2.12.0 

func (x *CommitId) Reset()

func (*CommitId) String added in v2.12.0 

func (x *CommitId) String() string

type CommitId_Changelist added in v2.12.0 

type CommitId_Changelist struct {
	// Submitted changelist.
	Changelist int64 `protobuf:"varint,1,opt,name=changelist,proto3,oneof"`
}

type CommitId_Hash added in v2.12.0 

type CommitId_Hash struct {
	Hash string `protobuf:"bytes,2,opt,name=hash,proto3,oneof"`
}

type ConvolutionDescriptorProto added in v2.12.0 

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 DataType `` /* 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 added in v2.12.0 

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

Deprecated: Use ConvolutionDescriptorProto.ProtoReflect.Descriptor instead.

func (*ConvolutionDescriptorProto) GetComputeMode added in v2.12.0 

func (x *ConvolutionDescriptorProto) GetComputeMode() DataType

func (*ConvolutionDescriptorProto) GetConvolutionMode added in v2.12.0 

func (x *ConvolutionDescriptorProto) GetConvolutionMode() ConvolutionMode

func (*ConvolutionDescriptorProto) GetDilations added in v2.12.0 

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

func (*ConvolutionDescriptorProto) GetGroupCount added in v2.12.0 

func (x *ConvolutionDescriptorProto) GetGroupCount() int32

func (*ConvolutionDescriptorProto) GetName added in v2.12.0 

func (x *ConvolutionDescriptorProto) GetName() string

func (*ConvolutionDescriptorProto) GetPaddings added in v2.12.0 

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

func (*ConvolutionDescriptorProto) GetStrides added in v2.12.0 

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

func (*ConvolutionDescriptorProto) ProtoMessage added in v2.12.0 

func (*ConvolutionDescriptorProto) ProtoMessage()

func (*ConvolutionDescriptorProto) ProtoReflect added in v2.12.0 

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

func (*ConvolutionDescriptorProto) Reset added in v2.12.0 

func (x *ConvolutionDescriptorProto) Reset()

func (*ConvolutionDescriptorProto) String added in v2.12.0 

func (x *ConvolutionDescriptorProto) String() string

type ConvolutionKind added in v2.12.0 

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
	ConvolutionKind_FORWARD_GRAPH           ConvolutionKind = 5
)

func (ConvolutionKind) Descriptor added in v2.12.0 

func (ConvolutionKind) Descriptor() protoreflect.EnumDescriptor

func (ConvolutionKind) Enum added in v2.12.0 

func (x ConvolutionKind) Enum() *ConvolutionKind

func (ConvolutionKind) EnumDescriptor deprecated added in v2.12.0 

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

Deprecated: Use ConvolutionKind.Descriptor instead.

func (ConvolutionKind) Number added in v2.12.0 

func (x ConvolutionKind) Number() protoreflect.EnumNumber

func (ConvolutionKind) String added in v2.12.0 

func (x ConvolutionKind) String() string

func (ConvolutionKind) Type added in v2.12.0 

func (ConvolutionKind) Type() protoreflect.EnumType

type ConvolutionMode added in v2.12.0 

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 added in v2.12.0 

func (ConvolutionMode) Descriptor() protoreflect.EnumDescriptor

func (ConvolutionMode) Enum added in v2.12.0 

func (x ConvolutionMode) Enum() *ConvolutionMode

func (ConvolutionMode) EnumDescriptor deprecated added in v2.12.0 

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

Deprecated: Use ConvolutionMode.Descriptor instead.

func (ConvolutionMode) Number added in v2.12.0 

func (x ConvolutionMode) Number() protoreflect.EnumNumber

func (ConvolutionMode) String added in v2.12.0 

func (x ConvolutionMode) String() string

func (ConvolutionMode) Type added in v2.12.0 

func (ConvolutionMode) Type() protoreflect.EnumType

type CoordinatedJob added in v2.12.0 

type CoordinatedJob struct {
	Name     string `protobuf:"bytes,1,opt,name=name,proto3" json:"name,omitempty"`
	NumTasks int32  `protobuf:"varint,2,opt,name=num_tasks,json=numTasks,proto3" json:"num_tasks,omitempty"`
	// contains filtered or unexported fields
}

Represents a job type and the number of tasks under this job. For example, ("worker", 20) implies that there will be 20 worker tasks.

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

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

Deprecated: Use CoordinatedJob.ProtoReflect.Descriptor instead.

func (*CoordinatedJob) GetName added in v2.12.0 

func (x *CoordinatedJob) GetName() string

func (*CoordinatedJob) GetNumTasks added in v2.12.0 

func (x *CoordinatedJob) GetNumTasks() int32

func (*CoordinatedJob) ProtoMessage added in v2.12.0 

func (*CoordinatedJob) ProtoMessage()

func (*CoordinatedJob) ProtoReflect added in v2.12.0 

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

func (*CoordinatedJob) Reset added in v2.12.0 

func (x *CoordinatedJob) Reset()

func (*CoordinatedJob) String added in v2.12.0 

func (x *CoordinatedJob) String() string

type CoordinatedTask added in v2.12.0 

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 added in v2.12.0 

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

Deprecated: Use CoordinatedTask.ProtoReflect.Descriptor instead.

func (*CoordinatedTask) GetJobName added in v2.12.0 

func (x *CoordinatedTask) GetJobName() string

func (*CoordinatedTask) GetTaskId added in v2.12.0 

func (x *CoordinatedTask) GetTaskId() int32

func (*CoordinatedTask) ProtoMessage added in v2.12.0 

func (*CoordinatedTask) ProtoMessage()

func (*CoordinatedTask) ProtoReflect added in v2.12.0 

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

func (*CoordinatedTask) Reset added in v2.12.0 

func (x *CoordinatedTask) Reset()

func (*CoordinatedTask) String added in v2.12.0 

func (x *CoordinatedTask) String() string

type CoordinatedTaskState added in v2.12.0 

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 added in v2.12.0 

func (CoordinatedTaskState) Descriptor() protoreflect.EnumDescriptor

func (CoordinatedTaskState) Enum added in v2.12.0 

func (x CoordinatedTaskState) Enum() *CoordinatedTaskState

func (CoordinatedTaskState) EnumDescriptor deprecated added in v2.12.0 

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

Deprecated: Use CoordinatedTaskState.Descriptor instead.

func (CoordinatedTaskState) Number added in v2.12.0 

func (x CoordinatedTaskState) Number() protoreflect.EnumNumber

func (CoordinatedTaskState) String added in v2.12.0 

func (x CoordinatedTaskState) String() string

func (CoordinatedTaskState) Type added in v2.12.0 

func (CoordinatedTaskState) Type() protoreflect.EnumType

type CoordinatedTaskStateInfo added in v2.12.0 

type CoordinatedTaskStateInfo struct {
	Task         *CoordinatedTask          `protobuf:"bytes,1,opt,name=task,proto3" json:"task,omitempty"`
	State        CoordinatedTaskState      `protobuf:"varint,2,opt,name=state,proto3,enum=tensorflow.CoordinatedTaskState" json:"state,omitempty"`
	ErrorCode    int32                     `protobuf:"varint,3,opt,name=error_code,json=errorCode,proto3" json:"error_code,omitempty"`
	ErrorMessage string                    `protobuf:"bytes,4,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
}

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

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

Deprecated: Use CoordinatedTaskStateInfo.ProtoReflect.Descriptor instead.

func (*CoordinatedTaskStateInfo) GetErrorCode added in v2.12.0 

func (x *CoordinatedTaskStateInfo) GetErrorCode() int32

func (*CoordinatedTaskStateInfo) GetErrorMessage added in v2.12.0 

func (x *CoordinatedTaskStateInfo) GetErrorMessage() string

func (*CoordinatedTaskStateInfo) GetErrorPayload added in v2.12.0 

func (x *CoordinatedTaskStateInfo) GetErrorPayload() *CoordinationServiceError

func (*CoordinatedTaskStateInfo) GetState added in v2.12.0 

func (x *CoordinatedTaskStateInfo) GetState() CoordinatedTaskState

func (*CoordinatedTaskStateInfo) GetTask added in v2.12.0 

func (x *CoordinatedTaskStateInfo) GetTask() *CoordinatedTask

func (*CoordinatedTaskStateInfo) ProtoMessage added in v2.12.0 

func (*CoordinatedTaskStateInfo) ProtoMessage()

func (*CoordinatedTaskStateInfo) ProtoReflect added in v2.12.0 

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

func (*CoordinatedTaskStateInfo) Reset added in v2.12.0 

func (x *CoordinatedTaskStateInfo) Reset()

func (*CoordinatedTaskStateInfo) String added in v2.12.0 

func (x *CoordinatedTaskStateInfo) String() string

type CoordinationServiceClient added in v2.12.0 

type CoordinationServiceClient interface {
	// Register task to coordination service so that the service starts to track
	// liveness of the task. RPC blocks and returns only when it registers to
	// the service successfully, or error happens in the registering process.
	RegisterTask(ctx context.Context, in *RegisterTaskRequest, opts ...grpc.CallOption) (*RegisterTaskResponse, error)
	// Heartbeat message from task to coordination service. Heartbeat is sent from
	// a task to refresh its timestamp on leader to avoid it becoming stale.
	// RPC responds immediately after refreshing the timestamp on leader.
	Heartbeat(ctx context.Context, in *HeartbeatRequest, opts ...grpc.CallOption) (*HeartbeatResponse, error)
	// Wait for all tasks in the cluster to be up and running. The RPC request
	// only gets responded when all tasks have registered, or some error occurs.
	WaitForAllTasks(ctx context.Context, in *WaitForAllTasksRequest, opts ...grpc.CallOption) (*WaitForAllTasksResponse, error)
	// Disconnects task from the service. If `shutdown_barrier_timeout_in_ms` is
	// specified in the config, blocks until all tasks reach the barrier before
	// disconnecting together. If the barrier times out, tasks at the barrier will
	// still disconnect, while an error is reported to tasks that did not reach
	// the barrier on time.
	ShutdownTask(ctx context.Context, in *ShutdownTaskRequest, opts ...grpc.CallOption) (*ShutdownTaskResponse, error)
	// Disconnects task from the service if it is in an ERROR state, thereby
	// allowing it to reconnect via RegisterTask() in the future.
	ResetTask(ctx context.Context, in *ResetTaskRequest, opts ...grpc.CallOption) (*ResetTaskResponse, error)
	// Report error to the task. RPC sets the receiving instance of coordination
	// service agent to error state permanently.
	// TODO(b/195990880): Consider splitting this into a different RPC service.
	ReportErrorToTask(ctx context.Context, in *ReportErrorToTaskRequest, opts ...grpc.CallOption) (*ReportErrorToTaskResponse, error)
	// Report task error to coordination service. RPC sets the service-side task
	// state to error, and propagate the error to other tasks in the cluster.
	ReportErrorToService(ctx context.Context, in *ReportErrorToServiceRequest, opts ...grpc.CallOption) (*ReportErrorToServiceResponse, error)
	// Get the state of a remote task. Specifically, RPC returns a
	// CoordinatedTaskState, and if the task is in an error status, returns a
	// non-OK error code, non-empty error message and error payload.
	GetTaskState(ctx context.Context, in *GetTaskStateRequest, opts ...grpc.CallOption) (*GetTaskStateResponse, error)
	// Insert configuration key-value that will be accessible to all cluster
	// tasks. The key can be formatted as Unix file path with hierarchy. The
	// coordination service key-value store should only be used for cluster
	// configuration data.
	InsertKeyValue(ctx context.Context, in *InsertKeyValueRequest, opts ...grpc.CallOption) (*InsertKeyValueResponse, error)
	// Get configuration key-value. The request blocks until the key-value data
	// becomes available (i.e., set by a task in the cluster).
	GetKeyValue(ctx context.Context, in *GetKeyValueRequest, opts ...grpc.CallOption) (*GetKeyValueResponse, error)
	// Get configuration key-value. The request does not block, but returns an
	// error if the requested key does not exist.
	TryGetKeyValue(ctx context.Context, in *TryGetKeyValueRequest, opts ...grpc.CallOption) (*TryGetKeyValueResponse, error)
	// Same as GetKeyValue, but returns all values that have keys which are
	// prefixed with the directory key.
	GetKeyValueDir(ctx context.Context, in *GetKeyValueDirRequest, opts ...grpc.CallOption) (*GetKeyValueDirResponse, error)
	// Delete configuration key-value. If is_directory is set in request,
	// recursively clean up all key-values under the path specified by `key`.
	DeleteKeyValue(ctx context.Context, in *DeleteKeyValueRequest, opts ...grpc.CallOption) (*DeleteKeyValueResponse, error)
	// Blocks until all (or a subset of) tasks are at the barrier or the barrier
	// fails.
	//
	// `barrier_id` should be unique across barriers. Once the barrier has passed
	// or failed, subsequent calls will not block, and immediately respond with
	// the previous response.
	//
	// The first WaitAtBarrier() call received by the service for a particular
	// barrier id is special in that it determines the barrier deadline based on
	// timeout duration.
	// However, if subsequent calls by different agents specify a different set of
	// `tasks` for the same `barrier_id`, the barrier will fail instantly.
	//
	// If no tasks are specified (default), the barrier will block for all the
	// connected tasks.
	//
	// Possible service errors:
	//   - DeadlineExceeded: Timed out waiting for specified tasks at the barrier.
	//     Deadline is determined by the server timestamp when it receives the
	//     first WaitAtBarrier() + timeout duration.
	//   - Cancelled: One of the tasks called CancelBarrier().
	//   - Aborted: Service is shutting down.
	//   - Internal: Any participating task is in ERROR state.
	//   - InvalidArgument: (1) Conflicting tasks specified by different agents
	//     for the same barrier, (2) one of the participating tasks is not in
	//     the cluster, or (3) task making the request is not included in the
	//     list of participating tasks.
	Barrier(ctx context.Context, in *BarrierRequest, opts ...grpc.CallOption) (*BarrierResponse, error)
	// Aborts the barrier if it is ongoing.
	// Current and future WaitAtBarrier() calls with the same id will return a
	// CANCELLED error status.
	// Possible service errors:
	//   - FailedPrecondition: Barrier has already been passed.
	CancelBarrier(ctx context.Context, in *CancelBarrierRequest, opts ...grpc.CallOption) (*CancelBarrierResponse, error)
}

CoordinationServiceClient is the client API for CoordinationService service.

For semantics around ctx use and closing/ending streaming RPCs, please refer to https://pkg.go.dev/google.golang.org/grpc/?tab=doc#ClientConn.NewStream.

func NewCoordinationServiceClient added in v2.12.0 

func NewCoordinationServiceClient(cc grpc.ClientConnInterface) CoordinationServiceClient

type CoordinationServiceConfig added in v2.12.0 

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 */
	CoordinatedJobList   []*CoordinatedJob `protobuf:"bytes,10,rep,name=coordinated_job_list,json=coordinatedJobList,proto3" json:"coordinated_job_list,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"`
	// If a task restarts with a new incarnation, we may allow it to reconnect
	// silently. This is useful when we know that a task can immediately resume
	// work upon re-connecting to the service.
	AllowNewIncarnationToReconnect bool `` /* 159-byte string literal not displayed */
	// contains filtered or unexported fields
}

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

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

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

Deprecated: Use CoordinationServiceConfig.ProtoReflect.Descriptor instead.

func (*CoordinationServiceConfig) GetAgentDestructionWithoutShutdown added in v2.12.0 

func (x *CoordinationServiceConfig) GetAgentDestructionWithoutShutdown() bool

func (*CoordinationServiceConfig) GetAllowNewIncarnationToReconnect added in v2.13.0 

func (x *CoordinationServiceConfig) GetAllowNewIncarnationToReconnect() bool

func (*CoordinationServiceConfig) GetClusterRegisterTimeoutInMs added in v2.12.0 

func (x *CoordinationServiceConfig) GetClusterRegisterTimeoutInMs() int64

func (*CoordinationServiceConfig) GetCoordinatedJobList added in v2.12.0 

func (x *CoordinationServiceConfig) GetCoordinatedJobList() []*CoordinatedJob

func (*CoordinationServiceConfig) GetEnableHealthCheck added in v2.12.0 

func (x *CoordinationServiceConfig) GetEnableHealthCheck() bool

func (*CoordinationServiceConfig) GetHeartbeatTimeoutInMs added in v2.12.0 

func (x *CoordinationServiceConfig) GetHeartbeatTimeoutInMs() int64

func (*CoordinationServiceConfig) GetRecoverableJobs added in v2.12.0 

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

func (*CoordinationServiceConfig) GetServiceLeader added in v2.12.0 

func (x *CoordinationServiceConfig) GetServiceLeader() string

func (*CoordinationServiceConfig) GetServiceType added in v2.12.0 

func (x *CoordinationServiceConfig) GetServiceType() string

func (*CoordinationServiceConfig) GetShutdownBarrierTimeoutInMs added in v2.12.0 

func (x *CoordinationServiceConfig) GetShutdownBarrierTimeoutInMs() int64

func (*CoordinationServiceConfig) ProtoMessage added in v2.12.0 

func (*CoordinationServiceConfig) ProtoMessage()

func (*CoordinationServiceConfig) ProtoReflect added in v2.12.0 

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

func (*CoordinationServiceConfig) Reset added in v2.12.0 

func (x *CoordinationServiceConfig) Reset()

func (*CoordinationServiceConfig) String added in v2.12.0 

func (x *CoordinationServiceConfig) String() string

type CoordinationServiceError added in v2.12.0 

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 added in v2.12.0 

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

Deprecated: Use CoordinationServiceError.ProtoReflect.Descriptor instead.

func (*CoordinationServiceError) GetIsReportedError added in v2.12.0 

func (x *CoordinationServiceError) GetIsReportedError() bool

func (*CoordinationServiceError) GetSourceTask added in v2.12.0 

func (x *CoordinationServiceError) GetSourceTask() *CoordinatedTask

func (*CoordinationServiceError) ProtoMessage added in v2.12.0 

func (*CoordinationServiceError) ProtoMessage()

func (*CoordinationServiceError) ProtoReflect added in v2.12.0 

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

func (*CoordinationServiceError) Reset added in v2.12.0 

func (x *CoordinationServiceError) Reset()

func (*CoordinationServiceError) String added in v2.12.0 

func (x *CoordinationServiceError) String() string

type CoordinationServiceServer added in v2.12.0 

type CoordinationServiceServer interface {
	// Register task to coordination service so that the service starts to track
	// liveness of the task. RPC blocks and returns only when it registers to
	// the service successfully, or error happens in the registering process.
	RegisterTask(context.Context, *RegisterTaskRequest) (*RegisterTaskResponse, error)
	// Heartbeat message from task to coordination service. Heartbeat is sent from
	// a task to refresh its timestamp on leader to avoid it becoming stale.
	// RPC responds immediately after refreshing the timestamp on leader.
	Heartbeat(context.Context, *HeartbeatRequest) (*HeartbeatResponse, error)
	// Wait for all tasks in the cluster to be up and running. The RPC request
	// only gets responded when all tasks have registered, or some error occurs.
	WaitForAllTasks(context.Context, *WaitForAllTasksRequest) (*WaitForAllTasksResponse, error)
	// Disconnects task from the service. If `shutdown_barrier_timeout_in_ms` is
	// specified in the config, blocks until all tasks reach the barrier before
	// disconnecting together. If the barrier times out, tasks at the barrier will
	// still disconnect, while an error is reported to tasks that did not reach
	// the barrier on time.
	ShutdownTask(context.Context, *ShutdownTaskRequest) (*ShutdownTaskResponse, error)
	// Disconnects task from the service if it is in an ERROR state, thereby
	// allowing it to reconnect via RegisterTask() in the future.
	ResetTask(context.Context, *ResetTaskRequest) (*ResetTaskResponse, error)
	// Report error to the task. RPC sets the receiving instance of coordination
	// service agent to error state permanently.
	// TODO(b/195990880): Consider splitting this into a different RPC service.
	ReportErrorToTask(context.Context, *ReportErrorToTaskRequest) (*ReportErrorToTaskResponse, error)
	// Report task error to coordination service. RPC sets the service-side task
	// state to error, and propagate the error to other tasks in the cluster.
	ReportErrorToService(context.Context, *ReportErrorToServiceRequest) (*ReportErrorToServiceResponse, error)
	// Get the state of a remote task. Specifically, RPC returns a
	// CoordinatedTaskState, and if the task is in an error status, returns a
	// non-OK error code, non-empty error message and error payload.
	GetTaskState(context.Context, *GetTaskStateRequest) (*GetTaskStateResponse, error)
	// Insert configuration key-value that will be accessible to all cluster
	// tasks. The key can be formatted as Unix file path with hierarchy. The
	// coordination service key-value store should only be used for cluster
	// configuration data.
	InsertKeyValue(context.Context, *InsertKeyValueRequest) (*InsertKeyValueResponse, error)
	// Get configuration key-value. The request blocks until the key-value data
	// becomes available (i.e., set by a task in the cluster).
	GetKeyValue(context.Context, *GetKeyValueRequest) (*GetKeyValueResponse, error)
	// Get configuration key-value. The request does not block, but returns an
	// error if the requested key does not exist.
	TryGetKeyValue(context.Context, *TryGetKeyValueRequest) (*TryGetKeyValueResponse, error)
	// Same as GetKeyValue, but returns all values that have keys which are
	// prefixed with the directory key.
	GetKeyValueDir(context.Context, *GetKeyValueDirRequest) (*GetKeyValueDirResponse, error)
	// Delete configuration key-value. If is_directory is set in request,
	// recursively clean up all key-values under the path specified by `key`.
	DeleteKeyValue(context.Context, *DeleteKeyValueRequest) (*DeleteKeyValueResponse, error)
	// Blocks until all (or a subset of) tasks are at the barrier or the barrier
	// fails.
	//
	// `barrier_id` should be unique across barriers. Once the barrier has passed
	// or failed, subsequent calls will not block, and immediately respond with
	// the previous response.
	//
	// The first WaitAtBarrier() call received by the service for a particular
	// barrier id is special in that it determines the barrier deadline based on
	// timeout duration.
	// However, if subsequent calls by different agents specify a different set of
	// `tasks` for the same `barrier_id`, the barrier will fail instantly.
	//
	// If no tasks are specified (default), the barrier will block for all the
	// connected tasks.
	//
	// Possible service errors:
	//   - DeadlineExceeded: Timed out waiting for specified tasks at the barrier.
	//     Deadline is determined by the server timestamp when it receives the
	//     first WaitAtBarrier() + timeout duration.
	//   - Cancelled: One of the tasks called CancelBarrier().
	//   - Aborted: Service is shutting down.
	//   - Internal: Any participating task is in ERROR state.
	//   - InvalidArgument: (1) Conflicting tasks specified by different agents
	//     for the same barrier, (2) one of the participating tasks is not in
	//     the cluster, or (3) task making the request is not included in the
	//     list of participating tasks.
	Barrier(context.Context, *BarrierRequest) (*BarrierResponse, error)
	// Aborts the barrier if it is ongoing.
	// Current and future WaitAtBarrier() calls with the same id will return a
	// CANCELLED error status.
	// Possible service errors:
	//   - FailedPrecondition: Barrier has already been passed.
	CancelBarrier(context.Context, *CancelBarrierRequest) (*CancelBarrierResponse, error)
	// contains filtered or unexported methods
}

CoordinationServiceServer is the server API for CoordinationService service. All implementations must embed UnimplementedCoordinationServiceServer for forward compatibility

type DataLayout added in v2.12.0 

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 added in v2.12.0 

func (DataLayout) Descriptor() protoreflect.EnumDescriptor

func (DataLayout) Enum added in v2.12.0 

func (x DataLayout) Enum() *DataLayout

func (DataLayout) EnumDescriptor deprecated added in v2.12.0 

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

Deprecated: Use DataLayout.Descriptor instead.

func (DataLayout) Number added in v2.12.0 

func (x DataLayout) Number() protoreflect.EnumNumber

func (DataLayout) String added in v2.12.0 

func (x DataLayout) String() string

func (DataLayout) Type added in v2.12.0 

func (DataLayout) Type() protoreflect.EnumType

type DataType added in v2.12.0 

type DataType int32

Specifies the data type used by an operation. 

const (
	DataType_kFloat         DataType = 0
	DataType_kDouble        DataType = 1
	DataType_kHalf          DataType = 2
	DataType_kInt8          DataType = 3
	DataType_kInt32         DataType = 4
	DataType_kComplexFloat  DataType = 5
	DataType_kComplexDouble DataType = 6
	DataType_kBF16          DataType = 7
	DataType_kF8E5M2        DataType = 8
	DataType_kF8E4M3FN      DataType = 9
	DataType_kF8E5M2FNUZ    DataType = 10
	DataType_kF8E4M3FNUZ    DataType = 11
	DataType_kInt64         DataType = 12
)

func (DataType) Descriptor added in v2.12.0 

func (DataType) Descriptor() protoreflect.EnumDescriptor

func (DataType) Enum added in v2.12.0 

func (x DataType) Enum() *DataType

func (DataType) EnumDescriptor deprecated added in v2.12.0 

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

Deprecated: Use DataType.Descriptor instead.

func (DataType) Number added in v2.12.0 

func (x DataType) Number() protoreflect.EnumNumber

func (DataType) String added in v2.12.0 

func (x DataType) String() string

func (DataType) Type added in v2.12.0 

func (DataType) Type() protoreflect.EnumType

type DeleteKeyValueRequest added in v2.12.0 

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 added in v2.12.0 

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

Deprecated: Use DeleteKeyValueRequest.ProtoReflect.Descriptor instead.

func (*DeleteKeyValueRequest) GetIsDirectory added in v2.12.0 

func (x *DeleteKeyValueRequest) GetIsDirectory() bool

func (*DeleteKeyValueRequest) GetKey added in v2.12.0 

func (x *DeleteKeyValueRequest) GetKey() string

func (*DeleteKeyValueRequest) ProtoMessage added in v2.12.0 

func (*DeleteKeyValueRequest) ProtoMessage()

func (*DeleteKeyValueRequest) ProtoReflect added in v2.12.0 

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

func (*DeleteKeyValueRequest) Reset added in v2.12.0 

func (x *DeleteKeyValueRequest) Reset()

func (*DeleteKeyValueRequest) String added in v2.12.0 

func (x *DeleteKeyValueRequest) String() string

type DeleteKeyValueResponse added in v2.12.0 

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

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

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

Deprecated: Use DeleteKeyValueResponse.ProtoReflect.Descriptor instead.

func (*DeleteKeyValueResponse) ProtoMessage added in v2.12.0 

func (*DeleteKeyValueResponse) ProtoMessage()

func (*DeleteKeyValueResponse) ProtoReflect added in v2.12.0 

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

func (*DeleteKeyValueResponse) Reset added in v2.12.0 

func (x *DeleteKeyValueResponse) Reset()

func (*DeleteKeyValueResponse) String added in v2.12.0 

func (x *DeleteKeyValueResponse) String() string

type DeviceInfo added in v2.12.0 

type DeviceInfo struct {
	Device []*anypb.Any `protobuf:"bytes,1,rep,name=device,proto3" json:"device,omitempty"`
	// contains filtered or unexported fields
}

Placeholder message to be extended by other runtimes' device representations.

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

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

Deprecated: Use DeviceInfo.ProtoReflect.Descriptor instead.

func (*DeviceInfo) GetDevice added in v2.12.0 

func (x *DeviceInfo) GetDevice() []*anypb.Any

func (*DeviceInfo) ProtoMessage added in v2.12.0 

func (*DeviceInfo) ProtoMessage()

func (*DeviceInfo) ProtoReflect added in v2.12.0 

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

func (*DeviceInfo) Reset added in v2.12.0 

func (x *DeviceInfo) Reset()

func (*DeviceInfo) String added in v2.12.0 

func (x *DeviceInfo) String() string

type EntryValue added in v2.12.0 

type EntryValue struct {

	// Types that are assignable to Kind:
	//
	//	*EntryValue_DoubleValue
	//	*EntryValue_StringValue
	Kind isEntryValue_Kind `protobuf_oneof:"kind"`
	// contains filtered or unexported fields
}

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

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

Deprecated: Use EntryValue.ProtoReflect.Descriptor instead.

func (*EntryValue) GetDoubleValue added in v2.12.0 

func (x *EntryValue) GetDoubleValue() float64

func (*EntryValue) GetKind added in v2.12.0 

func (m *EntryValue) GetKind() isEntryValue_Kind

func (*EntryValue) GetStringValue added in v2.12.0 

func (x *EntryValue) GetStringValue() string

func (*EntryValue) ProtoMessage added in v2.12.0 

func (*EntryValue) ProtoMessage()

func (*EntryValue) ProtoReflect added in v2.12.0 

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

func (*EntryValue) Reset added in v2.12.0 

func (x *EntryValue) Reset()

func (*EntryValue) String added in v2.12.0 

func (x *EntryValue) String() string

type EntryValue_DoubleValue added in v2.12.0 

type EntryValue_DoubleValue struct {
	DoubleValue float64 `protobuf:"fixed64,1,opt,name=double_value,json=doubleValue,proto3,oneof"`
}

type EntryValue_StringValue added in v2.12.0 

type EntryValue_StringValue struct {
	StringValue string `protobuf:"bytes,2,opt,name=string_value,json=stringValue,proto3,oneof"`
}

type FilterLayout added in v2.12.0 

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
	// cuDNN-specific filter reordering (using `cudnnReorderFilterAndBias`)
	// When the filter is reordered, so is the bias (if present).
	FilterLayout_kOutputInputYX32_CudnnReordered FilterLayout = 6
	FilterLayout_kInputYXOutput                  FilterLayout = 3
	FilterLayout_kYXInputOutput                  FilterLayout = 4
)

func (FilterLayout) Descriptor added in v2.12.0 

func (FilterLayout) Descriptor() protoreflect.EnumDescriptor

func (FilterLayout) Enum added in v2.12.0 

func (x FilterLayout) Enum() *FilterLayout

func (FilterLayout) EnumDescriptor deprecated added in v2.12.0 

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

Deprecated: Use FilterLayout.Descriptor instead.

func (FilterLayout) Number added in v2.12.0 

func (x FilterLayout) Number() protoreflect.EnumNumber

func (FilterLayout) String added in v2.12.0 

func (x FilterLayout) String() string

func (FilterLayout) Type added in v2.12.0 

func (FilterLayout) Type() protoreflect.EnumType

type FusedMHAKind added in v2.13.0 

type FusedMHAKind int32

FusedMHAKind kind 

const (
	FusedMHAKind_BMM1_OUTPUT_UNKNOWN    FusedMHAKind = 0
	FusedMHAKind_BMM1_OUTPUT_INPUT_TYPE FusedMHAKind = 1
	FusedMHAKind_BMM1_OUTPUT_FLOAT      FusedMHAKind = 2
)

func (FusedMHAKind) Descriptor added in v2.13.0 

func (FusedMHAKind) Descriptor() protoreflect.EnumDescriptor

func (FusedMHAKind) Enum added in v2.13.0 

func (x FusedMHAKind) Enum() *FusedMHAKind

func (FusedMHAKind) EnumDescriptor deprecated added in v2.13.0 

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

Deprecated: Use FusedMHAKind.Descriptor instead.

func (FusedMHAKind) Number added in v2.13.0 

func (x FusedMHAKind) Number() protoreflect.EnumNumber

func (FusedMHAKind) String added in v2.13.0 

func (x FusedMHAKind) String() string

func (FusedMHAKind) Type added in v2.13.0 

func (FusedMHAKind) Type() protoreflect.EnumType

type GPUInfo added in v2.12.0 

type GPUInfo struct {
	Model string `protobuf:"bytes,1,opt,name=model,proto3" json:"model,omitempty"`              // e.g. "Tesla K40c"
	Uuid  string `protobuf:"bytes,2,opt,name=uuid,proto3" json:"uuid,omitempty"`                // Final entry in output of "nvidia-smi -L"
	BusId string `protobuf:"bytes,3,opt,name=bus_id,json=busId,proto3" json:"bus_id,omitempty"` // e.g. "0000:04:00.0"
	// contains filtered or unexported fields
}

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

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

Deprecated: Use GPUInfo.ProtoReflect.Descriptor instead.

func (*GPUInfo) GetBusId added in v2.12.0 

func (x *GPUInfo) GetBusId() string

func (*GPUInfo) GetModel added in v2.12.0 

func (x *GPUInfo) GetModel() string

func (*GPUInfo) GetUuid added in v2.12.0 

func (x *GPUInfo) GetUuid() string

func (*GPUInfo) ProtoMessage added in v2.12.0 

func (*GPUInfo) ProtoMessage()

func (*GPUInfo) ProtoReflect added in v2.12.0 

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

func (*GPUInfo) Reset added in v2.12.0 

func (x *GPUInfo) Reset()

func (*GPUInfo) String added in v2.12.0 

func (x *GPUInfo) String() string

type GetKeyValueDirRequest added in v2.12.0 

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 added in v2.12.0 

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

Deprecated: Use GetKeyValueDirRequest.ProtoReflect.Descriptor instead.

func (*GetKeyValueDirRequest) GetDirectoryKey added in v2.12.0 

func (x *GetKeyValueDirRequest) GetDirectoryKey() string

func (*GetKeyValueDirRequest) ProtoMessage added in v2.12.0 

func (*GetKeyValueDirRequest) ProtoMessage()

func (*GetKeyValueDirRequest) ProtoReflect added in v2.12.0 

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

func (*GetKeyValueDirRequest) Reset added in v2.12.0 

func (x *GetKeyValueDirRequest) Reset()

func (*GetKeyValueDirRequest) String added in v2.12.0 

func (x *GetKeyValueDirRequest) String() string

type GetKeyValueDirResponse added in v2.12.0 

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 added in v2.12.0 

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

Deprecated: Use GetKeyValueDirResponse.ProtoReflect.Descriptor instead.

func (*GetKeyValueDirResponse) GetDirectoryKey added in v2.12.0 

func (x *GetKeyValueDirResponse) GetDirectoryKey() string

func (*GetKeyValueDirResponse) GetKv added in v2.12.0 

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

func (*GetKeyValueDirResponse) ProtoMessage added in v2.12.0 

func (*GetKeyValueDirResponse) ProtoMessage()

func (*GetKeyValueDirResponse) ProtoReflect added in v2.12.0 

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

func (*GetKeyValueDirResponse) Reset added in v2.12.0 

func (x *GetKeyValueDirResponse) Reset()

func (*GetKeyValueDirResponse) String added in v2.12.0 

func (x *GetKeyValueDirResponse) String() string

type GetKeyValueRequest added in v2.12.0 

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 added in v2.12.0 

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

Deprecated: Use GetKeyValueRequest.ProtoReflect.Descriptor instead.

func (*GetKeyValueRequest) GetKey added in v2.12.0 

func (x *GetKeyValueRequest) GetKey() string

func (*GetKeyValueRequest) ProtoMessage added in v2.12.0 

func (*GetKeyValueRequest) ProtoMessage()

func (*GetKeyValueRequest) ProtoReflect added in v2.12.0 

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

func (*GetKeyValueRequest) Reset added in v2.12.0 

func (x *GetKeyValueRequest) Reset()

func (*GetKeyValueRequest) String added in v2.12.0 

func (x *GetKeyValueRequest) String() string

type GetKeyValueResponse added in v2.12.0 

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

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

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

Deprecated: Use GetKeyValueResponse.ProtoReflect.Descriptor instead.

func (*GetKeyValueResponse) GetKv added in v2.12.0 

func (x *GetKeyValueResponse) GetKv() *KeyValueEntry

func (*GetKeyValueResponse) ProtoMessage added in v2.12.0 

func (*GetKeyValueResponse) ProtoMessage()

func (*GetKeyValueResponse) ProtoReflect added in v2.12.0 

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

func (*GetKeyValueResponse) Reset added in v2.12.0 

func (x *GetKeyValueResponse) Reset()

func (*GetKeyValueResponse) String added in v2.12.0 

func (x *GetKeyValueResponse) String() string

type GetTaskStateRequest added in v2.12.0 

type GetTaskStateRequest struct {
	SourceTask []*CoordinatedTask `protobuf:"bytes,1,rep,name=source_task,json=sourceTask,proto3" json:"source_task,omitempty"`
	// contains filtered or unexported fields
}

Request and response messages for getting state of a remote task.

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

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

Deprecated: Use GetTaskStateRequest.ProtoReflect.Descriptor instead.

func (*GetTaskStateRequest) GetSourceTask added in v2.12.0 

func (x *GetTaskStateRequest) GetSourceTask() []*CoordinatedTask

func (*GetTaskStateRequest) ProtoMessage added in v2.12.0 

func (*GetTaskStateRequest) ProtoMessage()

func (*GetTaskStateRequest) ProtoReflect added in v2.12.0 

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

func (*GetTaskStateRequest) Reset added in v2.12.0 

func (x *GetTaskStateRequest) Reset()

func (*GetTaskStateRequest) String added in v2.12.0 

func (x *GetTaskStateRequest) String() string

type GetTaskStateResponse added in v2.12.0 

type GetTaskStateResponse struct {
	TaskState []*CoordinatedTaskStateInfo `protobuf:"bytes,1,rep,name=task_state,json=taskState,proto3" json:"task_state,omitempty"`
	// contains filtered or unexported fields
}

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

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

Deprecated: Use GetTaskStateResponse.ProtoReflect.Descriptor instead.

func (*GetTaskStateResponse) GetTaskState added in v2.12.0 

func (x *GetTaskStateResponse) GetTaskState() []*CoordinatedTaskStateInfo

func (*GetTaskStateResponse) ProtoMessage added in v2.12.0 

func (*GetTaskStateResponse) ProtoMessage()

func (*GetTaskStateResponse) ProtoReflect added in v2.12.0 

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

func (*GetTaskStateResponse) Reset added in v2.12.0 

func (x *GetTaskStateResponse) Reset()

func (*GetTaskStateResponse) String added in v2.12.0 

func (x *GetTaskStateResponse) String() string

type GrpcPayloadContainer added in v2.12.0 

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 added in v2.12.0 

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

Deprecated: Use GrpcPayloadContainer.ProtoReflect.Descriptor instead.

func (*GrpcPayloadContainer) GetPayloads added in v2.12.0 

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

func (*GrpcPayloadContainer) ProtoMessage added in v2.12.0 

func (*GrpcPayloadContainer) ProtoMessage()

func (*GrpcPayloadContainer) ProtoReflect added in v2.12.0 

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

func (*GrpcPayloadContainer) Reset added in v2.12.0 

func (x *GrpcPayloadContainer) Reset()

func (*GrpcPayloadContainer) String added in v2.12.0 

func (x *GrpcPayloadContainer) String() string

type GrpcPayloadsLost added in v2.12.0 

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 added in v2.12.0 

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

Deprecated: Use GrpcPayloadsLost.ProtoReflect.Descriptor instead.

func (*GrpcPayloadsLost) ProtoMessage added in v2.12.0 

func (*GrpcPayloadsLost) ProtoMessage()

func (*GrpcPayloadsLost) ProtoReflect added in v2.12.0 

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

func (*GrpcPayloadsLost) Reset added in v2.12.0 

func (x *GrpcPayloadsLost) Reset()

func (*GrpcPayloadsLost) String added in v2.12.0 

func (x *GrpcPayloadsLost) String() string

type HeartbeatRequest added in v2.12.0 

type HeartbeatRequest struct {
	Incarnation uint64           `protobuf:"fixed64,3,opt,name=incarnation,proto3" json:"incarnation,omitempty"`
	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 sending heartbeats.

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

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

Deprecated: Use HeartbeatRequest.ProtoReflect.Descriptor instead.

func (*HeartbeatRequest) GetIncarnation added in v2.12.0 

func (x *HeartbeatRequest) GetIncarnation() uint64

func (*HeartbeatRequest) GetSourceTask added in v2.12.0 

func (x *HeartbeatRequest) GetSourceTask() *CoordinatedTask

func (*HeartbeatRequest) ProtoMessage added in v2.12.0 

func (*HeartbeatRequest) ProtoMessage()

func (*HeartbeatRequest) ProtoReflect added in v2.12.0 

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

func (*HeartbeatRequest) Reset added in v2.12.0 

func (x *HeartbeatRequest) Reset()

func (*HeartbeatRequest) String added in v2.12.0 

func (x *HeartbeatRequest) String() string

type HeartbeatResponse added in v2.12.0 

type HeartbeatResponse struct {
	LeaderIncarnation uint64 `protobuf:"fixed64,1,opt,name=leader_incarnation,json=leaderIncarnation,proto3" json:"leader_incarnation,omitempty"`
	// contains filtered or unexported fields
}

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

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

Deprecated: Use HeartbeatResponse.ProtoReflect.Descriptor instead.

func (*HeartbeatResponse) GetLeaderIncarnation added in v2.12.0 

func (x *HeartbeatResponse) GetLeaderIncarnation() uint64

func (*HeartbeatResponse) ProtoMessage added in v2.12.0 

func (*HeartbeatResponse) ProtoMessage()

func (*HeartbeatResponse) ProtoReflect added in v2.12.0 

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

func (*HeartbeatResponse) Reset added in v2.12.0 

func (x *HeartbeatResponse) Reset()

func (*HeartbeatResponse) String added in v2.12.0 

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 tsl/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 added in v2.12.0 

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 added in v2.12.0 

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

Deprecated: Use InsertKeyValueRequest.ProtoReflect.Descriptor instead.

func (*InsertKeyValueRequest) GetKv added in v2.12.0 

func (x *InsertKeyValueRequest) GetKv() *KeyValueEntry

func (*InsertKeyValueRequest) ProtoMessage added in v2.12.0 

func (*InsertKeyValueRequest) ProtoMessage()

func (*InsertKeyValueRequest) ProtoReflect added in v2.12.0 

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

func (*InsertKeyValueRequest) Reset added in v2.12.0 

func (x *InsertKeyValueRequest) Reset()

func (*InsertKeyValueRequest) String added in v2.12.0 

func (x *InsertKeyValueRequest) String() string

type InsertKeyValueResponse added in v2.12.0 

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

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

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

Deprecated: Use InsertKeyValueResponse.ProtoReflect.Descriptor instead.

func (*InsertKeyValueResponse) ProtoMessage added in v2.12.0 

func (*InsertKeyValueResponse) ProtoMessage()

func (*InsertKeyValueResponse) ProtoReflect added in v2.12.0 

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

func (*InsertKeyValueResponse) Reset added in v2.12.0 

func (x *InsertKeyValueResponse) Reset()

func (*InsertKeyValueResponse) String added in v2.12.0 

func (x *InsertKeyValueResponse) String() string

type KeyValueEntry added in v2.12.0 

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 added in v2.12.0 

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

Deprecated: Use KeyValueEntry.ProtoReflect.Descriptor instead.

func (*KeyValueEntry) GetKey added in v2.12.0 

func (x *KeyValueEntry) GetKey() string

func (*KeyValueEntry) GetValue added in v2.12.0 

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

func (*KeyValueEntry) ProtoMessage added in v2.12.0 

func (*KeyValueEntry) ProtoMessage()

func (*KeyValueEntry) ProtoReflect added in v2.12.0 

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

func (*KeyValueEntry) Reset added in v2.12.0 

func (x *KeyValueEntry) Reset()

func (*KeyValueEntry) String added in v2.12.0 

func (x *KeyValueEntry) String() string

type MachineConfiguration added in v2.12.0 

type MachineConfiguration struct {

	// Host name of machine that ran the benchmark.
	Hostname string `protobuf:"bytes,1,opt,name=hostname,proto3" json:"hostname,omitempty"`
	// Unique serial number of the machine.
	SerialIdentifier string `protobuf:"bytes,7,opt,name=serial_identifier,json=serialIdentifier,proto3" json:"serial_identifier,omitempty"`
	// Additional platform information.
	PlatformInfo *PlatformInfo `protobuf:"bytes,2,opt,name=platform_info,json=platformInfo,proto3" json:"platform_info,omitempty"`
	// CPU Information.
	CpuInfo *CPUInfo `protobuf:"bytes,3,opt,name=cpu_info,json=cpuInfo,proto3" json:"cpu_info,omitempty"`
	// Other devices that are attached and relevant (e.g. GPUInfo).
	DeviceInfo []*anypb.Any `protobuf:"bytes,4,rep,name=device_info,json=deviceInfo,proto3" json:"device_info,omitempty"`
	// Devices accessible to the test (e.g. as given by list_local_devices).
	AvailableDeviceInfo []*AvailableDeviceInfo `protobuf:"bytes,5,rep,name=available_device_info,json=availableDeviceInfo,proto3" json:"available_device_info,omitempty"`
	MemoryInfo          *MemoryInfo            `protobuf:"bytes,6,opt,name=memory_info,json=memoryInfo,proto3" json:"memory_info,omitempty"`
	// contains filtered or unexported fields
}

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

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

Deprecated: Use MachineConfiguration.ProtoReflect.Descriptor instead.

func (*MachineConfiguration) GetAvailableDeviceInfo added in v2.12.0 

func (x *MachineConfiguration) GetAvailableDeviceInfo() []*AvailableDeviceInfo

func (*MachineConfiguration) GetCpuInfo added in v2.12.0 

func (x *MachineConfiguration) GetCpuInfo() *CPUInfo

func (*MachineConfiguration) GetDeviceInfo added in v2.12.0 

func (x *MachineConfiguration) GetDeviceInfo() []*anypb.Any

func (*MachineConfiguration) GetHostname added in v2.12.0 

func (x *MachineConfiguration) GetHostname() string

func (*MachineConfiguration) GetMemoryInfo added in v2.12.0 

func (x *MachineConfiguration) GetMemoryInfo() *MemoryInfo

func (*MachineConfiguration) GetPlatformInfo added in v2.12.0 

func (x *MachineConfiguration) GetPlatformInfo() *PlatformInfo

func (*MachineConfiguration) GetSerialIdentifier added in v2.12.0 

func (x *MachineConfiguration) GetSerialIdentifier() string

func (*MachineConfiguration) ProtoMessage added in v2.12.0 

func (*MachineConfiguration) ProtoMessage()

func (*MachineConfiguration) ProtoReflect added in v2.12.0 

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

func (*MachineConfiguration) Reset added in v2.12.0 

func (x *MachineConfiguration) Reset()

func (*MachineConfiguration) String added in v2.12.0 

func (x *MachineConfiguration) String() string

type MemAllocatorStats added in v2.12.0 

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 added in v2.12.0 

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

Deprecated: Use MemAllocatorStats.ProtoReflect.Descriptor instead.

func (*MemAllocatorStats) GetBytesInUse added in v2.12.0 

func (x *MemAllocatorStats) GetBytesInUse() int64

func (*MemAllocatorStats) GetFragmentationMetric added in v2.12.0 

func (x *MemAllocatorStats) GetFragmentationMetric() float32

func (*MemAllocatorStats) GetLargestAllocSize added in v2.12.0 

func (x *MemAllocatorStats) GetLargestAllocSize() int64

func (*MemAllocatorStats) GetNumAllocs added in v2.12.0 

func (x *MemAllocatorStats) GetNumAllocs() int64

func (*MemAllocatorStats) GetPeakBytesInUse added in v2.12.0 

func (x *MemAllocatorStats) GetPeakBytesInUse() int64

func (*MemAllocatorStats) ProtoMessage added in v2.12.0 

func (*MemAllocatorStats) ProtoMessage()

func (*MemAllocatorStats) ProtoReflect added in v2.12.0 

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

func (*MemAllocatorStats) Reset added in v2.12.0 

func (x *MemAllocatorStats) Reset()

func (*MemAllocatorStats) String added in v2.12.0 

func (x *MemAllocatorStats) String() string

type MemChunk added in v2.12.0 

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 added in v2.12.0 

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

Deprecated: Use MemChunk.ProtoReflect.Descriptor instead.

func (*MemChunk) GetActionCount added in v2.12.0 

func (x *MemChunk) GetActionCount() uint64

func (*MemChunk) GetAddress added in v2.12.0 

func (x *MemChunk) GetAddress() uint64

func (*MemChunk) GetBin added in v2.12.0 

func (x *MemChunk) GetBin() int32

func (*MemChunk) GetFreedAtCount added in v2.12.0 

func (x *MemChunk) GetFreedAtCount() uint64

func (*MemChunk) GetInUse added in v2.12.0 

func (x *MemChunk) GetInUse() bool

func (*MemChunk) GetOpName added in v2.12.0 

func (x *MemChunk) GetOpName() string

func (*MemChunk) GetRequestedSize added in v2.12.0 

func (x *MemChunk) GetRequestedSize() int64

func (*MemChunk) GetSize added in v2.12.0 

func (x *MemChunk) GetSize() int64

func (*MemChunk) GetStepId added in v2.12.0 

func (x *MemChunk) GetStepId() uint64

func (*MemChunk) ProtoMessage added in v2.12.0 

func (*MemChunk) ProtoMessage()

func (*MemChunk) ProtoReflect added in v2.12.0 

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

func (*MemChunk) Reset added in v2.12.0 

func (x *MemChunk) Reset()

func (*MemChunk) String added in v2.12.0 

func (x *MemChunk) String() string

type MemoryDump added in v2.12.0 

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 added in v2.12.0 

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

Deprecated: Use MemoryDump.ProtoReflect.Descriptor instead.

func (*MemoryDump) GetAllocatorName added in v2.12.0 

func (x *MemoryDump) GetAllocatorName() string

func (*MemoryDump) GetBinSummary added in v2.12.0 

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

func (*MemoryDump) GetChunk added in v2.12.0 

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

func (*MemoryDump) GetSnapShot added in v2.12.0 

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

func (*MemoryDump) GetStats added in v2.12.0 

func (x *MemoryDump) GetStats() *MemAllocatorStats

func (*MemoryDump) ProtoMessage added in v2.12.0 

func (*MemoryDump) ProtoMessage()

func (*MemoryDump) ProtoReflect added in v2.12.0 

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

func (*MemoryDump) Reset added in v2.12.0 

func (x *MemoryDump) Reset()

func (*MemoryDump) String added in v2.12.0 

func (x *MemoryDump) String() string

type MemoryInfo added in v2.12.0 

type MemoryInfo struct {
	Total     int64 `protobuf:"varint,1,opt,name=total,proto3" json:"total,omitempty"`         // Total virtual memory in bytes
	Available int64 `protobuf:"varint,2,opt,name=available,proto3" json:"available,omitempty"` // Immediately available memory in bytes
	// contains filtered or unexported fields
}

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

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

Deprecated: Use MemoryInfo.ProtoReflect.Descriptor instead.

func (*MemoryInfo) GetAvailable added in v2.12.0 

func (x *MemoryInfo) GetAvailable() int64

func (*MemoryInfo) GetTotal added in v2.12.0 

func (x *MemoryInfo) GetTotal() int64

func (*MemoryInfo) ProtoMessage added in v2.12.0 

func (*MemoryInfo) ProtoMessage()

func (*MemoryInfo) ProtoReflect added in v2.12.0 

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

func (*MemoryInfo) Reset added in v2.12.0 

func (x *MemoryInfo) Reset()

func (*MemoryInfo) String added in v2.12.0 

func (x *MemoryInfo) String() string

type MetricEntry added in v2.12.0 

type MetricEntry struct {

	// Metric name
	Name string `protobuf:"bytes,1,opt,name=name,proto3" json:"name,omitempty"`
	// Metric value
	Value float64 `protobuf:"fixed64,2,opt,name=value,proto3" json:"value,omitempty"`
	// The minimum acceptable value for the metric if specified
	MinValue *wrapperspb.DoubleValue `protobuf:"bytes,3,opt,name=min_value,json=minValue,proto3" json:"min_value,omitempty"`
	// The maximum acceptable value for the metric if specified
	MaxValue *wrapperspb.DoubleValue `protobuf:"bytes,4,opt,name=max_value,json=maxValue,proto3" json:"max_value,omitempty"`
	// contains filtered or unexported fields
}

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

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

Deprecated: Use MetricEntry.ProtoReflect.Descriptor instead.

func (*MetricEntry) GetMaxValue added in v2.12.0 

func (x *MetricEntry) GetMaxValue() *wrapperspb.DoubleValue

func (*MetricEntry) GetMinValue added in v2.12.0 

func (x *MetricEntry) GetMinValue() *wrapperspb.DoubleValue

func (*MetricEntry) GetName added in v2.12.0 

func (x *MetricEntry) GetName() string

func (*MetricEntry) GetValue added in v2.12.0 

func (x *MetricEntry) GetValue() float64

func (*MetricEntry) ProtoMessage added in v2.12.0 

func (*MetricEntry) ProtoMessage()

func (*MetricEntry) ProtoReflect added in v2.12.0 

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

func (*MetricEntry) Reset added in v2.12.0 

func (x *MetricEntry) Reset()

func (*MetricEntry) String added in v2.12.0 

func (x *MetricEntry) String() string

type PlatformInfo added in v2.12.0 

type PlatformInfo struct {
	Bits    string `protobuf:"bytes,1,opt,name=bits,proto3" json:"bits,omitempty"`       // e.g. '64bit'
	Linkage string `protobuf:"bytes,2,opt,name=linkage,proto3" json:"linkage,omitempty"` // e.g. 'ELF'
	Machine string `protobuf:"bytes,3,opt,name=machine,proto3" json:"machine,omitempty"` // e.g. 'i386'
	Release string `protobuf:"bytes,4,opt,name=release,proto3" json:"release,omitempty"` // e.g. '3.13.0-76-generic'
	System  string `protobuf:"bytes,5,opt,name=system,proto3" json:"system,omitempty"`   // e.g. 'Linux'
	Version string `protobuf:"bytes,6,opt,name=version,proto3" json:"version,omitempty"` // e.g. '#120-Ubuntu SMP Mon Jan 18 15:59:10 UTC 2016'
	// contains filtered or unexported fields
}

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

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

Deprecated: Use PlatformInfo.ProtoReflect.Descriptor instead.

func (*PlatformInfo) GetBits added in v2.12.0 

func (x *PlatformInfo) GetBits() string

func (*PlatformInfo) GetLinkage added in v2.12.0 

func (x *PlatformInfo) GetLinkage() string

func (*PlatformInfo) GetMachine added in v2.12.0 

func (x *PlatformInfo) GetMachine() string

func (*PlatformInfo) GetRelease added in v2.12.0 

func (x *PlatformInfo) GetRelease() string

func (*PlatformInfo) GetSystem added in v2.12.0 

func (x *PlatformInfo) GetSystem() string

func (*PlatformInfo) GetVersion added in v2.12.0 

func (x *PlatformInfo) GetVersion() string

func (*PlatformInfo) ProtoMessage added in v2.12.0 

func (*PlatformInfo) ProtoMessage()

func (*PlatformInfo) ProtoReflect added in v2.12.0 

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

func (*PlatformInfo) Reset added in v2.12.0 

func (x *PlatformInfo) Reset()

func (*PlatformInfo) String added in v2.12.0 

func (x *PlatformInfo) String() string

type RPCOptions added in v2.12.0 

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
}

RPC options for distributed runtime.

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

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

Deprecated: Use RPCOptions.ProtoReflect.Descriptor instead.

func (*RPCOptions) GetCacheRpcResponse added in v2.12.0 

func (x *RPCOptions) GetCacheRpcResponse() bool

func (*RPCOptions) GetCompressionAlgorithm added in v2.12.0 

func (x *RPCOptions) GetCompressionAlgorithm() string

func (*RPCOptions) GetCompressionLevel added in v2.12.0 

func (x *RPCOptions) GetCompressionLevel() int32

func (*RPCOptions) GetDisableSessionConnectionSharing added in v2.12.0 

func (x *RPCOptions) GetDisableSessionConnectionSharing() bool

func (*RPCOptions) GetNumChannelsPerTarget added in v2.12.0 

func (x *RPCOptions) GetNumChannelsPerTarget() int32

func (*RPCOptions) GetUseRpcForInprocessMaster added in v2.12.0 

func (x *RPCOptions) GetUseRpcForInprocessMaster() bool

func (*RPCOptions) ProtoMessage added in v2.12.0 

func (*RPCOptions) ProtoMessage()

func (*RPCOptions) ProtoReflect added in v2.12.0 

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

func (*RPCOptions) Reset added in v2.12.0 

func (x *RPCOptions) Reset()

func (*RPCOptions) String added in v2.12.0 

func (x *RPCOptions) String() string

type RegisterTaskRequest added in v2.12.0 

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 added in v2.12.0 

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

Deprecated: Use RegisterTaskRequest.ProtoReflect.Descriptor instead.

func (*RegisterTaskRequest) GetIncarnation added in v2.12.0 

func (x *RegisterTaskRequest) GetIncarnation() uint64

func (*RegisterTaskRequest) GetSourceTask added in v2.12.0 

func (x *RegisterTaskRequest) GetSourceTask() *CoordinatedTask

func (*RegisterTaskRequest) ProtoMessage added in v2.12.0 

func (*RegisterTaskRequest) ProtoMessage()

func (*RegisterTaskRequest) ProtoReflect added in v2.12.0 

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

func (*RegisterTaskRequest) Reset added in v2.12.0 

func (x *RegisterTaskRequest) Reset()

func (*RegisterTaskRequest) String added in v2.12.0 

func (x *RegisterTaskRequest) String() string

type RegisterTaskResponse added in v2.12.0 

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 added in v2.12.0 

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

Deprecated: Use RegisterTaskResponse.ProtoReflect.Descriptor instead.

func (*RegisterTaskResponse) GetLeaderIncarnation added in v2.12.0 

func (x *RegisterTaskResponse) GetLeaderIncarnation() uint64

func (*RegisterTaskResponse) ProtoMessage added in v2.12.0 

func (*RegisterTaskResponse) ProtoMessage()

func (*RegisterTaskResponse) ProtoReflect added in v2.12.0 

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

func (*RegisterTaskResponse) Reset added in v2.12.0 

func (x *RegisterTaskResponse) Reset()

func (*RegisterTaskResponse) String added in v2.12.0 

func (x *RegisterTaskResponse) String() string

type ReportErrorToServiceRequest added in v2.12.0 

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 added in v2.12.0 

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

Deprecated: Use ReportErrorToServiceRequest.ProtoReflect.Descriptor instead.

func (*ReportErrorToServiceRequest) GetErrorCode added in v2.12.0 

func (x *ReportErrorToServiceRequest) GetErrorCode() int32

func (*ReportErrorToServiceRequest) GetErrorMessage added in v2.12.0 

func (x *ReportErrorToServiceRequest) GetErrorMessage() string

func (*ReportErrorToServiceRequest) GetErrorOrigin added in v2.12.0 

func (x *ReportErrorToServiceRequest) GetErrorOrigin() *CoordinatedTask

func (*ReportErrorToServiceRequest) ProtoMessage added in v2.12.0 

func (*ReportErrorToServiceRequest) ProtoMessage()

func (*ReportErrorToServiceRequest) ProtoReflect added in v2.12.0 

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

func (*ReportErrorToServiceRequest) Reset added in v2.12.0 

func (x *ReportErrorToServiceRequest) Reset()

func (*ReportErrorToServiceRequest) String added in v2.12.0 

func (x *ReportErrorToServiceRequest) String() string

type ReportErrorToServiceResponse added in v2.12.0 

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

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

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

Deprecated: Use ReportErrorToServiceResponse.ProtoReflect.Descriptor instead.

func (*ReportErrorToServiceResponse) ProtoMessage added in v2.12.0 

func (*ReportErrorToServiceResponse) ProtoMessage()

func (*ReportErrorToServiceResponse) ProtoReflect added in v2.12.0 

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

func (*ReportErrorToServiceResponse) Reset added in v2.12.0 

func (x *ReportErrorToServiceResponse) Reset()

func (*ReportErrorToServiceResponse) String added in v2.12.0 

func (x *ReportErrorToServiceResponse) String() string

type ReportErrorToTaskRequest added in v2.12.0 

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 added in v2.12.0 

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

Deprecated: Use ReportErrorToTaskRequest.ProtoReflect.Descriptor instead.

func (*ReportErrorToTaskRequest) GetErrorCode added in v2.12.0 

func (x *ReportErrorToTaskRequest) GetErrorCode() int32

func (*ReportErrorToTaskRequest) GetErrorMessage added in v2.12.0 

func (x *ReportErrorToTaskRequest) GetErrorMessage() string

func (*ReportErrorToTaskRequest) GetErrorPayload added in v2.12.0 

func (x *ReportErrorToTaskRequest) GetErrorPayload() *CoordinationServiceError

func (*ReportErrorToTaskRequest) ProtoMessage added in v2.12.0 

func (*ReportErrorToTaskRequest) ProtoMessage()

func (*ReportErrorToTaskRequest) ProtoReflect added in v2.12.0 

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

func (*ReportErrorToTaskRequest) Reset added in v2.12.0 

func (x *ReportErrorToTaskRequest) Reset()

func (*ReportErrorToTaskRequest) String added in v2.12.0 

func (x *ReportErrorToTaskRequest) String() string

type ReportErrorToTaskResponse added in v2.12.0 

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

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

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

Deprecated: Use ReportErrorToTaskResponse.ProtoReflect.Descriptor instead.

func (*ReportErrorToTaskResponse) ProtoMessage added in v2.12.0 

func (*ReportErrorToTaskResponse) ProtoMessage()

func (*ReportErrorToTaskResponse) ProtoReflect added in v2.12.0 

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

func (*ReportErrorToTaskResponse) Reset added in v2.12.0 

func (x *ReportErrorToTaskResponse) Reset()

func (*ReportErrorToTaskResponse) String added in v2.12.0 

func (x *ReportErrorToTaskResponse) String() string

type ResetTaskRequest added in v2.12.0 

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 added in v2.12.0 

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

Deprecated: Use ResetTaskRequest.ProtoReflect.Descriptor instead.

func (*ResetTaskRequest) GetSourceTask added in v2.12.0 

func (x *ResetTaskRequest) GetSourceTask() *CoordinatedTask

func (*ResetTaskRequest) ProtoMessage added in v2.12.0 

func (*ResetTaskRequest) ProtoMessage()

func (*ResetTaskRequest) ProtoReflect added in v2.12.0 

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

func (*ResetTaskRequest) Reset added in v2.12.0 

func (x *ResetTaskRequest) Reset()

func (*ResetTaskRequest) String added in v2.12.0 

func (x *ResetTaskRequest) String() string

type ResetTaskResponse added in v2.12.0 

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

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

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

Deprecated: Use ResetTaskResponse.ProtoReflect.Descriptor instead.

func (*ResetTaskResponse) ProtoMessage added in v2.12.0 

func (*ResetTaskResponse) ProtoMessage()

func (*ResetTaskResponse) ProtoReflect added in v2.12.0 

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

func (*ResetTaskResponse) Reset added in v2.12.0 

func (x *ResetTaskResponse) Reset()

func (*ResetTaskResponse) String added in v2.12.0 

func (x *ResetTaskResponse) String() string

type RunConfiguration added in v2.12.0 

type RunConfiguration struct {
	Argument []string `protobuf:"bytes,1,rep,name=argument,proto3" json:"argument,omitempty"`
	// Environment variables used to run the test/benchmark.
	EnvVars map[string]string `` /* 170-byte string literal not displayed */
	// contains filtered or unexported fields
}

Run-specific items such as arguments to the test / benchmark.

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

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

Deprecated: Use RunConfiguration.ProtoReflect.Descriptor instead.

func (*RunConfiguration) GetArgument added in v2.12.0 

func (x *RunConfiguration) GetArgument() []string

func (*RunConfiguration) GetEnvVars added in v2.12.0 

func (x *RunConfiguration) GetEnvVars() map[string]string

func (*RunConfiguration) ProtoMessage added in v2.12.0 

func (*RunConfiguration) ProtoMessage()

func (*RunConfiguration) ProtoReflect added in v2.12.0 

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

func (*RunConfiguration) Reset added in v2.12.0 

func (x *RunConfiguration) Reset()

func (*RunConfiguration) String added in v2.12.0 

func (x *RunConfiguration) String() string

type ShutdownTaskRequest added in v2.12.0 

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 added in v2.12.0 

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

Deprecated: Use ShutdownTaskRequest.ProtoReflect.Descriptor instead.

func (*ShutdownTaskRequest) GetSourceTask added in v2.12.0 

func (x *ShutdownTaskRequest) GetSourceTask() *CoordinatedTask

func (*ShutdownTaskRequest) ProtoMessage added in v2.12.0 

func (*ShutdownTaskRequest) ProtoMessage()

func (*ShutdownTaskRequest) ProtoReflect added in v2.12.0 

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

func (*ShutdownTaskRequest) Reset added in v2.12.0 

func (x *ShutdownTaskRequest) Reset()

func (*ShutdownTaskRequest) String added in v2.12.0 

func (x *ShutdownTaskRequest) String() string

type ShutdownTaskResponse added in v2.12.0 

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

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

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

Deprecated: Use ShutdownTaskResponse.ProtoReflect.Descriptor instead.

func (*ShutdownTaskResponse) ProtoMessage added in v2.12.0 

func (*ShutdownTaskResponse) ProtoMessage()

func (*ShutdownTaskResponse) ProtoReflect added in v2.12.0 

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

func (*ShutdownTaskResponse) Reset added in v2.12.0 

func (x *ShutdownTaskResponse) Reset()

func (*ShutdownTaskResponse) String added in v2.12.0 

func (x *ShutdownTaskResponse) String() string

type SnapShot added in v2.12.0 

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 added in v2.12.0 

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

Deprecated: Use SnapShot.ProtoReflect.Descriptor instead.

func (*SnapShot) GetActionCount added in v2.12.0 

func (x *SnapShot) GetActionCount() uint64

func (*SnapShot) GetSize added in v2.12.0 

func (x *SnapShot) GetSize() int64

func (*SnapShot) ProtoMessage added in v2.12.0 

func (*SnapShot) ProtoMessage()

func (*SnapShot) ProtoReflect added in v2.12.0 

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

func (*SnapShot) Reset added in v2.12.0 

func (x *SnapShot) Reset()

func (*SnapShot) String added in v2.12.0 

func (x *SnapShot) String() string

type StatusProto added in v2.13.0 

type StatusProto struct {

	// Status code as defined in tensorflow/tsl/protobuf/error_codes.proto.
	Code Code `protobuf:"varint,1,opt,name=code,proto3,enum=tensorflow.error.Code" json:"code,omitempty"`
	// Detail error message.
	Message string `protobuf:"bytes,2,opt,name=message,proto3" json:"message,omitempty"`
	// contains filtered or unexported fields
}

Wire-format for Status. Next tag: 3

func (*StatusProto) Descriptor deprecated added in v2.13.0 

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

Deprecated: Use StatusProto.ProtoReflect.Descriptor instead.

func (*StatusProto) GetCode added in v2.13.0 

func (x *StatusProto) GetCode() Code

func (*StatusProto) GetMessage added in v2.13.0 

func (x *StatusProto) GetMessage() string

func (*StatusProto) ProtoMessage added in v2.13.0 

func (*StatusProto) ProtoMessage()

func (*StatusProto) ProtoReflect added in v2.13.0 

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

func (*StatusProto) Reset added in v2.13.0 

func (x *StatusProto) Reset()

func (*StatusProto) String added in v2.13.0 

func (x *StatusProto) String() string

type TensorDescriptorProto added in v2.12.0 

type TensorDescriptorProto struct {
	Dimensions []int64  `protobuf:"varint,1,rep,packed,name=dimensions,proto3" json:"dimensions,omitempty"`
	DataType   DataType `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 added in v2.12.0 

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

Deprecated: Use TensorDescriptorProto.ProtoReflect.Descriptor instead.

func (*TensorDescriptorProto) GetDataLayout added in v2.12.0 

func (x *TensorDescriptorProto) GetDataLayout() DataLayout

func (*TensorDescriptorProto) GetDataType added in v2.12.0 

func (x *TensorDescriptorProto) GetDataType() DataType

func (*TensorDescriptorProto) GetDimensions added in v2.12.0 

func (x *TensorDescriptorProto) GetDimensions() []int64

func (*TensorDescriptorProto) GetFilterLayout added in v2.12.0 

func (x *TensorDescriptorProto) GetFilterLayout() FilterLayout

func (*TensorDescriptorProto) GetLayoutOneof added in v2.12.0 

func (m *TensorDescriptorProto) GetLayoutOneof() isTensorDescriptorProto_LayoutOneof

func (*TensorDescriptorProto) ProtoMessage added in v2.12.0 

func (*TensorDescriptorProto) ProtoMessage()

func (*TensorDescriptorProto) ProtoReflect added in v2.12.0 

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

func (*TensorDescriptorProto) Reset added in v2.12.0 

func (x *TensorDescriptorProto) Reset()

func (*TensorDescriptorProto) String added in v2.12.0 

func (x *TensorDescriptorProto) String() string

type TensorDescriptorProto_DataLayout added in v2.12.0 

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 added in v2.12.0 

type TensorDescriptorProto_FilterLayout struct {
	FilterLayout FilterLayout `protobuf:"varint,4,opt,name=filter_layout,json=filterLayout,proto3,enum=stream_executor.dnn.FilterLayout,oneof"`
}

type TestResults added in v2.12.0 

type TestResults struct {

	// The target of the run, e.g.:
	//
	//	//tensorflow/core:kernels_adjust_contrast_op_benchmark_test
	Target string `protobuf:"bytes,1,opt,name=target,proto3" json:"target,omitempty"`
	// The list of tests or benchmarks in this run.
	Entries *BenchmarkEntries `protobuf:"bytes,2,opt,name=entries,proto3" json:"entries,omitempty"`
	// The configuration of the build (compiled opt? with cuda? any copts?)
	BuildConfiguration *BuildConfiguration `protobuf:"bytes,3,opt,name=build_configuration,json=buildConfiguration,proto3" json:"build_configuration,omitempty"`
	// The commit id (git hash or changelist)
	CommitId *CommitId `protobuf:"bytes,4,opt,name=commit_id,json=commitId,proto3" json:"commit_id,omitempty"`
	// The time the run started (in seconds of UTC time since Unix epoch)
	StartTime int64 `protobuf:"varint,5,opt,name=start_time,json=startTime,proto3" json:"start_time,omitempty"`
	// The amount of time the total run took (wall time in seconds)
	RunTime float64 `protobuf:"fixed64,6,opt,name=run_time,json=runTime,proto3" json:"run_time,omitempty"`
	// Machine-specific parameters (Platform and CPU info)
	MachineConfiguration *MachineConfiguration `protobuf:"bytes,7,opt,name=machine_configuration,json=machineConfiguration,proto3" json:"machine_configuration,omitempty"`
	// Run-specific parameters (arguments, etc)
	RunConfiguration *RunConfiguration `protobuf:"bytes,8,opt,name=run_configuration,json=runConfiguration,proto3" json:"run_configuration,omitempty"`
	// Benchmark target identifier.
	Name          string                    `protobuf:"bytes,9,opt,name=name,proto3" json:"name,omitempty"`
	BenchmarkType TestResults_BenchmarkType `` /* 144-byte string literal not displayed */
	// Used for differentiating between continuous and debug builds.
	// Must be one of:
	// * cbuild: results from continuous build.
	// * presubmit: results from oneshot requests.
	// * culprit: results from culprit finder rerun.
	RunMode string `protobuf:"bytes,11,opt,name=run_mode,json=runMode,proto3" json:"run_mode,omitempty"`
	// TensorFlow version this benchmark runs against.
	// This can be either set to full version or just the major version.
	TfVersion string `protobuf:"bytes,12,opt,name=tf_version,json=tfVersion,proto3" json:"tf_version,omitempty"`
	// contains filtered or unexported fields
}

The output of one benchmark / test run. Each run contains a list of tests or benchmarks, stored as BenchmarkEntry messages.

This message should be emitted by the reporter (which runs the test / BM in a subprocess and then reads the emitted BenchmarkEntry messages; usually from a serialized json file, finally collecting them along with additional information about the test run.

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

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

Deprecated: Use TestResults.ProtoReflect.Descriptor instead.

func (*TestResults) GetBenchmarkType added in v2.12.0 

func (x *TestResults) GetBenchmarkType() TestResults_BenchmarkType

func (*TestResults) GetBuildConfiguration added in v2.12.0 

func (x *TestResults) GetBuildConfiguration() *BuildConfiguration

func (*TestResults) GetCommitId added in v2.12.0 

func (x *TestResults) GetCommitId() *CommitId

func (*TestResults) GetEntries added in v2.12.0 

func (x *TestResults) GetEntries() *BenchmarkEntries

func (*TestResults) GetMachineConfiguration added in v2.12.0 

func (x *TestResults) GetMachineConfiguration() *MachineConfiguration

func (*TestResults) GetName added in v2.12.0 

func (x *TestResults) GetName() string

func (*TestResults) GetRunConfiguration added in v2.12.0 

func (x *TestResults) GetRunConfiguration() *RunConfiguration

func (*TestResults) GetRunMode added in v2.12.0 

func (x *TestResults) GetRunMode() string

func (*TestResults) GetRunTime added in v2.12.0 

func (x *TestResults) GetRunTime() float64

func (*TestResults) GetStartTime added in v2.12.0 

func (x *TestResults) GetStartTime() int64

func (*TestResults) GetTarget added in v2.12.0 

func (x *TestResults) GetTarget() string

func (*TestResults) GetTfVersion added in v2.12.0 

func (x *TestResults) GetTfVersion() string

func (*TestResults) ProtoMessage added in v2.12.0 

func (*TestResults) ProtoMessage()

func (*TestResults) ProtoReflect added in v2.12.0 

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

func (*TestResults) Reset added in v2.12.0 

func (x *TestResults) Reset()

func (*TestResults) String added in v2.12.0 

func (x *TestResults) String() string

type TestResults_BenchmarkType added in v2.12.0 

type TestResults_BenchmarkType int32

The type of benchmark. 

const (
	TestResults_UNKNOWN            TestResults_BenchmarkType = 0 // Fallback for protos written before Type was introduced.
	TestResults_CPP_MICROBENCHMARK TestResults_BenchmarkType = 1
	TestResults_PYTHON_BENCHMARK   TestResults_BenchmarkType = 2
	TestResults_ANDROID_BENCHMARK  TestResults_BenchmarkType = 3
	TestResults_EDGE_BENCHMARK     TestResults_BenchmarkType = 4
	TestResults_IOS_BENCHMARK      TestResults_BenchmarkType = 5
)

func (TestResults_BenchmarkType) Descriptor added in v2.12.0 

func (TestResults_BenchmarkType) Descriptor() protoreflect.EnumDescriptor

func (TestResults_BenchmarkType) Enum added in v2.12.0 

func (x TestResults_BenchmarkType) Enum() *TestResults_BenchmarkType

func (TestResults_BenchmarkType) EnumDescriptor deprecated added in v2.12.0 

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

Deprecated: Use TestResults_BenchmarkType.Descriptor instead.

func (TestResults_BenchmarkType) Number added in v2.12.0 

func (x TestResults_BenchmarkType) Number() protoreflect.EnumNumber

func (TestResults_BenchmarkType) String added in v2.12.0 

func (x TestResults_BenchmarkType) String() string

func (TestResults_BenchmarkType) Type added in v2.12.0 

func (TestResults_BenchmarkType) Type() protoreflect.EnumType

type TryGetKeyValueRequest added in v2.12.0 

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

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

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

Deprecated: Use TryGetKeyValueRequest.ProtoReflect.Descriptor instead.

func (*TryGetKeyValueRequest) GetKey added in v2.12.0 

func (x *TryGetKeyValueRequest) GetKey() string

func (*TryGetKeyValueRequest) ProtoMessage added in v2.12.0 

func (*TryGetKeyValueRequest) ProtoMessage()

func (*TryGetKeyValueRequest) ProtoReflect added in v2.12.0 

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

func (*TryGetKeyValueRequest) Reset added in v2.12.0 

func (x *TryGetKeyValueRequest) Reset()

func (*TryGetKeyValueRequest) String added in v2.12.0 

func (x *TryGetKeyValueRequest) String() string

type TryGetKeyValueResponse added in v2.12.0 

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

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

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

Deprecated: Use TryGetKeyValueResponse.ProtoReflect.Descriptor instead.

func (*TryGetKeyValueResponse) GetKv added in v2.12.0 

func (x *TryGetKeyValueResponse) GetKv() *KeyValueEntry

func (*TryGetKeyValueResponse) ProtoMessage added in v2.12.0 

func (*TryGetKeyValueResponse) ProtoMessage()

func (*TryGetKeyValueResponse) ProtoReflect added in v2.12.0 

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

func (*TryGetKeyValueResponse) Reset added in v2.12.0 

func (x *TryGetKeyValueResponse) Reset()

func (*TryGetKeyValueResponse) String added in v2.12.0 

func (x *TryGetKeyValueResponse) String() string

type UnimplementedCoordinationServiceServer added in v2.12.0 

type UnimplementedCoordinationServiceServer struct {
}

UnimplementedCoordinationServiceServer must be embedded to have forward compatible implementations.

func (UnimplementedCoordinationServiceServer) Barrier added in v2.12.0 

func (UnimplementedCoordinationServiceServer) Barrier(context.Context, *BarrierRequest) (*BarrierResponse, error)

func (UnimplementedCoordinationServiceServer) CancelBarrier added in v2.12.0 

func (UnimplementedCoordinationServiceServer) CancelBarrier(context.Context, *CancelBarrierRequest) (*CancelBarrierResponse, error)

func (UnimplementedCoordinationServiceServer) DeleteKeyValue added in v2.12.0 

func (UnimplementedCoordinationServiceServer) DeleteKeyValue(context.Context, *DeleteKeyValueRequest) (*DeleteKeyValueResponse, error)

func (UnimplementedCoordinationServiceServer) GetKeyValue added in v2.12.0 

func (UnimplementedCoordinationServiceServer) GetKeyValue(context.Context, *GetKeyValueRequest) (*GetKeyValueResponse, error)

func (UnimplementedCoordinationServiceServer) GetKeyValueDir added in v2.12.0 

func (UnimplementedCoordinationServiceServer) GetKeyValueDir(context.Context, *GetKeyValueDirRequest) (*GetKeyValueDirResponse, error)

func (UnimplementedCoordinationServiceServer) GetTaskState added in v2.12.0 

func (UnimplementedCoordinationServiceServer) GetTaskState(context.Context, *GetTaskStateRequest) (*GetTaskStateResponse, error)

func (UnimplementedCoordinationServiceServer) Heartbeat added in v2.12.0 

func (UnimplementedCoordinationServiceServer) Heartbeat(context.Context, *HeartbeatRequest) (*HeartbeatResponse, error)

func (UnimplementedCoordinationServiceServer) InsertKeyValue added in v2.12.0 

func (UnimplementedCoordinationServiceServer) InsertKeyValue(context.Context, *InsertKeyValueRequest) (*InsertKeyValueResponse, error)

func (UnimplementedCoordinationServiceServer) RegisterTask added in v2.12.0 

func (UnimplementedCoordinationServiceServer) RegisterTask(context.Context, *RegisterTaskRequest) (*RegisterTaskResponse, error)

func (UnimplementedCoordinationServiceServer) ReportErrorToService added in v2.12.0 

func (UnimplementedCoordinationServiceServer) ReportErrorToService(context.Context, *ReportErrorToServiceRequest) (*ReportErrorToServiceResponse, error)

func (UnimplementedCoordinationServiceServer) ReportErrorToTask added in v2.12.0 

func (UnimplementedCoordinationServiceServer) ReportErrorToTask(context.Context, *ReportErrorToTaskRequest) (*ReportErrorToTaskResponse, error)

func (UnimplementedCoordinationServiceServer) ResetTask added in v2.12.0 

func (UnimplementedCoordinationServiceServer) ResetTask(context.Context, *ResetTaskRequest) (*ResetTaskResponse, error)

func (UnimplementedCoordinationServiceServer) ShutdownTask added in v2.12.0 

func (UnimplementedCoordinationServiceServer) ShutdownTask(context.Context, *ShutdownTaskRequest) (*ShutdownTaskResponse, error)

func (UnimplementedCoordinationServiceServer) TryGetKeyValue added in v2.12.0 

func (UnimplementedCoordinationServiceServer) TryGetKeyValue(context.Context, *TryGetKeyValueRequest) (*TryGetKeyValueResponse, error)

func (UnimplementedCoordinationServiceServer) WaitForAllTasks added in v2.12.0 

func (UnimplementedCoordinationServiceServer) WaitForAllTasks(context.Context, *WaitForAllTasksRequest) (*WaitForAllTasksResponse, error)

type UnsafeCoordinationServiceServer added in v2.12.0 

type UnsafeCoordinationServiceServer interface {
	// contains filtered or unexported methods
}

UnsafeCoordinationServiceServer may be embedded to opt out of forward compatibility for this service. Use of this interface is not recommended, as added methods to CoordinationServiceServer will result in compilation errors.

type WaitForAllTasksRequest added in v2.12.0 

type WaitForAllTasksRequest struct {
	SourceTask *CoordinatedTask `protobuf:"bytes,5,opt,name=source_task,json=sourceTask,proto3" json:"source_task,omitempty"`
	// All local device attributes on the request sender;
	DeviceInfo *DeviceInfo `protobuf:"bytes,6,opt,name=device_info,json=deviceInfo,proto3" json:"device_info,omitempty"`
	// contains filtered or unexported fields
}

Request and response messages for waiting for all tasks.

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

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

Deprecated: Use WaitForAllTasksRequest.ProtoReflect.Descriptor instead.

func (*WaitForAllTasksRequest) GetDeviceInfo added in v2.12.0 

func (x *WaitForAllTasksRequest) GetDeviceInfo() *DeviceInfo

func (*WaitForAllTasksRequest) GetSourceTask added in v2.12.0 

func (x *WaitForAllTasksRequest) GetSourceTask() *CoordinatedTask

func (*WaitForAllTasksRequest) ProtoMessage added in v2.12.0 

func (*WaitForAllTasksRequest) ProtoMessage()

func (*WaitForAllTasksRequest) ProtoReflect added in v2.12.0 

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

func (*WaitForAllTasksRequest) Reset added in v2.12.0 

func (x *WaitForAllTasksRequest) Reset()

func (*WaitForAllTasksRequest) String added in v2.12.0 

func (x *WaitForAllTasksRequest) String() string

type WaitForAllTasksResponse added in v2.12.0 

type WaitForAllTasksResponse struct {
	LeaderIncarnation uint64 `protobuf:"fixed64,1,opt,name=leader_incarnation,json=leaderIncarnation,proto3" json:"leader_incarnation,omitempty"`
	// All devices in the cluster.
	DeviceInfo *DeviceInfo `protobuf:"bytes,4,opt,name=device_info,json=deviceInfo,proto3" json:"device_info,omitempty"`
	// contains filtered or unexported fields
}

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

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

Deprecated: Use WaitForAllTasksResponse.ProtoReflect.Descriptor instead.

func (*WaitForAllTasksResponse) GetDeviceInfo added in v2.12.0 

func (x *WaitForAllTasksResponse) GetDeviceInfo() *DeviceInfo

func (*WaitForAllTasksResponse) GetLeaderIncarnation added in v2.12.0 

func (x *WaitForAllTasksResponse) GetLeaderIncarnation() uint64

func (*WaitForAllTasksResponse) ProtoMessage added in v2.12.0 

func (*WaitForAllTasksResponse) ProtoMessage()

func (*WaitForAllTasksResponse) ProtoReflect added in v2.12.0 

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

func (*WaitForAllTasksResponse) Reset added in v2.12.0 

func (x *WaitForAllTasksResponse) Reset()

func (*WaitForAllTasksResponse) String added in v2.12.0 

func (x *WaitForAllTasksResponse) String() string

type WorkerPossiblyRestarted added in v2.12.0 

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 added in v2.12.0 

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

Deprecated: Use WorkerPossiblyRestarted.ProtoReflect.Descriptor instead.

func (*WorkerPossiblyRestarted) ProtoMessage added in v2.12.0 

func (*WorkerPossiblyRestarted) ProtoMessage()

func (*WorkerPossiblyRestarted) ProtoReflect added in v2.12.0 

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

func (*WorkerPossiblyRestarted) Reset added in v2.12.0 

func (x *WorkerPossiblyRestarted) Reset()

func (*WorkerPossiblyRestarted) String added in v2.12.0 

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