diff options
author | Rachel Lim <rachelim@google.com> | 2018-09-28 16:10:45 -0700 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-09-28 16:17:55 -0700 |
commit | 478d370eb116ad2294134d75a886637a7d6da225 (patch) | |
tree | 279ef8e8a2c9abeeda583393a986f055b9be314c /tensorflow/core/grappler/optimizers/data/vectorization/BUILD | |
parent | a98bac521406bedef3ff2b9af9564b21ddda4d82 (diff) |
[tf.data] Use Graph instead of GraphDef/FunctionDef for vectorization transforms
PiperOrigin-RevId: 215011835
Diffstat (limited to 'tensorflow/core/grappler/optimizers/data/vectorization/BUILD')
-rw-r--r-- | tensorflow/core/grappler/optimizers/data/vectorization/BUILD | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/tensorflow/core/grappler/optimizers/data/vectorization/BUILD b/tensorflow/core/grappler/optimizers/data/vectorization/BUILD index 1462cb234d..37aa24b947 100644 --- a/tensorflow/core/grappler/optimizers/data/vectorization/BUILD +++ b/tensorflow/core/grappler/optimizers/data/vectorization/BUILD @@ -9,13 +9,14 @@ load("//tensorflow/core:platform/default/build_config.bzl", "tf_protos_all") VECTORIZER_DEPS = [ ":vectorizer_registry", - "//tensorflow/core/grappler/optimizers/data:function_utils", + "//tensorflow/core/grappler/optimizers/data:graph_utils", ] + tf_protos_all() cc_library( name = "vectorizer", hdrs = ["vectorizer.h"], deps = [ + "//tensorflow/core:core_cpu", "//tensorflow/core:lib", ] + tf_protos_all(), ) |