diff options
Diffstat (limited to 'tensorflow/tools/api/generator/api_gen.bzl')
-rw-r--r-- | tensorflow/tools/api/generator/api_gen.bzl | 7 |
1 files changed, 3 insertions, 4 deletions
diff --git a/tensorflow/tools/api/generator/api_gen.bzl b/tensorflow/tools/api/generator/api_gen.bzl index e5ec86c0a0..d746b5d3e4 100644 --- a/tensorflow/tools/api/generator/api_gen.bzl +++ b/tensorflow/tools/api/generator/api_gen.bzl @@ -134,7 +134,7 @@ def gen_api_init_files( package_dep = "//tensorflow/python:no_contrib"): root_init_template_flag = "" if root_init_template: - root_init_template_flag = "--root_init_template=$(location " + root_init_template + ")" + root_init_template_flag = "--root_init_template=$(location " + root_init_template + ")" api_gen_binary_target = "create_" + package + "_api" native.py_binary( @@ -154,9 +154,8 @@ def gen_api_init_files( outs = output_files, cmd = ( "$(location :" + api_gen_binary_target + ") " + - root_init_template_flag + " --apidir=$(@D) --apiname=" + api_name + " --package=" + package + " $(OUTS)" - ), + root_init_template_flag + " --apidir=$(@D) --apiname=" + api_name + " --package=" + package + " $(OUTS)"), srcs = srcs, - tools = [":" + api_gen_binary_target], + tools = [":" + api_gen_binary_target ], visibility = ["//tensorflow:__pkg__"], ) |