diff options
author | Rachel Lim <rachelim@google.com> | 2018-10-09 14:36:33 -0700 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-10-09 14:46:11 -0700 |
commit | 950cf87104bfee28e2165fe368f66337b8a1336d (patch) | |
tree | 59ace2e229776b79897c54b4be0705231d5ac9f2 /tensorflow/core/grappler/optimizers/data/vectorization/BUILD | |
parent | 35caff957424a60bd7d7e4e92a1ec87f617781c6 (diff) |
[tf.data vectorization] Add vectorizer for `Add` op
PiperOrigin-RevId: 216424512
Diffstat (limited to 'tensorflow/core/grappler/optimizers/data/vectorization/BUILD')
-rw-r--r-- | tensorflow/core/grappler/optimizers/data/vectorization/BUILD | 34 |
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(), -) |