1load("//tensorflow:tensorflow.bzl", "pybind_extension") 2 3package( 4 default_visibility = [ 5 "//visibility:public", 6 ], 7 licenses = ["notice"], 8) 9 10py_binary( 11 name = "modify_model_interface", 12 srcs = ["modify_model_interface.py"], 13 python_version = "PY3", 14 srcs_version = "PY3", 15 deps = [ 16 ":modify_model_interface_constants", 17 ":modify_model_interface_lib", 18 "@absl_py//absl:app", 19 "@absl_py//absl/flags", 20 ], 21) 22 23py_library( 24 name = "modify_model_interface_lib", 25 srcs = ["modify_model_interface_lib.py"], 26 srcs_version = "PY3", 27 deps = [ 28 ":_pywrap_modify_model_interface", 29 ":modify_model_interface_constants", 30 "//tensorflow:tensorflow_py", 31 "//tensorflow/lite/python:schema_py", 32 ], 33) 34 35py_test( 36 name = "modify_model_interface_lib_test", 37 srcs = ["modify_model_interface_lib_test.py"], 38 python_version = "PY3", 39 srcs_version = "PY3", 40 deps = [ 41 ":modify_model_interface_lib", 42 "//tensorflow:tensorflow_py", 43 "//tensorflow/python:client_testlib", 44 "//tensorflow/python:framework_test_lib", 45 "//third_party/py/numpy", 46 ], 47) 48 49py_library( 50 name = "modify_model_interface_constants", 51 srcs = ["modify_model_interface_constants.py"], 52 srcs_version = "PY3", 53 deps = ["//tensorflow/python:dtypes"], 54) 55 56pybind_extension( 57 name = "_pywrap_modify_model_interface", 58 srcs = ["modify_model_interface.cc"], 59 deps = [ 60 "//tensorflow/lite/schema:schema_fbs", 61 "//tensorflow/lite/tools/optimize:modify_model_interface", 62 "@pybind11", 63 ], 64) 65