protobuf

package
v1.0.0 Latest Latest
Warning

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

 Go to latest Published: May 6, 2024 License: Apache-2.0 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 (
	NormKind_name = map[int32]string{
		0: "LAYER_FWD_INFER",
		1: "LAYER_FWD_TRAIN",
		2: "LAYER_BWD",
	}
	NormKind_value = map[string]int32{
		"LAYER_FWD_INFER": 0,
		"LAYER_FWD_TRAIN": 1,
		"LAYER_BWD":       2,
	}
)

Enum value maps for NormKind.

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 (
	FMHAMaskKind_name = map[int32]string{
		0: "NO_MASK",
		1: "PADDING",
		2: "CAUSAL",
		3: "PADDING_CAUSAL",
		4: "ALIBI",
	}
	FMHAMaskKind_value = map[string]int32{
		"NO_MASK":        0,
		"PADDING":        1,
		"CAUSAL":         2,
		"PADDING_CAUSAL": 3,
		"ALIBI":          4,
	}
)

Enum value maps for FMHAMaskKind.

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: "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_tsl_protobuf_bfc_memory_map_proto protoreflect.FileDescriptor
View Source 
var File_tsl_protobuf_coordination_config_proto protoreflect.FileDescriptor
View Source 
var File_tsl_protobuf_coordination_service_proto protoreflect.FileDescriptor
View Source 
var File_tsl_protobuf_distributed_runtime_payloads_proto protoreflect.FileDescriptor
View Source 
var File_tsl_protobuf_dnn_proto protoreflect.FileDescriptor
View Source 
var File_tsl_protobuf_error_codes_proto protoreflect.FileDescriptor
View Source 
var File_tsl_protobuf_histogram_proto protoreflect.FileDescriptor
View Source 
var File_tsl_protobuf_rpc_options_proto protoreflect.FileDescriptor
View Source 
var File_tsl_protobuf_status_proto protoreflect.FileDescriptor
View Source 
var File_tsl_protobuf_test_log_proto protoreflect.FileDescriptor

Functions

func RegisterCoordinationServiceServer 

func RegisterCoordinationServiceServer(s grpc.ServiceRegistrar, srv CoordinationServiceServer)

Types

type ActivationMode 

type ActivationMode int32

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

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

func (ActivationMode) Descriptor() protoreflect.EnumDescriptor

func (ActivationMode) Enum 

func (x ActivationMode) Enum() *ActivationMode

func (ActivationMode) EnumDescriptor deprecated 

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

Deprecated: Use ActivationMode.Descriptor instead.

func (ActivationMode) Number 

func (x ActivationMode) Number() protoreflect.EnumNumber

func (ActivationMode) String 

func (x ActivationMode) String() string

func (ActivationMode) Type 

func (ActivationMode) Type() protoreflect.EnumType

type AlgorithmConfigProto 

type AlgorithmConfigProto struct {

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

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

func (*AlgorithmConfigProto) Descriptor deprecated 

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

Deprecated: Use AlgorithmConfigProto.ProtoReflect.Descriptor instead.

func (*AlgorithmConfigProto) GetAlgorithm 

func (x *AlgorithmConfigProto) GetAlgorithm() *AlgorithmProto

func (*AlgorithmConfigProto) GetAlgorithmNoScratch 

func (x *AlgorithmConfigProto) GetAlgorithmNoScratch() *AlgorithmProto

func (*AlgorithmConfigProto) GetOptionalAlgorithm 

func (m *AlgorithmConfigProto) GetOptionalAlgorithm() isAlgorithmConfigProto_OptionalAlgorithm

func (*AlgorithmConfigProto) GetOptionalAlgorithmNoScratch 

func (m *AlgorithmConfigProto) GetOptionalAlgorithmNoScratch() isAlgorithmConfigProto_OptionalAlgorithmNoScratch

func (*AlgorithmConfigProto) GetOptionalScratchSize 

func (m *AlgorithmConfigProto) GetOptionalScratchSize() isAlgorithmConfigProto_OptionalScratchSize

func (*AlgorithmConfigProto) GetScratchSize 

func (x *AlgorithmConfigProto) GetScratchSize() int64

func (*AlgorithmConfigProto) ProtoMessage 

func (*AlgorithmConfigProto) ProtoMessage()

func (*AlgorithmConfigProto) ProtoReflect 

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

func (*AlgorithmConfigProto) Reset 

func (x *AlgorithmConfigProto) Reset()

func (*AlgorithmConfigProto) String 

func (x *AlgorithmConfigProto) String() string

type AlgorithmConfigProto_Algorithm 

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

type AlgorithmConfigProto_AlgorithmNoScratch 

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

type AlgorithmConfigProto_ScratchSize 

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

type AlgorithmProto 

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

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

Deprecated: Use AlgorithmProto.ProtoReflect.Descriptor instead.

func (*AlgorithmProto) GetAlgoId 

func (x *AlgorithmProto) GetAlgoId() int64

func (*AlgorithmProto) GetIsCudnnFrontend 

func (x *AlgorithmProto) GetIsCudnnFrontend() bool

func (*AlgorithmProto) GetMathType 

func (x *AlgorithmProto) GetMathType() AlgorithmProto_MathType

func (*AlgorithmProto) GetTuningKnobs 

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

func (*AlgorithmProto) GetWorkspaceSize 

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

func (*AlgorithmProto) ProtoMessage 

func (*AlgorithmProto) ProtoMessage()

func (*AlgorithmProto) ProtoReflect 

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

func (*AlgorithmProto) Reset 

func (x *AlgorithmProto) Reset()

func (*AlgorithmProto) String 

func (x *AlgorithmProto) String() string

type AlgorithmProto_MathType 

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

func (AlgorithmProto_MathType) Descriptor 

func (AlgorithmProto_MathType) Descriptor() protoreflect.EnumDescriptor

func (AlgorithmProto_MathType) Enum 

func (x AlgorithmProto_MathType) Enum() *AlgorithmProto_MathType

func (AlgorithmProto_MathType) EnumDescriptor deprecated 

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

Deprecated: Use AlgorithmProto_MathType.Descriptor instead.

func (AlgorithmProto_MathType) Number 

func (x AlgorithmProto_MathType) Number() protoreflect.EnumNumber

func (AlgorithmProto_MathType) String 

func (x AlgorithmProto_MathType) String() string

func (AlgorithmProto_MathType) Type 

func (AlgorithmProto_MathType) Type() protoreflect.EnumType

type AvailableDeviceInfo 

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 

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

Deprecated: Use AvailableDeviceInfo.ProtoReflect.Descriptor instead.

func (*AvailableDeviceInfo) GetMemoryLimit 

func (x *AvailableDeviceInfo) GetMemoryLimit() int64

func (*AvailableDeviceInfo) GetName 

func (x *AvailableDeviceInfo) GetName() string

func (*AvailableDeviceInfo) GetPhysicalDescription 

func (x *AvailableDeviceInfo) GetPhysicalDescription() string

func (*AvailableDeviceInfo) GetType 

func (x *AvailableDeviceInfo) GetType() string

func (*AvailableDeviceInfo) ProtoMessage 

func (*AvailableDeviceInfo) ProtoMessage()

func (*AvailableDeviceInfo) ProtoReflect 

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

func (*AvailableDeviceInfo) Reset 

func (x *AvailableDeviceInfo) Reset()

func (*AvailableDeviceInfo) String 

func (x *AvailableDeviceInfo) String() string

type BarrierRequest 

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

Request and response messages for generic sync barriers.

func (*BarrierRequest) Descriptor deprecated 

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

Deprecated: Use BarrierRequest.ProtoReflect.Descriptor instead.

func (*BarrierRequest) GetBarrierId 

func (x *BarrierRequest) GetBarrierId() string

func (*BarrierRequest) GetBarrierTimeoutInMs 

func (x *BarrierRequest) GetBarrierTimeoutInMs() int64

func (*BarrierRequest) GetSourceTask 

func (x *BarrierRequest) GetSourceTask() *CoordinatedTask

func (*BarrierRequest) GetTasks 

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

func (*BarrierRequest) ProtoMessage 

func (*BarrierRequest) ProtoMessage()

func (*BarrierRequest) ProtoReflect 

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

func (*BarrierRequest) Reset 

func (x *BarrierRequest) Reset()

func (*BarrierRequest) String 

func (x *BarrierRequest) String() string

type BarrierResponse 

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

func (*BarrierResponse) Descriptor deprecated 

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

Deprecated: Use BarrierResponse.ProtoReflect.Descriptor instead.

func (*BarrierResponse) ProtoMessage 

func (*BarrierResponse) ProtoMessage()

func (*BarrierResponse) ProtoReflect 

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

func (*BarrierResponse) Reset 

func (x *BarrierResponse) Reset()

func (*BarrierResponse) String 

func (x *BarrierResponse) String() string

type BenchmarkEntries 

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

func (*BenchmarkEntries) Descriptor deprecated 

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

Deprecated: Use BenchmarkEntries.ProtoReflect.Descriptor instead.

func (*BenchmarkEntries) GetEntry 

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

func (*BenchmarkEntries) ProtoMessage 

func (*BenchmarkEntries) ProtoMessage()

func (*BenchmarkEntries) ProtoReflect 

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

func (*BenchmarkEntries) Reset 

func (x *BenchmarkEntries) Reset()

func (*BenchmarkEntries) String 

func (x *BenchmarkEntries) String() string

type BenchmarkEntry 

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 

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

Deprecated: Use BenchmarkEntry.ProtoReflect.Descriptor instead.

func (*BenchmarkEntry) GetCpuTime 

func (x *BenchmarkEntry) GetCpuTime() float64

func (*BenchmarkEntry) GetExtras 

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

func (*BenchmarkEntry) GetIters 

func (x *BenchmarkEntry) GetIters() int64

func (*BenchmarkEntry) GetMetrics 

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

func (*BenchmarkEntry) GetName 

func (x *BenchmarkEntry) GetName() string

func (*BenchmarkEntry) GetThroughput 

func (x *BenchmarkEntry) GetThroughput() float64

func (*BenchmarkEntry) GetWallTime 

func (x *BenchmarkEntry) GetWallTime() float64

func (*BenchmarkEntry) ProtoMessage 

func (*BenchmarkEntry) ProtoMessage()

func (*BenchmarkEntry) ProtoReflect 

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

func (*BenchmarkEntry) Reset 

func (x *BenchmarkEntry) Reset()

func (*BenchmarkEntry) String 

func (x *BenchmarkEntry) String() string

type BinSummary 

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

func (*BinSummary) Descriptor deprecated 

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

Deprecated: Use BinSummary.ProtoReflect.Descriptor instead.

func (*BinSummary) GetBin 

func (x *BinSummary) GetBin() int32

func (*BinSummary) GetTotalBytesInBin 

func (x *BinSummary) GetTotalBytesInBin() int64

func (*BinSummary) GetTotalBytesInUse 

func (x *BinSummary) GetTotalBytesInUse() int64

func (*BinSummary) GetTotalChunksInBin 

func (x *BinSummary) GetTotalChunksInBin() int64

func (*BinSummary) GetTotalChunksInUse 

func (x *BinSummary) GetTotalChunksInUse() int64

func (*BinSummary) ProtoMessage 

func (*BinSummary) ProtoMessage()

func (*BinSummary) ProtoReflect 

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

func (*BinSummary) Reset 

func (x *BinSummary) Reset()

func (*BinSummary) String 

func (x *BinSummary) String() string

type BuildConfiguration 

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 

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

Deprecated: Use BuildConfiguration.ProtoReflect.Descriptor instead.

func (*BuildConfiguration) GetCcFlags 

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

func (*BuildConfiguration) GetMode 

func (x *BuildConfiguration) GetMode() string

func (*BuildConfiguration) GetOpts 

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

func (*BuildConfiguration) ProtoMessage 

func (*BuildConfiguration) ProtoMessage()

func (*BuildConfiguration) ProtoReflect 

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

func (*BuildConfiguration) Reset 

func (x *BuildConfiguration) Reset()

func (*BuildConfiguration) String 

func (x *BuildConfiguration) String() string

type CPUInfo 

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 

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

Deprecated: Use CPUInfo.ProtoReflect.Descriptor instead.

func (*CPUInfo) GetCacheSize 

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

func (*CPUInfo) GetCpuGovernor 

func (x *CPUInfo) GetCpuGovernor() string

func (*CPUInfo) GetCpuInfo 

func (x *CPUInfo) GetCpuInfo() string

func (*CPUInfo) GetMhzPerCpu 

func (x *CPUInfo) GetMhzPerCpu() float64

func (*CPUInfo) GetNumCores 

func (x *CPUInfo) GetNumCores() int64

func (*CPUInfo) GetNumCoresAllowed 

func (x *CPUInfo) GetNumCoresAllowed() int64

func (*CPUInfo) ProtoMessage 

func (*CPUInfo) ProtoMessage()

func (*CPUInfo) ProtoReflect 

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

func (*CPUInfo) Reset 

func (x *CPUInfo) Reset()

func (*CPUInfo) String 

func (x *CPUInfo) String() string

type CancelBarrierRequest 

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

Request and response messages for cancelling generic sync barriers.

func (*CancelBarrierRequest) Descriptor deprecated 

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

Deprecated: Use CancelBarrierRequest.ProtoReflect.Descriptor instead.

func (*CancelBarrierRequest) GetBarrierId 

func (x *CancelBarrierRequest) GetBarrierId() string

func (*CancelBarrierRequest) GetSourceTask 

func (x *CancelBarrierRequest) GetSourceTask() *CoordinatedTask

func (*CancelBarrierRequest) ProtoMessage 

func (*CancelBarrierRequest) ProtoMessage()

func (*CancelBarrierRequest) ProtoReflect 

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

func (*CancelBarrierRequest) Reset 

func (x *CancelBarrierRequest) Reset()

func (*CancelBarrierRequest) String 

func (x *CancelBarrierRequest) String() string

type CancelBarrierResponse 

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

func (*CancelBarrierResponse) Descriptor deprecated 

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

Deprecated: Use CancelBarrierResponse.ProtoReflect.Descriptor instead.

func (*CancelBarrierResponse) ProtoMessage 

func (*CancelBarrierResponse) ProtoMessage()

func (*CancelBarrierResponse) ProtoReflect 

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

func (*CancelBarrierResponse) Reset 

func (x *CancelBarrierResponse) Reset()

func (*CancelBarrierResponse) String 

func (x *CancelBarrierResponse) String() string

type 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 

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 

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

Deprecated: Use CommitId.ProtoReflect.Descriptor instead.

func (*CommitId) GetChangelist 

func (x *CommitId) GetChangelist() int64

func (*CommitId) GetHash 

func (x *CommitId) GetHash() string

func (*CommitId) GetKind 

func (m *CommitId) GetKind() isCommitId_Kind

func (*CommitId) GetPendingChangelist 

func (x *CommitId) GetPendingChangelist() int64

func (*CommitId) GetSnapshot 

func (x *CommitId) GetSnapshot() string

func (*CommitId) ProtoMessage 

func (*CommitId) ProtoMessage()

func (*CommitId) ProtoReflect 

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

func (*CommitId) Reset 

func (x *CommitId) Reset()

func (*CommitId) String 

func (x *CommitId) String() string

type CommitId_Changelist 

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

type CommitId_Hash 

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

type ConvolutionDescriptorProto 

type ConvolutionDescriptorProto struct {
	Paddings  []int64 `protobuf:"varint,1,rep,packed,name=paddings,proto3" json:"paddings,omitempty"`
	Strides   []int64 `protobuf:"varint,2,rep,packed,name=strides,proto3" json:"strides,omitempty"`
	Dilations []int64 `protobuf:"varint,3,rep,packed,name=dilations,proto3" json:"dilations,omitempty"`
	// The "accumulator" type. For example, use F32 as an accumulator for F16
	// convolutions.
	// See cuDNN's cudnnConvolutionMode_t.
	ComputeMode 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 

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

Deprecated: Use ConvolutionDescriptorProto.ProtoReflect.Descriptor instead.

func (*ConvolutionDescriptorProto) GetComputeMode 

func (x *ConvolutionDescriptorProto) GetComputeMode() DataType

func (*ConvolutionDescriptorProto) GetConvolutionMode 

func (x *ConvolutionDescriptorProto) GetConvolutionMode() ConvolutionMode

func (*ConvolutionDescriptorProto) GetDilations 

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

func (*ConvolutionDescriptorProto) GetGroupCount 

func (x *ConvolutionDescriptorProto) GetGroupCount() int32

func (*ConvolutionDescriptorProto) GetName 

func (x *ConvolutionDescriptorProto) GetName() string

func (*ConvolutionDescriptorProto) GetPaddings 

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

func (*ConvolutionDescriptorProto) GetStrides 

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

func (*ConvolutionDescriptorProto) ProtoMessage 

func (*ConvolutionDescriptorProto) ProtoMessage()

func (*ConvolutionDescriptorProto) ProtoReflect 

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

func (*ConvolutionDescriptorProto) Reset 

func (x *ConvolutionDescriptorProto) Reset()

func (*ConvolutionDescriptorProto) String 

func (x *ConvolutionDescriptorProto) String() string

type ConvolutionKind 

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

func (ConvolutionKind) Descriptor 

func (ConvolutionKind) Descriptor() protoreflect.EnumDescriptor

func (ConvolutionKind) Enum 

func (x ConvolutionKind) Enum() *ConvolutionKind

func (ConvolutionKind) EnumDescriptor deprecated 

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

Deprecated: Use ConvolutionKind.Descriptor instead.

func (ConvolutionKind) Number 

func (x ConvolutionKind) Number() protoreflect.EnumNumber

func (ConvolutionKind) String 

func (x ConvolutionKind) String() string

func (ConvolutionKind) Type 

func (ConvolutionKind) Type() protoreflect.EnumType

type ConvolutionMode 

type ConvolutionMode int32

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

const (
	ConvolutionMode_CROSS_CORRELATION ConvolutionMode = 0
	ConvolutionMode_CONVOLUTION       ConvolutionMode = 1
)

func (ConvolutionMode) Descriptor 

func (ConvolutionMode) Descriptor() protoreflect.EnumDescriptor

func (ConvolutionMode) Enum 

func (x ConvolutionMode) Enum() *ConvolutionMode

func (ConvolutionMode) EnumDescriptor deprecated 

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

Deprecated: Use ConvolutionMode.Descriptor instead.

func (ConvolutionMode) Number 

func (x ConvolutionMode) Number() protoreflect.EnumNumber

func (ConvolutionMode) String 

func (x ConvolutionMode) String() string

func (ConvolutionMode) Type 

func (ConvolutionMode) Type() protoreflect.EnumType

type CoordinatedJob 

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 

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

Deprecated: Use CoordinatedJob.ProtoReflect.Descriptor instead.

func (*CoordinatedJob) GetName 

func (x *CoordinatedJob) GetName() string

func (*CoordinatedJob) GetNumTasks 

func (x *CoordinatedJob) GetNumTasks() int32

func (*CoordinatedJob) ProtoMessage 

func (*CoordinatedJob) ProtoMessage()

func (*CoordinatedJob) ProtoReflect 

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

func (*CoordinatedJob) Reset 

func (x *CoordinatedJob) Reset()

func (*CoordinatedJob) String 

func (x *CoordinatedJob) String() string

type CoordinatedTask 

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

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

func (*CoordinatedTask) Descriptor deprecated 

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

Deprecated: Use CoordinatedTask.ProtoReflect.Descriptor instead.

func (*CoordinatedTask) GetJobName 

func (x *CoordinatedTask) GetJobName() string

func (*CoordinatedTask) GetTaskId 

func (x *CoordinatedTask) GetTaskId() int32

func (*CoordinatedTask) ProtoMessage 

func (*CoordinatedTask) ProtoMessage()

func (*CoordinatedTask) ProtoReflect 

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

func (*CoordinatedTask) Reset 

func (x *CoordinatedTask) Reset()

func (*CoordinatedTask) String 

func (x *CoordinatedTask) String() string

type CoordinatedTaskState 

type CoordinatedTaskState int32

Represents the state of a remote worker 

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

func (CoordinatedTaskState) Descriptor 

func (CoordinatedTaskState) Descriptor() protoreflect.EnumDescriptor

func (CoordinatedTaskState) Enum 

func (x CoordinatedTaskState) Enum() *CoordinatedTaskState

func (CoordinatedTaskState) EnumDescriptor deprecated 

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

Deprecated: Use CoordinatedTaskState.Descriptor instead.

func (CoordinatedTaskState) Number 

func (x CoordinatedTaskState) Number() protoreflect.EnumNumber

func (CoordinatedTaskState) String 

func (x CoordinatedTaskState) String() string

func (CoordinatedTaskState) Type 

func (CoordinatedTaskState) Type() protoreflect.EnumType

type CoordinatedTaskStateInfo 

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 

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

Deprecated: Use CoordinatedTaskStateInfo.ProtoReflect.Descriptor instead.

func (*CoordinatedTaskStateInfo) GetErrorCode 

func (x *CoordinatedTaskStateInfo) GetErrorCode() int32

func (*CoordinatedTaskStateInfo) GetErrorMessage 

func (x *CoordinatedTaskStateInfo) GetErrorMessage() string

func (*CoordinatedTaskStateInfo) GetErrorPayload 

func (x *CoordinatedTaskStateInfo) GetErrorPayload() *CoordinationServiceError

func (*CoordinatedTaskStateInfo) GetState 

func (x *CoordinatedTaskStateInfo) GetState() CoordinatedTaskState

func (*CoordinatedTaskStateInfo) GetTask 

func (x *CoordinatedTaskStateInfo) GetTask() *CoordinatedTask

func (*CoordinatedTaskStateInfo) ProtoMessage 

func (*CoordinatedTaskStateInfo) ProtoMessage()

func (*CoordinatedTaskStateInfo) ProtoReflect 

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

func (*CoordinatedTaskStateInfo) Reset 

func (x *CoordinatedTaskStateInfo) Reset()

func (*CoordinatedTaskStateInfo) String 

func (x *CoordinatedTaskStateInfo) String() string

type CoordinationServiceClient 

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 

func NewCoordinationServiceClient(cc grpc.ClientConnInterface) CoordinationServiceClient

type CoordinationServiceConfig 

type CoordinationServiceConfig struct {

	// Type of coordination service implementation to enable.
	// For example, setting the service type as "standalone" starts a service
	// instance on the leader task to provide the coordination services such as
	// heartbeats and consistent key-value store.
	ServiceType string `protobuf:"bytes,1,opt,name=service_type,json=serviceType,proto3" json:"service_type,omitempty"`
	// Address where the coordination service instance is hosted.
	ServiceLeader string `protobuf:"bytes,2,opt,name=service_leader,json=serviceLeader,proto3" json:"service_leader,omitempty"`
	// Whether to enable the health check mechanism.
	EnableHealthCheck bool `protobuf:"varint,3,opt,name=enable_health_check,json=enableHealthCheck,proto3" json:"enable_health_check,omitempty"`
	// Maximum wait time for all members in the cluster to be registered.
	ClusterRegisterTimeoutInMs int64 `` /* 146-byte string literal not displayed */
	// Heartbeat timeout, if a task does not record heartbeat in this time
	// window, it will be considered disconnected.
	// Note: This is also used as a grace period to accept any heartbeats after
	// the agent has disconnected, to account for the lag time between the service
	// recording the state change and the agent stopping heartbeats.
	HeartbeatTimeoutInMs int64             `` /* 126-byte string literal not displayed */
	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 */
	// Disables coordination service.
	// Some libraries enable coordination service by default even if the user did
	// not specify any config. This field allows users to explicitly disable
	// coordination service under all situations.
	ForceDisable bool `protobuf:"varint,12,opt,name=force_disable,json=forceDisable,proto3" json:"force_disable,omitempty"`
	// contains filtered or unexported fields
}

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

func (*CoordinationServiceConfig) Descriptor deprecated 

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

Deprecated: Use CoordinationServiceConfig.ProtoReflect.Descriptor instead.

func (*CoordinationServiceConfig) GetAgentDestructionWithoutShutdown 

func (x *CoordinationServiceConfig) GetAgentDestructionWithoutShutdown() bool

func (*CoordinationServiceConfig) GetAllowNewIncarnationToReconnect 

func (x *CoordinationServiceConfig) GetAllowNewIncarnationToReconnect() bool

func (*CoordinationServiceConfig) GetClusterRegisterTimeoutInMs 

func (x *CoordinationServiceConfig) GetClusterRegisterTimeoutInMs() int64

func (*CoordinationServiceConfig) GetCoordinatedJobList 

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

func (*CoordinationServiceConfig) GetEnableHealthCheck 

func (x *CoordinationServiceConfig) GetEnableHealthCheck() bool

func (*CoordinationServiceConfig) GetForceDisable 

func (x *CoordinationServiceConfig) GetForceDisable() bool

func (*CoordinationServiceConfig) GetHeartbeatTimeoutInMs 

func (x *CoordinationServiceConfig) GetHeartbeatTimeoutInMs() int64

func (*CoordinationServiceConfig) GetRecoverableJobs 

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

func (*CoordinationServiceConfig) GetServiceLeader 

func (x *CoordinationServiceConfig) GetServiceLeader() string

func (*CoordinationServiceConfig) GetServiceType 

func (x *CoordinationServiceConfig) GetServiceType() string

func (*CoordinationServiceConfig) GetShutdownBarrierTimeoutInMs 

func (x *CoordinationServiceConfig) GetShutdownBarrierTimeoutInMs() int64

func (*CoordinationServiceConfig) ProtoMessage 

func (*CoordinationServiceConfig) ProtoMessage()

func (*CoordinationServiceConfig) ProtoReflect 

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

func (*CoordinationServiceConfig) Reset 

func (x *CoordinationServiceConfig) Reset()

func (*CoordinationServiceConfig) String 

func (x *CoordinationServiceConfig) String() string

type CoordinationServiceError 

type CoordinationServiceError struct {

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

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

func (*CoordinationServiceError) Descriptor deprecated 

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

Deprecated: Use CoordinationServiceError.ProtoReflect.Descriptor instead.

func (*CoordinationServiceError) GetIsReportedError 

func (x *CoordinationServiceError) GetIsReportedError() bool

func (*CoordinationServiceError) GetSourceTask 

func (x *CoordinationServiceError) GetSourceTask() *CoordinatedTask

func (*CoordinationServiceError) ProtoMessage 

func (*CoordinationServiceError) ProtoMessage()

func (*CoordinationServiceError) ProtoReflect 

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

func (*CoordinationServiceError) Reset 

func (x *CoordinationServiceError) Reset()

func (*CoordinationServiceError) String 

func (x *CoordinationServiceError) String() string

type CoordinationServiceServer 

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 

type DataLayout int32

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

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

func (DataLayout) Descriptor 

func (DataLayout) Descriptor() protoreflect.EnumDescriptor

func (DataLayout) Enum 

func (x DataLayout) Enum() *DataLayout

func (DataLayout) EnumDescriptor deprecated 

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

Deprecated: Use DataLayout.Descriptor instead.

func (DataLayout) Number 

func (x DataLayout) Number() protoreflect.EnumNumber

func (DataLayout) String 

func (x DataLayout) String() string

func (DataLayout) Type 

func (DataLayout) Type() protoreflect.EnumType

type DataType 

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 

func (DataType) Descriptor() protoreflect.EnumDescriptor

func (DataType) Enum 

func (x DataType) Enum() *DataType

func (DataType) EnumDescriptor deprecated 

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

Deprecated: Use DataType.Descriptor instead.

func (DataType) Number 

func (x DataType) Number() protoreflect.EnumNumber

func (DataType) String 

func (x DataType) String() string

func (DataType) Type 

func (DataType) Type() protoreflect.EnumType

type DeleteKeyValueRequest 

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

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

func (*DeleteKeyValueRequest) Descriptor deprecated 

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

Deprecated: Use DeleteKeyValueRequest.ProtoReflect.Descriptor instead.

func (*DeleteKeyValueRequest) GetIsDirectory 

func (x *DeleteKeyValueRequest) GetIsDirectory() bool

func (*DeleteKeyValueRequest) GetKey 

func (x *DeleteKeyValueRequest) GetKey() string

func (*DeleteKeyValueRequest) ProtoMessage 

func (*DeleteKeyValueRequest) ProtoMessage()

func (*DeleteKeyValueRequest) ProtoReflect 

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

func (*DeleteKeyValueRequest) Reset 

func (x *DeleteKeyValueRequest) Reset()

func (*DeleteKeyValueRequest) String 

func (x *DeleteKeyValueRequest) String() string

type DeleteKeyValueResponse 

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

func (*DeleteKeyValueResponse) Descriptor deprecated 

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

Deprecated: Use DeleteKeyValueResponse.ProtoReflect.Descriptor instead.

func (*DeleteKeyValueResponse) ProtoMessage 

func (*DeleteKeyValueResponse) ProtoMessage()

func (*DeleteKeyValueResponse) ProtoReflect 

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

func (*DeleteKeyValueResponse) Reset 

func (x *DeleteKeyValueResponse) Reset()

func (*DeleteKeyValueResponse) String 

func (x *DeleteKeyValueResponse) String() string

type DeviceInfo 

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 

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

Deprecated: Use DeviceInfo.ProtoReflect.Descriptor instead.

func (*DeviceInfo) GetDevice 

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

func (*DeviceInfo) ProtoMessage 

func (*DeviceInfo) ProtoMessage()

func (*DeviceInfo) ProtoReflect 

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

func (*DeviceInfo) Reset 

func (x *DeviceInfo) Reset()

func (*DeviceInfo) String 

func (x *DeviceInfo) String() string

type EntryValue 

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 

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

Deprecated: Use EntryValue.ProtoReflect.Descriptor instead.

func (*EntryValue) GetDoubleValue 

func (x *EntryValue) GetDoubleValue() float64

func (*EntryValue) GetKind 

func (m *EntryValue) GetKind() isEntryValue_Kind

func (*EntryValue) GetStringValue 

func (x *EntryValue) GetStringValue() string

func (*EntryValue) ProtoMessage 

func (*EntryValue) ProtoMessage()

func (*EntryValue) ProtoReflect 

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

func (*EntryValue) Reset 

func (x *EntryValue) Reset()

func (*EntryValue) String 

func (x *EntryValue) String() string

type EntryValue_DoubleValue 

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

type EntryValue_StringValue 

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

type FMHAMaskKind 

type FMHAMaskKind int32

FusedMHAMaskKind kind 

const (
	FMHAMaskKind_NO_MASK        FMHAMaskKind = 0
	FMHAMaskKind_PADDING        FMHAMaskKind = 1
	FMHAMaskKind_CAUSAL         FMHAMaskKind = 2
	FMHAMaskKind_PADDING_CAUSAL FMHAMaskKind = 3
	FMHAMaskKind_ALIBI          FMHAMaskKind = 4
)

func (FMHAMaskKind) Descriptor 

func (FMHAMaskKind) Descriptor() protoreflect.EnumDescriptor

func (FMHAMaskKind) Enum 

func (x FMHAMaskKind) Enum() *FMHAMaskKind

func (FMHAMaskKind) EnumDescriptor deprecated 

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

Deprecated: Use FMHAMaskKind.Descriptor instead.

func (FMHAMaskKind) Number 

func (x FMHAMaskKind) Number() protoreflect.EnumNumber

func (FMHAMaskKind) String 

func (x FMHAMaskKind) String() string

func (FMHAMaskKind) Type 

func (FMHAMaskKind) Type() protoreflect.EnumType

type FilterLayout 

type FilterLayout int32

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

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

func (FilterLayout) Descriptor() protoreflect.EnumDescriptor

func (FilterLayout) Enum 

func (x FilterLayout) Enum() *FilterLayout

func (FilterLayout) EnumDescriptor deprecated 

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

Deprecated: Use FilterLayout.Descriptor instead.

func (FilterLayout) Number 

func (x FilterLayout) Number() protoreflect.EnumNumber

func (FilterLayout) String 

func (x FilterLayout) String() string

func (FilterLayout) Type 

func (FilterLayout) Type() protoreflect.EnumType

type FusedMHAKind 

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 

func (FusedMHAKind) Descriptor() protoreflect.EnumDescriptor

func (FusedMHAKind) Enum 

func (x FusedMHAKind) Enum() *FusedMHAKind

func (FusedMHAKind) EnumDescriptor deprecated 

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

Deprecated: Use FusedMHAKind.Descriptor instead.

func (FusedMHAKind) Number 

func (x FusedMHAKind) Number() protoreflect.EnumNumber

func (FusedMHAKind) String 

func (x FusedMHAKind) String() string

func (FusedMHAKind) Type 

func (FusedMHAKind) Type() protoreflect.EnumType

type GPUInfo 

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 

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

Deprecated: Use GPUInfo.ProtoReflect.Descriptor instead.

func (*GPUInfo) GetBusId 

func (x *GPUInfo) GetBusId() string

func (*GPUInfo) GetModel 

func (x *GPUInfo) GetModel() string

func (*GPUInfo) GetUuid 

func (x *GPUInfo) GetUuid() string

func (*GPUInfo) ProtoMessage 

func (*GPUInfo) ProtoMessage()

func (*GPUInfo) ProtoReflect 

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

func (*GPUInfo) Reset 

func (x *GPUInfo) Reset()

func (*GPUInfo) String 

func (x *GPUInfo) String() string

type GetKeyValueDirRequest 

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

func (*GetKeyValueDirRequest) Descriptor deprecated 

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

Deprecated: Use GetKeyValueDirRequest.ProtoReflect.Descriptor instead.

func (*GetKeyValueDirRequest) GetDirectoryKey 

func (x *GetKeyValueDirRequest) GetDirectoryKey() string

func (*GetKeyValueDirRequest) ProtoMessage 

func (*GetKeyValueDirRequest) ProtoMessage()

func (*GetKeyValueDirRequest) ProtoReflect 

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

func (*GetKeyValueDirRequest) Reset 

func (x *GetKeyValueDirRequest) Reset()

func (*GetKeyValueDirRequest) String 

func (x *GetKeyValueDirRequest) String() string

type GetKeyValueDirResponse 

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

func (*GetKeyValueDirResponse) Descriptor deprecated 

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

Deprecated: Use GetKeyValueDirResponse.ProtoReflect.Descriptor instead.

func (*GetKeyValueDirResponse) GetDirectoryKey 

func (x *GetKeyValueDirResponse) GetDirectoryKey() string

func (*GetKeyValueDirResponse) GetKv 

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

func (*GetKeyValueDirResponse) ProtoMessage 

func (*GetKeyValueDirResponse) ProtoMessage()

func (*GetKeyValueDirResponse) ProtoReflect 

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

func (*GetKeyValueDirResponse) Reset 

func (x *GetKeyValueDirResponse) Reset()

func (*GetKeyValueDirResponse) String 

func (x *GetKeyValueDirResponse) String() string

type GetKeyValueRequest 

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

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

func (*GetKeyValueRequest) Descriptor deprecated 

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

Deprecated: Use GetKeyValueRequest.ProtoReflect.Descriptor instead.

func (*GetKeyValueRequest) GetKey 

func (x *GetKeyValueRequest) GetKey() string

func (*GetKeyValueRequest) ProtoMessage 

func (*GetKeyValueRequest) ProtoMessage()

func (*GetKeyValueRequest) ProtoReflect 

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

func (*GetKeyValueRequest) Reset 

func (x *GetKeyValueRequest) Reset()

func (*GetKeyValueRequest) String 

func (x *GetKeyValueRequest) String() string

type GetKeyValueResponse 

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

func (*GetKeyValueResponse) Descriptor deprecated 

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

Deprecated: Use GetKeyValueResponse.ProtoReflect.Descriptor instead.

func (*GetKeyValueResponse) GetKv 

func (x *GetKeyValueResponse) GetKv() *KeyValueEntry

func (*GetKeyValueResponse) ProtoMessage 

func (*GetKeyValueResponse) ProtoMessage()

func (*GetKeyValueResponse) ProtoReflect 

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

func (*GetKeyValueResponse) Reset 

func (x *GetKeyValueResponse) Reset()

func (*GetKeyValueResponse) String 

func (x *GetKeyValueResponse) String() string

type GetTaskStateRequest 

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 

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

Deprecated: Use GetTaskStateRequest.ProtoReflect.Descriptor instead.

func (*GetTaskStateRequest) GetSourceTask 

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

func (*GetTaskStateRequest) ProtoMessage 

func (*GetTaskStateRequest) ProtoMessage()

func (*GetTaskStateRequest) ProtoReflect 

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

func (*GetTaskStateRequest) Reset 

func (x *GetTaskStateRequest) Reset()

func (*GetTaskStateRequest) String 

func (x *GetTaskStateRequest) String() string

type GetTaskStateResponse 

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 

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

Deprecated: Use GetTaskStateResponse.ProtoReflect.Descriptor instead.

func (*GetTaskStateResponse) GetTaskState 

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

func (*GetTaskStateResponse) ProtoMessage 

func (*GetTaskStateResponse) ProtoMessage()

func (*GetTaskStateResponse) ProtoReflect 

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

func (*GetTaskStateResponse) Reset 

func (x *GetTaskStateResponse) Reset()

func (*GetTaskStateResponse) String 

func (x *GetTaskStateResponse) String() string

type GrpcPayloadContainer 

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

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

func (*GrpcPayloadContainer) Descriptor deprecated 

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

Deprecated: Use GrpcPayloadContainer.ProtoReflect.Descriptor instead.

func (*GrpcPayloadContainer) GetPayloads 

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

func (*GrpcPayloadContainer) ProtoMessage 

func (*GrpcPayloadContainer) ProtoMessage()

func (*GrpcPayloadContainer) ProtoReflect 

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

func (*GrpcPayloadContainer) Reset 

func (x *GrpcPayloadContainer) Reset()

func (*GrpcPayloadContainer) String 

func (x *GrpcPayloadContainer) String() string

type GrpcPayloadsLost 

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

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

func (*GrpcPayloadsLost) Descriptor deprecated 

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

Deprecated: Use GrpcPayloadsLost.ProtoReflect.Descriptor instead.

func (*GrpcPayloadsLost) ProtoMessage 

func (*GrpcPayloadsLost) ProtoMessage()

func (*GrpcPayloadsLost) ProtoReflect 

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

func (*GrpcPayloadsLost) Reset 

func (x *GrpcPayloadsLost) Reset()

func (*GrpcPayloadsLost) String 

func (x *GrpcPayloadsLost) String() string

type HeartbeatRequest 

type HeartbeatRequest struct {
	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 

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

Deprecated: Use HeartbeatRequest.ProtoReflect.Descriptor instead.

func (*HeartbeatRequest) GetIncarnation 

func (x *HeartbeatRequest) GetIncarnation() uint64

func (*HeartbeatRequest) GetSourceTask 

func (x *HeartbeatRequest) GetSourceTask() *CoordinatedTask

func (*HeartbeatRequest) ProtoMessage 

func (*HeartbeatRequest) ProtoMessage()

func (*HeartbeatRequest) ProtoReflect 

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

func (*HeartbeatRequest) Reset 

func (x *HeartbeatRequest) Reset()

func (*HeartbeatRequest) String 

func (x *HeartbeatRequest) String() string

type HeartbeatResponse 

type HeartbeatResponse struct {
	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 

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

Deprecated: Use HeartbeatResponse.ProtoReflect.Descriptor instead.

func (*HeartbeatResponse) GetLeaderIncarnation 

func (x *HeartbeatResponse) GetLeaderIncarnation() uint64

func (*HeartbeatResponse) ProtoMessage 

func (*HeartbeatResponse) ProtoMessage()

func (*HeartbeatResponse) ProtoReflect 

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

func (*HeartbeatResponse) Reset 

func (x *HeartbeatResponse) Reset()

func (*HeartbeatResponse) String 

func (x *HeartbeatResponse) String() string

type HistogramProto 

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

Serialization format for histogram module in 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 

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

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

func (*InsertKeyValueRequest) Descriptor deprecated 

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

Deprecated: Use InsertKeyValueRequest.ProtoReflect.Descriptor instead.

func (*InsertKeyValueRequest) GetKv 

func (x *InsertKeyValueRequest) GetKv() *KeyValueEntry

func (*InsertKeyValueRequest) ProtoMessage 

func (*InsertKeyValueRequest) ProtoMessage()

func (*InsertKeyValueRequest) ProtoReflect 

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

func (*InsertKeyValueRequest) Reset 

func (x *InsertKeyValueRequest) Reset()

func (*InsertKeyValueRequest) String 

func (x *InsertKeyValueRequest) String() string

type InsertKeyValueResponse 

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

func (*InsertKeyValueResponse) Descriptor deprecated 

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

Deprecated: Use InsertKeyValueResponse.ProtoReflect.Descriptor instead.

func (*InsertKeyValueResponse) ProtoMessage 

func (*InsertKeyValueResponse) ProtoMessage()

func (*InsertKeyValueResponse) ProtoReflect 

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

func (*InsertKeyValueResponse) Reset 

func (x *InsertKeyValueResponse) Reset()

func (*InsertKeyValueResponse) String 

func (x *InsertKeyValueResponse) String() string

type KeyValueEntry 

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

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

func (*KeyValueEntry) Descriptor deprecated 

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

Deprecated: Use KeyValueEntry.ProtoReflect.Descriptor instead.

func (*KeyValueEntry) GetKey 

func (x *KeyValueEntry) GetKey() string

func (*KeyValueEntry) GetValue 

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

func (*KeyValueEntry) ProtoMessage 

func (*KeyValueEntry) ProtoMessage()

func (*KeyValueEntry) ProtoReflect 

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

func (*KeyValueEntry) Reset 

func (x *KeyValueEntry) Reset()

func (*KeyValueEntry) String 

func (x *KeyValueEntry) String() string

type MachineConfiguration 

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 

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

Deprecated: Use MachineConfiguration.ProtoReflect.Descriptor instead.

func (*MachineConfiguration) GetAvailableDeviceInfo 

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

func (*MachineConfiguration) GetCpuInfo 

func (x *MachineConfiguration) GetCpuInfo() *CPUInfo

func (*MachineConfiguration) GetDeviceInfo 

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

func (*MachineConfiguration) GetHostname 

func (x *MachineConfiguration) GetHostname() string

func (*MachineConfiguration) GetMemoryInfo 

func (x *MachineConfiguration) GetMemoryInfo() *MemoryInfo

func (*MachineConfiguration) GetPlatformInfo 

func (x *MachineConfiguration) GetPlatformInfo() *PlatformInfo

func (*MachineConfiguration) GetSerialIdentifier 

func (x *MachineConfiguration) GetSerialIdentifier() string

func (*MachineConfiguration) ProtoMessage 

func (*MachineConfiguration) ProtoMessage()

func (*MachineConfiguration) ProtoReflect 

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

func (*MachineConfiguration) Reset 

func (x *MachineConfiguration) Reset()

func (*MachineConfiguration) String 

func (x *MachineConfiguration) String() string

type MemAllocatorStats 

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

Some of the data from AllocatorStats

func (*MemAllocatorStats) Descriptor deprecated 

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

Deprecated: Use MemAllocatorStats.ProtoReflect.Descriptor instead.

func (*MemAllocatorStats) GetBytesInUse 

func (x *MemAllocatorStats) GetBytesInUse() int64

func (*MemAllocatorStats) GetFragmentationMetric 

func (x *MemAllocatorStats) GetFragmentationMetric() float32

func (*MemAllocatorStats) GetLargestAllocSize 

func (x *MemAllocatorStats) GetLargestAllocSize() int64

func (*MemAllocatorStats) GetNumAllocs 

func (x *MemAllocatorStats) GetNumAllocs() int64

func (*MemAllocatorStats) GetPeakBytesInUse 

func (x *MemAllocatorStats) GetPeakBytesInUse() int64

func (*MemAllocatorStats) ProtoMessage 

func (*MemAllocatorStats) ProtoMessage()

func (*MemAllocatorStats) ProtoReflect 

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

func (*MemAllocatorStats) Reset 

func (x *MemAllocatorStats) Reset()

func (*MemAllocatorStats) String 

func (x *MemAllocatorStats) String() string

type MemChunk 

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

func (*MemChunk) Descriptor deprecated 

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

Deprecated: Use MemChunk.ProtoReflect.Descriptor instead.

func (*MemChunk) GetActionCount 

func (x *MemChunk) GetActionCount() uint64

func (*MemChunk) GetAddress 

func (x *MemChunk) GetAddress() uint64

func (*MemChunk) GetBin 

func (x *MemChunk) GetBin() int32

func (*MemChunk) GetFreedAtCount 

func (x *MemChunk) GetFreedAtCount() uint64

func (*MemChunk) GetInUse 

func (x *MemChunk) GetInUse() bool

func (*MemChunk) GetOpName 

func (x *MemChunk) GetOpName() string

func (*MemChunk) GetRequestedSize 

func (x *MemChunk) GetRequestedSize() int64

func (*MemChunk) GetSize 

func (x *MemChunk) GetSize() int64

func (*MemChunk) GetStepId 

func (x *MemChunk) GetStepId() uint64

func (*MemChunk) ProtoMessage 

func (*MemChunk) ProtoMessage()

func (*MemChunk) ProtoReflect 

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

func (*MemChunk) Reset 

func (x *MemChunk) Reset()

func (*MemChunk) String 

func (x *MemChunk) String() string

type MemoryDump 

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

func (*MemoryDump) Descriptor deprecated 

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

Deprecated: Use MemoryDump.ProtoReflect.Descriptor instead.

func (*MemoryDump) GetAllocatorName 

func (x *MemoryDump) GetAllocatorName() string

func (*MemoryDump) GetBinSummary 

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

func (*MemoryDump) GetChunk 

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

func (*MemoryDump) GetSnapShot 

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

func (*MemoryDump) GetStats 

func (x *MemoryDump) GetStats() *MemAllocatorStats

func (*MemoryDump) ProtoMessage 

func (*MemoryDump) ProtoMessage()

func (*MemoryDump) ProtoReflect 

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

func (*MemoryDump) Reset 

func (x *MemoryDump) Reset()

func (*MemoryDump) String 

func (x *MemoryDump) String() string

type MemoryInfo 

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 

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

Deprecated: Use MemoryInfo.ProtoReflect.Descriptor instead.

func (*MemoryInfo) GetAvailable 

func (x *MemoryInfo) GetAvailable() int64

func (*MemoryInfo) GetTotal 

func (x *MemoryInfo) GetTotal() int64

func (*MemoryInfo) ProtoMessage 

func (*MemoryInfo) ProtoMessage()

func (*MemoryInfo) ProtoReflect 

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

func (*MemoryInfo) Reset 

func (x *MemoryInfo) Reset()

func (*MemoryInfo) String 

func (x *MemoryInfo) String() string

type MetricEntry 

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 

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

Deprecated: Use MetricEntry.ProtoReflect.Descriptor instead.

func (*MetricEntry) GetMaxValue 

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

func (*MetricEntry) GetMinValue 

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

func (*MetricEntry) GetName 

func (x *MetricEntry) GetName() string

func (*MetricEntry) GetValue 

func (x *MetricEntry) GetValue() float64

func (*MetricEntry) ProtoMessage 

func (*MetricEntry) ProtoMessage()

func (*MetricEntry) ProtoReflect 

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

func (*MetricEntry) Reset 

func (x *MetricEntry) Reset()

func (*MetricEntry) String 

func (x *MetricEntry) String() string

type NormKind 

type NormKind int32

NormKind kind 

const (
	NormKind_LAYER_FWD_INFER NormKind = 0
	NormKind_LAYER_FWD_TRAIN NormKind = 1
	NormKind_LAYER_BWD       NormKind = 2
)

func (NormKind) Descriptor 

func (NormKind) Descriptor() protoreflect.EnumDescriptor

func (NormKind) Enum 

func (x NormKind) Enum() *NormKind

func (NormKind) EnumDescriptor deprecated 

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

Deprecated: Use NormKind.Descriptor instead.

func (NormKind) Number 

func (x NormKind) Number() protoreflect.EnumNumber

func (NormKind) String 

func (x NormKind) String() string

func (NormKind) Type 

func (NormKind) Type() protoreflect.EnumType

type PlatformInfo 

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 

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

Deprecated: Use PlatformInfo.ProtoReflect.Descriptor instead.

func (*PlatformInfo) GetBits 

func (x *PlatformInfo) GetBits() string

func (*PlatformInfo) GetLinkage 

func (x *PlatformInfo) GetLinkage() string

func (*PlatformInfo) GetMachine 

func (x *PlatformInfo) GetMachine() string

func (*PlatformInfo) GetRelease 

func (x *PlatformInfo) GetRelease() string

func (*PlatformInfo) GetSystem 

func (x *PlatformInfo) GetSystem() string

func (*PlatformInfo) GetVersion 

func (x *PlatformInfo) GetVersion() string

func (*PlatformInfo) ProtoMessage 

func (*PlatformInfo) ProtoMessage()

func (*PlatformInfo) ProtoReflect 

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

func (*PlatformInfo) Reset 

func (x *PlatformInfo) Reset()

func (*PlatformInfo) String 

func (x *PlatformInfo) String() string

type RPCOptions 

type RPCOptions struct {

	// If true, always use RPC to contact the session target.
	//
	// If false (the default option), TensorFlow may use an optimized
	// transport for client-master communication that avoids the RPC
	// stack. This option is primarily for used testing the RPC stack.
	UseRpcForInprocessMaster bool `` /* 140-byte string literal not displayed */
	// The compression algorithm to be used. One of "deflate", "gzip".
	CompressionAlgorithm string `protobuf:"bytes,2,opt,name=compression_algorithm,json=compressionAlgorithm,proto3" json:"compression_algorithm,omitempty"`
	// If compression_algorithm is set, the compression level to be used.
	// From 0 (no compression), up to 3.
	CompressionLevel int32 `protobuf:"varint,3,opt,name=compression_level,json=compressionLevel,proto3" json:"compression_level,omitempty"`
	// Setting cache_rpc_response to true will enable sender side caching of
	// response for RecvTensorAsync and RecvBufAsync to allow receiver to retry
	// requests . This is only necessary when the network fabric is experiencing a
	// significant error rate.  Without it we'll fail a step on an network error,
	// while with it we'll be able to complete long steps (like complex
	// initializations) in the face of some network errors during RecvTensor.
	CacheRpcResponse bool `protobuf:"varint,4,opt,name=cache_rpc_response,json=cacheRpcResponse,proto3" json:"cache_rpc_response,omitempty"`
	// Disables TCP connection sharing when opening a new RPC channel.
	DisableSessionConnectionSharing bool `` /* 159-byte string literal not displayed */
	// Setting num_channels_per_target > 0 allows uses of multiple channels to
	// communicate to the same target. This can be used to improve the aggregate
	// throughput on high speed links (e.g 100G) where single connection is not
	// sufficient to maximize link utilization. Note that a single RPC only goes
	// on a single channel, this only helps in situations where there are multiple
	// transfers to the same target overlapping in time.
	NumChannelsPerTarget int32 `` /* 126-byte string literal not displayed */
	// contains filtered or unexported fields
}

RPC options for distributed runtime.

func (*RPCOptions) Descriptor deprecated 

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

Deprecated: Use RPCOptions.ProtoReflect.Descriptor instead.

func (*RPCOptions) GetCacheRpcResponse 

func (x *RPCOptions) GetCacheRpcResponse() bool

func (*RPCOptions) GetCompressionAlgorithm 

func (x *RPCOptions) GetCompressionAlgorithm() string

func (*RPCOptions) GetCompressionLevel 

func (x *RPCOptions) GetCompressionLevel() int32

func (*RPCOptions) GetDisableSessionConnectionSharing 

func (x *RPCOptions) GetDisableSessionConnectionSharing() bool

func (*RPCOptions) GetNumChannelsPerTarget 

func (x *RPCOptions) GetNumChannelsPerTarget() int32

func (*RPCOptions) GetUseRpcForInprocessMaster 

func (x *RPCOptions) GetUseRpcForInprocessMaster() bool

func (*RPCOptions) ProtoMessage 

func (*RPCOptions) ProtoMessage()

func (*RPCOptions) ProtoReflect 

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

func (*RPCOptions) Reset 

func (x *RPCOptions) Reset()

func (*RPCOptions) String 

func (x *RPCOptions) String() string

type RegisterTaskRequest 

type RegisterTaskRequest struct {
	Incarnation uint64           `protobuf:"fixed64,3,opt,name=incarnation,proto3" json:"incarnation,omitempty"`
	SourceTask  *CoordinatedTask `protobuf:"bytes,5,opt,name=source_task,json=sourceTask,proto3" json:"source_task,omitempty"`
	// contains filtered or unexported fields
}

Request and response messages for registering a task to the cluster leader. A task is uniquely represented by its `job_name`, `task_id` and `incarnation`. Leader responds with its `incarnation` to identify a leader process.

func (*RegisterTaskRequest) Descriptor deprecated 

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

Deprecated: Use RegisterTaskRequest.ProtoReflect.Descriptor instead.

func (*RegisterTaskRequest) GetIncarnation 

func (x *RegisterTaskRequest) GetIncarnation() uint64

func (*RegisterTaskRequest) GetSourceTask 

func (x *RegisterTaskRequest) GetSourceTask() *CoordinatedTask

func (*RegisterTaskRequest) ProtoMessage 

func (*RegisterTaskRequest) ProtoMessage()

func (*RegisterTaskRequest) ProtoReflect 

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

func (*RegisterTaskRequest) Reset 

func (x *RegisterTaskRequest) Reset()

func (*RegisterTaskRequest) String 

func (x *RegisterTaskRequest) String() string

type RegisterTaskResponse 

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

func (*RegisterTaskResponse) Descriptor deprecated 

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

Deprecated: Use RegisterTaskResponse.ProtoReflect.Descriptor instead.

func (*RegisterTaskResponse) GetLeaderIncarnation 

func (x *RegisterTaskResponse) GetLeaderIncarnation() uint64

func (*RegisterTaskResponse) ProtoMessage 

func (*RegisterTaskResponse) ProtoMessage()

func (*RegisterTaskResponse) ProtoReflect 

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

func (*RegisterTaskResponse) Reset 

func (x *RegisterTaskResponse) Reset()

func (*RegisterTaskResponse) String 

func (x *RegisterTaskResponse) String() string

type ReportErrorToServiceRequest 

type ReportErrorToServiceRequest struct {
	ErrorCode    int32            `protobuf:"varint,1,opt,name=error_code,json=errorCode,proto3" json:"error_code,omitempty"`
	ErrorMessage string           `protobuf:"bytes,2,opt,name=error_message,json=errorMessage,proto3" json:"error_message,omitempty"`
	ErrorOrigin  *CoordinatedTask `protobuf:"bytes,5,opt,name=error_origin,json=errorOrigin,proto3" json:"error_origin,omitempty"`
	// contains filtered or unexported fields
}

Request and response messages for reporting errors to service instance.

func (*ReportErrorToServiceRequest) Descriptor deprecated 

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

Deprecated: Use ReportErrorToServiceRequest.ProtoReflect.Descriptor instead.

func (*ReportErrorToServiceRequest) GetErrorCode 

func (x *ReportErrorToServiceRequest) GetErrorCode() int32

func (*ReportErrorToServiceRequest) GetErrorMessage 

func (x *ReportErrorToServiceRequest) GetErrorMessage() string

func (*ReportErrorToServiceRequest) GetErrorOrigin 

func (x *ReportErrorToServiceRequest) GetErrorOrigin() *CoordinatedTask

func (*ReportErrorToServiceRequest) ProtoMessage 

func (*ReportErrorToServiceRequest) ProtoMessage()

func (*ReportErrorToServiceRequest) ProtoReflect 

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

func (*ReportErrorToServiceRequest) Reset 

func (x *ReportErrorToServiceRequest) Reset()

func (*ReportErrorToServiceRequest) String 

func (x *ReportErrorToServiceRequest) String() string

type ReportErrorToServiceResponse 

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

func (*ReportErrorToServiceResponse) Descriptor deprecated 

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

Deprecated: Use ReportErrorToServiceResponse.ProtoReflect.Descriptor instead.

func (*ReportErrorToServiceResponse) ProtoMessage 

func (*ReportErrorToServiceResponse) ProtoMessage()

func (*ReportErrorToServiceResponse) ProtoReflect 

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

func (*ReportErrorToServiceResponse) Reset 

func (x *ReportErrorToServiceResponse) Reset()

func (*ReportErrorToServiceResponse) String 

func (x *ReportErrorToServiceResponse) String() string

type ReportErrorToTaskRequest 

type ReportErrorToTaskRequest struct {
	ErrorCode    int32                     `protobuf:"varint,1,opt,name=error_code,json=errorCode,proto3" json:"error_code,omitempty"`
	ErrorMessage string                    `protobuf:"bytes,2,opt,name=error_message,json=errorMessage,proto3" json:"error_message,omitempty"`
	ErrorPayload *CoordinationServiceError `protobuf:"bytes,5,opt,name=error_payload,json=errorPayload,proto3" json:"error_payload,omitempty"`
	// contains filtered or unexported fields
}

Request and response messages for reporting errors to task.

func (*ReportErrorToTaskRequest) Descriptor deprecated 

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

Deprecated: Use ReportErrorToTaskRequest.ProtoReflect.Descriptor instead.

func (*ReportErrorToTaskRequest) GetErrorCode 

func (x *ReportErrorToTaskRequest) GetErrorCode() int32

func (*ReportErrorToTaskRequest) GetErrorMessage 

func (x *ReportErrorToTaskRequest) GetErrorMessage() string

func (*ReportErrorToTaskRequest) GetErrorPayload 

func (x *ReportErrorToTaskRequest) GetErrorPayload() *CoordinationServiceError

func (*ReportErrorToTaskRequest) ProtoMessage 

func (*ReportErrorToTaskRequest) ProtoMessage()

func (*ReportErrorToTaskRequest) ProtoReflect 

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

func (*ReportErrorToTaskRequest) Reset 

func (x *ReportErrorToTaskRequest) Reset()

func (*ReportErrorToTaskRequest) String 

func (x *ReportErrorToTaskRequest) String() string

type ReportErrorToTaskResponse 

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

func (*ReportErrorToTaskResponse) Descriptor deprecated 

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

Deprecated: Use ReportErrorToTaskResponse.ProtoReflect.Descriptor instead.

func (*ReportErrorToTaskResponse) ProtoMessage 

func (*ReportErrorToTaskResponse) ProtoMessage()

func (*ReportErrorToTaskResponse) ProtoReflect 

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

func (*ReportErrorToTaskResponse) Reset 

func (x *ReportErrorToTaskResponse) Reset()

func (*ReportErrorToTaskResponse) String 

func (x *ReportErrorToTaskResponse) String() string

type ResetTaskRequest 

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

Request and response messages for resetting a task state in the service.

func (*ResetTaskRequest) Descriptor deprecated 

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

Deprecated: Use ResetTaskRequest.ProtoReflect.Descriptor instead.

func (*ResetTaskRequest) GetSourceTask 

func (x *ResetTaskRequest) GetSourceTask() *CoordinatedTask

func (*ResetTaskRequest) ProtoMessage 

func (*ResetTaskRequest) ProtoMessage()

func (*ResetTaskRequest) ProtoReflect 

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

func (*ResetTaskRequest) Reset 

func (x *ResetTaskRequest) Reset()

func (*ResetTaskRequest) String 

func (x *ResetTaskRequest) String() string

type ResetTaskResponse 

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

func (*ResetTaskResponse) Descriptor deprecated 

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

Deprecated: Use ResetTaskResponse.ProtoReflect.Descriptor instead.

func (*ResetTaskResponse) ProtoMessage 

func (*ResetTaskResponse) ProtoMessage()

func (*ResetTaskResponse) ProtoReflect 

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

func (*ResetTaskResponse) Reset 

func (x *ResetTaskResponse) Reset()

func (*ResetTaskResponse) String 

func (x *ResetTaskResponse) String() string

type RunConfiguration 

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 

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

Deprecated: Use RunConfiguration.ProtoReflect.Descriptor instead.

func (*RunConfiguration) GetArgument 

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

func (*RunConfiguration) GetEnvVars 

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

func (*RunConfiguration) ProtoMessage 

func (*RunConfiguration) ProtoMessage()

func (*RunConfiguration) ProtoReflect 

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

func (*RunConfiguration) Reset 

func (x *RunConfiguration) Reset()

func (*RunConfiguration) String 

func (x *RunConfiguration) String() string

type ShutdownTaskRequest 

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

Request and response messages for disconnecting a task from the service.

func (*ShutdownTaskRequest) Descriptor deprecated 

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

Deprecated: Use ShutdownTaskRequest.ProtoReflect.Descriptor instead.

func (*ShutdownTaskRequest) GetSourceTask 

func (x *ShutdownTaskRequest) GetSourceTask() *CoordinatedTask

func (*ShutdownTaskRequest) ProtoMessage 

func (*ShutdownTaskRequest) ProtoMessage()

func (*ShutdownTaskRequest) ProtoReflect 

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

func (*ShutdownTaskRequest) Reset 

func (x *ShutdownTaskRequest) Reset()

func (*ShutdownTaskRequest) String 

func (x *ShutdownTaskRequest) String() string

type ShutdownTaskResponse 

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

func (*ShutdownTaskResponse) Descriptor deprecated 

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

Deprecated: Use ShutdownTaskResponse.ProtoReflect.Descriptor instead.

func (*ShutdownTaskResponse) ProtoMessage 

func (*ShutdownTaskResponse) ProtoMessage()

func (*ShutdownTaskResponse) ProtoReflect 

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

func (*ShutdownTaskResponse) Reset 

func (x *ShutdownTaskResponse) Reset()

func (*ShutdownTaskResponse) String 

func (x *ShutdownTaskResponse) String() string

type SnapShot 

type SnapShot struct {
	ActionCount uint64 `protobuf:"varint,1,opt,name=action_count,json=actionCount,proto3" json:"action_count,omitempty"`
	Size        int64  `protobuf:"varint,2,opt,name=size,proto3" json:"size,omitempty"`
	// contains filtered or unexported fields
}

func (*SnapShot) Descriptor deprecated 

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

Deprecated: Use SnapShot.ProtoReflect.Descriptor instead.

func (*SnapShot) GetActionCount 

func (x *SnapShot) GetActionCount() uint64

func (*SnapShot) GetSize 

func (x *SnapShot) GetSize() int64

func (*SnapShot) ProtoMessage 

func (*SnapShot) ProtoMessage()

func (*SnapShot) ProtoReflect 

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

func (*SnapShot) Reset 

func (x *SnapShot) Reset()

func (*SnapShot) String 

func (x *SnapShot) String() string

type StatusProto 

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 

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

Deprecated: Use StatusProto.ProtoReflect.Descriptor instead.

func (*StatusProto) GetCode 

func (x *StatusProto) GetCode() Code

func (*StatusProto) GetMessage 

func (x *StatusProto) GetMessage() string

func (*StatusProto) ProtoMessage 

func (*StatusProto) ProtoMessage()

func (*StatusProto) ProtoReflect 

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

func (*StatusProto) Reset 

func (x *StatusProto) Reset()

func (*StatusProto) String 

func (x *StatusProto) String() string

type TensorDescriptorProto 

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 

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

Deprecated: Use TensorDescriptorProto.ProtoReflect.Descriptor instead.

func (*TensorDescriptorProto) GetDataLayout 

func (x *TensorDescriptorProto) GetDataLayout() DataLayout

func (*TensorDescriptorProto) GetDataType 

func (x *TensorDescriptorProto) GetDataType() DataType

func (*TensorDescriptorProto) GetDimensions 

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

func (*TensorDescriptorProto) GetFilterLayout 

func (x *TensorDescriptorProto) GetFilterLayout() FilterLayout

func (*TensorDescriptorProto) GetLayoutOneof 

func (m *TensorDescriptorProto) GetLayoutOneof() isTensorDescriptorProto_LayoutOneof

func (*TensorDescriptorProto) ProtoMessage 

func (*TensorDescriptorProto) ProtoMessage()

func (*TensorDescriptorProto) ProtoReflect 

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

func (*TensorDescriptorProto) Reset 

func (x *TensorDescriptorProto) Reset()

func (*TensorDescriptorProto) String 

func (x *TensorDescriptorProto) String() string

type TensorDescriptorProto_DataLayout 

type TensorDescriptorProto_DataLayout struct {
	DataLayout DataLayout `protobuf:"varint,3,opt,name=data_layout,json=dataLayout,proto3,enum=stream_executor.dnn.DataLayout,oneof"`
}

type TensorDescriptorProto_FilterLayout 

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

type TestResults 

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 

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

Deprecated: Use TestResults.ProtoReflect.Descriptor instead.

func (*TestResults) GetBenchmarkType 

func (x *TestResults) GetBenchmarkType() TestResults_BenchmarkType

func (*TestResults) GetBuildConfiguration 

func (x *TestResults) GetBuildConfiguration() *BuildConfiguration

func (*TestResults) GetCommitId 

func (x *TestResults) GetCommitId() *CommitId

func (*TestResults) GetEntries 

func (x *TestResults) GetEntries() *BenchmarkEntries

func (*TestResults) GetMachineConfiguration 

func (x *TestResults) GetMachineConfiguration() *MachineConfiguration

func (*TestResults) GetName 

func (x *TestResults) GetName() string

func (*TestResults) GetRunConfiguration 

func (x *TestResults) GetRunConfiguration() *RunConfiguration

func (*TestResults) GetRunMode 

func (x *TestResults) GetRunMode() string

func (*TestResults) GetRunTime 

func (x *TestResults) GetRunTime() float64

func (*TestResults) GetStartTime 

func (x *TestResults) GetStartTime() int64

func (*TestResults) GetTarget 

func (x *TestResults) GetTarget() string

func (*TestResults) GetTfVersion 

func (x *TestResults) GetTfVersion() string

func (*TestResults) ProtoMessage 

func (*TestResults) ProtoMessage()

func (*TestResults) ProtoReflect 

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

func (*TestResults) Reset 

func (x *TestResults) Reset()

func (*TestResults) String 

func (x *TestResults) String() string

type TestResults_BenchmarkType 

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 

func (TestResults_BenchmarkType) Descriptor() protoreflect.EnumDescriptor

func (TestResults_BenchmarkType) Enum 

func (x TestResults_BenchmarkType) Enum() *TestResults_BenchmarkType

func (TestResults_BenchmarkType) EnumDescriptor deprecated 

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

Deprecated: Use TestResults_BenchmarkType.Descriptor instead.

func (TestResults_BenchmarkType) Number 

func (x TestResults_BenchmarkType) Number() protoreflect.EnumNumber

func (TestResults_BenchmarkType) String 

func (x TestResults_BenchmarkType) String() string

func (TestResults_BenchmarkType) Type 

func (TestResults_BenchmarkType) Type() protoreflect.EnumType

type TryGetKeyValueRequest 

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

func (*TryGetKeyValueRequest) Descriptor deprecated 

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

Deprecated: Use TryGetKeyValueRequest.ProtoReflect.Descriptor instead.

func (*TryGetKeyValueRequest) GetKey 

func (x *TryGetKeyValueRequest) GetKey() string

func (*TryGetKeyValueRequest) ProtoMessage 

func (*TryGetKeyValueRequest) ProtoMessage()

func (*TryGetKeyValueRequest) ProtoReflect 

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

func (*TryGetKeyValueRequest) Reset 

func (x *TryGetKeyValueRequest) Reset()

func (*TryGetKeyValueRequest) String 

func (x *TryGetKeyValueRequest) String() string

type TryGetKeyValueResponse 

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

func (*TryGetKeyValueResponse) Descriptor deprecated 

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

Deprecated: Use TryGetKeyValueResponse.ProtoReflect.Descriptor instead.

func (*TryGetKeyValueResponse) GetKv 

func (x *TryGetKeyValueResponse) GetKv() *KeyValueEntry

func (*TryGetKeyValueResponse) ProtoMessage 

func (*TryGetKeyValueResponse) ProtoMessage()

func (*TryGetKeyValueResponse) ProtoReflect 

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

func (*TryGetKeyValueResponse) Reset 

func (x *TryGetKeyValueResponse) Reset()

func (*TryGetKeyValueResponse) String 

func (x *TryGetKeyValueResponse) String() string

type UnimplementedCoordinationServiceServer 

type UnimplementedCoordinationServiceServer struct {
}

UnimplementedCoordinationServiceServer must be embedded to have forward compatible implementations.

func (UnimplementedCoordinationServiceServer) Barrier 

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

func (UnimplementedCoordinationServiceServer) CancelBarrier 

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

func (UnimplementedCoordinationServiceServer) DeleteKeyValue 

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

func (UnimplementedCoordinationServiceServer) GetKeyValue 

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

func (UnimplementedCoordinationServiceServer) GetKeyValueDir 

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

func (UnimplementedCoordinationServiceServer) GetTaskState 

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

func (UnimplementedCoordinationServiceServer) Heartbeat 

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

func (UnimplementedCoordinationServiceServer) InsertKeyValue 

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

func (UnimplementedCoordinationServiceServer) RegisterTask 

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

func (UnimplementedCoordinationServiceServer) ReportErrorToService 

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

func (UnimplementedCoordinationServiceServer) ReportErrorToTask 

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

func (UnimplementedCoordinationServiceServer) ResetTask 

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

func (UnimplementedCoordinationServiceServer) ShutdownTask 

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

func (UnimplementedCoordinationServiceServer) TryGetKeyValue 

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

func (UnimplementedCoordinationServiceServer) WaitForAllTasks 

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

type UnsafeCoordinationServiceServer 

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 

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 

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

Deprecated: Use WaitForAllTasksRequest.ProtoReflect.Descriptor instead.

func (*WaitForAllTasksRequest) GetDeviceInfo 

func (x *WaitForAllTasksRequest) GetDeviceInfo() *DeviceInfo

func (*WaitForAllTasksRequest) GetSourceTask 

func (x *WaitForAllTasksRequest) GetSourceTask() *CoordinatedTask

func (*WaitForAllTasksRequest) ProtoMessage 

func (*WaitForAllTasksRequest) ProtoMessage()

func (*WaitForAllTasksRequest) ProtoReflect 

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

func (*WaitForAllTasksRequest) Reset 

func (x *WaitForAllTasksRequest) Reset()

func (*WaitForAllTasksRequest) String 

func (x *WaitForAllTasksRequest) String() string

type WaitForAllTasksResponse 

type WaitForAllTasksResponse struct {
	LeaderIncarnation uint64 `protobuf:"fixed64,1,opt,name=leader_incarnation,json=leaderIncarnation,proto3" json:"leader_incarnation,omitempty"`
	// All devices in the cluster.
	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 

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

Deprecated: Use WaitForAllTasksResponse.ProtoReflect.Descriptor instead.

func (*WaitForAllTasksResponse) GetDeviceInfo 

func (x *WaitForAllTasksResponse) GetDeviceInfo() *DeviceInfo

func (*WaitForAllTasksResponse) GetLeaderIncarnation 

func (x *WaitForAllTasksResponse) GetLeaderIncarnation() uint64

func (*WaitForAllTasksResponse) ProtoMessage 

func (*WaitForAllTasksResponse) ProtoMessage()

func (*WaitForAllTasksResponse) ProtoReflect 

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

func (*WaitForAllTasksResponse) Reset 

func (x *WaitForAllTasksResponse) Reset()

func (*WaitForAllTasksResponse) String 

func (x *WaitForAllTasksResponse) String() string

type WorkerPossiblyRestarted 

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

If included as a payload, this message flags the Status to be a possible outcome of a worker restart. URI: "type.googleapis.com/tensorflow.distributed_runtime.WorkerPossiblyRestarted"

func (*WorkerPossiblyRestarted) Descriptor deprecated 

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

Deprecated: Use WorkerPossiblyRestarted.ProtoReflect.Descriptor instead.

func (*WorkerPossiblyRestarted) ProtoMessage 

func (*WorkerPossiblyRestarted) ProtoMessage()

func (*WorkerPossiblyRestarted) ProtoReflect 

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

func (*WorkerPossiblyRestarted) Reset 

func (x *WorkerPossiblyRestarted) Reset()

func (*WorkerPossiblyRestarted) String 

func (x *WorkerPossiblyRestarted) String() string

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.