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

Generate headers with both ".proto.h" and ".pb.h" suffixes for android protos.

Browse Source 
Change: 124745447
This commit is contained in:
A. Unique TensorFlower 2016-06-13 10:09:56 -08:00 committed by TensorFlower Gardener
parent 1d8065d603
commit c41a43e24f
1 changed files with 5 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
tensorflow/tensorflow.bzl
View File

@ -71,11 +71,13 @@ def tf_android_core_proto_sources_relative():
"util/saved_tensor_slice.proto",
]
# Returns the list of pb.h headers that are generated for
# Returns the list of pb.h and proto.h headers that are generated for
# tf_android_core_proto_sources().
def tf_android_core_proto_headers():
return ["//tensorflow/core/" + p.replace(".proto", ".pb.h")
for p in tf_android_core_proto_sources_relative()]
return (["//tensorflow/core/" + p.replace(".proto", ".pb.h")
for p in tf_android_core_proto_sources_relative()] +
["//tensorflow/core/" + p.replace(".proto", ".proto.h")
for p in tf_android_core_proto_sources_relative()])
# Returns the list of protos for which proto_text headers should be generated.
def tf_proto_text_protos_relative():