Class TpuGrpc.TpuFutureStub (2.16.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 v2

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>

deleteNode(DeleteNodeRequest request)

public ListenableFuture<Operation> deleteNode(DeleteNodeRequest request)

Deletes a node.

Parameter
NameDescription
requestDeleteNodeRequest
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>

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>

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>

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>