Ensure linking with TensorFlow shared libraries that prodive symbol definitions used in Python extensions.
load("//tensorflow_federated/tools:build_defs.bzl", "tff_cc_cpu_gpu_test_with_tf_deps", "tff_cc_library_with_tf_deps", "tff_cc_library_with_tf_runtime_deps", "tff_cc_test_with_tf_deps", "tff_pybind_extension_with_tf_deps")
-load("@pybind11_bazel//:build_defs.bzl", "pybind_extension")
load("@rules_python//python:defs.bzl", "py_binary")
package(default_visibility = [
@@ -118,7 +117,7 @@ tff_cc_test_with_tf_deps(
],
)
-pybind_extension(
+tff_pybind_extension_with_tf_deps(
name = "executor_bindings",
srcs = ["executor_bindings.cc"],
deps = [
