Class TpuGrpc.TpuFutureStub (2.31.0)

public static final class TpuGrpc.TpuFutureStub extends AbstractFutureStub<TpuGrpc.TpuFutureStub>

A stub to allow clients to do ListenableFuture-style rpc calls to service Tpu.

Manages TPU nodes and other resources TPU API v2alpha1

Inheritance

java.lang.Object > io.grpc.stub.AbstractStub > io.grpc.stub.AbstractFutureStub > TpuGrpc.TpuFutureStub

Methods

build(Channel channel, CallOptions callOptions)

protected TpuGrpc.TpuFutureStub build(Channel channel, CallOptions callOptions)
Parameters
NameDescription
channelio.grpc.Channel
callOptionsio.grpc.CallOptions
Returns
TypeDescription
TpuGrpc.TpuFutureStub
Overrides
io.grpc.stub.AbstractStub.build(io.grpc.Channel,io.grpc.CallOptions)

createNode(CreateNodeRequest request)

public ListenableFuture<Operation> createNode(CreateNodeRequest request)

Creates a node.

Parameter
NameDescription
requestCreateNodeRequest
Returns
TypeDescription
com.google.common.util.concurrent.ListenableFuture<Operation>

createQueuedResource(CreateQueuedResourceRequest request)

public ListenableFuture<Operation> createQueuedResource(CreateQueuedResourceRequest request)

Creates a QueuedResource TPU instance.

Parameter
NameDescription
requestCreateQueuedResourceRequest
Returns
TypeDescription
com.google.common.util.concurrent.ListenableFuture<Operation>

deleteNode(DeleteNodeRequest request)

public ListenableFuture<Operation> deleteNode(DeleteNodeRequest request)

Deletes a node.

Parameter
NameDescription
requestDeleteNodeRequest
Returns
TypeDescription
com.google.common.util.concurrent.ListenableFuture<Operation>

deleteQueuedResource(DeleteQueuedResourceRequest request)

public ListenableFuture<Operation> deleteQueuedResource(DeleteQueuedResourceRequest request)

Deletes a QueuedResource TPU instance.

Parameter
NameDescription
requestDeleteQueuedResourceRequest
Returns
TypeDescription
com.google.common.util.concurrent.ListenableFuture<Operation>

generateServiceIdentity(GenerateServiceIdentityRequest request)

public ListenableFuture<GenerateServiceIdentityResponse> generateServiceIdentity(GenerateServiceIdentityRequest request)

Generates the Cloud TPU service identity for the project.

Parameter
NameDescription
requestGenerateServiceIdentityRequest
Returns
TypeDescription
com.google.common.util.concurrent.ListenableFuture<GenerateServiceIdentityResponse>

getAcceleratorType(GetAcceleratorTypeRequest request)

public ListenableFuture<AcceleratorType> getAcceleratorType(GetAcceleratorTypeRequest request)

Gets AcceleratorType.

Parameter
NameDescription
requestGetAcceleratorTypeRequest
Returns
TypeDescription
com.google.common.util.concurrent.ListenableFuture<AcceleratorType>

getGuestAttributes(GetGuestAttributesRequest request)

public ListenableFuture<GetGuestAttributesResponse> getGuestAttributes(GetGuestAttributesRequest request)

Retrieves the guest attributes for the node.

Parameter
NameDescription
requestGetGuestAttributesRequest
Returns
TypeDescription
com.google.common.util.concurrent.ListenableFuture<GetGuestAttributesResponse>

getNode(GetNodeRequest request)

public ListenableFuture<Node> getNode(GetNodeRequest request)

Gets the details of a node.

Parameter
NameDescription
requestGetNodeRequest
Returns
TypeDescription
com.google.common.util.concurrent.ListenableFuture<Node>

getQueuedResource(GetQueuedResourceRequest request)

public ListenableFuture<QueuedResource> getQueuedResource(GetQueuedResourceRequest request)

Gets details of a queued resource.

Parameter
NameDescription
requestGetQueuedResourceRequest
Returns
TypeDescription
com.google.common.util.concurrent.ListenableFuture<QueuedResource>

getRuntimeVersion(GetRuntimeVersionRequest request)

public ListenableFuture<RuntimeVersion> getRuntimeVersion(GetRuntimeVersionRequest request)

Gets a runtime version.

Parameter
NameDescription
requestGetRuntimeVersionRequest
Returns
TypeDescription
com.google.common.util.concurrent.ListenableFuture<RuntimeVersion>

listAcceleratorTypes(ListAcceleratorTypesRequest request)

public ListenableFuture<ListAcceleratorTypesResponse> listAcceleratorTypes(ListAcceleratorTypesRequest request)

Lists accelerator types supported by this API.

Parameter
NameDescription
requestListAcceleratorTypesRequest
Returns
TypeDescription
com.google.common.util.concurrent.ListenableFuture<ListAcceleratorTypesResponse>

listNodes(ListNodesRequest request)

public ListenableFuture<ListNodesResponse> listNodes(ListNodesRequest request)

Lists nodes.

Parameter
NameDescription
requestListNodesRequest
Returns
TypeDescription
com.google.common.util.concurrent.ListenableFuture<ListNodesResponse>

listQueuedResources(ListQueuedResourcesRequest request)

public ListenableFuture<ListQueuedResourcesResponse> listQueuedResources(ListQueuedResourcesRequest request)

Lists queued resources.

Parameter
NameDescription
requestListQueuedResourcesRequest
Returns
TypeDescription
com.google.common.util.concurrent.ListenableFuture<ListQueuedResourcesResponse>

listRuntimeVersions(ListRuntimeVersionsRequest request)

public ListenableFuture<ListRuntimeVersionsResponse> listRuntimeVersions(ListRuntimeVersionsRequest request)

Lists runtime versions supported by this API.

Parameter
NameDescription
requestListRuntimeVersionsRequest
Returns
TypeDescription
com.google.common.util.concurrent.ListenableFuture<ListRuntimeVersionsResponse>

resetQueuedResource(ResetQueuedResourceRequest request)

public ListenableFuture<Operation> resetQueuedResource(ResetQueuedResourceRequest request)

Resets a QueuedResource TPU instance

Parameter
NameDescription
requestResetQueuedResourceRequest
Returns
TypeDescription
com.google.common.util.concurrent.ListenableFuture<Operation>

simulateMaintenanceEvent(SimulateMaintenanceEventRequest request)

public ListenableFuture<Operation> simulateMaintenanceEvent(SimulateMaintenanceEventRequest request)

Simulates a maintenance event.

Parameter
NameDescription
requestSimulateMaintenanceEventRequest
Returns
TypeDescription
com.google.common.util.concurrent.ListenableFuture<Operation>

startNode(StartNodeRequest request)

public ListenableFuture<Operation> startNode(StartNodeRequest request)

Starts a node.

Parameter
NameDescription
requestStartNodeRequest
Returns
TypeDescription
com.google.common.util.concurrent.ListenableFuture<Operation>

stopNode(StopNodeRequest request)

public ListenableFuture<Operation> stopNode(StopNodeRequest request)

Stops a node. This operation is only available with single TPU nodes.

Parameter
NameDescription
requestStopNodeRequest
Returns
TypeDescription
com.google.common.util.concurrent.ListenableFuture<Operation>

updateNode(UpdateNodeRequest request)

public ListenableFuture<Operation> updateNode(UpdateNodeRequest request)

Updates the configurations of a node.

Parameter
NameDescription
requestUpdateNodeRequest
Returns
TypeDescription
com.google.common.util.concurrent.ListenableFuture<Operation>