Skip to content

Commit 31d10e3

Browse files
Yao Zhangtensorflower-gardener
Yao Zhang
authored andcommitted
Add shape and remapping optimization to the disabled optimizer
list. PiperOrigin-RevId: 210602483
1 parent b43dde0 commit 31d10e3

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

tensorflow/core/grappler/clusters/cluster.cc

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -81,6 +81,8 @@ void Cluster::DisableOptimizer(bool disable) {
8181
rewriter_config->set_dependency_optimization(RewriterConfig::OFF);
8282
rewriter_config->set_constant_folding(RewriterConfig::OFF);
8383
rewriter_config->set_memory_optimization(RewriterConfig::NO_MEM_OPT);
84+
rewriter_config->set_shape_optimization(RewriterConfig::OFF);
85+
rewriter_config->set_remapping(RewriterConfig::OFF);
8486
rewriter_config->mutable_auto_parallel()->set_enable(false);
8587
rewriter_config->clear_optimizers();
8688
} else {

0 commit comments

Comments
 (0)