diff options
| author |  Benoit Steiner <bsteiner@google.com> | 2017-08-11 18:21:24 -0700 |
|---|---|---|
| committer |  TensorFlower Gardener <gardener@tensorflow.org> | 2017-08-11 18:26:30 -0700 |
| commit | 566821c7fef5a7a9528beac2ee40277210f56b31 (patch) | |
| tree | bb9a6c170737069e0b24c3d636a6ce4085d905aa | |
| parent | 9d8c88a18aa5c148ddccf7c8ceda476e44713560 (diff) | |
Don't run constant folding when xla is turned on since it isn't stable enough.
PiperOrigin-RevId: 165049643
| -rw-r--r-- | tensorflow/core/grappler/optimizers/constant_folding.cc | 3 |
1 files changed, 3 insertions, 0 deletions
diff --git a/tensorflow/core/grappler/optimizers/constant_folding.cc b/tensorflow/core/grappler/optimizers/constant_folding.cc index 50d7dea21b..f568fd4f1f 100644 --- a/tensorflow/core/grappler/optimizers/constant_folding.cc +++ b/tensorflow/core/grappler/optimizers/constant_folding.cc @@ -268,6 +268,9 @@ bool ConstantFolding::IsFoldable(const NodeDef& node) const { if (op.find("Quantized") != string::npos || op.find("Sparse") == 0) { return false; } + if (node.attr().count("_XlaCompile") > 0) { + return false; + } // Don't fold stateful ops such as TruncatedNormal. const OpDef* op_def = nullptr; |