diff options
| author |  Rachel Lim <rachelim@google.com> | 2018-09-20 10:41:29 -0700 |
|---|---|---|
| committer |  TensorFlower Gardener <gardener@tensorflow.org> | 2018-09-20 10:46:06 -0700 |
| commit | 07c9ba4bba274d424404eedab14b3fab3f072350 (patch) | |
| tree | 68d8b9f1031fa7cbf08645d69c727b2498cb0d4f /tensorflow/core/grappler/optimizers/data/vectorization_utils.cc | |
| parent | 7eacd3278207baa64e0b67fe8463d4eee59b31de (diff) | |
[tf.data] Use vectorization_utils::VectorizeMapDefun in MapVectorization optimization
PiperOrigin-RevId: 213840320
Diffstat (limited to 'tensorflow/core/grappler/optimizers/data/vectorization_utils.cc')
| -rw-r--r-- | tensorflow/core/grappler/optimizers/data/vectorization_utils.cc | 5 |
1 files changed, 5 insertions, 0 deletions
diff --git a/tensorflow/core/grappler/optimizers/data/vectorization_utils.cc b/tensorflow/core/grappler/optimizers/data/vectorization_utils.cc index 6a59eb0d32..5dd9d00511 100644 --- a/tensorflow/core/grappler/optimizers/data/vectorization_utils.cc +++ b/tensorflow/core/grappler/optimizers/data/vectorization_utils.cc @@ -333,6 +333,11 @@ void Vectorization::Vectorize() { void VectorizeMapDefun(FunctionDef* outer_scope, FunctionDef* map_defun_fn, NodeDef* map_defun_node) { + if (map_defun_node->attr().at("f").func().name() != + map_defun_fn->signature().name()) { + LOG(ERROR) << "`map_defun_fn` and `map_defun_node` do not match"; + return; + } Vectorization(outer_scope, map_defun_fn, map_defun_node).Vectorize(); } |