diff options
| author |  James Keeling <jtkeeling@google.com> | 2018-07-24 03:09:47 -0700 |
|---|---|---|
| committer |  TensorFlower Gardener <gardener@tensorflow.org> | 2018-07-24 03:13:49 -0700 |
| commit | c21078f527023e3074b63109fb768413f82a8f8f (patch) | |
| tree | c8b593b525286a6b51beaca02c4c893f977d272e /tensorflow/python/BUILD | |
| parent | 33035bb79b6ecb408ef83cee5fc3e52ce058f39f (diff) | |
Add Python API functions to query kernels
This is part of the work to make available kernels easier to query at runtime.
PiperOrigin-RevId: 205802663
Diffstat (limited to 'tensorflow/python/BUILD')
| -rw-r--r-- | tensorflow/python/BUILD | 28 |
1 files changed, 28 insertions, 0 deletions
diff --git a/tensorflow/python/BUILD b/tensorflow/python/BUILD index 9c7f3b7b25..814239533c 100644 --- a/tensorflow/python/BUILD +++ b/tensorflow/python/BUILD @@ -96,6 +96,7 @@ py_library( ":image_ops", ":initializers_ns", ":io_ops", + ":kernels", ":layers", ":lib", ":list_ops", @@ -790,6 +791,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", @@ -1482,6 +1496,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 = [ |