From c5dec17afa367c73a646d0a785d4a269429e12c3 Mon Sep 17 00:00:00 2001 From: "A. Unique TensorFlower" Date: Fri, 3 May 2019 14:05:52 -0700 Subject: [PATCH] Update the profiler_analysis_pb2_grpc.py PiperOrigin-RevId: 246569715 --- tensorflow/python/tpu/profiler/BUILD | 6 +- .../profiler/profiler_analysis_pb2_grpc.py | 125 ++++++++++++++++ .../tpu_profiler_analysis_pb2_grpc.py | 138 ------------------ 3 files changed, 128 insertions(+), 141 deletions(-) create mode 100644 tensorflow/python/tpu/profiler/profiler_analysis_pb2_grpc.py delete mode 100644 tensorflow/python/tpu/profiler/tpu_profiler_analysis_pb2_grpc.py diff --git a/tensorflow/python/tpu/profiler/BUILD b/tensorflow/python/tpu/profiler/BUILD index 60ff0a5cd40..ddfc26eeea8 100644 --- a/tensorflow/python/tpu/profiler/BUILD +++ b/tensorflow/python/tpu/profiler/BUILD @@ -11,7 +11,7 @@ py_library( srcs = ["__init__.py"], srcs_version = "PY2AND3", deps = [ - ":tpu_profiler_analysis_pb2_grpc", + ":profiler_analysis_pb2_grpc", "//tensorflow/core:protos_all_py", "//tensorflow/core/profiler:profiler_analysis_proto_py", "//tensorflow/core/profiler:protos_all_py", @@ -20,8 +20,8 @@ py_library( ) py_library( - name = "tpu_profiler_analysis_pb2_grpc", - srcs = ["tpu_profiler_analysis_pb2_grpc.py"], + name = "profiler_analysis_pb2_grpc", + srcs = ["profiler_analysis_pb2_grpc.py"], srcs_version = "PY2AND3", visibility = ["//visibility:public"], deps = ["//tensorflow/core/profiler:profiler_analysis_proto_py"], diff --git a/tensorflow/python/tpu/profiler/profiler_analysis_pb2_grpc.py b/tensorflow/python/tpu/profiler/profiler_analysis_pb2_grpc.py new file mode 100644 index 00000000000..c28d1686de4 --- /dev/null +++ b/tensorflow/python/tpu/profiler/profiler_analysis_pb2_grpc.py @@ -0,0 +1,125 @@ +# Copyright 2019 The TensorFlow Authors. All Rights Reserved. +# +# Licensed under the Apache License, Version 2.0 (the "License"); +# you may not use this file except in compliance with the License. +# You may obtain a copy of the License at +# +# http://www.apache.org/licenses/LICENSE-2.0 +# +# Unless required by applicable law or agreed to in writing, software +# distributed under the License is distributed on an "AS IS" BASIS, +# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +# See the License for the specific language governing permissions and +# limitations under the License. +# ============================================================================== +# Generated by the gRPC Python protocol compiler plugin. DO NOT EDIT! +# +# Do not use pylint on generated code. +# pylint: disable=missing-docstring,g-short-docstring-punctuation,g-no-space-after-docstring-summary,invalid-name,line-too-long,unused-argument,g-doc-args +from __future__ import absolute_import +from __future__ import division +from __future__ import print_function + +import grpc + +from tensorflow.core.profiler import profiler_analysis_pb2 as third__party_dot_tensorflow_dot_core_dot_profiler_dot_profiler__analysis__pb2 + + +class ProfileAnalysisStub(object): + """////////////////////////////////////////////////////////////////////////////// + + ProfileAnalysis service provide entry point for profiling TPU and for + serving profiled data to Tensorboard through GRPC + ////////////////////////////////////////////////////////////////////////////// + """ + + def __init__(self, channel): + """Constructor. + + Args: + channel: A grpc.Channel. + """ + self.NewSession = channel.unary_unary( + '/tensorflow.ProfileAnalysis/NewSession', + request_serializer=third__party_dot_tensorflow_dot_core_dot_profiler_dot_profiler__analysis__pb2 + .NewProfileSessionRequest.SerializeToString, + response_deserializer=third__party_dot_tensorflow_dot_core_dot_profiler_dot_profiler__analysis__pb2 + .NewProfileSessionResponse.FromString, + ) + self.EnumSessions = channel.unary_unary( + '/tensorflow.ProfileAnalysis/EnumSessions', + request_serializer=third__party_dot_tensorflow_dot_core_dot_profiler_dot_profiler__analysis__pb2 + .EnumProfileSessionsAndToolsRequest.SerializeToString, + response_deserializer=third__party_dot_tensorflow_dot_core_dot_profiler_dot_profiler__analysis__pb2 + .EnumProfileSessionsAndToolsResponse.FromString, + ) + self.GetSessionToolData = channel.unary_unary( + '/tensorflow.ProfileAnalysis/GetSessionToolData', + request_serializer=third__party_dot_tensorflow_dot_core_dot_profiler_dot_profiler__analysis__pb2 + .ProfileSessionDataRequest.SerializeToString, + response_deserializer=third__party_dot_tensorflow_dot_core_dot_profiler_dot_profiler__analysis__pb2 + .ProfileSessionDataResponse.FromString, + ) + + +class ProfileAnalysisServicer(object): + """////////////////////////////////////////////////////////////////////////////// + + ProfileAnalysis service provide entry point for profiling TPU and for + serving profiled data to Tensorboard through GRPC + ////////////////////////////////////////////////////////////////////////////// + """ + + def NewSession(self, request, context): + """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. + """ + context.set_code(grpc.StatusCode.UNIMPLEMENTED) + context.set_details('Method not implemented!') + raise NotImplementedError('Method not implemented!') + + def EnumSessions(self, request, context): + """Enumerate existing sessions and return available profile tools.""" + context.set_code(grpc.StatusCode.UNIMPLEMENTED) + context.set_details('Method not implemented!') + raise NotImplementedError('Method not implemented!') + + def GetSessionToolData(self, request, context): + """Retrieve specific tool's data for specific session.""" + context.set_code(grpc.StatusCode.UNIMPLEMENTED) + context.set_details('Method not implemented!') + raise NotImplementedError('Method not implemented!') + + +def add_ProfileAnalysisServicer_to_server(servicer, server): + rpc_method_handlers = { + 'NewSession': + grpc.unary_unary_rpc_method_handler( + servicer.NewSession, + request_deserializer=third__party_dot_tensorflow_dot_core_dot_profiler_dot_profiler__analysis__pb2 + .NewProfileSessionRequest.FromString, + response_serializer=third__party_dot_tensorflow_dot_core_dot_profiler_dot_profiler__analysis__pb2 + .NewProfileSessionResponse.SerializeToString, + ), + 'EnumSessions': + grpc.unary_unary_rpc_method_handler( + servicer.EnumSessions, + request_deserializer=third__party_dot_tensorflow_dot_core_dot_profiler_dot_profiler__analysis__pb2 + .EnumProfileSessionsAndToolsRequest.FromString, + response_serializer=third__party_dot_tensorflow_dot_core_dot_profiler_dot_profiler__analysis__pb2 + .EnumProfileSessionsAndToolsResponse.SerializeToString, + ), + 'GetSessionToolData': + grpc.unary_unary_rpc_method_handler( + servicer.GetSessionToolData, + request_deserializer=third__party_dot_tensorflow_dot_core_dot_profiler_dot_profiler__analysis__pb2 + .ProfileSessionDataRequest.FromString, + response_serializer=third__party_dot_tensorflow_dot_core_dot_profiler_dot_profiler__analysis__pb2 + .ProfileSessionDataResponse.SerializeToString, + ), + } + generic_handler = grpc.method_handlers_generic_handler( + 'tensorflow.ProfileAnalysis', rpc_method_handlers) + server.add_generic_rpc_handlers((generic_handler,)) diff --git a/tensorflow/python/tpu/profiler/tpu_profiler_analysis_pb2_grpc.py b/tensorflow/python/tpu/profiler/tpu_profiler_analysis_pb2_grpc.py deleted file mode 100644 index 8f514882887..00000000000 --- a/tensorflow/python/tpu/profiler/tpu_profiler_analysis_pb2_grpc.py +++ /dev/null @@ -1,138 +0,0 @@ -# Copyright 2017 The TensorFlow Authors. All Rights Reserved. -# -# Licensed under the Apache License, Version 2.0 (the "License"); -# you may not use this file except in compliance with the License. -# You may obtain a copy of the License at -# -# http://www.apache.org/licenses/LICENSE-2.0 -# -# Unless required by applicable law or agreed to in writing, software -# distributed under the License is distributed on an "AS IS" BASIS, -# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. -# See the License for the specific language governing permissions and -# limitations under the License. -# ============================================================================== -# Generated by the gRPC Python protocol compiler plugin. DO NOT EDIT! -# -# Do not use pylint on generated code. -# pylint: disable=missing-docstring,g-short-docstring-punctuation,g-no-space-after-docstring-summary,invalid-name,line-too-long,unused-argument,g-doc-args -from __future__ import absolute_import -from __future__ import division -from __future__ import print_function - -import grpc - -from tensorflow.contrib.tpu.profiler import tpu_profiler_analysis_pb2 as third__party_dot_tensorflow_dot_contrib_dot_tpu_dot_profiler_dot_tpu__profiler__analysis__pb2 - - -class TPUProfileAnalysisStub(object): - """////////////////////////////////////////////////////////////////////////////// - - TPUProfileAnalysis service provide entry point for profiling TPU and for - serving profiled data to Tensorboard through GRPC - ////////////////////////////////////////////////////////////////////////////// - """ - - def __init__(self, channel): - """Constructor. - - Args: - channel: A grpc.Channel. - """ - self.NewSession = channel.unary_unary( - '/tensorflow.TPUProfileAnalysis/NewSession', - request_serializer= - third__party_dot_tensorflow_dot_contrib_dot_tpu_dot_profiler_dot_tpu__profiler__analysis__pb2. - NewProfileSessionRequest.SerializeToString, - response_deserializer= - third__party_dot_tensorflow_dot_contrib_dot_tpu_dot_profiler_dot_tpu__profiler__analysis__pb2. - NewProfileSessionResponse.FromString, - ) - self.EnumSessions = channel.unary_unary( - '/tensorflow.TPUProfileAnalysis/EnumSessions', - request_serializer= - third__party_dot_tensorflow_dot_contrib_dot_tpu_dot_profiler_dot_tpu__profiler__analysis__pb2. - EnumProfileSessionsAndToolsRequest.SerializeToString, - response_deserializer= - third__party_dot_tensorflow_dot_contrib_dot_tpu_dot_profiler_dot_tpu__profiler__analysis__pb2. - EnumProfileSessionsAndToolsResponse.FromString, - ) - self.GetSessionToolData = channel.unary_unary( - '/tensorflow.TPUProfileAnalysis/GetSessionToolData', - request_serializer= - third__party_dot_tensorflow_dot_contrib_dot_tpu_dot_profiler_dot_tpu__profiler__analysis__pb2. - ProfileSessionDataRequest.SerializeToString, - response_deserializer= - third__party_dot_tensorflow_dot_contrib_dot_tpu_dot_profiler_dot_tpu__profiler__analysis__pb2. - ProfileSessionDataResponse.FromString, - ) - - -class TPUProfileAnalysisServicer(object): - """////////////////////////////////////////////////////////////////////////////// - - TPUProfileAnalysis service provide entry point for profiling TPU and for - serving profiled data to Tensorboard through GRPC - ////////////////////////////////////////////////////////////////////////////// - """ - - def NewSession(self, request, context): - """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. - """ - context.set_code(grpc.StatusCode.UNIMPLEMENTED) - context.set_details('Method not implemented!') - raise NotImplementedError('Method not implemented!') - - def EnumSessions(self, request, context): - """Enumerate existing sessions and return available profile tools. - """ - context.set_code(grpc.StatusCode.UNIMPLEMENTED) - context.set_details('Method not implemented!') - raise NotImplementedError('Method not implemented!') - - def GetSessionToolData(self, request, context): - """Retrieve specific tool's data for specific session. - """ - context.set_code(grpc.StatusCode.UNIMPLEMENTED) - context.set_details('Method not implemented!') - raise NotImplementedError('Method not implemented!') - - -def add_TPUProfileAnalysisServicer_to_server(servicer, server): - rpc_method_handlers = { - 'NewSession': - grpc.unary_unary_rpc_method_handler( - servicer.NewSession, - request_deserializer= - third__party_dot_tensorflow_dot_contrib_dot_tpu_dot_profiler_dot_tpu__profiler__analysis__pb2. - NewProfileSessionRequest.FromString, - response_serializer= - third__party_dot_tensorflow_dot_contrib_dot_tpu_dot_profiler_dot_tpu__profiler__analysis__pb2. - NewProfileSessionResponse.SerializeToString, - ), - 'EnumSessions': - grpc.unary_unary_rpc_method_handler( - servicer.EnumSessions, - request_deserializer= - third__party_dot_tensorflow_dot_contrib_dot_tpu_dot_profiler_dot_tpu__profiler__analysis__pb2. - EnumProfileSessionsAndToolsRequest.FromString, - response_serializer= - third__party_dot_tensorflow_dot_contrib_dot_tpu_dot_profiler_dot_tpu__profiler__analysis__pb2. - EnumProfileSessionsAndToolsResponse.SerializeToString, - ), - 'GetSessionToolData': - grpc.unary_unary_rpc_method_handler( - servicer.GetSessionToolData, - request_deserializer= - third__party_dot_tensorflow_dot_contrib_dot_tpu_dot_profiler_dot_tpu__profiler__analysis__pb2. - ProfileSessionDataRequest.FromString, - response_serializer= - third__party_dot_tensorflow_dot_contrib_dot_tpu_dot_profiler_dot_tpu__profiler__analysis__pb2. - ProfileSessionDataResponse.SerializeToString, - ), - } - generic_handler = grpc.method_handlers_generic_handler( - 'tensorflow.TPUProfileAnalysis', rpc_method_handlers) - server.add_generic_rpc_handlers((generic_handler,))