public class GrpcTpuStub extends TpuStub
gRPC stub implementation for the Tpu service API.
This class is for advanced usage and reflects the underlying API directly.
Static Methods
create(ClientContext clientContext)
public static final GrpcTpuStub create(ClientContext clientContext)
Parameter
Returns
Exceptions
create(ClientContext clientContext, GrpcStubCallableFactory callableFactory)
public static final GrpcTpuStub create(ClientContext clientContext, GrpcStubCallableFactory callableFactory)
Parameters
Returns
Exceptions
create(TpuStubSettings settings)
public static final GrpcTpuStub create(TpuStubSettings settings)
Parameter
Returns
Exceptions
Constructors
GrpcTpuStub(TpuStubSettings settings, ClientContext clientContext)
protected GrpcTpuStub(TpuStubSettings settings, ClientContext clientContext)
Constructs an instance of GrpcTpuStub, using the given settings. This is protected so that it
is easy to make a subclass, but otherwise, the static factory methods should be preferred.
Parameters
GrpcTpuStub(TpuStubSettings settings, ClientContext clientContext, GrpcStubCallableFactory callableFactory)
protected GrpcTpuStub(TpuStubSettings settings, ClientContext clientContext, GrpcStubCallableFactory callableFactory)
Constructs an instance of GrpcTpuStub, using the given settings. This is protected so that it
is easy to make a subclass, but otherwise, the static factory methods should be preferred.
Parameters
Methods
awaitTermination(long duration, TimeUnit unit)
public boolean awaitTermination(long duration, TimeUnit unit)
Parameters
Returns
Exceptions
close()
public final void close()
Overrides
createNodeCallable()
public UnaryCallable<CreateNodeRequest,Operation> createNodeCallable()
Returns
Overrides
createNodeOperationCallable()
public OperationCallable<CreateNodeRequest,Node,OperationMetadata> createNodeOperationCallable()
Returns
Overrides
deleteNodeCallable()
public UnaryCallable<DeleteNodeRequest,Operation> deleteNodeCallable()
Returns
Overrides
deleteNodeOperationCallable()
public OperationCallable<DeleteNodeRequest,Node,OperationMetadata> deleteNodeOperationCallable()
Returns
Overrides
getAcceleratorTypeCallable()
public UnaryCallable<GetAcceleratorTypeRequest,AcceleratorType> getAcceleratorTypeCallable()
Returns
Overrides
getNodeCallable()
public UnaryCallable<GetNodeRequest,Node> getNodeCallable()
Returns
Overrides
getOperationsStub()
public GrpcOperationsStub getOperationsStub()
Returns
Overrides
getTensorFlowVersionCallable()
public UnaryCallable<GetTensorFlowVersionRequest,TensorFlowVersion> getTensorFlowVersionCallable()
Returns
Overrides
isShutdown()
public boolean isShutdown()
Returns
isTerminated()
public boolean isTerminated()
Returns
listAcceleratorTypesCallable()
public UnaryCallable<ListAcceleratorTypesRequest,ListAcceleratorTypesResponse> listAcceleratorTypesCallable()
Returns
Overrides
listAcceleratorTypesPagedCallable()
public UnaryCallable<ListAcceleratorTypesRequest,TpuClient.ListAcceleratorTypesPagedResponse> listAcceleratorTypesPagedCallable()
Returns
Overrides
listNodesCallable()
public UnaryCallable<ListNodesRequest,ListNodesResponse> listNodesCallable()
Returns
Overrides
listNodesPagedCallable()
public UnaryCallable<ListNodesRequest,TpuClient.ListNodesPagedResponse> listNodesPagedCallable()
Returns
Overrides
listTensorFlowVersionsCallable()
public UnaryCallable<ListTensorFlowVersionsRequest,ListTensorFlowVersionsResponse> listTensorFlowVersionsCallable()
Returns
Overrides
listTensorFlowVersionsPagedCallable()
public UnaryCallable<ListTensorFlowVersionsRequest,TpuClient.ListTensorFlowVersionsPagedResponse> listTensorFlowVersionsPagedCallable()
Returns
Overrides
reimageNodeCallable()
public UnaryCallable<ReimageNodeRequest,Operation> reimageNodeCallable()
Returns
Overrides
reimageNodeOperationCallable()
public OperationCallable<ReimageNodeRequest,Node,OperationMetadata> reimageNodeOperationCallable()
Returns
Overrides
shutdown()
shutdownNow()
public void shutdownNow()
startNodeCallable()
public UnaryCallable<StartNodeRequest,Operation> startNodeCallable()
Returns
Overrides
startNodeOperationCallable()
public OperationCallable<StartNodeRequest,Node,OperationMetadata> startNodeOperationCallable()
Returns
Overrides
stopNodeCallable()
public UnaryCallable<StopNodeRequest,Operation> stopNodeCallable()
Returns
Overrides
stopNodeOperationCallable()
public OperationCallable<StopNodeRequest,Node,OperationMetadata> stopNodeOperationCallable()
Returns
Overrides