for_core_protos_go_proto

package
v0.0.0-...-6e98487 Latest Latest
Warning

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

Go to latest
Published: Jul 2, 2023 License: Apache-2.0 Imports: 10 Imported by: 0

Documentation

Index

Constants

This section is empty.

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 (
	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_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 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 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) Enum

func (CoordinatedTaskState) EnumDescriptor deprecated

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

Deprecated: Use CoordinatedTaskState.Descriptor instead.

func (CoordinatedTaskState) Number

func (CoordinatedTaskState) String

func (x CoordinatedTaskState) String() string

func (CoordinatedTaskState) Type

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 (*CoordinatedTaskStateInfo) GetTask

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.

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 */
	// 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) 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 (*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 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 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 (*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 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 (*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 (*ReportErrorToServiceResponse) Reset

func (x *ReportErrorToServiceResponse) Reset()

func (*ReportErrorToServiceResponse) 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 (*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 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) Enum

func (TestResults_BenchmarkType) EnumDescriptor deprecated

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

Deprecated: Use TestResults_BenchmarkType.Descriptor instead.

func (TestResults_BenchmarkType) Number

func (TestResults_BenchmarkType) String

func (x TestResults_BenchmarkType) String() string

func (TestResults_BenchmarkType) Type

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 (*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) CancelBarrier

func (UnimplementedCoordinationServiceServer) DeleteKeyValue

func (UnimplementedCoordinationServiceServer) GetKeyValue

func (UnimplementedCoordinationServiceServer) GetKeyValueDir

func (UnimplementedCoordinationServiceServer) GetTaskState

func (UnimplementedCoordinationServiceServer) Heartbeat

func (UnimplementedCoordinationServiceServer) InsertKeyValue

func (UnimplementedCoordinationServiceServer) RegisterTask

func (UnimplementedCoordinationServiceServer) ReportErrorToService

func (UnimplementedCoordinationServiceServer) ReportErrorToTask

func (UnimplementedCoordinationServiceServer) ResetTask

func (UnimplementedCoordinationServiceServer) ShutdownTask

func (UnimplementedCoordinationServiceServer) TryGetKeyValue

func (UnimplementedCoordinationServiceServer) WaitForAllTasks

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

Jump to

Keyboard shortcuts

? : This menu
/ : Search site
f or F : Jump to
y or Y : Canonical URL