Update the profiler_analysis_pb2_grpc.py
PiperOrigin-RevId: 246569715
This commit is contained in:
A. Unique TensorFlower
TensorFlower Gardener
parent
ebdd0432f4
commit
c5dec17afa
@ -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"],
|
||||
|
||||
125
tensorflow/python/tpu/profiler/profiler_analysis_pb2_grpc.py
Normal file
125
tensorflow/python/tpu/profiler/profiler_analysis_pb2_grpc.py
Normal file
@ -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,))
|
||||
@ -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,))
|
||||
Loading…
Reference in New Issue
Block a user