diff options
author | Avijit <Avijit.Chakraborty@intel.com> | 2018-07-25 01:08:01 -0700 |
---|---|---|
committer | Avijit <Avijit.Chakraborty@intel.com> | 2018-07-25 01:08:01 -0700 |
commit | 1cdacb8b10d0b4687387be5fd8be978d68602a1d (patch) | |
tree | a2bf88798854a426f073325eb85d85b3ab914418 /tensorflow/python/BUILD | |
parent | f88a6f93bee89c610fa8b399d037c7a33c1a0a3e (diff) | |
parent | 3f454e4060d855f43eebe0cdc27d8c24f906d430 (diff) |
Merge remote-tracking branch 'upstream/master'
Diffstat (limited to 'tensorflow/python/BUILD')
-rw-r--r-- | tensorflow/python/BUILD | 97 |
1 files changed, 93 insertions, 4 deletions
diff --git a/tensorflow/python/BUILD b/tensorflow/python/BUILD index f2ab2f80e6..62020d3de2 100644 --- a/tensorflow/python/BUILD +++ b/tensorflow/python/BUILD @@ -74,7 +74,7 @@ py_library( visibility = [ "//tensorflow:__pkg__", "//tensorflow/python/tools:__pkg__", - "//tensorflow/tools/api/generator:__pkg__", + "//tensorflow/python/tools/api/generator:__pkg__", ], deps = [ ":array_ops", @@ -97,6 +97,7 @@ py_library( ":image_ops", ":initializers_ns", ":io_ops", + ":kernels", ":layers", ":lib", ":list_ops", @@ -706,7 +707,9 @@ py_library( "framework/error_interpolation.py", ], srcs_version = "PY2AND3", - deps = [], + deps = [ + ":util", + ], ) py_library( @@ -744,8 +747,8 @@ py_library( srcs_version = "PY2AND3", deps = [ ":framework", + ":framework_ops", ":function", - ":op_def_registry", ":tensor_shape", ":versions", "//tensorflow/core:protos_all_py", @@ -761,8 +764,10 @@ py_test( deps = [ ":array_ops", ":client_testlib", + ":constant_op", ":dtypes", ":framework_ops", + ":function", ":function_def_to_graph", ":graph_to_function_def", ":math_ops", @@ -787,6 +792,19 @@ py_library( ) py_library( + name = "kernels", + srcs = [ + "framework/kernels.py", + ], + srcs_version = "PY2AND3", + deps = [ + ":pywrap_tensorflow", + ":util", + "//tensorflow/core:protos_all_py", + ], +) + +py_library( name = "op_def_library", srcs = ["framework/op_def_library.py"], srcs_version = "PY2AND3", @@ -823,6 +841,7 @@ py_library( ":platform", ":registry", ":tensor_shape", + ":traceable_stack", ":util", ":versions", "//tensorflow/core:protos_all_py", @@ -888,6 +907,17 @@ py_library( ], ) +# This target is maintained separately from :util to provide separate visibility +# for legacy users who were granted visibility when the functions were private +# members of ops.Graph. +py_library( + name = "tf_stack", + srcs = ["util/tf_stack.py"], + srcs_version = "PY2AND3", + visibility = ["//visibility:public"], + deps = [], +) + py_library( name = "tensor_shape", srcs = ["framework/tensor_shape.py"], @@ -923,6 +953,16 @@ py_library( ) py_library( + name = "traceable_stack", + srcs = ["framework/traceable_stack.py"], + srcs_version = "PY2AND3", + visibility = ["//visibility:public"], + deps = [ + ":util", + ], +) + +py_library( name = "versions", srcs = ["framework/versions.py"], srcs_version = "PY2AND3", @@ -1019,7 +1059,9 @@ py_test( srcs_version = "PY2AND3", deps = [ ":client_testlib", + ":constant_op", ":error_interpolation", + ":traceable_stack", ], ) @@ -1208,6 +1250,21 @@ py_test( ], ) +py_test( + name = "framework_traceable_stack_test", + size = "small", + srcs = ["framework/traceable_stack_test.py"], + main = "framework/traceable_stack_test.py", + srcs_version = "PY2AND3", + deps = [ + ":framework_test_lib", + ":platform_test", + ":test_ops", + ":traceable_stack", + ":util", + ], +) + tf_gen_op_wrapper_py( name = "test_ops", out = "framework/test_ops.py", @@ -1440,6 +1497,20 @@ py_test( ], ) +py_test( + name = "framework_kernels_test", + size = "small", + srcs = ["framework/kernels_test.py"], + main = "framework/kernels_test.py", + srcs_version = "PY2AND3", + deps = [ + ":framework_test_lib", + ":kernels", + ":platform_test", + ":test_ops", + ], +) + tf_gen_op_wrapper_private_py( name = "array_ops_gen", visibility = [ @@ -2097,8 +2168,8 @@ py_library( ":linalg_ops_gen", ":linalg_ops_impl", ":math_ops", - ":nn_ops", ":random_ops", + ":util", "//third_party/py/numpy", ], ) @@ -3007,6 +3078,20 @@ cuda_py_test( ) cuda_py_test( + name = "init_ops_test", + size = "small", + srcs = ["ops/init_ops_test.py"], + additional_deps = [ + ":client_testlib", + ":init_ops", + ":framework_ops", + ":resource_variable_ops", + "//third_party/py/numpy", + "//tensorflow/python/eager:context", + ], +) + +cuda_py_test( name = "math_grad_test", size = "small", srcs = ["ops/math_grad_test.py"], @@ -3298,6 +3383,9 @@ py_library( ], ), srcs_version = "PY2AND3", + visibility = visibility + [ + "//tensorflow:__pkg__", + ], deps = [ "//third_party/py/numpy", "@org_python_pypi_backports_weakref", @@ -3320,6 +3408,7 @@ py_test( ":math_ops", ":util", "//third_party/py/numpy", + "@absl_py//absl/testing:parameterized", ], ) |