diff options
author | A. Unique TensorFlower <gardener@tensorflow.org> | 2018-04-06 12:18:04 -0700 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-04-06 12:21:28 -0700 |
commit | 4d90c62824a2e4e445efab58d2c5829774a884ea (patch) | |
tree | 0da4301318b73186270d8d21e5f1afcc1b723b0d /tensorflow/core/grappler/optimizers/arithmetic_optimizer.h | |
parent | a056c115e83e6f07fd3dbb5d6439658828025024 (diff) |
Fix a few bugs in ArithmeticOptimizer and make it robust to failures of shape inference.
PiperOrigin-RevId: 191922788
Diffstat (limited to 'tensorflow/core/grappler/optimizers/arithmetic_optimizer.h')
-rw-r--r-- | tensorflow/core/grappler/optimizers/arithmetic_optimizer.h | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/tensorflow/core/grappler/optimizers/arithmetic_optimizer.h b/tensorflow/core/grappler/optimizers/arithmetic_optimizer.h index 7e81ed0a1f..39b89dedba 100644 --- a/tensorflow/core/grappler/optimizers/arithmetic_optimizer.h +++ b/tensorflow/core/grappler/optimizers/arithmetic_optimizer.h @@ -105,7 +105,7 @@ class ArithmeticOptimizer : public GraphOptimizer { // Runs peep-hole optimizations on `optimized_graph`, e.g., removing inverse // transposes. - Status SimplifyArithmeticOps(); + Status SimplifyArithmeticOps(bool can_use_shapes); // Tries to simplify the expression that roots at `node` and replaces the uses // of `node` to the simplified expression. Returns the name of the simplified // tensor (e.g. "split:1") or an emtpy string if no simplification is |