
tensorflow.TPUProfileAnalysisGrpc Maven / Gradle / Ivy
package tensorflow;
import static io.grpc.MethodDescriptor.generateFullMethodName;
import static io.grpc.stub.ClientCalls.asyncBidiStreamingCall;
import static io.grpc.stub.ClientCalls.asyncClientStreamingCall;
import static io.grpc.stub.ClientCalls.asyncServerStreamingCall;
import static io.grpc.stub.ClientCalls.asyncUnaryCall;
import static io.grpc.stub.ClientCalls.blockingServerStreamingCall;
import static io.grpc.stub.ClientCalls.blockingUnaryCall;
import static io.grpc.stub.ClientCalls.futureUnaryCall;
import static io.grpc.stub.ServerCalls.asyncBidiStreamingCall;
import static io.grpc.stub.ServerCalls.asyncClientStreamingCall;
import static io.grpc.stub.ServerCalls.asyncServerStreamingCall;
import static io.grpc.stub.ServerCalls.asyncUnaryCall;
import static io.grpc.stub.ServerCalls.asyncUnimplementedStreamingCall;
import static io.grpc.stub.ServerCalls.asyncUnimplementedUnaryCall;
/**
*
*//////////////////////////////////////////////////////////////////////////////
* TPUProfileAnalysis service provide entry point for profiling TPU and for
* serving profiled data to Tensorboard through GRPC
* //////////////////////////////////////////////////////////////////////////////
*
*/
@javax.annotation.Generated(
value = "by gRPC proto compiler (version 1.18.0)",
comments = "Source: tensorflow/contrib/tpu/profiler/tpu_profiler_analysis.proto")
public final class TPUProfileAnalysisGrpc {
private TPUProfileAnalysisGrpc() {}
public static final String SERVICE_NAME = "tensorflow.TPUProfileAnalysis";
// Static method descriptors that strictly reflect the proto.
private static volatile io.grpc.MethodDescriptor getNewSessionMethod;
@io.grpc.stub.annotations.RpcMethod(
fullMethodName = SERVICE_NAME + '/' + "NewSession",
requestType = tensorflow.TpuProfilerAnalysis.NewProfileSessionRequest.class,
responseType = tensorflow.TpuProfilerAnalysis.NewProfileSessionResponse.class,
methodType = io.grpc.MethodDescriptor.MethodType.UNARY)
public static io.grpc.MethodDescriptor getNewSessionMethod() {
io.grpc.MethodDescriptor getNewSessionMethod;
if ((getNewSessionMethod = TPUProfileAnalysisGrpc.getNewSessionMethod) == null) {
synchronized (TPUProfileAnalysisGrpc.class) {
if ((getNewSessionMethod = TPUProfileAnalysisGrpc.getNewSessionMethod) == null) {
TPUProfileAnalysisGrpc.getNewSessionMethod = getNewSessionMethod =
io.grpc.MethodDescriptor.newBuilder()
.setType(io.grpc.MethodDescriptor.MethodType.UNARY)
.setFullMethodName(generateFullMethodName(
"tensorflow.TPUProfileAnalysis", "NewSession"))
.setSampledToLocalTracing(true)
.setRequestMarshaller(io.grpc.protobuf.ProtoUtils.marshaller(
tensorflow.TpuProfilerAnalysis.NewProfileSessionRequest.getDefaultInstance()))
.setResponseMarshaller(io.grpc.protobuf.ProtoUtils.marshaller(
tensorflow.TpuProfilerAnalysis.NewProfileSessionResponse.getDefaultInstance()))
.setSchemaDescriptor(new TPUProfileAnalysisMethodDescriptorSupplier("NewSession"))
.build();
}
}
}
return getNewSessionMethod;
}
private static volatile io.grpc.MethodDescriptor getEnumSessionsMethod;
@io.grpc.stub.annotations.RpcMethod(
fullMethodName = SERVICE_NAME + '/' + "EnumSessions",
requestType = tensorflow.TpuProfilerAnalysis.EnumProfileSessionsAndToolsRequest.class,
responseType = tensorflow.TpuProfilerAnalysis.EnumProfileSessionsAndToolsResponse.class,
methodType = io.grpc.MethodDescriptor.MethodType.UNARY)
public static io.grpc.MethodDescriptor getEnumSessionsMethod() {
io.grpc.MethodDescriptor getEnumSessionsMethod;
if ((getEnumSessionsMethod = TPUProfileAnalysisGrpc.getEnumSessionsMethod) == null) {
synchronized (TPUProfileAnalysisGrpc.class) {
if ((getEnumSessionsMethod = TPUProfileAnalysisGrpc.getEnumSessionsMethod) == null) {
TPUProfileAnalysisGrpc.getEnumSessionsMethod = getEnumSessionsMethod =
io.grpc.MethodDescriptor.newBuilder()
.setType(io.grpc.MethodDescriptor.MethodType.UNARY)
.setFullMethodName(generateFullMethodName(
"tensorflow.TPUProfileAnalysis", "EnumSessions"))
.setSampledToLocalTracing(true)
.setRequestMarshaller(io.grpc.protobuf.ProtoUtils.marshaller(
tensorflow.TpuProfilerAnalysis.EnumProfileSessionsAndToolsRequest.getDefaultInstance()))
.setResponseMarshaller(io.grpc.protobuf.ProtoUtils.marshaller(
tensorflow.TpuProfilerAnalysis.EnumProfileSessionsAndToolsResponse.getDefaultInstance()))
.setSchemaDescriptor(new TPUProfileAnalysisMethodDescriptorSupplier("EnumSessions"))
.build();
}
}
}
return getEnumSessionsMethod;
}
private static volatile io.grpc.MethodDescriptor getGetSessionToolDataMethod;
@io.grpc.stub.annotations.RpcMethod(
fullMethodName = SERVICE_NAME + '/' + "GetSessionToolData",
requestType = tensorflow.TpuProfilerAnalysis.ProfileSessionDataRequest.class,
responseType = tensorflow.TpuProfilerAnalysis.ProfileSessionDataResponse.class,
methodType = io.grpc.MethodDescriptor.MethodType.UNARY)
public static io.grpc.MethodDescriptor getGetSessionToolDataMethod() {
io.grpc.MethodDescriptor getGetSessionToolDataMethod;
if ((getGetSessionToolDataMethod = TPUProfileAnalysisGrpc.getGetSessionToolDataMethod) == null) {
synchronized (TPUProfileAnalysisGrpc.class) {
if ((getGetSessionToolDataMethod = TPUProfileAnalysisGrpc.getGetSessionToolDataMethod) == null) {
TPUProfileAnalysisGrpc.getGetSessionToolDataMethod = getGetSessionToolDataMethod =
io.grpc.MethodDescriptor.newBuilder()
.setType(io.grpc.MethodDescriptor.MethodType.UNARY)
.setFullMethodName(generateFullMethodName(
"tensorflow.TPUProfileAnalysis", "GetSessionToolData"))
.setSampledToLocalTracing(true)
.setRequestMarshaller(io.grpc.protobuf.ProtoUtils.marshaller(
tensorflow.TpuProfilerAnalysis.ProfileSessionDataRequest.getDefaultInstance()))
.setResponseMarshaller(io.grpc.protobuf.ProtoUtils.marshaller(
tensorflow.TpuProfilerAnalysis.ProfileSessionDataResponse.getDefaultInstance()))
.setSchemaDescriptor(new TPUProfileAnalysisMethodDescriptorSupplier("GetSessionToolData"))
.build();
}
}
}
return getGetSessionToolDataMethod;
}
/**
* Creates a new async stub that supports all call types for the service
*/
public static TPUProfileAnalysisStub newStub(io.grpc.Channel channel) {
return new TPUProfileAnalysisStub(channel);
}
/**
* Creates a new blocking-style stub that supports unary and streaming output calls on the service
*/
public static TPUProfileAnalysisBlockingStub newBlockingStub(
io.grpc.Channel channel) {
return new TPUProfileAnalysisBlockingStub(channel);
}
/**
* Creates a new ListenableFuture-style stub that supports unary calls on the service
*/
public static TPUProfileAnalysisFutureStub newFutureStub(
io.grpc.Channel channel) {
return new TPUProfileAnalysisFutureStub(channel);
}
/**
*
*//////////////////////////////////////////////////////////////////////////////
* TPUProfileAnalysis service provide entry point for profiling TPU and for
* serving profiled data to Tensorboard through GRPC
* //////////////////////////////////////////////////////////////////////////////
*
*/
public static abstract class TPUProfileAnalysisImplBase implements io.grpc.BindableService {
/**
*
* Starts a profiling session, blocks until it completes.
* TPUProfileAnalysis service delegate this to TPUProfiler service.
* Populate the profiled data in repository, then return status to caller.
*
*/
public void newSession(tensorflow.TpuProfilerAnalysis.NewProfileSessionRequest request,
io.grpc.stub.StreamObserver responseObserver) {
asyncUnimplementedUnaryCall(getNewSessionMethod(), responseObserver);
}
/**
*
* Enumerate existing sessions and return available profile tools.
*
*/
public void enumSessions(tensorflow.TpuProfilerAnalysis.EnumProfileSessionsAndToolsRequest request,
io.grpc.stub.StreamObserver responseObserver) {
asyncUnimplementedUnaryCall(getEnumSessionsMethod(), responseObserver);
}
/**
*
* Retrieve specific tool's data for specific session.
*
*/
public void getSessionToolData(tensorflow.TpuProfilerAnalysis.ProfileSessionDataRequest request,
io.grpc.stub.StreamObserver responseObserver) {
asyncUnimplementedUnaryCall(getGetSessionToolDataMethod(), responseObserver);
}
@java.lang.Override public final io.grpc.ServerServiceDefinition bindService() {
return io.grpc.ServerServiceDefinition.builder(getServiceDescriptor())
.addMethod(
getNewSessionMethod(),
asyncUnaryCall(
new MethodHandlers<
tensorflow.TpuProfilerAnalysis.NewProfileSessionRequest,
tensorflow.TpuProfilerAnalysis.NewProfileSessionResponse>(
this, METHODID_NEW_SESSION)))
.addMethod(
getEnumSessionsMethod(),
asyncUnaryCall(
new MethodHandlers<
tensorflow.TpuProfilerAnalysis.EnumProfileSessionsAndToolsRequest,
tensorflow.TpuProfilerAnalysis.EnumProfileSessionsAndToolsResponse>(
this, METHODID_ENUM_SESSIONS)))
.addMethod(
getGetSessionToolDataMethod(),
asyncUnaryCall(
new MethodHandlers<
tensorflow.TpuProfilerAnalysis.ProfileSessionDataRequest,
tensorflow.TpuProfilerAnalysis.ProfileSessionDataResponse>(
this, METHODID_GET_SESSION_TOOL_DATA)))
.build();
}
}
/**
*
*//////////////////////////////////////////////////////////////////////////////
* TPUProfileAnalysis service provide entry point for profiling TPU and for
* serving profiled data to Tensorboard through GRPC
* //////////////////////////////////////////////////////////////////////////////
*
*/
public static final class TPUProfileAnalysisStub extends io.grpc.stub.AbstractStub {
private TPUProfileAnalysisStub(io.grpc.Channel channel) {
super(channel);
}
private TPUProfileAnalysisStub(io.grpc.Channel channel,
io.grpc.CallOptions callOptions) {
super(channel, callOptions);
}
@java.lang.Override
protected TPUProfileAnalysisStub build(io.grpc.Channel channel,
io.grpc.CallOptions callOptions) {
return new TPUProfileAnalysisStub(channel, callOptions);
}
/**
*
* Starts a profiling session, blocks until it completes.
* TPUProfileAnalysis service delegate this to TPUProfiler service.
* Populate the profiled data in repository, then return status to caller.
*
*/
public void newSession(tensorflow.TpuProfilerAnalysis.NewProfileSessionRequest request,
io.grpc.stub.StreamObserver responseObserver) {
asyncUnaryCall(
getChannel().newCall(getNewSessionMethod(), getCallOptions()), request, responseObserver);
}
/**
*
* Enumerate existing sessions and return available profile tools.
*
*/
public void enumSessions(tensorflow.TpuProfilerAnalysis.EnumProfileSessionsAndToolsRequest request,
io.grpc.stub.StreamObserver responseObserver) {
asyncUnaryCall(
getChannel().newCall(getEnumSessionsMethod(), getCallOptions()), request, responseObserver);
}
/**
*
* Retrieve specific tool's data for specific session.
*
*/
public void getSessionToolData(tensorflow.TpuProfilerAnalysis.ProfileSessionDataRequest request,
io.grpc.stub.StreamObserver responseObserver) {
asyncUnaryCall(
getChannel().newCall(getGetSessionToolDataMethod(), getCallOptions()), request, responseObserver);
}
}
/**
*
*//////////////////////////////////////////////////////////////////////////////
* TPUProfileAnalysis service provide entry point for profiling TPU and for
* serving profiled data to Tensorboard through GRPC
* //////////////////////////////////////////////////////////////////////////////
*
*/
public static final class TPUProfileAnalysisBlockingStub extends io.grpc.stub.AbstractStub {
private TPUProfileAnalysisBlockingStub(io.grpc.Channel channel) {
super(channel);
}
private TPUProfileAnalysisBlockingStub(io.grpc.Channel channel,
io.grpc.CallOptions callOptions) {
super(channel, callOptions);
}
@java.lang.Override
protected TPUProfileAnalysisBlockingStub build(io.grpc.Channel channel,
io.grpc.CallOptions callOptions) {
return new TPUProfileAnalysisBlockingStub(channel, callOptions);
}
/**
*
* Starts a profiling session, blocks until it completes.
* TPUProfileAnalysis service delegate this to TPUProfiler service.
* Populate the profiled data in repository, then return status to caller.
*
*/
public tensorflow.TpuProfilerAnalysis.NewProfileSessionResponse newSession(tensorflow.TpuProfilerAnalysis.NewProfileSessionRequest request) {
return blockingUnaryCall(
getChannel(), getNewSessionMethod(), getCallOptions(), request);
}
/**
*
* Enumerate existing sessions and return available profile tools.
*
*/
public tensorflow.TpuProfilerAnalysis.EnumProfileSessionsAndToolsResponse enumSessions(tensorflow.TpuProfilerAnalysis.EnumProfileSessionsAndToolsRequest request) {
return blockingUnaryCall(
getChannel(), getEnumSessionsMethod(), getCallOptions(), request);
}
/**
*
* Retrieve specific tool's data for specific session.
*
*/
public tensorflow.TpuProfilerAnalysis.ProfileSessionDataResponse getSessionToolData(tensorflow.TpuProfilerAnalysis.ProfileSessionDataRequest request) {
return blockingUnaryCall(
getChannel(), getGetSessionToolDataMethod(), getCallOptions(), request);
}
}
/**
*
*//////////////////////////////////////////////////////////////////////////////
* TPUProfileAnalysis service provide entry point for profiling TPU and for
* serving profiled data to Tensorboard through GRPC
* //////////////////////////////////////////////////////////////////////////////
*
*/
public static final class TPUProfileAnalysisFutureStub extends io.grpc.stub.AbstractStub {
private TPUProfileAnalysisFutureStub(io.grpc.Channel channel) {
super(channel);
}
private TPUProfileAnalysisFutureStub(io.grpc.Channel channel,
io.grpc.CallOptions callOptions) {
super(channel, callOptions);
}
@java.lang.Override
protected TPUProfileAnalysisFutureStub build(io.grpc.Channel channel,
io.grpc.CallOptions callOptions) {
return new TPUProfileAnalysisFutureStub(channel, callOptions);
}
/**
*
* Starts a profiling session, blocks until it completes.
* TPUProfileAnalysis service delegate this to TPUProfiler service.
* Populate the profiled data in repository, then return status to caller.
*
*/
public com.google.common.util.concurrent.ListenableFuture newSession(
tensorflow.TpuProfilerAnalysis.NewProfileSessionRequest request) {
return futureUnaryCall(
getChannel().newCall(getNewSessionMethod(), getCallOptions()), request);
}
/**
*
* Enumerate existing sessions and return available profile tools.
*
*/
public com.google.common.util.concurrent.ListenableFuture enumSessions(
tensorflow.TpuProfilerAnalysis.EnumProfileSessionsAndToolsRequest request) {
return futureUnaryCall(
getChannel().newCall(getEnumSessionsMethod(), getCallOptions()), request);
}
/**
*
* Retrieve specific tool's data for specific session.
*
*/
public com.google.common.util.concurrent.ListenableFuture getSessionToolData(
tensorflow.TpuProfilerAnalysis.ProfileSessionDataRequest request) {
return futureUnaryCall(
getChannel().newCall(getGetSessionToolDataMethod(), getCallOptions()), request);
}
}
private static final int METHODID_NEW_SESSION = 0;
private static final int METHODID_ENUM_SESSIONS = 1;
private static final int METHODID_GET_SESSION_TOOL_DATA = 2;
private static final class MethodHandlers implements
io.grpc.stub.ServerCalls.UnaryMethod,
io.grpc.stub.ServerCalls.ServerStreamingMethod,
io.grpc.stub.ServerCalls.ClientStreamingMethod,
io.grpc.stub.ServerCalls.BidiStreamingMethod {
private final TPUProfileAnalysisImplBase serviceImpl;
private final int methodId;
MethodHandlers(TPUProfileAnalysisImplBase serviceImpl, int methodId) {
this.serviceImpl = serviceImpl;
this.methodId = methodId;
}
@java.lang.Override
@java.lang.SuppressWarnings("unchecked")
public void invoke(Req request, io.grpc.stub.StreamObserver responseObserver) {
switch (methodId) {
case METHODID_NEW_SESSION:
serviceImpl.newSession((tensorflow.TpuProfilerAnalysis.NewProfileSessionRequest) request,
(io.grpc.stub.StreamObserver) responseObserver);
break;
case METHODID_ENUM_SESSIONS:
serviceImpl.enumSessions((tensorflow.TpuProfilerAnalysis.EnumProfileSessionsAndToolsRequest) request,
(io.grpc.stub.StreamObserver) responseObserver);
break;
case METHODID_GET_SESSION_TOOL_DATA:
serviceImpl.getSessionToolData((tensorflow.TpuProfilerAnalysis.ProfileSessionDataRequest) request,
(io.grpc.stub.StreamObserver) responseObserver);
break;
default:
throw new AssertionError();
}
}
@java.lang.Override
@java.lang.SuppressWarnings("unchecked")
public io.grpc.stub.StreamObserver invoke(
io.grpc.stub.StreamObserver responseObserver) {
switch (methodId) {
default:
throw new AssertionError();
}
}
}
private static abstract class TPUProfileAnalysisBaseDescriptorSupplier
implements io.grpc.protobuf.ProtoFileDescriptorSupplier, io.grpc.protobuf.ProtoServiceDescriptorSupplier {
TPUProfileAnalysisBaseDescriptorSupplier() {}
@java.lang.Override
public com.google.protobuf.Descriptors.FileDescriptor getFileDescriptor() {
return tensorflow.TpuProfilerAnalysis.getDescriptor();
}
@java.lang.Override
public com.google.protobuf.Descriptors.ServiceDescriptor getServiceDescriptor() {
return getFileDescriptor().findServiceByName("TPUProfileAnalysis");
}
}
private static final class TPUProfileAnalysisFileDescriptorSupplier
extends TPUProfileAnalysisBaseDescriptorSupplier {
TPUProfileAnalysisFileDescriptorSupplier() {}
}
private static final class TPUProfileAnalysisMethodDescriptorSupplier
extends TPUProfileAnalysisBaseDescriptorSupplier
implements io.grpc.protobuf.ProtoMethodDescriptorSupplier {
private final String methodName;
TPUProfileAnalysisMethodDescriptorSupplier(String methodName) {
this.methodName = methodName;
}
@java.lang.Override
public com.google.protobuf.Descriptors.MethodDescriptor getMethodDescriptor() {
return getServiceDescriptor().findMethodByName(methodName);
}
}
private static volatile io.grpc.ServiceDescriptor serviceDescriptor;
public static io.grpc.ServiceDescriptor getServiceDescriptor() {
io.grpc.ServiceDescriptor result = serviceDescriptor;
if (result == null) {
synchronized (TPUProfileAnalysisGrpc.class) {
result = serviceDescriptor;
if (result == null) {
serviceDescriptor = result = io.grpc.ServiceDescriptor.newBuilder(SERVICE_NAME)
.setSchemaDescriptor(new TPUProfileAnalysisFileDescriptorSupplier())
.addMethod(getNewSessionMethod())
.addMethod(getEnumSessionsMethod())
.addMethod(getGetSessionToolDataMethod())
.build();
}
}
}
return result;
}
}
© 2015 - 2025 Weber Informatics LLC | Privacy Policy