diff options
| author |  A. Unique TensorFlower <gardener@tensorflow.org> | 2018-05-02 16:04:09 -0700 |
|---|---|---|
| committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-05-02 16:52:33 -0700 |
| commit | 30927ec6b625121bae1b89b07f9faeaebaed321f (patch) | |
| tree | 0f54ab601134eb818ae72eb032286034245cb218 /tensorflow/python/grappler | |
| parent | 49f2afe21e3cada8951205d00e877c873a33754c (diff) | |
Mark all nodes processed by AddOpsRewrite/MinBCast stages with a tag.
PiperOrigin-RevId: 195167597
Diffstat (limited to 'tensorflow/python/grappler')
| -rw-r--r-- | tensorflow/python/grappler/layout_optimizer_test.py | 8 |
1 files changed, 6 insertions, 2 deletions
diff --git a/tensorflow/python/grappler/layout_optimizer_test.py b/tensorflow/python/grappler/layout_optimizer_test.py index e3dd4b0bdf..2d6925d1a8 100644 --- a/tensorflow/python/grappler/layout_optimizer_test.py +++ b/tensorflow/python/grappler/layout_optimizer_test.py @@ -150,10 +150,14 @@ def _loop_with_vec_and_4d(): def _get_config(layout_optimizer=True): if layout_optimizer: rewrite_options = rewriter_config_pb2.RewriterConfig( - layout_optimizer=rewriter_config_pb2.RewriterConfig.ON) + layout_optimizer=rewriter_config_pb2.RewriterConfig.ON, + # do not remove duplicated nodes + arithmetic_optimization=rewriter_config_pb2.RewriterConfig.OFF) else: rewrite_options = rewriter_config_pb2.RewriterConfig( - layout_optimizer=rewriter_config_pb2.RewriterConfig.OFF) + layout_optimizer=rewriter_config_pb2.RewriterConfig.OFF, + # do not remove duplicated nodes + arithmetic_optimization=rewriter_config_pb2.RewriterConfig.OFF) graph_options = config_pb2.GraphOptions( rewrite_options=rewrite_options, build_cost_model=1) config = config_pb2.ConfigProto(graph_options=graph_options) |