experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Transition Lite pybind_extension targets to use the new link_in_framework argument to dynamically link to libtensorflow_framework.so.

Browse Source 
That is what `tf_binary_additional_srcs()` was doing, but now we can specify that cleanly using an argument to the bazel macro.

PiperOrigin-RevId: 301172587
Change-Id: Ia85d12b38e481638fcd598ed2d94690fa1b99b00
This commit is contained in:
Amit Patankar 2020-03-16 09:21:34 -07:00 committed by TensorFlower Gardener
parent f46ac90866
commit 8864f9738d
3 changed files with 8 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
tensorflow/lite/python/interpreter_wrapper/BUILD
View File

@ -1,5 +1,4 @@
load("//tensorflow:tensorflow.bzl", "pybind_extension") load("//tensorflow:tensorflow.bzl", "pybind_extension")
load("//tensorflow:tensorflow.bzl", "tf_binary_additional_srcs")
package( package(
default_visibility = ["//visibility:public"], default_visibility = ["//visibility:public"],
@ -66,8 +65,9 @@ pybind_extension(
name = "_pywrap_tensorflow_interpreter_wrapper", name = "_pywrap_tensorflow_interpreter_wrapper",
srcs = [ srcs = [
"interpreter_wrapper_pybind11.cc", "interpreter_wrapper_pybind11.cc",
] + tf_binary_additional_srcs(), ],
hdrs = ["interpreter_wrapper.h"], hdrs = ["interpreter_wrapper.h"],
link_in_framework = True,
module_name = "_pywrap_tensorflow_interpreter_wrapper", module_name = "_pywrap_tensorflow_interpreter_wrapper",
deps = [ deps = [
":interpreter_wrapper_lib", ":interpreter_wrapper_lib",

7
tensorflow/lite/python/optimize/BUILD
View File

@ -1,5 +1,4 @@
load("//tensorflow:tensorflow.bzl", "pybind_extension") load("//tensorflow:tensorflow.bzl", "pybind_extension")
load("//tensorflow:tensorflow.bzl", "tf_binary_additional_srcs")
package( package(
default_visibility = ["//visibility:public"], default_visibility = ["//visibility:public"],
@ -46,8 +45,9 @@ pybind_extension(
name = "_pywrap_tensorflow_lite_calibration_wrapper", name = "_pywrap_tensorflow_lite_calibration_wrapper",
srcs = [ srcs = [
"calibration_wrapper_pybind11.cc", "calibration_wrapper_pybind11.cc",
] + tf_binary_additional_srcs(), ],
hdrs = ["calibration_wrapper.h"], hdrs = ["calibration_wrapper.h"],
link_in_framework = True,
module_name = "_pywrap_tensorflow_lite_calibration_wrapper", module_name = "_pywrap_tensorflow_lite_calibration_wrapper",
deps = [ deps = [
":calibration_wrapper_lib", ":calibration_wrapper_lib",
@ -61,8 +61,9 @@ pybind_extension(
name = "_pywrap_tensorflow_lite_sparsification_wrapper", name = "_pywrap_tensorflow_lite_sparsification_wrapper",
srcs = [ srcs = [
"sparsification_wrapper_pybind11.cc", "sparsification_wrapper_pybind11.cc",
] + tf_binary_additional_srcs(), ],
hdrs = ["sparsification_wrapper.h"], hdrs = ["sparsification_wrapper.h"],
link_in_framework = True,
module_name = "_pywrap_tensorflow_lite_sparsification_wrapper", module_name = "_pywrap_tensorflow_lite_sparsification_wrapper",
deps = [ deps = [
":sparsification_wrapper_lib", ":sparsification_wrapper_lib",

4
tensorflow/lite/python/testdata/BUILD vendored
View File

@ -1,6 +1,5 @@
load("//tensorflow/lite:build_def.bzl", "tf_to_tflite") load("//tensorflow/lite:build_def.bzl", "tf_to_tflite")
load("//tensorflow:tensorflow.bzl", "pybind_extension") load("//tensorflow:tensorflow.bzl", "pybind_extension")
load("//tensorflow:tensorflow.bzl", "tf_binary_additional_srcs")
package( package(
default_visibility = ["//tensorflow:internal"], default_visibility = ["//tensorflow:internal"],
@ -92,9 +91,10 @@ pybind_extension(
name = "_pywrap_test_registerer", name = "_pywrap_test_registerer",
srcs = [ srcs = [
"test_registerer_wrapper.cc", "test_registerer_wrapper.cc",
] + tf_binary_additional_srcs(), ],
hdrs = ["test_registerer.h"], hdrs = ["test_registerer.h"],
additional_exported_symbols = ["TF_TestRegisterer"], additional_exported_symbols = ["TF_TestRegisterer"],
link_in_framework = True,
module_name = "_pywrap_test_registerer", module_name = "_pywrap_test_registerer",
deps = [ deps = [
":test_registerer", ":test_registerer",