tpu_driver

package
v0.0.0-...-3233e8b Latest Latest
Warning

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

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

Documentation

Index

Constants

View Source 
const (
	Default_TpuDriverConfig_GrpcConfig_ConnectionTimeoutSecs = int64(30)
	Default_TpuDriverConfig_GrpcConfig_KeepaliveTimeoutSecs  = int64(30)
)

Default values for TpuDriverConfig_GrpcConfig fields.

View Source 
const (
	CloudTpuDriver_Open_FullMethodName            = "/tpu_driver.CloudTpuDriver/Open"
	CloudTpuDriver_Close_FullMethodName           = "/tpu_driver.CloudTpuDriver/Close"
	CloudTpuDriver_Reset_FullMethodName           = "/tpu_driver.CloudTpuDriver/Reset"
	CloudTpuDriver_QuerySystemInfo_FullMethodName = "/tpu_driver.CloudTpuDriver/QuerySystemInfo"
	CloudTpuDriver_StreamExecute_FullMethodName   = "/tpu_driver.CloudTpuDriver/StreamExecute"
)
View Source 
const (
	Default_OpenRequest_ClientVersion = int32(0)
)

Default values for OpenRequest fields.

View Source 
const (
	Default_OpenResponse_MaxIdleTimeSeconds = int32(3600)
)

Default values for OpenResponse fields.

Variables

View Source 
var (
	MemoryRegion_name = map[int32]string{
		1: "HBM",
	}
	MemoryRegion_value = map[string]int32{
		"HBM": 1,
	}
)

Enum value maps for MemoryRegion.

View Source 
var CloudTpuDriver_ServiceDesc = grpc.ServiceDesc{
	ServiceName: "tpu_driver.CloudTpuDriver",
	HandlerType: (*CloudTpuDriverServer)(nil),
	Methods: []grpc.MethodDesc{
		{
			MethodName: "Open",
			Handler:    _CloudTpuDriver_Open_Handler,
		},
		{
			MethodName: "Close",
			Handler:    _CloudTpuDriver_Close_Handler,
		},
		{
			MethodName: "Reset",
			Handler:    _CloudTpuDriver_Reset_Handler,
		},
		{
			MethodName: "QuerySystemInfo",
			Handler:    _CloudTpuDriver_QuerySystemInfo_Handler,
		},
	},
	Streams: []grpc.StreamDesc{
		{
			StreamName:    "StreamExecute",
			Handler:       _CloudTpuDriver_StreamExecute_Handler,
			ServerStreams: true,
			ClientStreams: true,
		},
	},
	Metadata: "tensorflow/compiler/xla/python/tpu_driver/tpu_service.proto",
}

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

View Source 
var File_tensorflow_compiler_xla_python_tpu_driver_tpu_driver_proto protoreflect.FileDescriptor
View Source 
var File_tensorflow_compiler_xla_python_tpu_driver_tpu_service_proto protoreflect.FileDescriptor

Functions

func RegisterCloudTpuDriverServer 

func RegisterCloudTpuDriverServer(s grpc.ServiceRegistrar, srv CloudTpuDriverServer)

Types

type AllocateRequest 

type AllocateRequest struct {
	CoreId *int32        `protobuf:"varint,1,req,name=core_id,json=coreId" json:"core_id,omitempty"`
	Region *MemoryRegion `protobuf:"varint,2,req,name=region,enum=tpu_driver.MemoryRegion" json:"region,omitempty"`
	// Types that are assignable to Size:
	//
	//	*AllocateRequest_NumBytes
	//	*AllocateRequest_Shape
	Size isAllocateRequest_Size `protobuf_oneof:"size"`
	// contains filtered or unexported fields
}

func (*AllocateRequest) Descriptor deprecated 

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

Deprecated: Use AllocateRequest.ProtoReflect.Descriptor instead.

func (*AllocateRequest) GetCoreId 

func (x *AllocateRequest) GetCoreId() int32

func (*AllocateRequest) GetNumBytes 

func (x *AllocateRequest) GetNumBytes() int64

func (*AllocateRequest) GetRegion 

func (x *AllocateRequest) GetRegion() MemoryRegion

func (*AllocateRequest) GetShape 

func (x *AllocateRequest) GetShape() *data.ShapeProto

func (*AllocateRequest) GetSize 

func (m *AllocateRequest) GetSize() isAllocateRequest_Size

func (*AllocateRequest) ProtoMessage 

func (*AllocateRequest) ProtoMessage()

func (*AllocateRequest) ProtoReflect 

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

func (*AllocateRequest) Reset 

func (x *AllocateRequest) Reset()

func (*AllocateRequest) String 

func (x *AllocateRequest) String() string

type AllocateRequest_NumBytes 

type AllocateRequest_NumBytes struct {
	NumBytes int64 `protobuf:"varint,3,opt,name=num_bytes,json=numBytes,oneof"`
}

type AllocateRequest_Shape 

type AllocateRequest_Shape struct {
	Shape *data.ShapeProto `protobuf:"bytes,4,opt,name=shape,oneof"`
}

type AllocateTupleRequest 

type AllocateTupleRequest struct {
	CoreId   *int32        `protobuf:"varint,1,req,name=core_id,json=coreId" json:"core_id,omitempty"`
	Region   *MemoryRegion `protobuf:"varint,2,req,name=region,enum=tpu_driver.MemoryRegion" json:"region,omitempty"`
	Children []int64       `protobuf:"varint,3,rep,name=children" json:"children,omitempty"`
	// contains filtered or unexported fields
}

func (*AllocateTupleRequest) Descriptor deprecated 

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

Deprecated: Use AllocateTupleRequest.ProtoReflect.Descriptor instead.

func (*AllocateTupleRequest) GetChildren 

func (x *AllocateTupleRequest) GetChildren() []int64

func (*AllocateTupleRequest) GetCoreId 

func (x *AllocateTupleRequest) GetCoreId() int32

func (*AllocateTupleRequest) GetRegion 

func (x *AllocateTupleRequest) GetRegion() MemoryRegion

func (*AllocateTupleRequest) ProtoMessage 

func (*AllocateTupleRequest) ProtoMessage()

func (*AllocateTupleRequest) ProtoReflect 

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

func (*AllocateTupleRequest) Reset 

func (x *AllocateTupleRequest) Reset()

func (*AllocateTupleRequest) String 

func (x *AllocateTupleRequest) String() string

type ChipCoordinate 

type ChipCoordinate struct {
	X *int32 `protobuf:"varint,1,req,name=x" json:"x,omitempty"`
	Y *int32 `protobuf:"varint,2,req,name=y" json:"y,omitempty"`
	Z *int32 `protobuf:"varint,3,req,name=z" json:"z,omitempty"`
	// contains filtered or unexported fields
}

func (*ChipCoordinate) Descriptor deprecated 

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

Deprecated: Use ChipCoordinate.ProtoReflect.Descriptor instead.

func (*ChipCoordinate) GetX 

func (x *ChipCoordinate) GetX() int32

func (*ChipCoordinate) GetY 

func (x *ChipCoordinate) GetY() int32

func (*ChipCoordinate) GetZ 

func (x *ChipCoordinate) GetZ() int32

func (*ChipCoordinate) ProtoMessage 

func (*ChipCoordinate) ProtoMessage()

func (*ChipCoordinate) ProtoReflect 

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

func (*ChipCoordinate) Reset 

func (x *ChipCoordinate) Reset()

func (*ChipCoordinate) String 

func (x *ChipCoordinate) String() string

type CloseRequest 

type CloseRequest struct {
	ClientId *uint32 `protobuf:"fixed32,1,req,name=client_id,json=clientId" json:"client_id,omitempty"`
	// contains filtered or unexported fields
}

func (*CloseRequest) Descriptor deprecated 

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

Deprecated: Use CloseRequest.ProtoReflect.Descriptor instead.

func (*CloseRequest) GetClientId 

func (x *CloseRequest) GetClientId() uint32

func (*CloseRequest) ProtoMessage 

func (*CloseRequest) ProtoMessage()

func (*CloseRequest) ProtoReflect 

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

func (*CloseRequest) Reset 

func (x *CloseRequest) Reset()

func (*CloseRequest) String 

func (x *CloseRequest) String() string

type CloseResponse 

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

func (*CloseResponse) Descriptor deprecated 

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

Deprecated: Use CloseResponse.ProtoReflect.Descriptor instead.

func (*CloseResponse) ProtoMessage 

func (*CloseResponse) ProtoMessage()

func (*CloseResponse) ProtoReflect 

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

func (*CloseResponse) Reset 

func (x *CloseResponse) Reset()

func (*CloseResponse) String 

func (x *CloseResponse) String() string

type CloudTpuDriverClient 

type CloudTpuDriverClient interface {
	// Open the driver. If the driver is already open, return an error.
	Open(ctx context.Context, in *OpenRequest, opts ...grpc.CallOption) (*OpenResponse, error)
	// Close the driver. Any outstanding requests will be terminated.
	Close(ctx context.Context, in *CloseRequest, opts ...grpc.CallOption) (*CloseResponse, error)
	// Reset the driver. All connected clients will be disconnected.
	Reset(ctx context.Context, in *ResetRequest, opts ...grpc.CallOption) (*ResetResponse, error)
	// Query the driver for current system performance information.
	QuerySystemInfo(ctx context.Context, in *QuerySystemInfoRequest, opts ...grpc.CallOption) (*QuerySystemInfoResponse, error)
	// Enqueue an operation to be executed when its dependencies are satisfied.
	StreamExecute(ctx context.Context, opts ...grpc.CallOption) (CloudTpuDriver_StreamExecuteClient, error)
}

CloudTpuDriverClient is the client API for CloudTpuDriver service.

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

func NewCloudTpuDriverClient 

func NewCloudTpuDriverClient(cc grpc.ClientConnInterface) CloudTpuDriverClient

type CloudTpuDriverServer 

type CloudTpuDriverServer interface {
	// Open the driver. If the driver is already open, return an error.
	Open(context.Context, *OpenRequest) (*OpenResponse, error)
	// Close the driver. Any outstanding requests will be terminated.
	Close(context.Context, *CloseRequest) (*CloseResponse, error)
	// Reset the driver. All connected clients will be disconnected.
	Reset(context.Context, *ResetRequest) (*ResetResponse, error)
	// Query the driver for current system performance information.
	QuerySystemInfo(context.Context, *QuerySystemInfoRequest) (*QuerySystemInfoResponse, error)
	// Enqueue an operation to be executed when its dependencies are satisfied.
	StreamExecute(CloudTpuDriver_StreamExecuteServer) error
	// contains filtered or unexported methods
}

CloudTpuDriverServer is the server API for CloudTpuDriver service. All implementations must embed UnimplementedCloudTpuDriverServer for forward compatibility

type CloudTpuDriver_StreamExecuteClient 

type CloudTpuDriver_StreamExecuteClient interface {
	Send(*StreamRequest) error
	Recv() (*StreamResponse, error)
	grpc.ClientStream
}

type CloudTpuDriver_StreamExecuteServer 

type CloudTpuDriver_StreamExecuteServer interface {
	Send(*StreamResponse) error
	Recv() (*StreamRequest, error)
	grpc.ServerStream
}

type CompileRequest 

type CompileRequest struct {
	HloProgram  *service.HloProto `protobuf:"bytes,1,req,name=hlo_program,json=hloProgram" json:"hlo_program,omitempty"`
	NumReplicas *int64            `protobuf:"varint,2,opt,name=num_replicas,json=numReplicas" json:"num_replicas,omitempty"`
	// contains filtered or unexported fields
}

func (*CompileRequest) Descriptor deprecated 

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

Deprecated: Use CompileRequest.ProtoReflect.Descriptor instead.

func (*CompileRequest) GetHloProgram 

func (x *CompileRequest) GetHloProgram() *service.HloProto

func (*CompileRequest) GetNumReplicas 

func (x *CompileRequest) GetNumReplicas() int64

func (*CompileRequest) ProtoMessage 

func (*CompileRequest) ProtoMessage()

func (*CompileRequest) ProtoReflect 

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

func (*CompileRequest) Reset 

func (x *CompileRequest) Reset()

func (*CompileRequest) String 

func (x *CompileRequest) String() string

type CompileResponse 

type CompileResponse struct {
	Metadata *CompiledProgramMetadata `protobuf:"bytes,1,req,name=metadata" json:"metadata,omitempty"`
	// contains filtered or unexported fields
}

func (*CompileResponse) Descriptor deprecated 

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

Deprecated: Use CompileResponse.ProtoReflect.Descriptor instead.

func (*CompileResponse) GetMetadata 

func (x *CompileResponse) GetMetadata() *CompiledProgramMetadata

func (*CompileResponse) ProtoMessage 

func (*CompileResponse) ProtoMessage()

func (*CompileResponse) ProtoReflect 

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

func (*CompileResponse) Reset 

func (x *CompileResponse) Reset()

func (*CompileResponse) String 

func (x *CompileResponse) String() string

type CompiledProgramMetadata 

type CompiledProgramMetadata struct {
	ProgramShape *data.ProgramShapeProto `protobuf:"bytes,1,req,name=program_shape,json=programShape" json:"program_shape,omitempty"`
	// contains filtered or unexported fields
}

func (*CompiledProgramMetadata) Descriptor deprecated 

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

Deprecated: Use CompiledProgramMetadata.ProtoReflect.Descriptor instead.

func (*CompiledProgramMetadata) GetProgramShape 

func (x *CompiledProgramMetadata) GetProgramShape() *data.ProgramShapeProto

func (*CompiledProgramMetadata) ProtoMessage 

func (*CompiledProgramMetadata) ProtoMessage()

func (*CompiledProgramMetadata) ProtoReflect 

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

func (*CompiledProgramMetadata) Reset 

func (x *CompiledProgramMetadata) Reset()

func (*CompiledProgramMetadata) String 

func (x *CompiledProgramMetadata) String() string

type CpuInfo 

type CpuInfo struct {
	NumCpuCores         *int32   `protobuf:"varint,1,req,name=num_cpu_cores,json=numCpuCores" json:"num_cpu_cores,omitempty"`
	CpuLoadAverage_1Min *float32 `protobuf:"fixed32,2,req,name=cpu_load_average_1min,json=cpuLoadAverage1min" json:"cpu_load_average_1min,omitempty"`
	RamBytesTotal       *int64   `protobuf:"varint,100,req,name=ram_bytes_total,json=ramBytesTotal" json:"ram_bytes_total,omitempty"`
	RamBytesAvailable   *int64   `protobuf:"varint,101,req,name=ram_bytes_available,json=ramBytesAvailable" json:"ram_bytes_available,omitempty"`
	// contains filtered or unexported fields
}

func (*CpuInfo) Descriptor deprecated 

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

Deprecated: Use CpuInfo.ProtoReflect.Descriptor instead.

func (*CpuInfo) GetCpuLoadAverage_1Min 

func (x *CpuInfo) GetCpuLoadAverage_1Min() float32

func (*CpuInfo) GetNumCpuCores 

func (x *CpuInfo) GetNumCpuCores() int32

func (*CpuInfo) GetRamBytesAvailable 

func (x *CpuInfo) GetRamBytesAvailable() int64

func (*CpuInfo) GetRamBytesTotal 

func (x *CpuInfo) GetRamBytesTotal() 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 DeallocateRequest 

type DeallocateRequest struct {
	Handle *int64 `protobuf:"varint,1,req,name=handle" json:"handle,omitempty"`
	// contains filtered or unexported fields
}

func (*DeallocateRequest) Descriptor deprecated 

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

Deprecated: Use DeallocateRequest.ProtoReflect.Descriptor instead.

func (*DeallocateRequest) GetHandle 

func (x *DeallocateRequest) GetHandle() int64

func (*DeallocateRequest) ProtoMessage 

func (*DeallocateRequest) ProtoMessage()

func (*DeallocateRequest) ProtoReflect 

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

func (*DeallocateRequest) Reset 

func (x *DeallocateRequest) Reset()

func (*DeallocateRequest) String 

func (x *DeallocateRequest) String() string

type ExecuteRequest 

type ExecuteRequest struct {
	LoadedProgramHandle *int64                      `protobuf:"varint,1,req,name=loaded_program_handle,json=loadedProgramHandle" json:"loaded_program_handle,omitempty"`
	InputHandle         []int64                     `protobuf:"varint,2,rep,name=input_handle,json=inputHandle" json:"input_handle,omitempty"`
	OutputHandle        []int64                     `protobuf:"varint,3,rep,name=output_handle,json=outputHandle" json:"output_handle,omitempty"`
	DeviceAssignment    *data.DeviceAssignmentProto `protobuf:"bytes,4,opt,name=device_assignment,json=deviceAssignment" json:"device_assignment,omitempty"`
	// contains filtered or unexported fields
}

func (*ExecuteRequest) Descriptor deprecated 

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

Deprecated: Use ExecuteRequest.ProtoReflect.Descriptor instead.

func (*ExecuteRequest) GetDeviceAssignment 

func (x *ExecuteRequest) GetDeviceAssignment() *data.DeviceAssignmentProto

func (*ExecuteRequest) GetInputHandle 

func (x *ExecuteRequest) GetInputHandle() []int64

func (*ExecuteRequest) GetLoadedProgramHandle 

func (x *ExecuteRequest) GetLoadedProgramHandle() int64

func (*ExecuteRequest) GetOutputHandle 

func (x *ExecuteRequest) GetOutputHandle() []int64

func (*ExecuteRequest) ProtoMessage 

func (*ExecuteRequest) ProtoMessage()

func (*ExecuteRequest) ProtoReflect 

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

func (*ExecuteRequest) Reset 

func (x *ExecuteRequest) Reset()

func (*ExecuteRequest) String 

func (x *ExecuteRequest) String() string

type LoadProgramRequest 

type LoadProgramRequest struct {
	CoreId                *int32 `protobuf:"varint,1,req,name=core_id,json=coreId" json:"core_id,omitempty"`
	CompiledProgramHandle *int64 `protobuf:"varint,2,req,name=compiled_program_handle,json=compiledProgramHandle" json:"compiled_program_handle,omitempty"`
	// contains filtered or unexported fields
}

func (*LoadProgramRequest) Descriptor deprecated 

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

Deprecated: Use LoadProgramRequest.ProtoReflect.Descriptor instead.

func (*LoadProgramRequest) GetCompiledProgramHandle 

func (x *LoadProgramRequest) GetCompiledProgramHandle() int64

func (*LoadProgramRequest) GetCoreId 

func (x *LoadProgramRequest) GetCoreId() int32

func (*LoadProgramRequest) ProtoMessage 

func (*LoadProgramRequest) ProtoMessage()

func (*LoadProgramRequest) ProtoReflect 

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

func (*LoadProgramRequest) Reset 

func (x *LoadProgramRequest) Reset()

func (*LoadProgramRequest) String 

func (x *LoadProgramRequest) String() string

type MemoryRegion 

type MemoryRegion int32
const (
	MemoryRegion_HBM MemoryRegion = 1
)

func (MemoryRegion) Descriptor 

func (MemoryRegion) Descriptor() protoreflect.EnumDescriptor

func (MemoryRegion) Enum 

func (x MemoryRegion) Enum() *MemoryRegion

func (MemoryRegion) EnumDescriptor deprecated 

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

Deprecated: Use MemoryRegion.Descriptor instead.

func (MemoryRegion) Number 

func (x MemoryRegion) Number() protoreflect.EnumNumber

func (MemoryRegion) String 

func (x MemoryRegion) String() string

func (MemoryRegion) Type 

func (MemoryRegion) Type() protoreflect.EnumType

func (*MemoryRegion) UnmarshalJSON deprecated 

func (x *MemoryRegion) UnmarshalJSON(b []byte) error

Deprecated: Do not use.

type OpenRequest 

type OpenRequest struct {

	// The version number for this client. Versions are bumped in case of
	// backwards incompatible client-server protocol changes. Servers will reject
	// clients with an unsupported version.
	ClientVersion *int32 `protobuf:"varint,1,opt,name=client_version,json=clientVersion,def=0" json:"client_version,omitempty"`
	// contains filtered or unexported fields
}

func (*OpenRequest) Descriptor deprecated 

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

Deprecated: Use OpenRequest.ProtoReflect.Descriptor instead.

func (*OpenRequest) GetClientVersion 

func (x *OpenRequest) GetClientVersion() int32

func (*OpenRequest) ProtoMessage 

func (*OpenRequest) ProtoMessage()

func (*OpenRequest) ProtoReflect 

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

func (*OpenRequest) Reset 

func (x *OpenRequest) Reset()

func (*OpenRequest) String 

func (x *OpenRequest) String() string

type OpenResponse 

type OpenResponse struct {
	ClientId *uint32 `protobuf:"varint,1,req,name=client_id,json=clientId" json:"client_id,omitempty"`
	// Maximum time this client can be idle before it is GC'ed and all resources
	// released.
	MaxIdleTimeSeconds *int32 `protobuf:"varint,2,opt,name=max_idle_time_seconds,json=maxIdleTimeSeconds,def=3600" json:"max_idle_time_seconds,omitempty"`
	// contains filtered or unexported fields
}

func (*OpenResponse) Descriptor deprecated 

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

Deprecated: Use OpenResponse.ProtoReflect.Descriptor instead.

func (*OpenResponse) GetClientId 

func (x *OpenResponse) GetClientId() uint32

func (*OpenResponse) GetMaxIdleTimeSeconds 

func (x *OpenResponse) GetMaxIdleTimeSeconds() int32

func (*OpenResponse) ProtoMessage 

func (*OpenResponse) ProtoMessage()

func (*OpenResponse) ProtoReflect 

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

func (*OpenResponse) Reset 

func (x *OpenResponse) Reset()

func (*OpenResponse) String 

func (x *OpenResponse) String() string

type QuerySystemInfoRequest 

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

func (*QuerySystemInfoRequest) Descriptor deprecated 

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

Deprecated: Use QuerySystemInfoRequest.ProtoReflect.Descriptor instead.

func (*QuerySystemInfoRequest) ProtoMessage 

func (*QuerySystemInfoRequest) ProtoMessage()

func (*QuerySystemInfoRequest) ProtoReflect 

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

func (*QuerySystemInfoRequest) Reset 

func (x *QuerySystemInfoRequest) Reset()

func (*QuerySystemInfoRequest) String 

func (x *QuerySystemInfoRequest) String() string

type QuerySystemInfoResponse 

type QuerySystemInfoResponse struct {
	SystemInfo *SystemInfo `protobuf:"bytes,1,req,name=system_info,json=systemInfo" json:"system_info,omitempty"`
	// contains filtered or unexported fields
}

func (*QuerySystemInfoResponse) Descriptor deprecated 

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

Deprecated: Use QuerySystemInfoResponse.ProtoReflect.Descriptor instead.

func (*QuerySystemInfoResponse) GetSystemInfo 

func (x *QuerySystemInfoResponse) GetSystemInfo() *SystemInfo

func (*QuerySystemInfoResponse) ProtoMessage 

func (*QuerySystemInfoResponse) ProtoMessage()

func (*QuerySystemInfoResponse) ProtoReflect 

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

func (*QuerySystemInfoResponse) Reset 

func (x *QuerySystemInfoResponse) Reset()

func (*QuerySystemInfoResponse) String 

func (x *QuerySystemInfoResponse) String() string

type ResetRequest 

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

func (*ResetRequest) Descriptor deprecated 

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

Deprecated: Use ResetRequest.ProtoReflect.Descriptor instead.

func (*ResetRequest) ProtoMessage 

func (*ResetRequest) ProtoMessage()

func (*ResetRequest) ProtoReflect 

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

func (*ResetRequest) Reset 

func (x *ResetRequest) Reset()

func (*ResetRequest) String 

func (x *ResetRequest) String() string

type ResetResponse 

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

func (*ResetResponse) Descriptor deprecated 

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

Deprecated: Use ResetResponse.ProtoReflect.Descriptor instead.

func (*ResetResponse) ProtoMessage 

func (*ResetResponse) ProtoMessage()

func (*ResetResponse) ProtoReflect 

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

func (*ResetResponse) Reset 

func (x *ResetResponse) Reset()

func (*ResetResponse) String 

func (x *ResetResponse) String() string

type StatusMessage 

type StatusMessage struct {
	Code    *int32  `protobuf:"varint,1,req,name=code" json:"code,omitempty"`
	Message *string `protobuf:"bytes,2,opt,name=message" json:"message,omitempty"`
	// contains filtered or unexported fields
}

func (*StatusMessage) Descriptor deprecated 

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

Deprecated: Use StatusMessage.ProtoReflect.Descriptor instead.

func (*StatusMessage) GetCode 

func (x *StatusMessage) GetCode() int32

func (*StatusMessage) GetMessage 

func (x *StatusMessage) GetMessage() string

func (*StatusMessage) ProtoMessage 

func (*StatusMessage) ProtoMessage()

func (*StatusMessage) ProtoReflect 

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

func (*StatusMessage) Reset 

func (x *StatusMessage) Reset()

func (*StatusMessage) String 

func (x *StatusMessage) String() string

type StreamRequest 

type StreamRequest struct {
	Entry []*StreamRequest_Entry `protobuf:"bytes,30,rep,name=entry" json:"entry,omitempty"`
	// contains filtered or unexported fields
}

func (*StreamRequest) Descriptor deprecated 

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

Deprecated: Use StreamRequest.ProtoReflect.Descriptor instead.

func (*StreamRequest) GetEntry 

func (x *StreamRequest) GetEntry() []*StreamRequest_Entry

func (*StreamRequest) ProtoMessage 

func (*StreamRequest) ProtoMessage()

func (*StreamRequest) ProtoReflect 

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

func (*StreamRequest) Reset 

func (x *StreamRequest) Reset()

func (*StreamRequest) String 

func (x *StreamRequest) String() string

type StreamRequest_Entry 

type StreamRequest_Entry struct {

	// Types that are assignable to Request:
	//
	//	*StreamRequest_Entry_Alloc
	//	*StreamRequest_Entry_AllocTuple
	//	*StreamRequest_Entry_Dealloc
	//	*StreamRequest_Entry_TransferTo
	//	*StreamRequest_Entry_TransferFrom
	//	*StreamRequest_Entry_TransferFromTo
	//	*StreamRequest_Entry_Compile
	//	*StreamRequest_Entry_Load
	//	*StreamRequest_Entry_Unload
	//	*StreamRequest_Entry_Execute
	Request isStreamRequest_Entry_Request `protobuf_oneof:"request"`
	// If specified, a list of encoded EventId values.
	WaitForId []int64 `protobuf:"varint,20,rep,name=wait_for_id,json=waitForId" json:"wait_for_id,omitempty"`
	// A unique, encoded EventId value.
	// For Allocate, Compile, and Load, this also defines the result handle.
	OperationId *int64 `protobuf:"varint,21,req,name=operation_id,json=operationId" json:"operation_id,omitempty"`
	// A unique identifier for the thread that issued this request. Currently
	// for debugging purposes only.
	ThreadId *int64 `protobuf:"varint,22,opt,name=thread_id,json=threadId" json:"thread_id,omitempty"`
	// contains filtered or unexported fields
}

func (*StreamRequest_Entry) Descriptor deprecated 

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

Deprecated: Use StreamRequest_Entry.ProtoReflect.Descriptor instead.

func (*StreamRequest_Entry) GetAlloc 

func (x *StreamRequest_Entry) GetAlloc() *AllocateRequest

func (*StreamRequest_Entry) GetAllocTuple 

func (x *StreamRequest_Entry) GetAllocTuple() *AllocateTupleRequest

func (*StreamRequest_Entry) GetCompile 

func (x *StreamRequest_Entry) GetCompile() *CompileRequest

func (*StreamRequest_Entry) GetDealloc 

func (x *StreamRequest_Entry) GetDealloc() *DeallocateRequest

func (*StreamRequest_Entry) GetExecute 

func (x *StreamRequest_Entry) GetExecute() *ExecuteRequest

func (*StreamRequest_Entry) GetLoad 

func (x *StreamRequest_Entry) GetLoad() *LoadProgramRequest

func (*StreamRequest_Entry) GetOperationId 

func (x *StreamRequest_Entry) GetOperationId() int64

func (*StreamRequest_Entry) GetRequest 

func (m *StreamRequest_Entry) GetRequest() isStreamRequest_Entry_Request

func (*StreamRequest_Entry) GetThreadId 

func (x *StreamRequest_Entry) GetThreadId() int64

func (*StreamRequest_Entry) GetTransferFrom 

func (x *StreamRequest_Entry) GetTransferFrom() *TransferFromDeviceRequest

func (*StreamRequest_Entry) GetTransferFromTo 

func (x *StreamRequest_Entry) GetTransferFromTo() *TransferFromDeviceToDeviceRequest

func (*StreamRequest_Entry) GetTransferTo 

func (x *StreamRequest_Entry) GetTransferTo() *TransferToDeviceRequest

func (*StreamRequest_Entry) GetUnload 

func (x *StreamRequest_Entry) GetUnload() *UnloadProgramRequest

func (*StreamRequest_Entry) GetWaitForId 

func (x *StreamRequest_Entry) GetWaitForId() []int64

func (*StreamRequest_Entry) ProtoMessage 

func (*StreamRequest_Entry) ProtoMessage()

func (*StreamRequest_Entry) ProtoReflect 

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

func (*StreamRequest_Entry) Reset 

func (x *StreamRequest_Entry) Reset()

func (*StreamRequest_Entry) String 

func (x *StreamRequest_Entry) String() string

type StreamRequest_Entry_Alloc 

type StreamRequest_Entry_Alloc struct {
	Alloc *AllocateRequest `protobuf:"bytes,1,opt,name=alloc,oneof"`
}

type StreamRequest_Entry_AllocTuple 

type StreamRequest_Entry_AllocTuple struct {
	AllocTuple *AllocateTupleRequest `protobuf:"bytes,2,opt,name=alloc_tuple,json=allocTuple,oneof"`
}

type StreamRequest_Entry_Compile 

type StreamRequest_Entry_Compile struct {
	Compile *CompileRequest `protobuf:"bytes,6,opt,name=compile,oneof"`
}

type StreamRequest_Entry_Dealloc 

type StreamRequest_Entry_Dealloc struct {
	Dealloc *DeallocateRequest `protobuf:"bytes,3,opt,name=dealloc,oneof"`
}

type StreamRequest_Entry_Execute 

type StreamRequest_Entry_Execute struct {
	Execute *ExecuteRequest `protobuf:"bytes,9,opt,name=execute,oneof"`
}

type StreamRequest_Entry_Load 

type StreamRequest_Entry_Load struct {
	Load *LoadProgramRequest `protobuf:"bytes,7,opt,name=load,oneof"`
}

type StreamRequest_Entry_TransferFrom 

type StreamRequest_Entry_TransferFrom struct {
	TransferFrom *TransferFromDeviceRequest `protobuf:"bytes,5,opt,name=transfer_from,json=transferFrom,oneof"`
}

type StreamRequest_Entry_TransferFromTo 

type StreamRequest_Entry_TransferFromTo struct {
	TransferFromTo *TransferFromDeviceToDeviceRequest `protobuf:"bytes,10,opt,name=transfer_from_to,json=transferFromTo,oneof"`
}

type StreamRequest_Entry_TransferTo 

type StreamRequest_Entry_TransferTo struct {
	TransferTo *TransferToDeviceRequest `protobuf:"bytes,4,opt,name=transfer_to,json=transferTo,oneof"`
}

type StreamRequest_Entry_Unload 

type StreamRequest_Entry_Unload struct {
	Unload *UnloadProgramRequest `protobuf:"bytes,8,opt,name=unload,oneof"`
}

type StreamResponse 

type StreamResponse struct {
	Entry []*StreamResponse_Entry `protobuf:"bytes,20,rep,name=entry" json:"entry,omitempty"`
	// contains filtered or unexported fields
}

func (*StreamResponse) Descriptor deprecated 

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

Deprecated: Use StreamResponse.ProtoReflect.Descriptor instead.

func (*StreamResponse) GetEntry 

func (x *StreamResponse) GetEntry() []*StreamResponse_Entry

func (*StreamResponse) ProtoMessage 

func (*StreamResponse) ProtoMessage()

func (*StreamResponse) ProtoReflect 

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

func (*StreamResponse) Reset 

func (x *StreamResponse) Reset()

func (*StreamResponse) String 

func (x *StreamResponse) String() string

type StreamResponse_Entry 

type StreamResponse_Entry struct {

	// Types that are assignable to Response:
	//
	//	*StreamResponse_Entry_TransferFrom
	//	*StreamResponse_Entry_Compile
	Response isStreamResponse_Entry_Response `protobuf_oneof:"response"`
	Status   *StatusMessage                  `protobuf:"bytes,10,req,name=status" json:"status,omitempty"`
	// Echos the given encoded EventId value.
	OperationId *int64 `protobuf:"varint,11,req,name=operation_id,json=operationId" json:"operation_id,omitempty"`
	// contains filtered or unexported fields
}

func (*StreamResponse_Entry) Descriptor deprecated 

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

Deprecated: Use StreamResponse_Entry.ProtoReflect.Descriptor instead.

func (*StreamResponse_Entry) GetCompile 

func (x *StreamResponse_Entry) GetCompile() *CompileResponse

func (*StreamResponse_Entry) GetOperationId 

func (x *StreamResponse_Entry) GetOperationId() int64

func (*StreamResponse_Entry) GetResponse 

func (m *StreamResponse_Entry) GetResponse() isStreamResponse_Entry_Response

func (*StreamResponse_Entry) GetStatus 

func (x *StreamResponse_Entry) GetStatus() *StatusMessage

func (*StreamResponse_Entry) GetTransferFrom 

func (x *StreamResponse_Entry) GetTransferFrom() *TransferFromDeviceResponse

func (*StreamResponse_Entry) ProtoMessage 

func (*StreamResponse_Entry) ProtoMessage()

func (*StreamResponse_Entry) ProtoReflect 

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

func (*StreamResponse_Entry) Reset 

func (x *StreamResponse_Entry) Reset()

func (*StreamResponse_Entry) String 

func (x *StreamResponse_Entry) String() string

type StreamResponse_Entry_Compile 

type StreamResponse_Entry_Compile struct {
	Compile *CompileResponse `protobuf:"bytes,4,opt,name=compile,oneof"`
}

type StreamResponse_Entry_TransferFrom 

type StreamResponse_Entry_TransferFrom struct {
	TransferFrom *TransferFromDeviceResponse `protobuf:"bytes,3,opt,name=transfer_from,json=transferFrom,oneof"`
}

type SystemInfo 

type SystemInfo struct {
	TpuChip   []*TpuChipInfo `protobuf:"bytes,1,rep,name=tpu_chip,json=tpuChip" json:"tpu_chip,omitempty"`
	Cpu       *CpuInfo       `protobuf:"bytes,2,req,name=cpu" json:"cpu,omitempty"`
	LocalCore []*TpuCoreInfo `protobuf:"bytes,3,rep,name=local_core,json=localCore" json:"local_core,omitempty"`
	HostId    *int32         `protobuf:"varint,4,opt,name=host_id,json=hostId" json:"host_id,omitempty"`
	HostCount *int32         `protobuf:"varint,5,opt,name=host_count,json=hostCount" json:"host_count,omitempty"`
	ChipCount *int32         `protobuf:"varint,6,opt,name=chip_count,json=chipCount" json:"chip_count,omitempty"`
	CoreCount *int32         `protobuf:"varint,7,opt,name=core_count,json=coreCount" json:"core_count,omitempty"`
	// contains filtered or unexported fields
}

func (*SystemInfo) Descriptor deprecated 

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

Deprecated: Use SystemInfo.ProtoReflect.Descriptor instead.

func (*SystemInfo) GetChipCount 

func (x *SystemInfo) GetChipCount() int32

func (*SystemInfo) GetCoreCount 

func (x *SystemInfo) GetCoreCount() int32

func (*SystemInfo) GetCpu 

func (x *SystemInfo) GetCpu() *CpuInfo

func (*SystemInfo) GetHostCount 

func (x *SystemInfo) GetHostCount() int32

func (*SystemInfo) GetHostId 

func (x *SystemInfo) GetHostId() int32

func (*SystemInfo) GetLocalCore 

func (x *SystemInfo) GetLocalCore() []*TpuCoreInfo

func (*SystemInfo) GetTpuChip 

func (x *SystemInfo) GetTpuChip() []*TpuChipInfo

func (*SystemInfo) ProtoMessage 

func (*SystemInfo) ProtoMessage()

func (*SystemInfo) ProtoReflect 

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

func (*SystemInfo) Reset 

func (x *SystemInfo) Reset()

func (*SystemInfo) String 

func (x *SystemInfo) String() string

type TpuChipInfo 

type TpuChipInfo struct {
	Core      []*TpuCoreInfo  `protobuf:"bytes,1,rep,name=core" json:"core,omitempty"`
	HostId    *int32          `protobuf:"varint,2,opt,name=host_id,json=hostId" json:"host_id,omitempty"`
	ChipCoord *ChipCoordinate `protobuf:"bytes,3,opt,name=chip_coord,json=chipCoord" json:"chip_coord,omitempty"`
	// contains filtered or unexported fields
}

func (*TpuChipInfo) Descriptor deprecated 

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

Deprecated: Use TpuChipInfo.ProtoReflect.Descriptor instead.

func (*TpuChipInfo) GetChipCoord 

func (x *TpuChipInfo) GetChipCoord() *ChipCoordinate

func (*TpuChipInfo) GetCore 

func (x *TpuChipInfo) GetCore() []*TpuCoreInfo

func (*TpuChipInfo) GetHostId 

func (x *TpuChipInfo) GetHostId() int32

func (*TpuChipInfo) ProtoMessage 

func (*TpuChipInfo) ProtoMessage()

func (*TpuChipInfo) ProtoReflect 

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

func (*TpuChipInfo) Reset 

func (x *TpuChipInfo) Reset()

func (*TpuChipInfo) String 

func (x *TpuChipInfo) String() string

type TpuCoreInfo 

type TpuCoreInfo struct {
	Id                  *int32 `protobuf:"varint,1,req,name=id" json:"id,omitempty"`
	CoreOnChipIndex     *int32 `protobuf:"varint,2,opt,name=core_on_chip_index,json=coreOnChipIndex" json:"core_on_chip_index,omitempty"`
	CoreOnHostIndex     *int32 `protobuf:"varint,3,opt,name=core_on_host_index,json=coreOnHostIndex" json:"core_on_host_index,omitempty"`
	HbmBytesAvailable   *int64 `protobuf:"varint,100,opt,name=hbm_bytes_available,json=hbmBytesAvailable" json:"hbm_bytes_available,omitempty"`
	HbmBytesAllocatable *int64 `protobuf:"varint,101,opt,name=hbm_bytes_allocatable,json=hbmBytesAllocatable" json:"hbm_bytes_allocatable,omitempty"`
	// contains filtered or unexported fields
}

func (*TpuCoreInfo) Descriptor deprecated 

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

Deprecated: Use TpuCoreInfo.ProtoReflect.Descriptor instead.

func (*TpuCoreInfo) GetCoreOnChipIndex 

func (x *TpuCoreInfo) GetCoreOnChipIndex() int32

func (*TpuCoreInfo) GetCoreOnHostIndex 

func (x *TpuCoreInfo) GetCoreOnHostIndex() int32

func (*TpuCoreInfo) GetHbmBytesAllocatable 

func (x *TpuCoreInfo) GetHbmBytesAllocatable() int64

func (*TpuCoreInfo) GetHbmBytesAvailable 

func (x *TpuCoreInfo) GetHbmBytesAvailable() int64

func (*TpuCoreInfo) GetId 

func (x *TpuCoreInfo) GetId() int32

func (*TpuCoreInfo) ProtoMessage 

func (*TpuCoreInfo) ProtoMessage()

func (*TpuCoreInfo) ProtoReflect 

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

func (*TpuCoreInfo) Reset 

func (x *TpuCoreInfo) Reset()

func (*TpuCoreInfo) String 

func (x *TpuCoreInfo) String() string

type TpuDriverConfig 

type TpuDriverConfig struct {
	Worker *string                     `protobuf:"bytes,1,opt,name=worker" json:"worker,omitempty"`
	Grpc   *TpuDriverConfig_GrpcConfig `protobuf:"bytes,2,opt,name=grpc" json:"grpc,omitempty"`
	// contains filtered or unexported fields
}

func (*TpuDriverConfig) Descriptor deprecated 

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

Deprecated: Use TpuDriverConfig.ProtoReflect.Descriptor instead.

func (*TpuDriverConfig) GetGrpc 

func (x *TpuDriverConfig) GetGrpc() *TpuDriverConfig_GrpcConfig

func (*TpuDriverConfig) GetWorker 

func (x *TpuDriverConfig) GetWorker() string

func (*TpuDriverConfig) ProtoMessage 

func (*TpuDriverConfig) ProtoMessage()

func (*TpuDriverConfig) ProtoReflect 

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

func (*TpuDriverConfig) Reset 

func (x *TpuDriverConfig) Reset()

func (*TpuDriverConfig) String 

func (x *TpuDriverConfig) String() string

type TpuDriverConfig_GrpcConfig 

type TpuDriverConfig_GrpcConfig struct {

	// Time in seconds before the initial connection to the server will timeout.
	ConnectionTimeoutSecs *int64 `` /* 127-byte string literal not displayed */
	// Time in seconds the server may be unresponsive before terminating the
	// connection.
	KeepaliveTimeoutSecs *int64 `protobuf:"varint,2,opt,name=keepalive_timeout_secs,json=keepaliveTimeoutSecs,def=30" json:"keepalive_timeout_secs,omitempty"`
	// contains filtered or unexported fields
}

func (*TpuDriverConfig_GrpcConfig) Descriptor deprecated 

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

Deprecated: Use TpuDriverConfig_GrpcConfig.ProtoReflect.Descriptor instead.

func (*TpuDriverConfig_GrpcConfig) GetConnectionTimeoutSecs 

func (x *TpuDriverConfig_GrpcConfig) GetConnectionTimeoutSecs() int64

func (*TpuDriverConfig_GrpcConfig) GetKeepaliveTimeoutSecs 

func (x *TpuDriverConfig_GrpcConfig) GetKeepaliveTimeoutSecs() int64

func (*TpuDriverConfig_GrpcConfig) ProtoMessage 

func (*TpuDriverConfig_GrpcConfig) ProtoMessage()

func (*TpuDriverConfig_GrpcConfig) ProtoReflect 

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

func (*TpuDriverConfig_GrpcConfig) Reset 

func (x *TpuDriverConfig_GrpcConfig) Reset()

func (*TpuDriverConfig_GrpcConfig) String 

func (x *TpuDriverConfig_GrpcConfig) String() string

type TransferFromDeviceRequest 

type TransferFromDeviceRequest struct {
	SourceHandle *int64 `protobuf:"varint,1,req,name=source_handle,json=sourceHandle" json:"source_handle,omitempty"`
	// contains filtered or unexported fields
}

func (*TransferFromDeviceRequest) Descriptor deprecated 

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

Deprecated: Use TransferFromDeviceRequest.ProtoReflect.Descriptor instead.

func (*TransferFromDeviceRequest) GetSourceHandle 

func (x *TransferFromDeviceRequest) GetSourceHandle() int64

func (*TransferFromDeviceRequest) ProtoMessage 

func (*TransferFromDeviceRequest) ProtoMessage()

func (*TransferFromDeviceRequest) ProtoReflect 

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

func (*TransferFromDeviceRequest) Reset 

func (x *TransferFromDeviceRequest) Reset()

func (*TransferFromDeviceRequest) String 

func (x *TransferFromDeviceRequest) String() string

type TransferFromDeviceResponse 

type TransferFromDeviceResponse struct {
	Data []byte `protobuf:"bytes,2,req,name=data" json:"data,omitempty"`
	// contains filtered or unexported fields
}

func (*TransferFromDeviceResponse) Descriptor deprecated 

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

Deprecated: Use TransferFromDeviceResponse.ProtoReflect.Descriptor instead.

func (*TransferFromDeviceResponse) GetData 

func (x *TransferFromDeviceResponse) GetData() []byte

func (*TransferFromDeviceResponse) ProtoMessage 

func (*TransferFromDeviceResponse) ProtoMessage()

func (*TransferFromDeviceResponse) ProtoReflect 

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

func (*TransferFromDeviceResponse) Reset 

func (x *TransferFromDeviceResponse) Reset()

func (*TransferFromDeviceResponse) String 

func (x *TransferFromDeviceResponse) String() string

type TransferFromDeviceToDeviceRequest 

type TransferFromDeviceToDeviceRequest struct {
	SourceHandle *int64 `protobuf:"varint,1,req,name=source_handle,json=sourceHandle" json:"source_handle,omitempty"`
	TargetHandle *int64 `protobuf:"varint,2,req,name=target_handle,json=targetHandle" json:"target_handle,omitempty"`
	// contains filtered or unexported fields
}

func (*TransferFromDeviceToDeviceRequest) Descriptor deprecated 

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

Deprecated: Use TransferFromDeviceToDeviceRequest.ProtoReflect.Descriptor instead.

func (*TransferFromDeviceToDeviceRequest) GetSourceHandle 

func (x *TransferFromDeviceToDeviceRequest) GetSourceHandle() int64

func (*TransferFromDeviceToDeviceRequest) GetTargetHandle 

func (x *TransferFromDeviceToDeviceRequest) GetTargetHandle() int64

func (*TransferFromDeviceToDeviceRequest) ProtoMessage 

func (*TransferFromDeviceToDeviceRequest) ProtoMessage()

func (*TransferFromDeviceToDeviceRequest) ProtoReflect 

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

func (*TransferFromDeviceToDeviceRequest) Reset 

func (x *TransferFromDeviceToDeviceRequest) Reset()

func (*TransferFromDeviceToDeviceRequest) String 

func (x *TransferFromDeviceToDeviceRequest) String() string

type TransferToDeviceRequest 

type TransferToDeviceRequest struct {
	TargetHandle *int64 `protobuf:"varint,1,req,name=target_handle,json=targetHandle" json:"target_handle,omitempty"`
	Data         []byte `protobuf:"bytes,2,req,name=data" json:"data,omitempty"`
	// contains filtered or unexported fields
}

func (*TransferToDeviceRequest) Descriptor deprecated 

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

Deprecated: Use TransferToDeviceRequest.ProtoReflect.Descriptor instead.

func (*TransferToDeviceRequest) GetData 

func (x *TransferToDeviceRequest) GetData() []byte

func (*TransferToDeviceRequest) GetTargetHandle 

func (x *TransferToDeviceRequest) GetTargetHandle() int64

func (*TransferToDeviceRequest) ProtoMessage 

func (*TransferToDeviceRequest) ProtoMessage()

func (*TransferToDeviceRequest) ProtoReflect 

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

func (*TransferToDeviceRequest) Reset 

func (x *TransferToDeviceRequest) Reset()

func (*TransferToDeviceRequest) String 

func (x *TransferToDeviceRequest) String() string

type UnimplementedCloudTpuDriverServer 

type UnimplementedCloudTpuDriverServer struct {
}

UnimplementedCloudTpuDriverServer must be embedded to have forward compatible implementations.

func (UnimplementedCloudTpuDriverServer) Close 

func (UnimplementedCloudTpuDriverServer) Close(context.Context, *CloseRequest) (*CloseResponse, error)

func (UnimplementedCloudTpuDriverServer) Open 

func (UnimplementedCloudTpuDriverServer) Open(context.Context, *OpenRequest) (*OpenResponse, error)

func (UnimplementedCloudTpuDriverServer) QuerySystemInfo 

func (UnimplementedCloudTpuDriverServer) QuerySystemInfo(context.Context, *QuerySystemInfoRequest) (*QuerySystemInfoResponse, error)

func (UnimplementedCloudTpuDriverServer) Reset 

func (UnimplementedCloudTpuDriverServer) Reset(context.Context, *ResetRequest) (*ResetResponse, error)

func (UnimplementedCloudTpuDriverServer) StreamExecute 

func (UnimplementedCloudTpuDriverServer) StreamExecute(CloudTpuDriver_StreamExecuteServer) error

type UnloadProgramRequest 

type UnloadProgramRequest struct {
	LoadedProgramHandle *int64 `protobuf:"varint,1,req,name=loaded_program_handle,json=loadedProgramHandle" json:"loaded_program_handle,omitempty"`
	// contains filtered or unexported fields
}

func (*UnloadProgramRequest) Descriptor deprecated 

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

Deprecated: Use UnloadProgramRequest.ProtoReflect.Descriptor instead.

func (*UnloadProgramRequest) GetLoadedProgramHandle 

func (x *UnloadProgramRequest) GetLoadedProgramHandle() int64

func (*UnloadProgramRequest) ProtoMessage 

func (*UnloadProgramRequest) ProtoMessage()

func (*UnloadProgramRequest) ProtoReflect 

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

func (*UnloadProgramRequest) Reset 

func (x *UnloadProgramRequest) Reset()

func (*UnloadProgramRequest) String 

func (x *UnloadProgramRequest) String() string

type UnsafeCloudTpuDriverServer 

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

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

Source Files

View all Source files

Jump to

Keyboard shortcuts

? : This menu
/ : Search site
f or F : Jump to
y or Y : Canonical URL
go.dev uses cookies from Google to deliver and enhance the quality of its services and to analyze traffic. Learn more.