experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Add go proto libs for profiler_analysis.proto and profiler_service.proto.

Browse Source 
PiperOrigin-RevId: 253336670
This commit is contained in:
A. Unique TensorFlower 2019-06-14 19:11:44 -07:00 committed by TensorFlower Gardener
parent 5e545bbb5c
commit 09c5e94be3
2 changed files with 11 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
tensorflow/core/profiler/BUILD
View File

@ -1,5 +1,5 @@
load("//tensorflow:tensorflow.bzl", "tf_cc_binary")
load("//tensorflow/core:platform/default/build_config.bzl", "tf_additional_all_protos", "tf_profiler_all_protos", "tf_proto_library")
load("//tensorflow/core:platform/default/build_config.bzl", "tf_proto_library", "tf_additional_all_protos")
package(
default_visibility = ["//visibility:public"],
@ -36,13 +36,19 @@ cc_library(
],
)
tf_proto_library(
name = "op_profile_proto",
srcs = ["op_profile.proto"],
cc_api_version = 2,
)
tf_proto_library(
name = "profiler_service_proto",
srcs = ["profiler_service.proto"],
has_services = 1,
cc_api_version = 2,
cc_grpc_version = 1,
protodeps = tf_profiler_all_protos() + tf_additional_all_protos(),
protodeps = [":op_profile_proto"] + tf_additional_all_protos(),
)
tf_proto_library(
@ -51,7 +57,7 @@ tf_proto_library(
has_services = 1,
cc_api_version = 2,
cc_grpc_version = 1,
protodeps = [":profiler_service_proto"] + tf_additional_all_protos(),
protodeps = [":profiler_service_proto"],
)
tf_proto_library(
@ -61,6 +67,7 @@ tf_proto_library(
exclude = [
"profiler_service.proto",
"profiler_analysis.proto",
"op_profile.proto",
],
),
cc_api_version = 2,

2
tensorflow/core/profiler/rpc/client/BUILD
View File

@ -36,7 +36,7 @@ cc_library(
"//tensorflow/core:grpc_services",
"//tensorflow/core:lib",
"//tensorflow/core:protos_all_cc",
"//tensorflow/core/profiler:protos_all_cc",
"//tensorflow/core/profiler:op_profile_proto_cc",
],
)