Class TpuGrpc.TpuFutureStub (2.56.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 v1

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
Name Description
channel io.grpc.Channel
callOptions io.grpc.CallOptions
Returns
Type Description
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
Name Description
request CreateNodeRequest
Returns
Type Description
com.google.common.util.concurrent.ListenableFuture<Operation>

deleteNode(DeleteNodeRequest request)

public ListenableFuture<Operation> deleteNode(DeleteNodeRequest request)

Deletes a node.

Parameter
Name Description
request DeleteNodeRequest
Returns
Type Description
com.google.common.util.concurrent.ListenableFuture<Operation>

getAcceleratorType(GetAcceleratorTypeRequest request)

public ListenableFuture<AcceleratorType> getAcceleratorType(GetAcceleratorTypeRequest request)

Gets AcceleratorType.

Parameter
Name Description
request GetAcceleratorTypeRequest
Returns
Type Description
com.google.common.util.concurrent.ListenableFuture<AcceleratorType>

getNode(GetNodeRequest request)

public ListenableFuture<Node> getNode(GetNodeRequest request)

Gets the details of a node.

Parameter
Name Description
request GetNodeRequest
Returns
Type Description
com.google.common.util.concurrent.ListenableFuture<Node>

getTensorFlowVersion(GetTensorFlowVersionRequest request)

public ListenableFuture<TensorFlowVersion> getTensorFlowVersion(GetTensorFlowVersionRequest request)

Gets TensorFlow Version.

Parameter
Name Description
request GetTensorFlowVersionRequest
Returns
Type Description
com.google.common.util.concurrent.ListenableFuture<TensorFlowVersion>

listAcceleratorTypes(ListAcceleratorTypesRequest request)

public ListenableFuture<ListAcceleratorTypesResponse> listAcceleratorTypes(ListAcceleratorTypesRequest request)

Lists accelerator types supported by this API.

Parameter
Name Description
request ListAcceleratorTypesRequest
Returns
Type Description
com.google.common.util.concurrent.ListenableFuture<ListAcceleratorTypesResponse>

listNodes(ListNodesRequest request)

public ListenableFuture<ListNodesResponse> listNodes(ListNodesRequest request)

Lists nodes.

Parameter
Name Description
request ListNodesRequest
Returns
Type Description
com.google.common.util.concurrent.ListenableFuture<ListNodesResponse>

listTensorFlowVersions(ListTensorFlowVersionsRequest request)

public ListenableFuture<ListTensorFlowVersionsResponse> listTensorFlowVersions(ListTensorFlowVersionsRequest request)

List TensorFlow versions supported by this API.

Parameter
Name Description
request ListTensorFlowVersionsRequest
Returns
Type Description
com.google.common.util.concurrent.ListenableFuture<ListTensorFlowVersionsResponse>

reimageNode(ReimageNodeRequest request)

public ListenableFuture<Operation> reimageNode(ReimageNodeRequest request)

Reimages a node's OS.

Parameter
Name Description
request ReimageNodeRequest
Returns
Type Description
com.google.common.util.concurrent.ListenableFuture<Operation>

startNode(StartNodeRequest request)

public ListenableFuture<Operation> startNode(StartNodeRequest request)

Starts a node.

Parameter
Name Description
request StartNodeRequest
Returns
Type Description
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
Name Description
request StopNodeRequest
Returns
Type Description
com.google.common.util.concurrent.ListenableFuture<Operation>