diff options
| author |  Benoit Steiner <bsteiner@google.com> | 2018-02-23 14:22:06 -0800 |
|---|---|---|
| committer |  TensorFlower Gardener <gardener@tensorflow.org> | 2018-02-23 14:26:13 -0800 |
| commit | a8213e7d032e676b3135f1ac8ec019f86f7fcd18 (patch) | |
| tree | 256a9349fc5c5dd66ecfb63eaedcf69c4ce77f6e /tensorflow/python/grappler | |
| parent | eba67dea5a5e83e2bc49a40202233823b7ea9973 (diff) | |
Preserve user placement as much as possible when optimizing the graph
PiperOrigin-RevId: 186822511
Diffstat (limited to 'tensorflow/python/grappler')
| -rw-r--r-- | tensorflow/python/grappler/tf_optimizer.i | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/tensorflow/python/grappler/tf_optimizer.i b/tensorflow/python/grappler/tf_optimizer.i index 1b657983a4..de9326ccfc 100644 --- a/tensorflow/python/grappler/tf_optimizer.i +++ b/tensorflow/python/grappler/tf_optimizer.i @@ -100,6 +100,7 @@ PyObject* TF_OptimizeGraph( tensorflow::grappler::ItemConfig item_config; item_config.inline_functions = false; item_config.apply_optimizations = false; + item_config.ignore_user_placement = false; std::unique_ptr<tensorflow::grappler::GrapplerItem> grappler_item = tensorflow::grappler::GrapplerItemFromMetaGraphDef(graph_id, metagraph, item_config); |