1load("//tensorflow:tensorflow.bzl", "tf_python_pybind_extension") 2 3# buildifier: disable=same-origin-load 4load("//tensorflow:tensorflow.bzl", "tf_py_test") 5 6package( 7 default_visibility = ["//tensorflow:internal"], 8 licenses = ["notice"], 9) 10 11tf_python_pybind_extension( 12 name = "_pywrap_server_lib", 13 srcs = ["server_lib_wrapper.cc"], 14 deps = [ 15 "//tensorflow/core:lib", 16 "//tensorflow/core/data/service:server_lib_headers_lib", 17 "//tensorflow/python:pybind11_lib", 18 "//tensorflow/python:pybind11_status", 19 "//third_party/python_runtime:headers", 20 "@com_google_absl//absl/strings", 21 "@pybind11", 22 ], 23) 24 25py_library( 26 name = "server_lib", 27 srcs = ["server_lib.py"], 28 srcs_version = "PY3", 29 visibility = [ 30 "//visibility:public", 31 ], 32 deps = [ 33 ":_pywrap_server_lib", 34 ":_pywrap_utils", 35 ], 36) 37 38tf_py_test( 39 name = "server_lib_test", 40 srcs = ["server_lib_test.py"], 41 deps = [ 42 ":server_lib", 43 "//tensorflow/core:protos_all_py", 44 "//tensorflow/python:errors", 45 "//tensorflow/python:platform_test", 46 "//tensorflow/python/profiler:profiler_client", 47 ], 48) 49 50tf_python_pybind_extension( 51 name = "_pywrap_utils", 52 srcs = ["utils_wrapper.cc"], 53 deps = [ 54 "//tensorflow/core/data/service:py_utils", 55 "//tensorflow/python:pybind11_lib", 56 "//third_party/python_runtime:headers", 57 "@pybind11", 58 ], 59) 60 61py_library( 62 name = "service", 63 srcs = ["__init__.py"], 64 srcs_version = "PY3", 65 deps = [ 66 ":server_lib", 67 "//tensorflow/python/data/experimental/ops:data_service_ops", 68 ], 69) 70