Versions in this module Expand all Collapse all v1 v1.0.0 May 6, 2024 Changes in this version + const CoordinationService_Barrier_FullMethodName + const CoordinationService_CancelBarrier_FullMethodName + const CoordinationService_DeleteKeyValue_FullMethodName + const CoordinationService_GetKeyValueDir_FullMethodName + const CoordinationService_GetKeyValue_FullMethodName + const CoordinationService_GetTaskState_FullMethodName + const CoordinationService_Heartbeat_FullMethodName + const CoordinationService_InsertKeyValue_FullMethodName + const CoordinationService_RegisterTask_FullMethodName + const CoordinationService_ReportErrorToService_FullMethodName + const CoordinationService_ReportErrorToTask_FullMethodName + const CoordinationService_ResetTask_FullMethodName + const CoordinationService_ShutdownTask_FullMethodName + const CoordinationService_TryGetKeyValue_FullMethodName + const CoordinationService_WaitForAllTasks_FullMethodName + var ActivationMode_name = map[int32]string + var ActivationMode_value = map[string]int32 + var AlgorithmProto_MathType_name = map[int32]string + var AlgorithmProto_MathType_value = map[string]int32 + var Code_name = map[int32]string + var Code_value = map[string]int32 + var ConvolutionKind_name = map[int32]string + var ConvolutionKind_value = map[string]int32 + var ConvolutionMode_name = map[int32]string + var ConvolutionMode_value = map[string]int32 + var CoordinatedTaskState_name = map[int32]string + var CoordinatedTaskState_value = map[string]int32 + var CoordinationService_ServiceDesc = grpc.ServiceDesc + var DataLayout_name = map[int32]string + var DataLayout_value = map[string]int32 + var DataType_name = map[int32]string + var DataType_value = map[string]int32 + var FMHAMaskKind_name = map[int32]string + var FMHAMaskKind_value = map[string]int32 + var File_tsl_protobuf_bfc_memory_map_proto protoreflect.FileDescriptor + var File_tsl_protobuf_coordination_config_proto protoreflect.FileDescriptor + var File_tsl_protobuf_coordination_service_proto protoreflect.FileDescriptor + var File_tsl_protobuf_distributed_runtime_payloads_proto protoreflect.FileDescriptor + var File_tsl_protobuf_dnn_proto protoreflect.FileDescriptor + var File_tsl_protobuf_error_codes_proto protoreflect.FileDescriptor + var File_tsl_protobuf_histogram_proto protoreflect.FileDescriptor + var File_tsl_protobuf_rpc_options_proto protoreflect.FileDescriptor + var File_tsl_protobuf_status_proto protoreflect.FileDescriptor + var File_tsl_protobuf_test_log_proto protoreflect.FileDescriptor + var FilterLayout_name = map[int32]string + var FilterLayout_value = map[string]int32 + var FusedMHAKind_name = map[int32]string + var FusedMHAKind_value = map[string]int32 + var NormKind_name = map[int32]string + var NormKind_value = map[string]int32 + var TestResults_BenchmarkType_name = map[int32]string + var TestResults_BenchmarkType_value = map[string]int32 + func RegisterCoordinationServiceServer(s grpc.ServiceRegistrar, srv CoordinationServiceServer) + type ActivationMode int32 + const ActivationMode_kBandPass + const ActivationMode_kElu + const ActivationMode_kGeluExact + const ActivationMode_kLeakyRelu + const ActivationMode_kNone + const ActivationMode_kRelu + const ActivationMode_kRelu6 + const ActivationMode_kReluX + const ActivationMode_kSigmoid + const ActivationMode_kTanh + func (ActivationMode) Descriptor() protoreflect.EnumDescriptor + func (ActivationMode) EnumDescriptor() ([]byte, []int) + func (ActivationMode) Type() protoreflect.EnumType + func (x ActivationMode) Enum() *ActivationMode + func (x ActivationMode) Number() protoreflect.EnumNumber + func (x ActivationMode) String() string + type AlgorithmConfigProto struct + OptionalAlgorithm isAlgorithmConfigProto_OptionalAlgorithm + OptionalAlgorithmNoScratch isAlgorithmConfigProto_OptionalAlgorithmNoScratch + OptionalScratchSize isAlgorithmConfigProto_OptionalScratchSize + func (*AlgorithmConfigProto) Descriptor() ([]byte, []int) + func (*AlgorithmConfigProto) ProtoMessage() + func (m *AlgorithmConfigProto) GetOptionalAlgorithm() isAlgorithmConfigProto_OptionalAlgorithm + func (m *AlgorithmConfigProto) GetOptionalAlgorithmNoScratch() isAlgorithmConfigProto_OptionalAlgorithmNoScratch + func (m *AlgorithmConfigProto) GetOptionalScratchSize() isAlgorithmConfigProto_OptionalScratchSize + func (x *AlgorithmConfigProto) GetAlgorithm() *AlgorithmProto + func (x *AlgorithmConfigProto) GetAlgorithmNoScratch() *AlgorithmProto + func (x *AlgorithmConfigProto) GetScratchSize() int64 + func (x *AlgorithmConfigProto) ProtoReflect() protoreflect.Message + func (x *AlgorithmConfigProto) Reset() + func (x *AlgorithmConfigProto) String() string + type AlgorithmConfigProto_Algorithm struct + Algorithm *AlgorithmProto + type AlgorithmConfigProto_AlgorithmNoScratch struct + AlgorithmNoScratch *AlgorithmProto + type AlgorithmConfigProto_ScratchSize struct + ScratchSize int64 + type AlgorithmProto struct + AlgoId int64 + IsCudnnFrontend bool + MathType AlgorithmProto_MathType + TuningKnobs map[int64]int64 + WorkspaceSize *wrapperspb.UInt64Value + func (*AlgorithmProto) Descriptor() ([]byte, []int) + func (*AlgorithmProto) ProtoMessage() + func (x *AlgorithmProto) GetAlgoId() int64 + func (x *AlgorithmProto) GetIsCudnnFrontend() bool + func (x *AlgorithmProto) GetMathType() AlgorithmProto_MathType + func (x *AlgorithmProto) GetTuningKnobs() map[int64]int64 + func (x *AlgorithmProto) GetWorkspaceSize() *wrapperspb.UInt64Value + func (x *AlgorithmProto) ProtoReflect() protoreflect.Message + func (x *AlgorithmProto) Reset() + func (x *AlgorithmProto) String() string + type AlgorithmProto_MathType int32 + const AlgorithmProto_DEFAULT_MATH + const AlgorithmProto_TENSOR_OP_MATH + func (AlgorithmProto_MathType) Descriptor() protoreflect.EnumDescriptor + func (AlgorithmProto_MathType) EnumDescriptor() ([]byte, []int) + func (AlgorithmProto_MathType) Type() protoreflect.EnumType + func (x AlgorithmProto_MathType) Enum() *AlgorithmProto_MathType + func (x AlgorithmProto_MathType) Number() protoreflect.EnumNumber + func (x AlgorithmProto_MathType) String() string + type AvailableDeviceInfo struct + MemoryLimit int64 + Name string + PhysicalDescription string + Type string + func (*AvailableDeviceInfo) Descriptor() ([]byte, []int) + func (*AvailableDeviceInfo) ProtoMessage() + func (x *AvailableDeviceInfo) GetMemoryLimit() int64 + func (x *AvailableDeviceInfo) GetName() string + func (x *AvailableDeviceInfo) GetPhysicalDescription() string + func (x *AvailableDeviceInfo) GetType() string + func (x *AvailableDeviceInfo) ProtoReflect() protoreflect.Message + func (x *AvailableDeviceInfo) Reset() + func (x *AvailableDeviceInfo) String() string + type BarrierRequest struct + BarrierId string + BarrierTimeoutInMs int64 + SourceTask *CoordinatedTask + Tasks []*CoordinatedTask + func (*BarrierRequest) Descriptor() ([]byte, []int) + func (*BarrierRequest) ProtoMessage() + func (x *BarrierRequest) GetBarrierId() string + func (x *BarrierRequest) GetBarrierTimeoutInMs() int64 + func (x *BarrierRequest) GetSourceTask() *CoordinatedTask + func (x *BarrierRequest) GetTasks() []*CoordinatedTask + func (x *BarrierRequest) ProtoReflect() protoreflect.Message + func (x *BarrierRequest) Reset() + func (x *BarrierRequest) String() string + type BarrierResponse struct + func (*BarrierResponse) Descriptor() ([]byte, []int) + func (*BarrierResponse) ProtoMessage() + func (x *BarrierResponse) ProtoReflect() protoreflect.Message + func (x *BarrierResponse) Reset() + func (x *BarrierResponse) String() string + type BenchmarkEntries struct + Entry []*BenchmarkEntry + func (*BenchmarkEntries) Descriptor() ([]byte, []int) + func (*BenchmarkEntries) ProtoMessage() + func (x *BenchmarkEntries) GetEntry() []*BenchmarkEntry + func (x *BenchmarkEntries) ProtoReflect() protoreflect.Message + func (x *BenchmarkEntries) Reset() + func (x *BenchmarkEntries) String() string + type BenchmarkEntry struct + CpuTime float64 + Extras map[string]*EntryValue + Iters int64 + Metrics []*MetricEntry + Name string + Throughput float64 + WallTime float64 + func (*BenchmarkEntry) Descriptor() ([]byte, []int) + func (*BenchmarkEntry) ProtoMessage() + func (x *BenchmarkEntry) GetCpuTime() float64 + func (x *BenchmarkEntry) GetExtras() map[string]*EntryValue + func (x *BenchmarkEntry) GetIters() int64 + func (x *BenchmarkEntry) GetMetrics() []*MetricEntry + func (x *BenchmarkEntry) GetName() string + func (x *BenchmarkEntry) GetThroughput() float64 + func (x *BenchmarkEntry) GetWallTime() float64 + func (x *BenchmarkEntry) ProtoReflect() protoreflect.Message + func (x *BenchmarkEntry) Reset() + func (x *BenchmarkEntry) String() string + type BinSummary struct + Bin int32 + TotalBytesInBin int64 + TotalBytesInUse int64 + TotalChunksInBin int64 + TotalChunksInUse int64 + func (*BinSummary) Descriptor() ([]byte, []int) + func (*BinSummary) ProtoMessage() + func (x *BinSummary) GetBin() int32 + func (x *BinSummary) GetTotalBytesInBin() int64 + func (x *BinSummary) GetTotalBytesInUse() int64 + func (x *BinSummary) GetTotalChunksInBin() int64 + func (x *BinSummary) GetTotalChunksInUse() int64 + func (x *BinSummary) ProtoReflect() protoreflect.Message + func (x *BinSummary) Reset() + func (x *BinSummary) String() string + type BuildConfiguration struct + CcFlags []string + Mode string + Opts []string + func (*BuildConfiguration) Descriptor() ([]byte, []int) + func (*BuildConfiguration) ProtoMessage() + func (x *BuildConfiguration) GetCcFlags() []string + func (x *BuildConfiguration) GetMode() string + func (x *BuildConfiguration) GetOpts() []string + func (x *BuildConfiguration) ProtoReflect() protoreflect.Message + func (x *BuildConfiguration) Reset() + func (x *BuildConfiguration) String() string + type CPUInfo struct + CacheSize map[string]int64 + CpuGovernor string + CpuInfo string + MhzPerCpu float64 + NumCores int64 + NumCoresAllowed int64 + func (*CPUInfo) Descriptor() ([]byte, []int) + func (*CPUInfo) ProtoMessage() + func (x *CPUInfo) GetCacheSize() map[string]int64 + func (x *CPUInfo) GetCpuGovernor() string + func (x *CPUInfo) GetCpuInfo() string + func (x *CPUInfo) GetMhzPerCpu() float64 + func (x *CPUInfo) GetNumCores() int64 + func (x *CPUInfo) GetNumCoresAllowed() int64 + func (x *CPUInfo) ProtoReflect() protoreflect.Message + func (x *CPUInfo) Reset() + func (x *CPUInfo) String() string + type CancelBarrierRequest struct + BarrierId string + SourceTask *CoordinatedTask + func (*CancelBarrierRequest) Descriptor() ([]byte, []int) + func (*CancelBarrierRequest) ProtoMessage() + func (x *CancelBarrierRequest) GetBarrierId() string + func (x *CancelBarrierRequest) GetSourceTask() *CoordinatedTask + func (x *CancelBarrierRequest) ProtoReflect() protoreflect.Message + func (x *CancelBarrierRequest) Reset() + func (x *CancelBarrierRequest) String() string + type CancelBarrierResponse struct + func (*CancelBarrierResponse) Descriptor() ([]byte, []int) + func (*CancelBarrierResponse) ProtoMessage() + func (x *CancelBarrierResponse) ProtoReflect() protoreflect.Message + func (x *CancelBarrierResponse) Reset() + func (x *CancelBarrierResponse) String() string + type Code int32 + const Code_ABORTED + const Code_ALREADY_EXISTS + const Code_CANCELLED + const Code_DATA_LOSS + const Code_DEADLINE_EXCEEDED + const Code_DO_NOT_USE_RESERVED_FOR_FUTURE_EXPANSION_USE_DEFAULT_IN_SWITCH_INSTEAD_ + const Code_FAILED_PRECONDITION + const Code_INTERNAL + const Code_INVALID_ARGUMENT + const Code_NOT_FOUND + const Code_OK + const Code_OUT_OF_RANGE + const Code_PERMISSION_DENIED + const Code_RESOURCE_EXHAUSTED + const Code_UNAUTHENTICATED + const Code_UNAVAILABLE + const Code_UNIMPLEMENTED + const Code_UNKNOWN + func (Code) Descriptor() protoreflect.EnumDescriptor + func (Code) EnumDescriptor() ([]byte, []int) + func (Code) Type() protoreflect.EnumType + func (x Code) Enum() *Code + func (x Code) Number() protoreflect.EnumNumber + func (x Code) String() string + type CommitId struct + Kind isCommitId_Kind + PendingChangelist int64 + Snapshot string + func (*CommitId) Descriptor() ([]byte, []int) + func (*CommitId) ProtoMessage() + func (m *CommitId) GetKind() isCommitId_Kind + func (x *CommitId) GetChangelist() int64 + func (x *CommitId) GetHash() string + func (x *CommitId) GetPendingChangelist() int64 + func (x *CommitId) GetSnapshot() string + func (x *CommitId) ProtoReflect() protoreflect.Message + func (x *CommitId) Reset() + func (x *CommitId) String() string + type CommitId_Changelist struct + Changelist int64 + type CommitId_Hash struct + Hash string + type ConvolutionDescriptorProto struct + ComputeMode DataType + ConvolutionMode ConvolutionMode + Dilations []int64 + GroupCount int32 + Name string + Paddings []int64 + Strides []int64 + func (*ConvolutionDescriptorProto) Descriptor() ([]byte, []int) + func (*ConvolutionDescriptorProto) ProtoMessage() + func (x *ConvolutionDescriptorProto) GetComputeMode() DataType + func (x *ConvolutionDescriptorProto) GetConvolutionMode() ConvolutionMode + func (x *ConvolutionDescriptorProto) GetDilations() []int64 + func (x *ConvolutionDescriptorProto) GetGroupCount() int32 + func (x *ConvolutionDescriptorProto) GetName() string + func (x *ConvolutionDescriptorProto) GetPaddings() []int64 + func (x *ConvolutionDescriptorProto) GetStrides() []int64 + func (x *ConvolutionDescriptorProto) ProtoReflect() protoreflect.Message + func (x *ConvolutionDescriptorProto) Reset() + func (x *ConvolutionDescriptorProto) String() string + type ConvolutionKind int32 + const ConvolutionKind_BACKWARD_DATA + const ConvolutionKind_BACKWARD_FILTER + const ConvolutionKind_FORWARD + const ConvolutionKind_FORWARD_BIAS_ACTIVATION + const ConvolutionKind_FORWARD_GRAPH + const ConvolutionKind_INVALID + func (ConvolutionKind) Descriptor() protoreflect.EnumDescriptor + func (ConvolutionKind) EnumDescriptor() ([]byte, []int) + func (ConvolutionKind) Type() protoreflect.EnumType + func (x ConvolutionKind) Enum() *ConvolutionKind + func (x ConvolutionKind) Number() protoreflect.EnumNumber + func (x ConvolutionKind) String() string + type ConvolutionMode int32 + const ConvolutionMode_CONVOLUTION + const ConvolutionMode_CROSS_CORRELATION + func (ConvolutionMode) Descriptor() protoreflect.EnumDescriptor + func (ConvolutionMode) EnumDescriptor() ([]byte, []int) + func (ConvolutionMode) Type() protoreflect.EnumType + func (x ConvolutionMode) Enum() *ConvolutionMode + func (x ConvolutionMode) Number() protoreflect.EnumNumber + func (x ConvolutionMode) String() string + type CoordinatedJob struct + Name string + NumTasks int32 + func (*CoordinatedJob) Descriptor() ([]byte, []int) + func (*CoordinatedJob) ProtoMessage() + func (x *CoordinatedJob) GetName() string + func (x *CoordinatedJob) GetNumTasks() int32 + func (x *CoordinatedJob) ProtoReflect() protoreflect.Message + func (x *CoordinatedJob) Reset() + func (x *CoordinatedJob) String() string + type CoordinatedTask struct + JobName string + TaskId int32 + func (*CoordinatedTask) Descriptor() ([]byte, []int) + func (*CoordinatedTask) ProtoMessage() + func (x *CoordinatedTask) GetJobName() string + func (x *CoordinatedTask) GetTaskId() int32 + func (x *CoordinatedTask) ProtoReflect() protoreflect.Message + func (x *CoordinatedTask) Reset() + func (x *CoordinatedTask) String() string + type CoordinatedTaskState int32 + const CoordinatedTaskState_TASKSTATE_CONNECTED + const CoordinatedTaskState_TASKSTATE_DISCONNECTED + const CoordinatedTaskState_TASKSTATE_ERROR + const CoordinatedTaskState_TASKSTATE_UNINITIALIZED + const CoordinatedTaskState_TASKSTATE_UNSPECIFIED + func (CoordinatedTaskState) Descriptor() protoreflect.EnumDescriptor + func (CoordinatedTaskState) EnumDescriptor() ([]byte, []int) + func (CoordinatedTaskState) Type() protoreflect.EnumType + func (x CoordinatedTaskState) Enum() *CoordinatedTaskState + func (x CoordinatedTaskState) Number() protoreflect.EnumNumber + func (x CoordinatedTaskState) String() string + type CoordinatedTaskStateInfo struct + ErrorCode int32 + ErrorMessage string + ErrorPayload *CoordinationServiceError + State CoordinatedTaskState + Task *CoordinatedTask + func (*CoordinatedTaskStateInfo) Descriptor() ([]byte, []int) + func (*CoordinatedTaskStateInfo) ProtoMessage() + func (x *CoordinatedTaskStateInfo) GetErrorCode() int32 + func (x *CoordinatedTaskStateInfo) GetErrorMessage() string + func (x *CoordinatedTaskStateInfo) GetErrorPayload() *CoordinationServiceError + func (x *CoordinatedTaskStateInfo) GetState() CoordinatedTaskState + func (x *CoordinatedTaskStateInfo) GetTask() *CoordinatedTask + func (x *CoordinatedTaskStateInfo) ProtoReflect() protoreflect.Message + func (x *CoordinatedTaskStateInfo) Reset() + func (x *CoordinatedTaskStateInfo) String() string + type CoordinationServiceClient interface + Barrier func(ctx context.Context, in *BarrierRequest, opts ...grpc.CallOption) (*BarrierResponse, error) + CancelBarrier func(ctx context.Context, in *CancelBarrierRequest, opts ...grpc.CallOption) (*CancelBarrierResponse, error) + DeleteKeyValue func(ctx context.Context, in *DeleteKeyValueRequest, opts ...grpc.CallOption) (*DeleteKeyValueResponse, error) + GetKeyValue func(ctx context.Context, in *GetKeyValueRequest, opts ...grpc.CallOption) (*GetKeyValueResponse, error) + GetKeyValueDir func(ctx context.Context, in *GetKeyValueDirRequest, opts ...grpc.CallOption) (*GetKeyValueDirResponse, error) + GetTaskState func(ctx context.Context, in *GetTaskStateRequest, opts ...grpc.CallOption) (*GetTaskStateResponse, error) + Heartbeat func(ctx context.Context, in *HeartbeatRequest, opts ...grpc.CallOption) (*HeartbeatResponse, error) + InsertKeyValue func(ctx context.Context, in *InsertKeyValueRequest, opts ...grpc.CallOption) (*InsertKeyValueResponse, error) + RegisterTask func(ctx context.Context, in *RegisterTaskRequest, opts ...grpc.CallOption) (*RegisterTaskResponse, error) + ReportErrorToService func(ctx context.Context, in *ReportErrorToServiceRequest, opts ...grpc.CallOption) (*ReportErrorToServiceResponse, error) + ReportErrorToTask func(ctx context.Context, in *ReportErrorToTaskRequest, opts ...grpc.CallOption) (*ReportErrorToTaskResponse, error) + ResetTask func(ctx context.Context, in *ResetTaskRequest, opts ...grpc.CallOption) (*ResetTaskResponse, error) + ShutdownTask func(ctx context.Context, in *ShutdownTaskRequest, opts ...grpc.CallOption) (*ShutdownTaskResponse, error) + TryGetKeyValue func(ctx context.Context, in *TryGetKeyValueRequest, opts ...grpc.CallOption) (*TryGetKeyValueResponse, error) + WaitForAllTasks func(ctx context.Context, in *WaitForAllTasksRequest, opts ...grpc.CallOption) (*WaitForAllTasksResponse, error) + func NewCoordinationServiceClient(cc grpc.ClientConnInterface) CoordinationServiceClient + type CoordinationServiceConfig struct + AgentDestructionWithoutShutdown bool + AllowNewIncarnationToReconnect bool + ClusterRegisterTimeoutInMs int64 + CoordinatedJobList []*CoordinatedJob + EnableHealthCheck bool + ForceDisable bool + HeartbeatTimeoutInMs int64 + RecoverableJobs []string + ServiceLeader string + ServiceType string + ShutdownBarrierTimeoutInMs int64 + func (*CoordinationServiceConfig) Descriptor() ([]byte, []int) + func (*CoordinationServiceConfig) ProtoMessage() + func (x *CoordinationServiceConfig) GetAgentDestructionWithoutShutdown() bool + func (x *CoordinationServiceConfig) GetAllowNewIncarnationToReconnect() bool + func (x *CoordinationServiceConfig) GetClusterRegisterTimeoutInMs() int64 + func (x *CoordinationServiceConfig) GetCoordinatedJobList() []*CoordinatedJob + func (x *CoordinationServiceConfig) GetEnableHealthCheck() bool + func (x *CoordinationServiceConfig) GetForceDisable() bool + func (x *CoordinationServiceConfig) GetHeartbeatTimeoutInMs() int64 + func (x *CoordinationServiceConfig) GetRecoverableJobs() []string + func (x *CoordinationServiceConfig) GetServiceLeader() string + func (x *CoordinationServiceConfig) GetServiceType() string + func (x *CoordinationServiceConfig) GetShutdownBarrierTimeoutInMs() int64 + func (x *CoordinationServiceConfig) ProtoReflect() protoreflect.Message + func (x *CoordinationServiceConfig) Reset() + func (x *CoordinationServiceConfig) String() string + type CoordinationServiceError struct + IsReportedError bool + SourceTask *CoordinatedTask + func (*CoordinationServiceError) Descriptor() ([]byte, []int) + func (*CoordinationServiceError) ProtoMessage() + func (x *CoordinationServiceError) GetIsReportedError() bool + func (x *CoordinationServiceError) GetSourceTask() *CoordinatedTask + func (x *CoordinationServiceError) ProtoReflect() protoreflect.Message + func (x *CoordinationServiceError) Reset() + func (x *CoordinationServiceError) String() string + type CoordinationServiceServer interface + Barrier func(context.Context, *BarrierRequest) (*BarrierResponse, error) + CancelBarrier func(context.Context, *CancelBarrierRequest) (*CancelBarrierResponse, error) + DeleteKeyValue func(context.Context, *DeleteKeyValueRequest) (*DeleteKeyValueResponse, error) + GetKeyValue func(context.Context, *GetKeyValueRequest) (*GetKeyValueResponse, error) + GetKeyValueDir func(context.Context, *GetKeyValueDirRequest) (*GetKeyValueDirResponse, error) + GetTaskState func(context.Context, *GetTaskStateRequest) (*GetTaskStateResponse, error) + Heartbeat func(context.Context, *HeartbeatRequest) (*HeartbeatResponse, error) + InsertKeyValue func(context.Context, *InsertKeyValueRequest) (*InsertKeyValueResponse, error) + RegisterTask func(context.Context, *RegisterTaskRequest) (*RegisterTaskResponse, error) + ReportErrorToService func(context.Context, *ReportErrorToServiceRequest) (*ReportErrorToServiceResponse, error) + ReportErrorToTask func(context.Context, *ReportErrorToTaskRequest) (*ReportErrorToTaskResponse, error) + ResetTask func(context.Context, *ResetTaskRequest) (*ResetTaskResponse, error) + ShutdownTask func(context.Context, *ShutdownTaskRequest) (*ShutdownTaskResponse, error) + TryGetKeyValue func(context.Context, *TryGetKeyValueRequest) (*TryGetKeyValueResponse, error) + WaitForAllTasks func(context.Context, *WaitForAllTasksRequest) (*WaitForAllTasksResponse, error) + type DataLayout int32 + const DataLayout_kBatchDepthYX + const DataLayout_kBatchDepthYX32 + const DataLayout_kBatchDepthYX4 + const DataLayout_kBatchYXDepth + const DataLayout_kYXBatchDepth + const DataLayout_kYXDepthBatch + func (DataLayout) Descriptor() protoreflect.EnumDescriptor + func (DataLayout) EnumDescriptor() ([]byte, []int) + func (DataLayout) Type() protoreflect.EnumType + func (x DataLayout) Enum() *DataLayout + func (x DataLayout) Number() protoreflect.EnumNumber + func (x DataLayout) String() string + type DataType int32 + const DataType_kBF16 + const DataType_kComplexDouble + const DataType_kComplexFloat + const DataType_kDouble + const DataType_kF8E4M3FN + const DataType_kF8E4M3FNUZ + const DataType_kF8E5M2 + const DataType_kF8E5M2FNUZ + const DataType_kFloat + const DataType_kHalf + const DataType_kInt32 + const DataType_kInt64 + const DataType_kInt8 + func (DataType) Descriptor() protoreflect.EnumDescriptor + func (DataType) EnumDescriptor() ([]byte, []int) + func (DataType) Type() protoreflect.EnumType + func (x DataType) Enum() *DataType + func (x DataType) Number() protoreflect.EnumNumber + func (x DataType) String() string + type DeleteKeyValueRequest struct + IsDirectory bool + Key string + func (*DeleteKeyValueRequest) Descriptor() ([]byte, []int) + func (*DeleteKeyValueRequest) ProtoMessage() + func (x *DeleteKeyValueRequest) GetIsDirectory() bool + func (x *DeleteKeyValueRequest) GetKey() string + func (x *DeleteKeyValueRequest) ProtoReflect() protoreflect.Message + func (x *DeleteKeyValueRequest) Reset() + func (x *DeleteKeyValueRequest) String() string + type DeleteKeyValueResponse struct + func (*DeleteKeyValueResponse) Descriptor() ([]byte, []int) + func (*DeleteKeyValueResponse) ProtoMessage() + func (x *DeleteKeyValueResponse) ProtoReflect() protoreflect.Message + func (x *DeleteKeyValueResponse) Reset() + func (x *DeleteKeyValueResponse) String() string + type DeviceInfo struct + Device []*anypb.Any + func (*DeviceInfo) Descriptor() ([]byte, []int) + func (*DeviceInfo) ProtoMessage() + func (x *DeviceInfo) GetDevice() []*anypb.Any + func (x *DeviceInfo) ProtoReflect() protoreflect.Message + func (x *DeviceInfo) Reset() + func (x *DeviceInfo) String() string + type EntryValue struct + Kind isEntryValue_Kind + func (*EntryValue) Descriptor() ([]byte, []int) + func (*EntryValue) ProtoMessage() + func (m *EntryValue) GetKind() isEntryValue_Kind + func (x *EntryValue) GetDoubleValue() float64 + func (x *EntryValue) GetStringValue() string + func (x *EntryValue) ProtoReflect() protoreflect.Message + func (x *EntryValue) Reset() + func (x *EntryValue) String() string + type EntryValue_DoubleValue struct + DoubleValue float64 + type EntryValue_StringValue struct + StringValue string + type FMHAMaskKind int32 + const FMHAMaskKind_ALIBI + const FMHAMaskKind_CAUSAL + const FMHAMaskKind_NO_MASK + const FMHAMaskKind_PADDING + const FMHAMaskKind_PADDING_CAUSAL + func (FMHAMaskKind) Descriptor() protoreflect.EnumDescriptor + func (FMHAMaskKind) EnumDescriptor() ([]byte, []int) + func (FMHAMaskKind) Type() protoreflect.EnumType + func (x FMHAMaskKind) Enum() *FMHAMaskKind + func (x FMHAMaskKind) Number() protoreflect.EnumNumber + func (x FMHAMaskKind) String() string + type FilterLayout int32 + const FilterLayout_kInputYXOutput + const FilterLayout_kOutputInputYX + const FilterLayout_kOutputInputYX32 + const FilterLayout_kOutputInputYX32_CudnnReordered + const FilterLayout_kOutputInputYX4 + const FilterLayout_kOutputYXInput + const FilterLayout_kYXInputOutput + func (FilterLayout) Descriptor() protoreflect.EnumDescriptor + func (FilterLayout) EnumDescriptor() ([]byte, []int) + func (FilterLayout) Type() protoreflect.EnumType + func (x FilterLayout) Enum() *FilterLayout + func (x FilterLayout) Number() protoreflect.EnumNumber + func (x FilterLayout) String() string + type FusedMHAKind int32 + const FusedMHAKind_BMM1_OUTPUT_FLOAT + const FusedMHAKind_BMM1_OUTPUT_INPUT_TYPE + const FusedMHAKind_BMM1_OUTPUT_UNKNOWN + func (FusedMHAKind) Descriptor() protoreflect.EnumDescriptor + func (FusedMHAKind) EnumDescriptor() ([]byte, []int) + func (FusedMHAKind) Type() protoreflect.EnumType + func (x FusedMHAKind) Enum() *FusedMHAKind + func (x FusedMHAKind) Number() protoreflect.EnumNumber + func (x FusedMHAKind) String() string + type GPUInfo struct + BusId string + Model string + Uuid string + func (*GPUInfo) Descriptor() ([]byte, []int) + func (*GPUInfo) ProtoMessage() + func (x *GPUInfo) GetBusId() string + func (x *GPUInfo) GetModel() string + func (x *GPUInfo) GetUuid() string + func (x *GPUInfo) ProtoReflect() protoreflect.Message + func (x *GPUInfo) Reset() + func (x *GPUInfo) String() string + type GetKeyValueDirRequest struct + DirectoryKey string + func (*GetKeyValueDirRequest) Descriptor() ([]byte, []int) + func (*GetKeyValueDirRequest) ProtoMessage() + func (x *GetKeyValueDirRequest) GetDirectoryKey() string + func (x *GetKeyValueDirRequest) ProtoReflect() protoreflect.Message + func (x *GetKeyValueDirRequest) Reset() + func (x *GetKeyValueDirRequest) String() string + type GetKeyValueDirResponse struct + DirectoryKey string + Kv []*KeyValueEntry + func (*GetKeyValueDirResponse) Descriptor() ([]byte, []int) + func (*GetKeyValueDirResponse) ProtoMessage() + func (x *GetKeyValueDirResponse) GetDirectoryKey() string + func (x *GetKeyValueDirResponse) GetKv() []*KeyValueEntry + func (x *GetKeyValueDirResponse) ProtoReflect() protoreflect.Message + func (x *GetKeyValueDirResponse) Reset() + func (x *GetKeyValueDirResponse) String() string + type GetKeyValueRequest struct + Key string + func (*GetKeyValueRequest) Descriptor() ([]byte, []int) + func (*GetKeyValueRequest) ProtoMessage() + func (x *GetKeyValueRequest) GetKey() string + func (x *GetKeyValueRequest) ProtoReflect() protoreflect.Message + func (x *GetKeyValueRequest) Reset() + func (x *GetKeyValueRequest) String() string + type GetKeyValueResponse struct + Kv *KeyValueEntry + func (*GetKeyValueResponse) Descriptor() ([]byte, []int) + func (*GetKeyValueResponse) ProtoMessage() + func (x *GetKeyValueResponse) GetKv() *KeyValueEntry + func (x *GetKeyValueResponse) ProtoReflect() protoreflect.Message + func (x *GetKeyValueResponse) Reset() + func (x *GetKeyValueResponse) String() string + type GetTaskStateRequest struct + SourceTask []*CoordinatedTask + func (*GetTaskStateRequest) Descriptor() ([]byte, []int) + func (*GetTaskStateRequest) ProtoMessage() + func (x *GetTaskStateRequest) GetSourceTask() []*CoordinatedTask + func (x *GetTaskStateRequest) ProtoReflect() protoreflect.Message + func (x *GetTaskStateRequest) Reset() + func (x *GetTaskStateRequest) String() string + type GetTaskStateResponse struct + TaskState []*CoordinatedTaskStateInfo + func (*GetTaskStateResponse) Descriptor() ([]byte, []int) + func (*GetTaskStateResponse) ProtoMessage() + func (x *GetTaskStateResponse) GetTaskState() []*CoordinatedTaskStateInfo + func (x *GetTaskStateResponse) ProtoReflect() protoreflect.Message + func (x *GetTaskStateResponse) Reset() + func (x *GetTaskStateResponse) String() string + type GrpcPayloadContainer struct + Payloads map[string][]byte + func (*GrpcPayloadContainer) Descriptor() ([]byte, []int) + func (*GrpcPayloadContainer) ProtoMessage() + func (x *GrpcPayloadContainer) GetPayloads() map[string][]byte + func (x *GrpcPayloadContainer) ProtoReflect() protoreflect.Message + func (x *GrpcPayloadContainer) Reset() + func (x *GrpcPayloadContainer) String() string + type GrpcPayloadsLost struct + func (*GrpcPayloadsLost) Descriptor() ([]byte, []int) + func (*GrpcPayloadsLost) ProtoMessage() + func (x *GrpcPayloadsLost) ProtoReflect() protoreflect.Message + func (x *GrpcPayloadsLost) Reset() + func (x *GrpcPayloadsLost) String() string + type HeartbeatRequest struct + Incarnation uint64 + SourceTask *CoordinatedTask + func (*HeartbeatRequest) Descriptor() ([]byte, []int) + func (*HeartbeatRequest) ProtoMessage() + func (x *HeartbeatRequest) GetIncarnation() uint64 + func (x *HeartbeatRequest) GetSourceTask() *CoordinatedTask + func (x *HeartbeatRequest) ProtoReflect() protoreflect.Message + func (x *HeartbeatRequest) Reset() + func (x *HeartbeatRequest) String() string + type HeartbeatResponse struct + LeaderIncarnation uint64 + func (*HeartbeatResponse) Descriptor() ([]byte, []int) + func (*HeartbeatResponse) ProtoMessage() + func (x *HeartbeatResponse) GetLeaderIncarnation() uint64 + func (x *HeartbeatResponse) ProtoReflect() protoreflect.Message + func (x *HeartbeatResponse) Reset() + func (x *HeartbeatResponse) String() string + type HistogramProto struct + Bucket []float64 + BucketLimit []float64 + Max float64 + Min float64 + Num float64 + Sum float64 + SumSquares float64 + func (*HistogramProto) Descriptor() ([]byte, []int) + func (*HistogramProto) ProtoMessage() + func (x *HistogramProto) GetBucket() []float64 + func (x *HistogramProto) GetBucketLimit() []float64 + func (x *HistogramProto) GetMax() float64 + func (x *HistogramProto) GetMin() float64 + func (x *HistogramProto) GetNum() float64 + func (x *HistogramProto) GetSum() float64 + func (x *HistogramProto) GetSumSquares() float64 + func (x *HistogramProto) ProtoReflect() protoreflect.Message + func (x *HistogramProto) Reset() + func (x *HistogramProto) String() string + type InsertKeyValueRequest struct + Kv *KeyValueEntry + func (*InsertKeyValueRequest) Descriptor() ([]byte, []int) + func (*InsertKeyValueRequest) ProtoMessage() + func (x *InsertKeyValueRequest) GetKv() *KeyValueEntry + func (x *InsertKeyValueRequest) ProtoReflect() protoreflect.Message + func (x *InsertKeyValueRequest) Reset() + func (x *InsertKeyValueRequest) String() string + type InsertKeyValueResponse struct + func (*InsertKeyValueResponse) Descriptor() ([]byte, []int) + func (*InsertKeyValueResponse) ProtoMessage() + func (x *InsertKeyValueResponse) ProtoReflect() protoreflect.Message + func (x *InsertKeyValueResponse) Reset() + func (x *InsertKeyValueResponse) String() string + type KeyValueEntry struct + Key string + Value []byte + func (*KeyValueEntry) Descriptor() ([]byte, []int) + func (*KeyValueEntry) ProtoMessage() + func (x *KeyValueEntry) GetKey() string + func (x *KeyValueEntry) GetValue() []byte + func (x *KeyValueEntry) ProtoReflect() protoreflect.Message + func (x *KeyValueEntry) Reset() + func (x *KeyValueEntry) String() string + type MachineConfiguration struct + AvailableDeviceInfo []*AvailableDeviceInfo + CpuInfo *CPUInfo + DeviceInfo []*anypb.Any + Hostname string + MemoryInfo *MemoryInfo + PlatformInfo *PlatformInfo + SerialIdentifier string + func (*MachineConfiguration) Descriptor() ([]byte, []int) + func (*MachineConfiguration) ProtoMessage() + func (x *MachineConfiguration) GetAvailableDeviceInfo() []*AvailableDeviceInfo + func (x *MachineConfiguration) GetCpuInfo() *CPUInfo + func (x *MachineConfiguration) GetDeviceInfo() []*anypb.Any + func (x *MachineConfiguration) GetHostname() string + func (x *MachineConfiguration) GetMemoryInfo() *MemoryInfo + func (x *MachineConfiguration) GetPlatformInfo() *PlatformInfo + func (x *MachineConfiguration) GetSerialIdentifier() string + func (x *MachineConfiguration) ProtoReflect() protoreflect.Message + func (x *MachineConfiguration) Reset() + func (x *MachineConfiguration) String() string + type MemAllocatorStats struct + BytesInUse int64 + FragmentationMetric float32 + LargestAllocSize int64 + NumAllocs int64 + PeakBytesInUse int64 + func (*MemAllocatorStats) Descriptor() ([]byte, []int) + func (*MemAllocatorStats) ProtoMessage() + func (x *MemAllocatorStats) GetBytesInUse() int64 + func (x *MemAllocatorStats) GetFragmentationMetric() float32 + func (x *MemAllocatorStats) GetLargestAllocSize() int64 + func (x *MemAllocatorStats) GetNumAllocs() int64 + func (x *MemAllocatorStats) GetPeakBytesInUse() int64 + func (x *MemAllocatorStats) ProtoReflect() protoreflect.Message + func (x *MemAllocatorStats) Reset() + func (x *MemAllocatorStats) String() string + type MemChunk struct + ActionCount uint64 + Address uint64 + Bin int32 + FreedAtCount uint64 + InUse bool + OpName string + RequestedSize int64 + Size int64 + StepId uint64 + func (*MemChunk) Descriptor() ([]byte, []int) + func (*MemChunk) ProtoMessage() + func (x *MemChunk) GetActionCount() uint64 + func (x *MemChunk) GetAddress() uint64 + func (x *MemChunk) GetBin() int32 + func (x *MemChunk) GetFreedAtCount() uint64 + func (x *MemChunk) GetInUse() bool + func (x *MemChunk) GetOpName() string + func (x *MemChunk) GetRequestedSize() int64 + func (x *MemChunk) GetSize() int64 + func (x *MemChunk) GetStepId() uint64 + func (x *MemChunk) ProtoReflect() protoreflect.Message + func (x *MemChunk) Reset() + func (x *MemChunk) String() string + type MemoryDump struct + AllocatorName string + BinSummary []*BinSummary + Chunk []*MemChunk + SnapShot []*SnapShot + Stats *MemAllocatorStats + func (*MemoryDump) Descriptor() ([]byte, []int) + func (*MemoryDump) ProtoMessage() + func (x *MemoryDump) GetAllocatorName() string + func (x *MemoryDump) GetBinSummary() []*BinSummary + func (x *MemoryDump) GetChunk() []*MemChunk + func (x *MemoryDump) GetSnapShot() []*SnapShot + func (x *MemoryDump) GetStats() *MemAllocatorStats + func (x *MemoryDump) ProtoReflect() protoreflect.Message + func (x *MemoryDump) Reset() + func (x *MemoryDump) String() string + type MemoryInfo struct + Available int64 + Total int64 + func (*MemoryInfo) Descriptor() ([]byte, []int) + func (*MemoryInfo) ProtoMessage() + func (x *MemoryInfo) GetAvailable() int64 + func (x *MemoryInfo) GetTotal() int64 + func (x *MemoryInfo) ProtoReflect() protoreflect.Message + func (x *MemoryInfo) Reset() + func (x *MemoryInfo) String() string + type MetricEntry struct + MaxValue *wrapperspb.DoubleValue + MinValue *wrapperspb.DoubleValue + Name string + Value float64 + func (*MetricEntry) Descriptor() ([]byte, []int) + func (*MetricEntry) ProtoMessage() + func (x *MetricEntry) GetMaxValue() *wrapperspb.DoubleValue + func (x *MetricEntry) GetMinValue() *wrapperspb.DoubleValue + func (x *MetricEntry) GetName() string + func (x *MetricEntry) GetValue() float64 + func (x *MetricEntry) ProtoReflect() protoreflect.Message + func (x *MetricEntry) Reset() + func (x *MetricEntry) String() string + type NormKind int32 + const NormKind_LAYER_BWD + const NormKind_LAYER_FWD_INFER + const NormKind_LAYER_FWD_TRAIN + func (NormKind) Descriptor() protoreflect.EnumDescriptor + func (NormKind) EnumDescriptor() ([]byte, []int) + func (NormKind) Type() protoreflect.EnumType + func (x NormKind) Enum() *NormKind + func (x NormKind) Number() protoreflect.EnumNumber + func (x NormKind) String() string + type PlatformInfo struct + Bits string + Linkage string + Machine string + Release string + System string + Version string + func (*PlatformInfo) Descriptor() ([]byte, []int) + func (*PlatformInfo) ProtoMessage() + func (x *PlatformInfo) GetBits() string + func (x *PlatformInfo) GetLinkage() string + func (x *PlatformInfo) GetMachine() string + func (x *PlatformInfo) GetRelease() string + func (x *PlatformInfo) GetSystem() string + func (x *PlatformInfo) GetVersion() string + func (x *PlatformInfo) ProtoReflect() protoreflect.Message + func (x *PlatformInfo) Reset() + func (x *PlatformInfo) String() string + type RPCOptions struct + CacheRpcResponse bool + CompressionAlgorithm string + CompressionLevel int32 + DisableSessionConnectionSharing bool + NumChannelsPerTarget int32 + UseRpcForInprocessMaster bool + func (*RPCOptions) Descriptor() ([]byte, []int) + func (*RPCOptions) ProtoMessage() + func (x *RPCOptions) GetCacheRpcResponse() bool + func (x *RPCOptions) GetCompressionAlgorithm() string + func (x *RPCOptions) GetCompressionLevel() int32 + func (x *RPCOptions) GetDisableSessionConnectionSharing() bool + func (x *RPCOptions) GetNumChannelsPerTarget() int32 + func (x *RPCOptions) GetUseRpcForInprocessMaster() bool + func (x *RPCOptions) ProtoReflect() protoreflect.Message + func (x *RPCOptions) Reset() + func (x *RPCOptions) String() string + type RegisterTaskRequest struct + Incarnation uint64 + SourceTask *CoordinatedTask + func (*RegisterTaskRequest) Descriptor() ([]byte, []int) + func (*RegisterTaskRequest) ProtoMessage() + func (x *RegisterTaskRequest) GetIncarnation() uint64 + func (x *RegisterTaskRequest) GetSourceTask() *CoordinatedTask + func (x *RegisterTaskRequest) ProtoReflect() protoreflect.Message + func (x *RegisterTaskRequest) Reset() + func (x *RegisterTaskRequest) String() string + type RegisterTaskResponse struct + LeaderIncarnation uint64 + func (*RegisterTaskResponse) Descriptor() ([]byte, []int) + func (*RegisterTaskResponse) ProtoMessage() + func (x *RegisterTaskResponse) GetLeaderIncarnation() uint64 + func (x *RegisterTaskResponse) ProtoReflect() protoreflect.Message + func (x *RegisterTaskResponse) Reset() + func (x *RegisterTaskResponse) String() string + type ReportErrorToServiceRequest struct + ErrorCode int32 + ErrorMessage string + ErrorOrigin *CoordinatedTask + func (*ReportErrorToServiceRequest) Descriptor() ([]byte, []int) + func (*ReportErrorToServiceRequest) ProtoMessage() + func (x *ReportErrorToServiceRequest) GetErrorCode() int32 + func (x *ReportErrorToServiceRequest) GetErrorMessage() string + func (x *ReportErrorToServiceRequest) GetErrorOrigin() *CoordinatedTask + func (x *ReportErrorToServiceRequest) ProtoReflect() protoreflect.Message + func (x *ReportErrorToServiceRequest) Reset() + func (x *ReportErrorToServiceRequest) String() string + type ReportErrorToServiceResponse struct + func (*ReportErrorToServiceResponse) Descriptor() ([]byte, []int) + func (*ReportErrorToServiceResponse) ProtoMessage() + func (x *ReportErrorToServiceResponse) ProtoReflect() protoreflect.Message + func (x *ReportErrorToServiceResponse) Reset() + func (x *ReportErrorToServiceResponse) String() string + type ReportErrorToTaskRequest struct + ErrorCode int32 + ErrorMessage string + ErrorPayload *CoordinationServiceError + func (*ReportErrorToTaskRequest) Descriptor() ([]byte, []int) + func (*ReportErrorToTaskRequest) ProtoMessage() + func (x *ReportErrorToTaskRequest) GetErrorCode() int32 + func (x *ReportErrorToTaskRequest) GetErrorMessage() string + func (x *ReportErrorToTaskRequest) GetErrorPayload() *CoordinationServiceError + func (x *ReportErrorToTaskRequest) ProtoReflect() protoreflect.Message + func (x *ReportErrorToTaskRequest) Reset() + func (x *ReportErrorToTaskRequest) String() string + type ReportErrorToTaskResponse struct + func (*ReportErrorToTaskResponse) Descriptor() ([]byte, []int) + func (*ReportErrorToTaskResponse) ProtoMessage() + func (x *ReportErrorToTaskResponse) ProtoReflect() protoreflect.Message + func (x *ReportErrorToTaskResponse) Reset() + func (x *ReportErrorToTaskResponse) String() string + type ResetTaskRequest struct + SourceTask *CoordinatedTask + func (*ResetTaskRequest) Descriptor() ([]byte, []int) + func (*ResetTaskRequest) ProtoMessage() + func (x *ResetTaskRequest) GetSourceTask() *CoordinatedTask + func (x *ResetTaskRequest) ProtoReflect() protoreflect.Message + func (x *ResetTaskRequest) Reset() + func (x *ResetTaskRequest) String() string + type ResetTaskResponse struct + func (*ResetTaskResponse) Descriptor() ([]byte, []int) + func (*ResetTaskResponse) ProtoMessage() + func (x *ResetTaskResponse) ProtoReflect() protoreflect.Message + func (x *ResetTaskResponse) Reset() + func (x *ResetTaskResponse) String() string + type RunConfiguration struct + Argument []string + EnvVars map[string]string + func (*RunConfiguration) Descriptor() ([]byte, []int) + func (*RunConfiguration) ProtoMessage() + func (x *RunConfiguration) GetArgument() []string + func (x *RunConfiguration) GetEnvVars() map[string]string + func (x *RunConfiguration) ProtoReflect() protoreflect.Message + func (x *RunConfiguration) Reset() + func (x *RunConfiguration) String() string + type ShutdownTaskRequest struct + SourceTask *CoordinatedTask + func (*ShutdownTaskRequest) Descriptor() ([]byte, []int) + func (*ShutdownTaskRequest) ProtoMessage() + func (x *ShutdownTaskRequest) GetSourceTask() *CoordinatedTask + func (x *ShutdownTaskRequest) ProtoReflect() protoreflect.Message + func (x *ShutdownTaskRequest) Reset() + func (x *ShutdownTaskRequest) String() string + type ShutdownTaskResponse struct + func (*ShutdownTaskResponse) Descriptor() ([]byte, []int) + func (*ShutdownTaskResponse) ProtoMessage() + func (x *ShutdownTaskResponse) ProtoReflect() protoreflect.Message + func (x *ShutdownTaskResponse) Reset() + func (x *ShutdownTaskResponse) String() string + type SnapShot struct + ActionCount uint64 + Size int64 + func (*SnapShot) Descriptor() ([]byte, []int) + func (*SnapShot) ProtoMessage() + func (x *SnapShot) GetActionCount() uint64 + func (x *SnapShot) GetSize() int64 + func (x *SnapShot) ProtoReflect() protoreflect.Message + func (x *SnapShot) Reset() + func (x *SnapShot) String() string + type StatusProto struct + Code Code + Message string + func (*StatusProto) Descriptor() ([]byte, []int) + func (*StatusProto) ProtoMessage() + func (x *StatusProto) GetCode() Code + func (x *StatusProto) GetMessage() string + func (x *StatusProto) ProtoReflect() protoreflect.Message + func (x *StatusProto) Reset() + func (x *StatusProto) String() string + type TensorDescriptorProto struct + DataType DataType + Dimensions []int64 + LayoutOneof isTensorDescriptorProto_LayoutOneof + func (*TensorDescriptorProto) Descriptor() ([]byte, []int) + func (*TensorDescriptorProto) ProtoMessage() + func (m *TensorDescriptorProto) GetLayoutOneof() isTensorDescriptorProto_LayoutOneof + func (x *TensorDescriptorProto) GetDataLayout() DataLayout + func (x *TensorDescriptorProto) GetDataType() DataType + func (x *TensorDescriptorProto) GetDimensions() []int64 + func (x *TensorDescriptorProto) GetFilterLayout() FilterLayout + func (x *TensorDescriptorProto) ProtoReflect() protoreflect.Message + func (x *TensorDescriptorProto) Reset() + func (x *TensorDescriptorProto) String() string + type TensorDescriptorProto_DataLayout struct + DataLayout DataLayout + type TensorDescriptorProto_FilterLayout struct + FilterLayout FilterLayout + type TestResults struct + BenchmarkType TestResults_BenchmarkType + BuildConfiguration *BuildConfiguration + CommitId *CommitId + Entries *BenchmarkEntries + MachineConfiguration *MachineConfiguration + Name string + RunConfiguration *RunConfiguration + RunMode string + RunTime float64 + StartTime int64 + Target string + TfVersion string + func (*TestResults) Descriptor() ([]byte, []int) + func (*TestResults) ProtoMessage() + func (x *TestResults) GetBenchmarkType() TestResults_BenchmarkType + func (x *TestResults) GetBuildConfiguration() *BuildConfiguration + func (x *TestResults) GetCommitId() *CommitId + func (x *TestResults) GetEntries() *BenchmarkEntries + func (x *TestResults) GetMachineConfiguration() *MachineConfiguration + func (x *TestResults) GetName() string + func (x *TestResults) GetRunConfiguration() *RunConfiguration + func (x *TestResults) GetRunMode() string + func (x *TestResults) GetRunTime() float64 + func (x *TestResults) GetStartTime() int64 + func (x *TestResults) GetTarget() string + func (x *TestResults) GetTfVersion() string + func (x *TestResults) ProtoReflect() protoreflect.Message + func (x *TestResults) Reset() + func (x *TestResults) String() string + type TestResults_BenchmarkType int32 + const TestResults_ANDROID_BENCHMARK + const TestResults_CPP_MICROBENCHMARK + const TestResults_EDGE_BENCHMARK + const TestResults_IOS_BENCHMARK + const TestResults_PYTHON_BENCHMARK + const TestResults_UNKNOWN + func (TestResults_BenchmarkType) Descriptor() protoreflect.EnumDescriptor + func (TestResults_BenchmarkType) EnumDescriptor() ([]byte, []int) + func (TestResults_BenchmarkType) Type() protoreflect.EnumType + func (x TestResults_BenchmarkType) Enum() *TestResults_BenchmarkType + func (x TestResults_BenchmarkType) Number() protoreflect.EnumNumber + func (x TestResults_BenchmarkType) String() string + type TryGetKeyValueRequest struct + Key string + func (*TryGetKeyValueRequest) Descriptor() ([]byte, []int) + func (*TryGetKeyValueRequest) ProtoMessage() + func (x *TryGetKeyValueRequest) GetKey() string + func (x *TryGetKeyValueRequest) ProtoReflect() protoreflect.Message + func (x *TryGetKeyValueRequest) Reset() + func (x *TryGetKeyValueRequest) String() string + type TryGetKeyValueResponse struct + Kv *KeyValueEntry + func (*TryGetKeyValueResponse) Descriptor() ([]byte, []int) + func (*TryGetKeyValueResponse) ProtoMessage() + func (x *TryGetKeyValueResponse) GetKv() *KeyValueEntry + func (x *TryGetKeyValueResponse) ProtoReflect() protoreflect.Message + func (x *TryGetKeyValueResponse) Reset() + func (x *TryGetKeyValueResponse) String() string + type UnimplementedCoordinationServiceServer struct + func (UnimplementedCoordinationServiceServer) Barrier(context.Context, *BarrierRequest) (*BarrierResponse, error) + func (UnimplementedCoordinationServiceServer) CancelBarrier(context.Context, *CancelBarrierRequest) (*CancelBarrierResponse, error) + func (UnimplementedCoordinationServiceServer) DeleteKeyValue(context.Context, *DeleteKeyValueRequest) (*DeleteKeyValueResponse, error) + func (UnimplementedCoordinationServiceServer) GetKeyValue(context.Context, *GetKeyValueRequest) (*GetKeyValueResponse, error) + func (UnimplementedCoordinationServiceServer) GetKeyValueDir(context.Context, *GetKeyValueDirRequest) (*GetKeyValueDirResponse, error) + func (UnimplementedCoordinationServiceServer) GetTaskState(context.Context, *GetTaskStateRequest) (*GetTaskStateResponse, error) + func (UnimplementedCoordinationServiceServer) Heartbeat(context.Context, *HeartbeatRequest) (*HeartbeatResponse, error) + func (UnimplementedCoordinationServiceServer) InsertKeyValue(context.Context, *InsertKeyValueRequest) (*InsertKeyValueResponse, error) + func (UnimplementedCoordinationServiceServer) RegisterTask(context.Context, *RegisterTaskRequest) (*RegisterTaskResponse, error) + func (UnimplementedCoordinationServiceServer) ReportErrorToService(context.Context, *ReportErrorToServiceRequest) (*ReportErrorToServiceResponse, error) + func (UnimplementedCoordinationServiceServer) ReportErrorToTask(context.Context, *ReportErrorToTaskRequest) (*ReportErrorToTaskResponse, error) + func (UnimplementedCoordinationServiceServer) ResetTask(context.Context, *ResetTaskRequest) (*ResetTaskResponse, error) + func (UnimplementedCoordinationServiceServer) ShutdownTask(context.Context, *ShutdownTaskRequest) (*ShutdownTaskResponse, error) + func (UnimplementedCoordinationServiceServer) TryGetKeyValue(context.Context, *TryGetKeyValueRequest) (*TryGetKeyValueResponse, error) + func (UnimplementedCoordinationServiceServer) WaitForAllTasks(context.Context, *WaitForAllTasksRequest) (*WaitForAllTasksResponse, error) + type UnsafeCoordinationServiceServer interface + type WaitForAllTasksRequest struct + DeviceInfo *DeviceInfo + SourceTask *CoordinatedTask + func (*WaitForAllTasksRequest) Descriptor() ([]byte, []int) + func (*WaitForAllTasksRequest) ProtoMessage() + func (x *WaitForAllTasksRequest) GetDeviceInfo() *DeviceInfo + func (x *WaitForAllTasksRequest) GetSourceTask() *CoordinatedTask + func (x *WaitForAllTasksRequest) ProtoReflect() protoreflect.Message + func (x *WaitForAllTasksRequest) Reset() + func (x *WaitForAllTasksRequest) String() string + type WaitForAllTasksResponse struct + DeviceInfo *DeviceInfo + LeaderIncarnation uint64 + func (*WaitForAllTasksResponse) Descriptor() ([]byte, []int) + func (*WaitForAllTasksResponse) ProtoMessage() + func (x *WaitForAllTasksResponse) GetDeviceInfo() *DeviceInfo + func (x *WaitForAllTasksResponse) GetLeaderIncarnation() uint64 + func (x *WaitForAllTasksResponse) ProtoReflect() protoreflect.Message + func (x *WaitForAllTasksResponse) Reset() + func (x *WaitForAllTasksResponse) String() string + type WorkerPossiblyRestarted struct + func (*WorkerPossiblyRestarted) Descriptor() ([]byte, []int) + func (*WorkerPossiblyRestarted) ProtoMessage() + func (x *WorkerPossiblyRestarted) ProtoReflect() protoreflect.Message + func (x *WorkerPossiblyRestarted) Reset() + func (x *WorkerPossiblyRestarted) String() string
Go to main page