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

getAcceleratorType(GetAcceleratorTypeRequest request)

public ListenableFuture<AcceleratorType> getAcceleratorType(GetAcceleratorTypeRequest request)

Gets AcceleratorType.

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

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>

getTensorFlowVersion(GetTensorFlowVersionRequest request)

public ListenableFuture<TensorFlowVersion> getTensorFlowVersion(GetTensorFlowVersionRequest request)

Gets TensorFlow Version.

Parameter
NameDescription
requestGetTensorFlowVersionRequest
Returns
TypeDescription
com.google.common.util.concurrent.ListenableFuture<TensorFlowVersion>

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>

listTensorFlowVersions(ListTensorFlowVersionsRequest request)

public ListenableFuture<ListTensorFlowVersionsResponse> listTensorFlowVersions(ListTensorFlowVersionsRequest request)

List TensorFlow versions supported by this API.

Parameter
NameDescription
requestListTensorFlowVersionsRequest
Returns
TypeDescription
com.google.common.util.concurrent.ListenableFuture<ListTensorFlowVersionsResponse>

reimageNode(ReimageNodeRequest request)

public ListenableFuture<Operation> reimageNode(ReimageNodeRequest request)

Reimages a node's OS.

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