Class TpuGrpc.TpuStub (2.19.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
NameDescription
channelio.grpc.Channel
callOptionsio.grpc.CallOptions
Returns
TypeDescription
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
NameDescription
requestCreateNodeRequest
responseObserverio.grpc.stub.StreamObserver<Operation>

deleteNode(DeleteNodeRequest request, StreamObserver<Operation> responseObserver)

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

Deletes a node.

Parameters
NameDescription
requestDeleteNodeRequest
responseObserverio.grpc.stub.StreamObserver<Operation>

getAcceleratorType(GetAcceleratorTypeRequest request, StreamObserver<AcceleratorType> responseObserver)

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

Gets AcceleratorType.

Parameters
NameDescription
requestGetAcceleratorTypeRequest
responseObserverio.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
NameDescription
requestGetNodeRequest
responseObserverio.grpc.stub.StreamObserver<Node>

getTensorFlowVersion(GetTensorFlowVersionRequest request, StreamObserver<TensorFlowVersion> responseObserver)

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

Gets TensorFlow Version.

Parameters
NameDescription
requestGetTensorFlowVersionRequest
responseObserverio.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
NameDescription
requestListAcceleratorTypesRequest
responseObserverio.grpc.stub.StreamObserver<ListAcceleratorTypesResponse>

listNodes(ListNodesRequest request, StreamObserver<ListNodesResponse> responseObserver)

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

Lists nodes.

Parameters
NameDescription
requestListNodesRequest
responseObserverio.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
NameDescription
requestListTensorFlowVersionsRequest
responseObserverio.grpc.stub.StreamObserver<ListTensorFlowVersionsResponse>

reimageNode(ReimageNodeRequest request, StreamObserver<Operation> responseObserver)

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

Reimages a node's OS.

Parameters
NameDescription
requestReimageNodeRequest
responseObserverio.grpc.stub.StreamObserver<Operation>

startNode(StartNodeRequest request, StreamObserver<Operation> responseObserver)

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

Starts a node.

Parameters
NameDescription
requestStartNodeRequest
responseObserverio.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
NameDescription
requestStopNodeRequest
responseObserverio.grpc.stub.StreamObserver<Operation>