diff options
| author |  A. Unique TensorFlower <gardener@tensorflow.org> | 2018-06-01 18:01:58 -0700 |
|---|---|---|
| committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-06-01 18:06:37 -0700 |
| commit | 6e5606fce0e4615880e2685a3674c498756b9cfb (patch) | |
| tree | 6e61421c775881913b7a80db9ffc86558fbe2845 /tensorflow/core/grappler/optimizers/arithmetic_optimizer.h | |
| parent | b33ba9a8e7e20e4b2378937204fe74af69982906 (diff) | |
Extract FoldMultiplyIntoConv optimization stage.
PiperOrigin-RevId: 198953044
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 0fce23a40a..ce3c633baf 100644 --- a/tensorflow/core/grappler/optimizers/arithmetic_optimizer.h +++ b/tensorflow/core/grappler/optimizers/arithmetic_optimizer.h @@ -61,6 +61,7 @@ class ArithmeticOptimizer : public GraphOptimizer { bool combine_add_to_addn = true; bool convert_sqrt_div_to_rsqrt_mul = false; bool dedup_computations = true; + bool fold_multiply_into_conv = true; bool hoist_common_factor_out_of_aggregation = true; bool hoist_cwise_unary_chains = false; bool minimize_broadcasts = true; |