Class TpuGrpc.TpuStub (2.56.0)

public static final class TpuGrpc.TpuStub extends AbstractAsyncStub<TpuGrpc.TpuStub>

A stub to allow clients to do asynchronous 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.AbstractAsyncStub > TpuGrpc.TpuStub

Methods

build(Channel channel, CallOptions callOptions)

protected TpuGrpc.TpuStub build(Channel channel, CallOptions callOptions)
Parameters
Name Description
channel io.grpc.Channel
callOptions io.grpc.CallOptions
Returns
Type Description
TpuGrpc.TpuStub
Overrides
io.grpc.stub.AbstractStub.build(io.grpc.Channel,io.grpc.CallOptions)

createNode(CreateNodeRequest request, StreamObserver<Operation> responseObserver)

public void createNode(CreateNodeRequest request, StreamObserver<Operation> responseObserver)

Creates a node.

Parameters
Name Description
request CreateNodeRequest
responseObserver io.grpc.stub.StreamObserver<Operation>

deleteNode(DeleteNodeRequest request, StreamObserver<Operation> responseObserver)

public void deleteNode(DeleteNodeRequest request, StreamObserver<Operation> responseObserver)

Deletes a node.

Parameters
Name Description
request DeleteNodeRequest
responseObserver io.grpc.stub.StreamObserver<Operation>

getAcceleratorType(GetAcceleratorTypeRequest request, StreamObserver<AcceleratorType> responseObserver)

public void getAcceleratorType(GetAcceleratorTypeRequest request, StreamObserver<AcceleratorType> responseObserver)

Gets AcceleratorType.

Parameters
Name Description
request GetAcceleratorTypeRequest
responseObserver io.grpc.stub.StreamObserver<AcceleratorType>

getNode(GetNodeRequest request, StreamObserver<Node> responseObserver)

public void getNode(GetNodeRequest request, StreamObserver<Node> responseObserver)

Gets the details of a node.

Parameters
Name Description
request GetNodeRequest
responseObserver io.grpc.stub.StreamObserver<Node>

getTensorFlowVersion(GetTensorFlowVersionRequest request, StreamObserver<TensorFlowVersion> responseObserver)

public void getTensorFlowVersion(GetTensorFlowVersionRequest request, StreamObserver<TensorFlowVersion> responseObserver)

Gets TensorFlow Version.

Parameters
Name Description
request GetTensorFlowVersionRequest
responseObserver io.grpc.stub.StreamObserver<TensorFlowVersion>

listAcceleratorTypes(ListAcceleratorTypesRequest request, StreamObserver<ListAcceleratorTypesResponse> responseObserver)

public void listAcceleratorTypes(ListAcceleratorTypesRequest request, StreamObserver<ListAcceleratorTypesResponse> responseObserver)

Lists accelerator types supported by this API.

Parameters
Name Description
request ListAcceleratorTypesRequest
responseObserver io.grpc.stub.StreamObserver<ListAcceleratorTypesResponse>

listNodes(ListNodesRequest request, StreamObserver<ListNodesResponse> responseObserver)

public void listNodes(ListNodesRequest request, StreamObserver<ListNodesResponse> responseObserver)

Lists nodes.

Parameters
Name Description
request ListNodesRequest
responseObserver io.grpc.stub.StreamObserver<ListNodesResponse>

listTensorFlowVersions(ListTensorFlowVersionsRequest request, StreamObserver<ListTensorFlowVersionsResponse> responseObserver)

public void listTensorFlowVersions(ListTensorFlowVersionsRequest request, StreamObserver<ListTensorFlowVersionsResponse> responseObserver)

List TensorFlow versions supported by this API.

Parameters
Name Description
request ListTensorFlowVersionsRequest
responseObserver io.grpc.stub.StreamObserver<ListTensorFlowVersionsResponse>

reimageNode(ReimageNodeRequest request, StreamObserver<Operation> responseObserver)

public void reimageNode(ReimageNodeRequest request, StreamObserver<Operation> responseObserver)

Reimages a node's OS.

Parameters
Name Description
request ReimageNodeRequest
responseObserver io.grpc.stub.StreamObserver<Operation>

startNode(StartNodeRequest request, StreamObserver<Operation> responseObserver)

public void startNode(StartNodeRequest request, StreamObserver<Operation> responseObserver)

Starts a node.

Parameters
Name Description
request StartNodeRequest
responseObserver io.grpc.stub.StreamObserver<Operation>

stopNode(StopNodeRequest request, StreamObserver<Operation> responseObserver)

public void stopNode(StopNodeRequest request, StreamObserver<Operation> responseObserver)

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

Parameters
Name Description
request StopNodeRequest
responseObserver io.grpc.stub.StreamObserver<Operation>