diff options
author | Eugene Zhulenev <ezhulenev@google.com> | 2018-06-05 12:19:43 -0700 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-06-05 12:22:39 -0700 |
commit | 2b5f598fbd822f911ad305ae1e57325aefd50826 (patch) | |
tree | 30ced01eceaa62a99ea7908688df5f79bf4c46d6 /tensorflow/core/grappler/optimizers/arithmetic_optimizer.h | |
parent | 920df27282b3f5d03d79f54ef05cea305c2a30d7 (diff) |
Move ReplaceMulWithSquare to a separate optimizer stage.
PiperOrigin-RevId: 199338297
Diffstat (limited to 'tensorflow/core/grappler/optimizers/arithmetic_optimizer.h')
-rw-r--r-- | tensorflow/core/grappler/optimizers/arithmetic_optimizer.h | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/tensorflow/core/grappler/optimizers/arithmetic_optimizer.h b/tensorflow/core/grappler/optimizers/arithmetic_optimizer.h index e6fc311929..8e00b83a70 100644 --- a/tensorflow/core/grappler/optimizers/arithmetic_optimizer.h +++ b/tensorflow/core/grappler/optimizers/arithmetic_optimizer.h @@ -74,6 +74,7 @@ class ArithmeticOptimizer : public GraphOptimizer { bool remove_redundant_cast = true; bool remove_redundant_reshape = true; bool reorder_cast_and_transpose = true; + bool replace_mul_with_square = true; // Choose which arithmetic optimizer stages will be enabled for a given // optimization level by default. |