diff options
| author |  Rachel Lim <rachelim@google.com> | 2018-10-05 17:01:01 -0700 |
|---|---|---|
| committer |  TensorFlower Gardener <gardener@tensorflow.org> | 2018-10-05 17:08:15 -0700 |
| commit | 15d399cd8590c18dc643d979883fe4201c8ea631 (patch) | |
| tree | 1c913d2a5a0a45a89f6d3343f8c8cf297ba6f8e3 /tensorflow/core/grappler/optimizers/data/vectorization/BUILD | |
| parent | ab97f1323bd2a98d20ed82dc3ff8585481961f0d (diff) | |
[tf.data vectorization] Feed inputs to vectorizers with notion of stackedness
PiperOrigin-RevId: 215989259
Diffstat (limited to 'tensorflow/core/grappler/optimizers/data/vectorization/BUILD')
| -rw-r--r-- | tensorflow/core/grappler/optimizers/data/vectorization/BUILD | 10 |
1 files changed, 10 insertions, 0 deletions
diff --git a/tensorflow/core/grappler/optimizers/data/vectorization/BUILD b/tensorflow/core/grappler/optimizers/data/vectorization/BUILD index 37aa24b947..985d6c6c3a 100644 --- a/tensorflow/core/grappler/optimizers/data/vectorization/BUILD +++ b/tensorflow/core/grappler/optimizers/data/vectorization/BUILD @@ -13,9 +13,19 @@ VECTORIZER_DEPS = [ ] + tf_protos_all() cc_library( + name = "wrapped_tensor", + hdrs = ["wrapped_tensor.h"], + deps = [ + "//tensorflow/core:core_cpu", + "//tensorflow/core:lib", + ], +) + +cc_library( name = "vectorizer", hdrs = ["vectorizer.h"], deps = [ + ":wrapped_tensor", "//tensorflow/core:core_cpu", "//tensorflow/core:lib", ] + tf_protos_all(), |