1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
| package(
| default_visibility = ["//visibility:public"],
| )
|
| licenses(["notice"]) # Apache 2.0
|
| load("//tensorflow:tensorflow.bzl", "tf_py_wrap_cc")
|
| cc_library(
| name = "numpy",
| srcs = ["numpy.cc"],
| hdrs = ["numpy.h"],
| deps = [
| "//third_party/py/numpy:headers",
| "//third_party/python_runtime:headers",
| ],
| )
|
| cc_library(
| name = "interpreter_wrapper_lib",
| srcs = ["interpreter_wrapper.cc"],
| hdrs = ["interpreter_wrapper.h"],
| deps = [
| ":numpy",
| ":python_error_reporter",
| ":python_utils",
| "//tensorflow/lite:framework",
| "//tensorflow/lite:string_util",
| "//tensorflow/lite/kernels:builtin_ops",
| "//third_party/py/numpy:headers",
| "//third_party/python_runtime:headers",
| "@com_google_absl//absl/memory",
| ],
| )
|
| cc_library(
| name = "python_error_reporter",
| srcs = ["python_error_reporter.cc"],
| hdrs = ["python_error_reporter.h"],
| deps = [
| "//tensorflow/lite/core/api",
| "//third_party/python_runtime:headers",
| ],
| )
|
| cc_library(
| name = "python_utils",
| srcs = ["python_utils.cc"],
| hdrs = ["python_utils.h"],
| deps = [
| ":numpy",
| "//tensorflow/lite:framework",
| "//tensorflow/lite:string_util",
| "//third_party/py/numpy:headers",
| "//third_party/python_runtime:headers",
| ],
| )
|
| tf_py_wrap_cc(
| name = "tensorflow_wrap_interpreter_wrapper",
| srcs = [
| "interpreter_wrapper.i",
| ],
| deps = [
| ":interpreter_wrapper_lib",
| "//third_party/python_runtime:headers",
| ],
| )
|
|
