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

Add an internal-only C++ wrapper around function_ops.

Browse Source 
Change: 135130656
This commit is contained in:
A. Unique TensorFlower 2016-10-04 10:29:14 -08:00 committed by TensorFlower Gardener
parent 60afe603d1
commit bae4d97a4d
2 changed files with 14 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
tensorflow/cc/BUILD
View File

@ -317,6 +317,15 @@ tf_gen_op_wrappers_cc(
pkg = "//tensorflow/core", pkg = "//tensorflow/core",
) )
tf_gen_op_wrappers_cc(
name = "function_ops",
op_lib_names = [
"function_ops",
],
pkg = "//tensorflow/core",
visibility = ["//tensorflow:internal"],
)
cc_library( cc_library(
name = "cc_op_gen_main", name = "cc_op_gen_main",
srcs = [ srcs = [

8
tensorflow/tensorflow.bzl
View File

@ -185,7 +185,7 @@ def tf_gen_op_wrapper_cc(name, out_ops_file, pkg="",
) )
# Run the op generator. # Run the op generator.
if name == "sendrecv_ops": if name == "sendrecv_ops" or name == "function_ops":
include_internal = "1" include_internal = "1"
else: else:
include_internal = "0" include_internal = "0"
@ -224,7 +224,8 @@ def tf_gen_op_wrappers_cc(name,
"//tensorflow/cc:scope", "//tensorflow/cc:scope",
"//tensorflow/cc:const_op", "//tensorflow/cc:const_op",
], ],
op_gen="//tensorflow/cc:cc_op_gen_main"): op_gen="//tensorflow/cc:cc_op_gen_main",
visibility=None):
subsrcs = other_srcs subsrcs = other_srcs
subhdrs = other_hdrs subhdrs = other_hdrs
for n in op_lib_names: for n in op_lib_names:
@ -242,7 +243,8 @@ def tf_gen_op_wrappers_cc(name,
"//tensorflow/core:protos_all_cc", "//tensorflow/core:protos_all_cc",
], ],
copts=tf_copts(), copts=tf_copts(),
alwayslink=1,) alwayslink=1,
visibility=visibility)
# Invoke this rule in .../tensorflow/python to build the wrapper library. # Invoke this rule in .../tensorflow/python to build the wrapper library.
def tf_gen_op_wrapper_py(name, out=None, hidden=None, visibility=None, deps=[], def tf_gen_op_wrapper_py(name, out=None, hidden=None, visibility=None, deps=[],