aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/contrib/lite/toco/tensorflow_graph_matching/BUILD
diff options
context:
space:
mode:
Diffstat (limited to 'tensorflow/contrib/lite/toco/tensorflow_graph_matching/BUILD')
-rw-r--r--tensorflow/contrib/lite/toco/tensorflow_graph_matching/BUILD102
1 files changed, 102 insertions, 0 deletions
diff --git a/tensorflow/contrib/lite/toco/tensorflow_graph_matching/BUILD b/tensorflow/contrib/lite/toco/tensorflow_graph_matching/BUILD
new file mode 100644
index 0000000000..0c1a1141fc
--- /dev/null
+++ b/tensorflow/contrib/lite/toco/tensorflow_graph_matching/BUILD
@@ -0,0 +1,102 @@
+package(default_visibility = ["//visibility:public"])
+
+licenses(["notice"]) # Apache 2.0
+
+load(
+ "//tensorflow:tensorflow.bzl",
+ "tf_cc_test",
+)
+
+cc_library(
+ name = "cluster_utils",
+ srcs = [
+ "cluster_utils.cc",
+ ],
+ hdrs = [
+ "cluster_utils.h",
+ ],
+ deps = [
+ "//tensorflow/contrib/lite/toco:toco_port",
+ ],
+)
+
+cc_library(
+ name = "cluster",
+ srcs = [
+ "cluster.cc",
+ ],
+ hdrs = [
+ "cluster.h",
+ ],
+ deps = [
+ ":cluster_utils",
+ "//tensorflow/contrib/lite/toco:model",
+ "//tensorflow/contrib/lite/toco:tooling_util",
+ "//tensorflow/core:protos_all_cc",
+ ],
+)
+
+cc_library(
+ name = "resolve_svdf",
+ srcs = [
+ "resolve_svdf.cc",
+ ],
+ hdrs = [
+ "resolve_svdf.h",
+ ],
+ visibility = ["//visibility:public"],
+ deps = [
+ ":cluster",
+ ":cluster_utils",
+ "//tensorflow/contrib/lite/toco:model",
+ "//tensorflow/contrib/lite/toco:toco_port",
+ "//tensorflow/contrib/lite/toco:tooling_util",
+ "//tensorflow/core:lib",
+ "//tensorflow/core:protos_all_cc",
+ "@protobuf_archive//:protobuf_headers",
+ ],
+)
+
+tf_cc_test(
+ name = "resolve_svdf_test",
+ srcs = ["resolve_svdf_test.cc"],
+ deps = [
+ ":cluster",
+ ":cluster_utils",
+ ":resolve_cluster",
+ ":resolve_svdf",
+ "//tensorflow/core:lib",
+ "//tensorflow/core:protos_all_cc",
+ "@com_google_googletest//:gtest_main",
+ ],
+)
+
+cc_library(
+ name = "resolve_cluster",
+ srcs = [
+ "resolve_cluster.cc",
+ ],
+ hdrs = [
+ "resolve_cluster.h",
+ ],
+ visibility = ["//visibility:public"],
+ deps = [
+ ":cluster",
+ ":cluster_utils",
+ ":resolve_svdf",
+ "//tensorflow/contrib/lite/toco:tooling_util",
+ "//tensorflow/core:protos_all_cc",
+ ],
+)
+
+filegroup(
+ name = "all_files",
+ srcs = glob(
+ ["**/*"],
+ exclude = [
+ "**/METADATA",
+ "**/OWNERS",
+ ],
+ ),
+ visibility = ["//tensorflow:__subpackages__"],
+)