tpupb

package
v1.8.4 Latest Latest
Warning

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

Go to latest
Published: Sep 22, 2025 License: Apache-2.0 Imports: 11 Imported by: 1

Documentation

Index

Constants

View Source
const (
	Tpu_ListNodes_FullMethodName              = "/google.cloud.tpu.v1.Tpu/ListNodes"
	Tpu_GetNode_FullMethodName                = "/google.cloud.tpu.v1.Tpu/GetNode"
	Tpu_CreateNode_FullMethodName             = "/google.cloud.tpu.v1.Tpu/CreateNode"
	Tpu_DeleteNode_FullMethodName             = "/google.cloud.tpu.v1.Tpu/DeleteNode"
	Tpu_ReimageNode_FullMethodName            = "/google.cloud.tpu.v1.Tpu/ReimageNode"
	Tpu_StopNode_FullMethodName               = "/google.cloud.tpu.v1.Tpu/StopNode"
	Tpu_StartNode_FullMethodName              = "/google.cloud.tpu.v1.Tpu/StartNode"
	Tpu_ListTensorFlowVersions_FullMethodName = "/google.cloud.tpu.v1.Tpu/ListTensorFlowVersions"
	Tpu_GetTensorFlowVersion_FullMethodName   = "/google.cloud.tpu.v1.Tpu/GetTensorFlowVersion"
	Tpu_ListAcceleratorTypes_FullMethodName   = "/google.cloud.tpu.v1.Tpu/ListAcceleratorTypes"
	Tpu_GetAcceleratorType_FullMethodName     = "/google.cloud.tpu.v1.Tpu/GetAcceleratorType"
)

Variables

View Source
var (
	Node_State_name = map[int32]string{
		0:  "STATE_UNSPECIFIED",
		1:  "CREATING",
		2:  "READY",
		3:  "RESTARTING",
		4:  "REIMAGING",
		5:  "DELETING",
		6:  "REPAIRING",
		8:  "STOPPED",
		9:  "STOPPING",
		10: "STARTING",
		11: "PREEMPTED",
		12: "TERMINATED",
		13: "HIDING",
		14: "HIDDEN",
		15: "UNHIDING",
		16: "UNKNOWN",
	}
	Node_State_value = map[string]int32{
		"STATE_UNSPECIFIED": 0,
		"CREATING":          1,
		"READY":             2,
		"RESTARTING":        3,
		"REIMAGING":         4,
		"DELETING":          5,
		"REPAIRING":         6,
		"STOPPED":           8,
		"STOPPING":          9,
		"STARTING":          10,
		"PREEMPTED":         11,
		"TERMINATED":        12,
		"HIDING":            13,
		"HIDDEN":            14,
		"UNHIDING":          15,
		"UNKNOWN":           16,
	}
)

Enum value maps for Node_State.

View Source
var (
	Node_Health_name = map[int32]string{
		0: "HEALTH_UNSPECIFIED",
		1: "HEALTHY",
		2: "DEPRECATED_UNHEALTHY",
		3: "TIMEOUT",
		4: "UNHEALTHY_TENSORFLOW",
		5: "UNHEALTHY_MAINTENANCE",
	}
	Node_Health_value = map[string]int32{
		"HEALTH_UNSPECIFIED":    0,
		"HEALTHY":               1,
		"DEPRECATED_UNHEALTHY":  2,
		"TIMEOUT":               3,
		"UNHEALTHY_TENSORFLOW":  4,
		"UNHEALTHY_MAINTENANCE": 5,
	}
)

Enum value maps for Node_Health.

View Source
var (
	Node_ApiVersion_name = map[int32]string{
		0: "API_VERSION_UNSPECIFIED",
		1: "V1_ALPHA1",
		2: "V1",
		3: "V2_ALPHA1",
	}
	Node_ApiVersion_value = map[string]int32{
		"API_VERSION_UNSPECIFIED": 0,
		"V1_ALPHA1":               1,
		"V1":                      2,
		"V2_ALPHA1":               3,
	}
)

Enum value maps for Node_ApiVersion.

View Source
var (
	Symptom_SymptomType_name = map[int32]string{
		0: "SYMPTOM_TYPE_UNSPECIFIED",
		1: "LOW_MEMORY",
		2: "OUT_OF_MEMORY",
		3: "EXECUTE_TIMED_OUT",
		4: "MESH_BUILD_FAIL",
		5: "HBM_OUT_OF_MEMORY",
		6: "PROJECT_ABUSE",
	}
	Symptom_SymptomType_value = map[string]int32{
		"SYMPTOM_TYPE_UNSPECIFIED": 0,
		"LOW_MEMORY":               1,
		"OUT_OF_MEMORY":            2,
		"EXECUTE_TIMED_OUT":        3,
		"MESH_BUILD_FAIL":          4,
		"HBM_OUT_OF_MEMORY":        5,
		"PROJECT_ABUSE":            6,
	}
)

Enum value maps for Symptom_SymptomType.

View Source
var File_google_cloud_tpu_v1_cloud_tpu_proto protoreflect.FileDescriptor
View Source
var Tpu_ServiceDesc = grpc.ServiceDesc{
	ServiceName: "google.cloud.tpu.v1.Tpu",
	HandlerType: (*TpuServer)(nil),
	Methods: []grpc.MethodDesc{
		{
			MethodName: "ListNodes",
			Handler:    _Tpu_ListNodes_Handler,
		},
		{
			MethodName: "GetNode",
			Handler:    _Tpu_GetNode_Handler,
		},
		{
			MethodName: "CreateNode",
			Handler:    _Tpu_CreateNode_Handler,
		},
		{
			MethodName: "DeleteNode",
			Handler:    _Tpu_DeleteNode_Handler,
		},
		{
			MethodName: "ReimageNode",
			Handler:    _Tpu_ReimageNode_Handler,
		},
		{
			MethodName: "StopNode",
			Handler:    _Tpu_StopNode_Handler,
		},
		{
			MethodName: "StartNode",
			Handler:    _Tpu_StartNode_Handler,
		},
		{
			MethodName: "ListTensorFlowVersions",
			Handler:    _Tpu_ListTensorFlowVersions_Handler,
		},
		{
			MethodName: "GetTensorFlowVersion",
			Handler:    _Tpu_GetTensorFlowVersion_Handler,
		},
		{
			MethodName: "ListAcceleratorTypes",
			Handler:    _Tpu_ListAcceleratorTypes_Handler,
		},
		{
			MethodName: "GetAcceleratorType",
			Handler:    _Tpu_GetAcceleratorType_Handler,
		},
	},
	Streams:  []grpc.StreamDesc{},
	Metadata: "google/cloud/tpu/v1/cloud_tpu.proto",
}

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

Functions

func RegisterTpuServer

func RegisterTpuServer(s grpc.ServiceRegistrar, srv TpuServer)

Types

type AcceleratorType

type AcceleratorType struct {

	// The resource name.
	Name string `protobuf:"bytes,1,opt,name=name,proto3" json:"name,omitempty"`
	// the accelerator type.
	Type string `protobuf:"bytes,2,opt,name=type,proto3" json:"type,omitempty"`
	// contains filtered or unexported fields
}

A accelerator type that a Node can be configured with.

func (*AcceleratorType) Descriptor deprecated

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

Deprecated: Use AcceleratorType.ProtoReflect.Descriptor instead.

func (*AcceleratorType) GetName

func (x *AcceleratorType) GetName() string

func (*AcceleratorType) GetType

func (x *AcceleratorType) GetType() string

func (*AcceleratorType) ProtoMessage

func (*AcceleratorType) ProtoMessage()

func (*AcceleratorType) ProtoReflect

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

func (*AcceleratorType) Reset

func (x *AcceleratorType) Reset()

func (*AcceleratorType) String

func (x *AcceleratorType) String() string

type CreateNodeRequest

type CreateNodeRequest struct {

	// Required. The parent resource name.
	Parent string `protobuf:"bytes,1,opt,name=parent,proto3" json:"parent,omitempty"`
	// The unqualified resource name.
	NodeId string `protobuf:"bytes,2,opt,name=node_id,json=nodeId,proto3" json:"node_id,omitempty"`
	// Required. The node.
	Node *Node `protobuf:"bytes,3,opt,name=node,proto3" json:"node,omitempty"`
	// contains filtered or unexported fields
}

Request for [CreateNode][google.cloud.tpu.v1.Tpu.CreateNode].

func (*CreateNodeRequest) Descriptor deprecated

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

Deprecated: Use CreateNodeRequest.ProtoReflect.Descriptor instead.

func (*CreateNodeRequest) GetNode

func (x *CreateNodeRequest) GetNode() *Node

func (*CreateNodeRequest) GetNodeId

func (x *CreateNodeRequest) GetNodeId() string

func (*CreateNodeRequest) GetParent

func (x *CreateNodeRequest) GetParent() string

func (*CreateNodeRequest) ProtoMessage

func (*CreateNodeRequest) ProtoMessage()

func (*CreateNodeRequest) ProtoReflect

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

func (*CreateNodeRequest) Reset

func (x *CreateNodeRequest) Reset()

func (*CreateNodeRequest) String

func (x *CreateNodeRequest) String() string

type DeleteNodeRequest

type DeleteNodeRequest struct {

	// Required. The resource name.
	Name string `protobuf:"bytes,1,opt,name=name,proto3" json:"name,omitempty"`
	// contains filtered or unexported fields
}

Request for [DeleteNode][google.cloud.tpu.v1.Tpu.DeleteNode].

func (*DeleteNodeRequest) Descriptor deprecated

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

Deprecated: Use DeleteNodeRequest.ProtoReflect.Descriptor instead.

func (*DeleteNodeRequest) GetName

func (x *DeleteNodeRequest) GetName() string

func (*DeleteNodeRequest) ProtoMessage

func (*DeleteNodeRequest) ProtoMessage()

func (*DeleteNodeRequest) ProtoReflect

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

func (*DeleteNodeRequest) Reset

func (x *DeleteNodeRequest) Reset()

func (*DeleteNodeRequest) String

func (x *DeleteNodeRequest) String() string

type GetAcceleratorTypeRequest

type GetAcceleratorTypeRequest struct {

	// Required. The resource name.
	Name string `protobuf:"bytes,1,opt,name=name,proto3" json:"name,omitempty"`
	// contains filtered or unexported fields
}

Request for [GetAcceleratorType][google.cloud.tpu.v1.Tpu.GetAcceleratorType].

func (*GetAcceleratorTypeRequest) Descriptor deprecated

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

Deprecated: Use GetAcceleratorTypeRequest.ProtoReflect.Descriptor instead.

func (*GetAcceleratorTypeRequest) GetName

func (x *GetAcceleratorTypeRequest) GetName() string

func (*GetAcceleratorTypeRequest) ProtoMessage

func (*GetAcceleratorTypeRequest) ProtoMessage()

func (*GetAcceleratorTypeRequest) ProtoReflect

func (*GetAcceleratorTypeRequest) Reset

func (x *GetAcceleratorTypeRequest) Reset()

func (*GetAcceleratorTypeRequest) String

func (x *GetAcceleratorTypeRequest) String() string

type GetNodeRequest

type GetNodeRequest struct {

	// Required. The resource name.
	Name string `protobuf:"bytes,1,opt,name=name,proto3" json:"name,omitempty"`
	// contains filtered or unexported fields
}

Request for [GetNode][google.cloud.tpu.v1.Tpu.GetNode].

func (*GetNodeRequest) Descriptor deprecated

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

Deprecated: Use GetNodeRequest.ProtoReflect.Descriptor instead.

func (*GetNodeRequest) GetName

func (x *GetNodeRequest) GetName() string

func (*GetNodeRequest) ProtoMessage

func (*GetNodeRequest) ProtoMessage()

func (*GetNodeRequest) ProtoReflect

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

func (*GetNodeRequest) Reset

func (x *GetNodeRequest) Reset()

func (*GetNodeRequest) String

func (x *GetNodeRequest) String() string

type GetTensorFlowVersionRequest

type GetTensorFlowVersionRequest struct {

	// Required. The resource name.
	Name string `protobuf:"bytes,1,opt,name=name,proto3" json:"name,omitempty"`
	// contains filtered or unexported fields
}

Request for [GetTensorFlowVersion][google.cloud.tpu.v1.Tpu.GetTensorFlowVersion].

func (*GetTensorFlowVersionRequest) Descriptor deprecated

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

Deprecated: Use GetTensorFlowVersionRequest.ProtoReflect.Descriptor instead.

func (*GetTensorFlowVersionRequest) GetName

func (x *GetTensorFlowVersionRequest) GetName() string

func (*GetTensorFlowVersionRequest) ProtoMessage

func (*GetTensorFlowVersionRequest) ProtoMessage()

func (*GetTensorFlowVersionRequest) ProtoReflect

func (*GetTensorFlowVersionRequest) Reset

func (x *GetTensorFlowVersionRequest) Reset()

func (*GetTensorFlowVersionRequest) String

func (x *GetTensorFlowVersionRequest) String() string

type ListAcceleratorTypesRequest

type ListAcceleratorTypesRequest struct {

	// Required. The parent resource name.
	Parent string `protobuf:"bytes,1,opt,name=parent,proto3" json:"parent,omitempty"`
	// The maximum number of items to return.
	PageSize int32 `protobuf:"varint,2,opt,name=page_size,json=pageSize,proto3" json:"page_size,omitempty"`
	// The next_page_token value returned from a previous List request, if any.
	PageToken string `protobuf:"bytes,3,opt,name=page_token,json=pageToken,proto3" json:"page_token,omitempty"`
	// List filter.
	Filter string `protobuf:"bytes,5,opt,name=filter,proto3" json:"filter,omitempty"`
	// Sort results.
	OrderBy string `protobuf:"bytes,6,opt,name=order_by,json=orderBy,proto3" json:"order_by,omitempty"`
	// contains filtered or unexported fields
}

Request for [ListAcceleratorTypes][google.cloud.tpu.v1.Tpu.ListAcceleratorTypes].

func (*ListAcceleratorTypesRequest) Descriptor deprecated

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

Deprecated: Use ListAcceleratorTypesRequest.ProtoReflect.Descriptor instead.

func (*ListAcceleratorTypesRequest) GetFilter

func (x *ListAcceleratorTypesRequest) GetFilter() string

func (*ListAcceleratorTypesRequest) GetOrderBy

func (x *ListAcceleratorTypesRequest) GetOrderBy() string

func (*ListAcceleratorTypesRequest) GetPageSize

func (x *ListAcceleratorTypesRequest) GetPageSize() int32

func (*ListAcceleratorTypesRequest) GetPageToken

func (x *ListAcceleratorTypesRequest) GetPageToken() string

func (*ListAcceleratorTypesRequest) GetParent

func (x *ListAcceleratorTypesRequest) GetParent() string

func (*ListAcceleratorTypesRequest) ProtoMessage

func (*ListAcceleratorTypesRequest) ProtoMessage()

func (*ListAcceleratorTypesRequest) ProtoReflect

func (*ListAcceleratorTypesRequest) Reset

func (x *ListAcceleratorTypesRequest) Reset()

func (*ListAcceleratorTypesRequest) String

func (x *ListAcceleratorTypesRequest) String() string

type ListAcceleratorTypesResponse

type ListAcceleratorTypesResponse struct {

	// The listed nodes.
	AcceleratorTypes []*AcceleratorType `protobuf:"bytes,1,rep,name=accelerator_types,json=acceleratorTypes,proto3" json:"accelerator_types,omitempty"`
	// The next page token or empty if none.
	NextPageToken string `protobuf:"bytes,2,opt,name=next_page_token,json=nextPageToken,proto3" json:"next_page_token,omitempty"`
	// Locations that could not be reached.
	Unreachable []string `protobuf:"bytes,3,rep,name=unreachable,proto3" json:"unreachable,omitempty"`
	// contains filtered or unexported fields
}

Response for [ListAcceleratorTypes][google.cloud.tpu.v1.Tpu.ListAcceleratorTypes].

func (*ListAcceleratorTypesResponse) Descriptor deprecated

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

Deprecated: Use ListAcceleratorTypesResponse.ProtoReflect.Descriptor instead.

func (*ListAcceleratorTypesResponse) GetAcceleratorTypes

func (x *ListAcceleratorTypesResponse) GetAcceleratorTypes() []*AcceleratorType

func (*ListAcceleratorTypesResponse) GetNextPageToken

func (x *ListAcceleratorTypesResponse) GetNextPageToken() string

func (*ListAcceleratorTypesResponse) GetUnreachable

func (x *ListAcceleratorTypesResponse) GetUnreachable() []string

func (*ListAcceleratorTypesResponse) ProtoMessage

func (*ListAcceleratorTypesResponse) ProtoMessage()

func (*ListAcceleratorTypesResponse) ProtoReflect

func (*ListAcceleratorTypesResponse) Reset

func (x *ListAcceleratorTypesResponse) Reset()

func (*ListAcceleratorTypesResponse) String

type ListNodesRequest

type ListNodesRequest struct {

	// Required. The parent resource name.
	Parent string `protobuf:"bytes,1,opt,name=parent,proto3" json:"parent,omitempty"`
	// The maximum number of items to return.
	PageSize int32 `protobuf:"varint,2,opt,name=page_size,json=pageSize,proto3" json:"page_size,omitempty"`
	// The next_page_token value returned from a previous List request, if any.
	PageToken string `protobuf:"bytes,3,opt,name=page_token,json=pageToken,proto3" json:"page_token,omitempty"`
	// contains filtered or unexported fields
}

Request for [ListNodes][google.cloud.tpu.v1.Tpu.ListNodes].

func (*ListNodesRequest) Descriptor deprecated

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

Deprecated: Use ListNodesRequest.ProtoReflect.Descriptor instead.

func (*ListNodesRequest) GetPageSize

func (x *ListNodesRequest) GetPageSize() int32

func (*ListNodesRequest) GetPageToken

func (x *ListNodesRequest) GetPageToken() string

func (*ListNodesRequest) GetParent

func (x *ListNodesRequest) GetParent() string

func (*ListNodesRequest) ProtoMessage

func (*ListNodesRequest) ProtoMessage()

func (*ListNodesRequest) ProtoReflect

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

func (*ListNodesRequest) Reset

func (x *ListNodesRequest) Reset()

func (*ListNodesRequest) String

func (x *ListNodesRequest) String() string

type ListNodesResponse

type ListNodesResponse struct {

	// The listed nodes.
	Nodes []*Node `protobuf:"bytes,1,rep,name=nodes,proto3" json:"nodes,omitempty"`
	// The next page token or empty if none.
	NextPageToken string `protobuf:"bytes,2,opt,name=next_page_token,json=nextPageToken,proto3" json:"next_page_token,omitempty"`
	// Locations that could not be reached.
	Unreachable []string `protobuf:"bytes,3,rep,name=unreachable,proto3" json:"unreachable,omitempty"`
	// contains filtered or unexported fields
}

Response for [ListNodes][google.cloud.tpu.v1.Tpu.ListNodes].

func (*ListNodesResponse) Descriptor deprecated

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

Deprecated: Use ListNodesResponse.ProtoReflect.Descriptor instead.

func (*ListNodesResponse) GetNextPageToken

func (x *ListNodesResponse) GetNextPageToken() string

func (*ListNodesResponse) GetNodes

func (x *ListNodesResponse) GetNodes() []*Node

func (*ListNodesResponse) GetUnreachable

func (x *ListNodesResponse) GetUnreachable() []string

func (*ListNodesResponse) ProtoMessage

func (*ListNodesResponse) ProtoMessage()

func (*ListNodesResponse) ProtoReflect

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

func (*ListNodesResponse) Reset

func (x *ListNodesResponse) Reset()

func (*ListNodesResponse) String

func (x *ListNodesResponse) String() string

type ListTensorFlowVersionsRequest

type ListTensorFlowVersionsRequest struct {

	// Required. The parent resource name.
	Parent string `protobuf:"bytes,1,opt,name=parent,proto3" json:"parent,omitempty"`
	// The maximum number of items to return.
	PageSize int32 `protobuf:"varint,2,opt,name=page_size,json=pageSize,proto3" json:"page_size,omitempty"`
	// The next_page_token value returned from a previous List request, if any.
	PageToken string `protobuf:"bytes,3,opt,name=page_token,json=pageToken,proto3" json:"page_token,omitempty"`
	// List filter.
	Filter string `protobuf:"bytes,5,opt,name=filter,proto3" json:"filter,omitempty"`
	// Sort results.
	OrderBy string `protobuf:"bytes,6,opt,name=order_by,json=orderBy,proto3" json:"order_by,omitempty"`
	// contains filtered or unexported fields
}

Request for [ListTensorFlowVersions][google.cloud.tpu.v1.Tpu.ListTensorFlowVersions].

func (*ListTensorFlowVersionsRequest) Descriptor deprecated

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

Deprecated: Use ListTensorFlowVersionsRequest.ProtoReflect.Descriptor instead.

func (*ListTensorFlowVersionsRequest) GetFilter

func (x *ListTensorFlowVersionsRequest) GetFilter() string

func (*ListTensorFlowVersionsRequest) GetOrderBy

func (x *ListTensorFlowVersionsRequest) GetOrderBy() string

func (*ListTensorFlowVersionsRequest) GetPageSize

func (x *ListTensorFlowVersionsRequest) GetPageSize() int32

func (*ListTensorFlowVersionsRequest) GetPageToken

func (x *ListTensorFlowVersionsRequest) GetPageToken() string

func (*ListTensorFlowVersionsRequest) GetParent

func (x *ListTensorFlowVersionsRequest) GetParent() string

func (*ListTensorFlowVersionsRequest) ProtoMessage

func (*ListTensorFlowVersionsRequest) ProtoMessage()

func (*ListTensorFlowVersionsRequest) ProtoReflect

func (*ListTensorFlowVersionsRequest) Reset

func (x *ListTensorFlowVersionsRequest) Reset()

func (*ListTensorFlowVersionsRequest) String

type ListTensorFlowVersionsResponse

type ListTensorFlowVersionsResponse struct {

	// The listed nodes.
	TensorflowVersions []*TensorFlowVersion `protobuf:"bytes,1,rep,name=tensorflow_versions,json=tensorflowVersions,proto3" json:"tensorflow_versions,omitempty"`
	// The next page token or empty if none.
	NextPageToken string `protobuf:"bytes,2,opt,name=next_page_token,json=nextPageToken,proto3" json:"next_page_token,omitempty"`
	// Locations that could not be reached.
	Unreachable []string `protobuf:"bytes,3,rep,name=unreachable,proto3" json:"unreachable,omitempty"`
	// contains filtered or unexported fields
}

Response for [ListTensorFlowVersions][google.cloud.tpu.v1.Tpu.ListTensorFlowVersions].

func (*ListTensorFlowVersionsResponse) Descriptor deprecated

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

Deprecated: Use ListTensorFlowVersionsResponse.ProtoReflect.Descriptor instead.

func (*ListTensorFlowVersionsResponse) GetNextPageToken

func (x *ListTensorFlowVersionsResponse) GetNextPageToken() string

func (*ListTensorFlowVersionsResponse) GetTensorflowVersions

func (x *ListTensorFlowVersionsResponse) GetTensorflowVersions() []*TensorFlowVersion

func (*ListTensorFlowVersionsResponse) GetUnreachable

func (x *ListTensorFlowVersionsResponse) GetUnreachable() []string

func (*ListTensorFlowVersionsResponse) ProtoMessage

func (*ListTensorFlowVersionsResponse) ProtoMessage()

func (*ListTensorFlowVersionsResponse) ProtoReflect

func (*ListTensorFlowVersionsResponse) Reset

func (x *ListTensorFlowVersionsResponse) Reset()

func (*ListTensorFlowVersionsResponse) String

type NetworkEndpoint

type NetworkEndpoint struct {

	// The IP address of this network endpoint.
	IpAddress string `protobuf:"bytes,1,opt,name=ip_address,json=ipAddress,proto3" json:"ip_address,omitempty"`
	// The port of this network endpoint.
	Port int32 `protobuf:"varint,2,opt,name=port,proto3" json:"port,omitempty"`
	// contains filtered or unexported fields
}

A network endpoint over which a TPU worker can be reached.

func (*NetworkEndpoint) Descriptor deprecated

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

Deprecated: Use NetworkEndpoint.ProtoReflect.Descriptor instead.

func (*NetworkEndpoint) GetIpAddress

func (x *NetworkEndpoint) GetIpAddress() string

func (*NetworkEndpoint) GetPort

func (x *NetworkEndpoint) GetPort() int32

func (*NetworkEndpoint) ProtoMessage

func (*NetworkEndpoint) ProtoMessage()

func (*NetworkEndpoint) ProtoReflect

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

func (*NetworkEndpoint) Reset

func (x *NetworkEndpoint) Reset()

func (*NetworkEndpoint) String

func (x *NetworkEndpoint) String() string

type Node

type Node struct {

	// Output only. Immutable. The name of the TPU
	Name string `protobuf:"bytes,1,opt,name=name,proto3" json:"name,omitempty"`
	// The user-supplied description of the TPU. Maximum of 512 characters.
	Description string `protobuf:"bytes,3,opt,name=description,proto3" json:"description,omitempty"`
	// Required. The type of hardware accelerators associated with this node.
	AcceleratorType string `protobuf:"bytes,5,opt,name=accelerator_type,json=acceleratorType,proto3" json:"accelerator_type,omitempty"`
	// Output only. DEPRECATED! Use network_endpoints instead.
	// The network address for the TPU Node as visible to Compute Engine
	// instances.
	//
	// Deprecated: Marked as deprecated in google/cloud/tpu/v1/cloud_tpu.proto.
	IpAddress string `protobuf:"bytes,8,opt,name=ip_address,json=ipAddress,proto3" json:"ip_address,omitempty"`
	// Output only. DEPRECATED! Use network_endpoints instead.
	// The network port for the TPU Node as visible to Compute Engine instances.
	//
	// Deprecated: Marked as deprecated in google/cloud/tpu/v1/cloud_tpu.proto.
	Port string `protobuf:"bytes,14,opt,name=port,proto3" json:"port,omitempty"`
	// Output only. The current state for the TPU Node.
	State Node_State `protobuf:"varint,9,opt,name=state,proto3,enum=google.cloud.tpu.v1.Node_State" json:"state,omitempty"`
	// Output only. If this field is populated, it contains a description of why
	// the TPU Node is unhealthy.
	HealthDescription string `protobuf:"bytes,10,opt,name=health_description,json=healthDescription,proto3" json:"health_description,omitempty"`
	// Required. The version of Tensorflow running in the Node.
	TensorflowVersion string `protobuf:"bytes,11,opt,name=tensorflow_version,json=tensorflowVersion,proto3" json:"tensorflow_version,omitempty"`
	// The name of a network they wish to peer the TPU node to. It must be a
	// preexisting Compute Engine network inside of the project on which this API
	// has been activated. If none is provided, "default" will be used.
	Network string `protobuf:"bytes,12,opt,name=network,proto3" json:"network,omitempty"`
	// The CIDR block that the TPU node will use when selecting an IP address.
	// This CIDR block must be a /29 block; the Compute Engine networks API
	// forbids a smaller block, and using a larger block would be wasteful (a
	// node can only consume one IP address). Errors will occur if the CIDR block
	// has already been used for a currently existing TPU node, the CIDR block
	// conflicts with any subnetworks in the user's provided network, or the
	// provided network is peered with another network that is using that CIDR
	// block.
	CidrBlock string `protobuf:"bytes,13,opt,name=cidr_block,json=cidrBlock,proto3" json:"cidr_block,omitempty"`
	// Output only. The service account used to run the tensor flow services
	// within the node. To share resources, including Google Cloud Storage data,
	// with the Tensorflow job running in the Node, this account must have
	// permissions to that data.
	ServiceAccount string `protobuf:"bytes,15,opt,name=service_account,json=serviceAccount,proto3" json:"service_account,omitempty"`
	// Output only. The time when the node was created.
	CreateTime *timestamppb.Timestamp `protobuf:"bytes,16,opt,name=create_time,json=createTime,proto3" json:"create_time,omitempty"`
	// The scheduling options for this node.
	SchedulingConfig *SchedulingConfig `protobuf:"bytes,17,opt,name=scheduling_config,json=schedulingConfig,proto3" json:"scheduling_config,omitempty"`
	// Output only. The network endpoints where TPU workers can be accessed and
	// sent work. It is recommended that Tensorflow clients of the node reach out
	// to the 0th entry in this map first.
	NetworkEndpoints []*NetworkEndpoint `protobuf:"bytes,21,rep,name=network_endpoints,json=networkEndpoints,proto3" json:"network_endpoints,omitempty"`
	// The health status of the TPU node.
	Health Node_Health `protobuf:"varint,22,opt,name=health,proto3,enum=google.cloud.tpu.v1.Node_Health" json:"health,omitempty"`
	// Resource labels to represent user-provided metadata.
	Labels map[string]string `` /* 154-byte string literal not displayed */
	// Whether the VPC peering for the node is set up through Service Networking
	// API. The VPC Peering should be set up before provisioning the node.
	// If this field is set, cidr_block field should not be specified. If the
	// network, that you want to peer the TPU Node to, is Shared VPC networks,
	// the node must be created with this this field enabled.
	UseServiceNetworking bool `protobuf:"varint,27,opt,name=use_service_networking,json=useServiceNetworking,proto3" json:"use_service_networking,omitempty"`
	// Output only. The API version that created this Node.
	ApiVersion Node_ApiVersion `` /* 134-byte string literal not displayed */
	// Output only. The Symptoms that have occurred to the TPU Node.
	Symptoms []*Symptom `protobuf:"bytes,39,rep,name=symptoms,proto3" json:"symptoms,omitempty"`
	// contains filtered or unexported fields
}

A TPU instance.

func (*Node) Descriptor deprecated

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

Deprecated: Use Node.ProtoReflect.Descriptor instead.

func (*Node) GetAcceleratorType

func (x *Node) GetAcceleratorType() string

func (*Node) GetApiVersion

func (x *Node) GetApiVersion() Node_ApiVersion

func (*Node) GetCidrBlock

func (x *Node) GetCidrBlock() string

func (*Node) GetCreateTime

func (x *Node) GetCreateTime() *timestamppb.Timestamp

func (*Node) GetDescription

func (x *Node) GetDescription() string

func (*Node) GetHealth

func (x *Node) GetHealth() Node_Health

func (*Node) GetHealthDescription

func (x *Node) GetHealthDescription() string

func (*Node) GetIpAddress deprecated

func (x *Node) GetIpAddress() string

Deprecated: Marked as deprecated in google/cloud/tpu/v1/cloud_tpu.proto.

func (*Node) GetLabels

func (x *Node) GetLabels() map[string]string

func (*Node) GetName

func (x *Node) GetName() string

func (*Node) GetNetwork

func (x *Node) GetNetwork() string

func (*Node) GetNetworkEndpoints

func (x *Node) GetNetworkEndpoints() []*NetworkEndpoint

func (*Node) GetPort deprecated

func (x *Node) GetPort() string

Deprecated: Marked as deprecated in google/cloud/tpu/v1/cloud_tpu.proto.

func (*Node) GetSchedulingConfig

func (x *Node) GetSchedulingConfig() *SchedulingConfig

func (*Node) GetServiceAccount

func (x *Node) GetServiceAccount() string

func (*Node) GetState

func (x *Node) GetState() Node_State

func (*Node) GetSymptoms

func (x *Node) GetSymptoms() []*Symptom

func (*Node) GetTensorflowVersion

func (x *Node) GetTensorflowVersion() string

func (*Node) GetUseServiceNetworking

func (x *Node) GetUseServiceNetworking() bool

func (*Node) ProtoMessage

func (*Node) ProtoMessage()

func (*Node) ProtoReflect

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

func (*Node) Reset

func (x *Node) Reset()

func (*Node) String

func (x *Node) String() string

type Node_ApiVersion

type Node_ApiVersion int32

TPU API Version.

const (
	// API version is unknown.
	Node_API_VERSION_UNSPECIFIED Node_ApiVersion = 0
	// TPU API V1Alpha1 version.
	Node_V1_ALPHA1 Node_ApiVersion = 1
	// TPU API V1 version.
	Node_V1 Node_ApiVersion = 2
	// TPU API V2Alpha1 version.
	Node_V2_ALPHA1 Node_ApiVersion = 3
)

func (Node_ApiVersion) Descriptor

func (Node_ApiVersion) Enum

func (x Node_ApiVersion) Enum() *Node_ApiVersion

func (Node_ApiVersion) EnumDescriptor deprecated

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

Deprecated: Use Node_ApiVersion.Descriptor instead.

func (Node_ApiVersion) Number

func (Node_ApiVersion) String

func (x Node_ApiVersion) String() string

func (Node_ApiVersion) Type

type Node_Health

type Node_Health int32

Health defines the status of a TPU node as reported by Health Monitor.

const (
	// Health status is unknown: not initialized or failed to retrieve.
	Node_HEALTH_UNSPECIFIED Node_Health = 0
	// The resource is healthy.
	Node_HEALTHY Node_Health = 1
	// The resource is unhealthy.
	Node_DEPRECATED_UNHEALTHY Node_Health = 2
	// The resource is unresponsive.
	Node_TIMEOUT Node_Health = 3
	// The in-guest ML stack is unhealthy.
	Node_UNHEALTHY_TENSORFLOW Node_Health = 4
	// The node is under maintenance/priority boost caused rescheduling and
	// will resume running once rescheduled.
	Node_UNHEALTHY_MAINTENANCE Node_Health = 5
)

func (Node_Health) Descriptor

func (Node_Health) Enum

func (x Node_Health) Enum() *Node_Health

func (Node_Health) EnumDescriptor deprecated

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

Deprecated: Use Node_Health.Descriptor instead.

func (Node_Health) Number

func (x Node_Health) Number() protoreflect.EnumNumber

func (Node_Health) String

func (x Node_Health) String() string

func (Node_Health) Type

type Node_State

type Node_State int32

Represents the different states of a TPU node during its lifecycle.

const (
	// TPU node state is not known/set.
	Node_STATE_UNSPECIFIED Node_State = 0
	// TPU node is being created.
	Node_CREATING Node_State = 1
	// TPU node has been created.
	Node_READY Node_State = 2
	// TPU node is restarting.
	Node_RESTARTING Node_State = 3
	// TPU node is undergoing reimaging.
	Node_REIMAGING Node_State = 4
	// TPU node is being deleted.
	Node_DELETING Node_State = 5
	// TPU node is being repaired and may be unusable. Details can be
	// found in the `help_description` field.
	Node_REPAIRING Node_State = 6
	// TPU node is stopped.
	Node_STOPPED Node_State = 8
	// TPU node is currently stopping.
	Node_STOPPING Node_State = 9
	// TPU node is currently starting.
	Node_STARTING Node_State = 10
	// TPU node has been preempted. Only applies to Preemptible TPU Nodes.
	Node_PREEMPTED Node_State = 11
	// TPU node has been terminated due to maintenance or has reached the end of
	// its life cycle (for preemptible nodes).
	Node_TERMINATED Node_State = 12
	// TPU node is currently hiding.
	Node_HIDING Node_State = 13
	// TPU node has been hidden.
	Node_HIDDEN Node_State = 14
	// TPU node is currently unhiding.
	Node_UNHIDING Node_State = 15
	// TPU node has unknown state after a failed repair.
	Node_UNKNOWN Node_State = 16
)

func (Node_State) Descriptor

func (Node_State) Descriptor() protoreflect.EnumDescriptor

func (Node_State) Enum

func (x Node_State) Enum() *Node_State

func (Node_State) EnumDescriptor deprecated

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

Deprecated: Use Node_State.Descriptor instead.

func (Node_State) Number

func (x Node_State) Number() protoreflect.EnumNumber

func (Node_State) String

func (x Node_State) String() string

func (Node_State) Type

type OperationMetadata

type OperationMetadata struct {

	// The time the operation was created.
	CreateTime *timestamppb.Timestamp `protobuf:"bytes,1,opt,name=create_time,json=createTime,proto3" json:"create_time,omitempty"`
	// The time the operation finished running.
	EndTime *timestamppb.Timestamp `protobuf:"bytes,2,opt,name=end_time,json=endTime,proto3" json:"end_time,omitempty"`
	// Target of the operation - for example
	// projects/project-1/connectivityTests/test-1
	Target string `protobuf:"bytes,3,opt,name=target,proto3" json:"target,omitempty"`
	// Name of the verb executed by the operation.
	Verb string `protobuf:"bytes,4,opt,name=verb,proto3" json:"verb,omitempty"`
	// Human-readable status of the operation, if any.
	StatusDetail string `protobuf:"bytes,5,opt,name=status_detail,json=statusDetail,proto3" json:"status_detail,omitempty"`
	// Specifies if cancellation was requested for the operation.
	CancelRequested bool `protobuf:"varint,6,opt,name=cancel_requested,json=cancelRequested,proto3" json:"cancel_requested,omitempty"`
	// API version.
	ApiVersion string `protobuf:"bytes,7,opt,name=api_version,json=apiVersion,proto3" json:"api_version,omitempty"`
	// contains filtered or unexported fields
}

Metadata describing an [Operation][google.longrunning.Operation]

func (*OperationMetadata) Descriptor deprecated

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

Deprecated: Use OperationMetadata.ProtoReflect.Descriptor instead.

func (*OperationMetadata) GetApiVersion

func (x *OperationMetadata) GetApiVersion() string

func (*OperationMetadata) GetCancelRequested

func (x *OperationMetadata) GetCancelRequested() bool

func (*OperationMetadata) GetCreateTime

func (x *OperationMetadata) GetCreateTime() *timestamppb.Timestamp

func (*OperationMetadata) GetEndTime

func (x *OperationMetadata) GetEndTime() *timestamppb.Timestamp

func (*OperationMetadata) GetStatusDetail

func (x *OperationMetadata) GetStatusDetail() string

func (*OperationMetadata) GetTarget

func (x *OperationMetadata) GetTarget() string

func (*OperationMetadata) GetVerb

func (x *OperationMetadata) GetVerb() string

func (*OperationMetadata) ProtoMessage

func (*OperationMetadata) ProtoMessage()

func (*OperationMetadata) ProtoReflect

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

func (*OperationMetadata) Reset

func (x *OperationMetadata) Reset()

func (*OperationMetadata) String

func (x *OperationMetadata) String() string

type ReimageNodeRequest

type ReimageNodeRequest struct {

	// The resource name.
	Name string `protobuf:"bytes,1,opt,name=name,proto3" json:"name,omitempty"`
	// The version for reimage to create.
	TensorflowVersion string `protobuf:"bytes,2,opt,name=tensorflow_version,json=tensorflowVersion,proto3" json:"tensorflow_version,omitempty"`
	// contains filtered or unexported fields
}

Request for [ReimageNode][google.cloud.tpu.v1.Tpu.ReimageNode].

func (*ReimageNodeRequest) Descriptor deprecated

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

Deprecated: Use ReimageNodeRequest.ProtoReflect.Descriptor instead.

func (*ReimageNodeRequest) GetName

func (x *ReimageNodeRequest) GetName() string

func (*ReimageNodeRequest) GetTensorflowVersion

func (x *ReimageNodeRequest) GetTensorflowVersion() string

func (*ReimageNodeRequest) ProtoMessage

func (*ReimageNodeRequest) ProtoMessage()

func (*ReimageNodeRequest) ProtoReflect

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

func (*ReimageNodeRequest) Reset

func (x *ReimageNodeRequest) Reset()

func (*ReimageNodeRequest) String

func (x *ReimageNodeRequest) String() string

type SchedulingConfig

type SchedulingConfig struct {

	// Defines whether the node is preemptible.
	Preemptible bool `protobuf:"varint,1,opt,name=preemptible,proto3" json:"preemptible,omitempty"`
	// Whether the node is created under a reservation.
	Reserved bool `protobuf:"varint,2,opt,name=reserved,proto3" json:"reserved,omitempty"`
	// contains filtered or unexported fields
}

Sets the scheduling options for this node.

func (*SchedulingConfig) Descriptor deprecated

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

Deprecated: Use SchedulingConfig.ProtoReflect.Descriptor instead.

func (*SchedulingConfig) GetPreemptible

func (x *SchedulingConfig) GetPreemptible() bool

func (*SchedulingConfig) GetReserved

func (x *SchedulingConfig) GetReserved() bool

func (*SchedulingConfig) ProtoMessage

func (*SchedulingConfig) ProtoMessage()

func (*SchedulingConfig) ProtoReflect

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

func (*SchedulingConfig) Reset

func (x *SchedulingConfig) Reset()

func (*SchedulingConfig) String

func (x *SchedulingConfig) String() string

type StartNodeRequest

type StartNodeRequest struct {

	// The resource name.
	Name string `protobuf:"bytes,1,opt,name=name,proto3" json:"name,omitempty"`
	// contains filtered or unexported fields
}

Request for [StartNode][google.cloud.tpu.v1.Tpu.StartNode].

func (*StartNodeRequest) Descriptor deprecated

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

Deprecated: Use StartNodeRequest.ProtoReflect.Descriptor instead.

func (*StartNodeRequest) GetName

func (x *StartNodeRequest) GetName() string

func (*StartNodeRequest) ProtoMessage

func (*StartNodeRequest) ProtoMessage()

func (*StartNodeRequest) ProtoReflect

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

func (*StartNodeRequest) Reset

func (x *StartNodeRequest) Reset()

func (*StartNodeRequest) String

func (x *StartNodeRequest) String() string

type StopNodeRequest

type StopNodeRequest struct {

	// The resource name.
	Name string `protobuf:"bytes,1,opt,name=name,proto3" json:"name,omitempty"`
	// contains filtered or unexported fields
}

Request for [StopNode][google.cloud.tpu.v1.Tpu.StopNode].

func (*StopNodeRequest) Descriptor deprecated

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

Deprecated: Use StopNodeRequest.ProtoReflect.Descriptor instead.

func (*StopNodeRequest) GetName

func (x *StopNodeRequest) GetName() string

func (*StopNodeRequest) ProtoMessage

func (*StopNodeRequest) ProtoMessage()

func (*StopNodeRequest) ProtoReflect

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

func (*StopNodeRequest) Reset

func (x *StopNodeRequest) Reset()

func (*StopNodeRequest) String

func (x *StopNodeRequest) String() string

type Symptom

type Symptom struct {

	// Timestamp when the Symptom is created.
	CreateTime *timestamppb.Timestamp `protobuf:"bytes,1,opt,name=create_time,json=createTime,proto3" json:"create_time,omitempty"`
	// Type of the Symptom.
	SymptomType Symptom_SymptomType `` /* 140-byte string literal not displayed */
	// Detailed information of the current Symptom.
	Details string `protobuf:"bytes,3,opt,name=details,proto3" json:"details,omitempty"`
	// A string used to uniquely distinguish a worker within a TPU node.
	WorkerId string `protobuf:"bytes,4,opt,name=worker_id,json=workerId,proto3" json:"worker_id,omitempty"`
	// contains filtered or unexported fields
}

A Symptom instance.

func (*Symptom) Descriptor deprecated

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

Deprecated: Use Symptom.ProtoReflect.Descriptor instead.

func (*Symptom) GetCreateTime

func (x *Symptom) GetCreateTime() *timestamppb.Timestamp

func (*Symptom) GetDetails

func (x *Symptom) GetDetails() string

func (*Symptom) GetSymptomType

func (x *Symptom) GetSymptomType() Symptom_SymptomType

func (*Symptom) GetWorkerId

func (x *Symptom) GetWorkerId() string

func (*Symptom) ProtoMessage

func (*Symptom) ProtoMessage()

func (*Symptom) ProtoReflect

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

func (*Symptom) Reset

func (x *Symptom) Reset()

func (*Symptom) String

func (x *Symptom) String() string

type Symptom_SymptomType

type Symptom_SymptomType int32

SymptomType represents the different types of Symptoms that a TPU can be at.

const (
	// Unspecified symptom.
	Symptom_SYMPTOM_TYPE_UNSPECIFIED Symptom_SymptomType = 0
	// TPU VM memory is low.
	Symptom_LOW_MEMORY Symptom_SymptomType = 1
	// TPU runtime is out of memory.
	Symptom_OUT_OF_MEMORY Symptom_SymptomType = 2
	// TPU runtime execution has timed out.
	Symptom_EXECUTE_TIMED_OUT Symptom_SymptomType = 3
	// TPU runtime fails to construct a mesh that recognizes each TPU device's
	// neighbors.
	Symptom_MESH_BUILD_FAIL Symptom_SymptomType = 4
	// TPU HBM is out of memory.
	Symptom_HBM_OUT_OF_MEMORY Symptom_SymptomType = 5
	// Abusive behaviors have been identified on the current project.
	Symptom_PROJECT_ABUSE Symptom_SymptomType = 6
)

func (Symptom_SymptomType) Descriptor

func (Symptom_SymptomType) Enum

func (Symptom_SymptomType) EnumDescriptor deprecated

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

Deprecated: Use Symptom_SymptomType.Descriptor instead.

func (Symptom_SymptomType) Number

func (Symptom_SymptomType) String

func (x Symptom_SymptomType) String() string

func (Symptom_SymptomType) Type

type TensorFlowVersion

type TensorFlowVersion struct {

	// The resource name.
	Name string `protobuf:"bytes,1,opt,name=name,proto3" json:"name,omitempty"`
	// the tensorflow version.
	Version string `protobuf:"bytes,2,opt,name=version,proto3" json:"version,omitempty"`
	// contains filtered or unexported fields
}

A tensorflow version that a Node can be configured with.

func (*TensorFlowVersion) Descriptor deprecated

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

Deprecated: Use TensorFlowVersion.ProtoReflect.Descriptor instead.

func (*TensorFlowVersion) GetName

func (x *TensorFlowVersion) GetName() string

func (*TensorFlowVersion) GetVersion

func (x *TensorFlowVersion) GetVersion() string

func (*TensorFlowVersion) ProtoMessage

func (*TensorFlowVersion) ProtoMessage()

func (*TensorFlowVersion) ProtoReflect

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

func (*TensorFlowVersion) Reset

func (x *TensorFlowVersion) Reset()

func (*TensorFlowVersion) String

func (x *TensorFlowVersion) String() string

type TpuClient

type TpuClient interface {
	// Lists nodes.
	ListNodes(ctx context.Context, in *ListNodesRequest, opts ...grpc.CallOption) (*ListNodesResponse, error)
	// Gets the details of a node.
	GetNode(ctx context.Context, in *GetNodeRequest, opts ...grpc.CallOption) (*Node, error)
	// Creates a node.
	CreateNode(ctx context.Context, in *CreateNodeRequest, opts ...grpc.CallOption) (*longrunningpb.Operation, error)
	// Deletes a node.
	DeleteNode(ctx context.Context, in *DeleteNodeRequest, opts ...grpc.CallOption) (*longrunningpb.Operation, error)
	// Reimages a node's OS.
	ReimageNode(ctx context.Context, in *ReimageNodeRequest, opts ...grpc.CallOption) (*longrunningpb.Operation, error)
	// Stops a node, this operation is only available with single TPU nodes.
	StopNode(ctx context.Context, in *StopNodeRequest, opts ...grpc.CallOption) (*longrunningpb.Operation, error)
	// Starts a node.
	StartNode(ctx context.Context, in *StartNodeRequest, opts ...grpc.CallOption) (*longrunningpb.Operation, error)
	// List TensorFlow versions supported by this API.
	ListTensorFlowVersions(ctx context.Context, in *ListTensorFlowVersionsRequest, opts ...grpc.CallOption) (*ListTensorFlowVersionsResponse, error)
	// Gets TensorFlow Version.
	GetTensorFlowVersion(ctx context.Context, in *GetTensorFlowVersionRequest, opts ...grpc.CallOption) (*TensorFlowVersion, error)
	// Lists accelerator types supported by this API.
	ListAcceleratorTypes(ctx context.Context, in *ListAcceleratorTypesRequest, opts ...grpc.CallOption) (*ListAcceleratorTypesResponse, error)
	// Gets AcceleratorType.
	GetAcceleratorType(ctx context.Context, in *GetAcceleratorTypeRequest, opts ...grpc.CallOption) (*AcceleratorType, error)
}

TpuClient is the client API for Tpu 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 NewTpuClient

func NewTpuClient(cc grpc.ClientConnInterface) TpuClient

type TpuServer

type TpuServer interface {
	// Lists nodes.
	ListNodes(context.Context, *ListNodesRequest) (*ListNodesResponse, error)
	// Gets the details of a node.
	GetNode(context.Context, *GetNodeRequest) (*Node, error)
	// Creates a node.
	CreateNode(context.Context, *CreateNodeRequest) (*longrunningpb.Operation, error)
	// Deletes a node.
	DeleteNode(context.Context, *DeleteNodeRequest) (*longrunningpb.Operation, error)
	// Reimages a node's OS.
	ReimageNode(context.Context, *ReimageNodeRequest) (*longrunningpb.Operation, error)
	// Stops a node, this operation is only available with single TPU nodes.
	StopNode(context.Context, *StopNodeRequest) (*longrunningpb.Operation, error)
	// Starts a node.
	StartNode(context.Context, *StartNodeRequest) (*longrunningpb.Operation, error)
	// List TensorFlow versions supported by this API.
	ListTensorFlowVersions(context.Context, *ListTensorFlowVersionsRequest) (*ListTensorFlowVersionsResponse, error)
	// Gets TensorFlow Version.
	GetTensorFlowVersion(context.Context, *GetTensorFlowVersionRequest) (*TensorFlowVersion, error)
	// Lists accelerator types supported by this API.
	ListAcceleratorTypes(context.Context, *ListAcceleratorTypesRequest) (*ListAcceleratorTypesResponse, error)
	// Gets AcceleratorType.
	GetAcceleratorType(context.Context, *GetAcceleratorTypeRequest) (*AcceleratorType, error)
}

TpuServer is the server API for Tpu service. All implementations should embed UnimplementedTpuServer for forward compatibility

type UnimplementedTpuServer

type UnimplementedTpuServer struct {
}

UnimplementedTpuServer should be embedded to have forward compatible implementations.

func (UnimplementedTpuServer) CreateNode

func (UnimplementedTpuServer) DeleteNode

func (UnimplementedTpuServer) GetAcceleratorType

func (UnimplementedTpuServer) GetNode

func (UnimplementedTpuServer) GetTensorFlowVersion

func (UnimplementedTpuServer) ListNodes

func (UnimplementedTpuServer) ReimageNode

func (UnimplementedTpuServer) StartNode

func (UnimplementedTpuServer) StopNode

type UnsafeTpuServer added in v1.8.4

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

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

Jump to

Keyboard shortcuts

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