diff options
author | 2018-05-11 09:27:13 -0700 | |
---|---|---|
committer | 2018-05-11 09:30:19 -0700 | |
commit | 4aa456ef505f60fed357b9e321703468471304c7 (patch) | |
tree | 168690cbaff3ddc0efd9bccdb7c73aeac2a004b2 /tensorflow/core/grappler/optimizers/arithmetic_optimizer.cc | |
parent | 6a43945520afbf4a6e54923402ae65c1e8361dfa (diff) |
ArithmeticOptimizer assumes valid feeds in aggressive mode.
ArithmeticOptimizer depends heavily on shapes in some stages.
PiperOrigin-RevId: 196264319
Diffstat (limited to 'tensorflow/core/grappler/optimizers/arithmetic_optimizer.cc')
-rw-r--r-- | tensorflow/core/grappler/optimizers/arithmetic_optimizer.cc | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/tensorflow/core/grappler/optimizers/arithmetic_optimizer.cc b/tensorflow/core/grappler/optimizers/arithmetic_optimizer.cc index 26eca9b820..30da23d212 100644 --- a/tensorflow/core/grappler/optimizers/arithmetic_optimizer.cc +++ b/tensorflow/core/grappler/optimizers/arithmetic_optimizer.cc @@ -2526,7 +2526,8 @@ Status ArithmeticOptimizer::Optimize(Cluster* /*cluster*/, TF_RETURN_IF_ERROR(TopologicalSort(optimized_graph_)); graph_properties_.reset(new GraphProperties(optimized_item)); - const Status status = graph_properties_->InferStatically(false); + const bool assume_valid_feeds = opt_level_ == RewriterConfig::AGGRESSIVE; + const Status status = graph_properties_->InferStatically(assume_valid_feeds); const bool can_use_shapes = status.ok(); if (!can_use_shapes) { VLOG(1) << "Shape inference failed." << status.error_message(); |