Remove hourglass imports from kernel_tests

Change: 142080137

1 files changed, 7 insertions, 4 deletions

diff --git a/tensorflow/python/build_defs.bzl b/tensorflow/python/build_defs.bzl
index 71adab99e0..1efac5738c 100644
--- a/tensorflow/python/build_defs.bzl
+++ b/tensorflow/python/build_defs.bzl
@@ -3,24 +3,27 @@
 load("//tensorflow:tensorflow.bzl", "tf_gen_op_wrapper_py")
 
 # Intended only for use within this directory.
-# Generated python wrappers are private visibility, users should depend on the
+# Generated python wrappers are "private" visibility, users should depend on the
 # full python code that incorporates the wrappers.  The generated targets have
 # a _gen suffix, so that the full python version can use the bare name.
 # We also hard code the hidden_file here to reduce duplication.
 #
 # We should consider moving the "out" default pattern into here, many other
-# consumers of the tf_gen_op_wrapper_py rule would be simplified if we don't 
+# consumers of the tf_gen_op_wrapper_py rule would be simplified if we don't
 # hard code the ops/ directory.
 
 def tf_gen_op_wrapper_private_py(name, out=None, deps=[],
-                                 require_shape_functions=False):
+                                 require_shape_functions=False,
+                                 visibility=[]):
   if not name.endswith("_gen"):
     fail("name must end in _gen")
+  if not visibility:
+    visibility = ["//visibility:private"]
   bare_op_name = name[:-4] # Strip of the _gen
   tf_gen_op_wrapper_py(name=bare_op_name,
     out=out,
     hidden_file="ops/hidden_ops.txt",
-    visibility=["//visibility:private"],
+    visibility=visibility,
     deps=deps,
     require_shape_functions=require_shape_functions,
     generated_target_name=name,