aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/core/grappler/optimizers/data/vectorization/BUILD
diff options
context:
space:
mode:
Diffstat (limited to 'tensorflow/core/grappler/optimizers/data/vectorization/BUILD')
-rw-r--r--tensorflow/core/grappler/optimizers/data/vectorization/BUILD34
1 files changed, 23 insertions, 11 deletions
diff --git a/tensorflow/core/grappler/optimizers/data/vectorization/BUILD b/tensorflow/core/grappler/optimizers/data/vectorization/BUILD
index 985d6c6c3a..09018d0124 100644
--- a/tensorflow/core/grappler/optimizers/data/vectorization/BUILD
+++ b/tensorflow/core/grappler/optimizers/data/vectorization/BUILD
@@ -9,7 +9,11 @@ load("//tensorflow/core:platform/default/build_config.bzl", "tf_protos_all")
VECTORIZER_DEPS = [
":vectorizer_registry",
+ "//tensorflow/cc:ops",
"//tensorflow/core/grappler/optimizers/data:graph_utils",
+ "//tensorflow/core:core_cpu",
+ "//tensorflow/cc:scope_internal",
+ "//tensorflow/cc:cc_ops",
] + tf_protos_all()
cc_library(
@@ -42,6 +46,24 @@ cc_library(
],
)
+tf_cc_test(
+ name = "vectorizer_registry_test",
+ srcs = ["vectorizer_registry_test.cc"],
+ deps = [
+ ":vectorizer_registry",
+ "//tensorflow/core:test",
+ "//tensorflow/core:test_main",
+ "//tensorflow/core:testlib",
+ ] + tf_protos_all(),
+)
+
+cc_library(
+ name = "add_vectorizer",
+ srcs = ["add_vectorizer.cc"],
+ deps = VECTORIZER_DEPS,
+ alwayslink = 1,
+)
+
cc_library(
name = "cast_vectorizer",
srcs = ["cast_vectorizer.cc"],
@@ -61,20 +83,10 @@ cc_library(
hdrs = ["vectorizer_registry.h"],
visibility = ["//visibility:public"],
deps = [
+ ":add_vectorizer",
":cast_vectorizer",
":unpack_vectorizer",
":vectorizer",
":vectorizer_registry",
],
)
-
-tf_cc_test(
- name = "vectorizer_registry_test",
- srcs = ["vectorizer_registry_test.cc"],
- deps = [
- ":vectorizer_registry",
- "//tensorflow/core:test",
- "//tensorflow/core:test_main",
- "//tensorflow/core:testlib",
- ] + tf_protos_all(),
-)