experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Merge pull request #22453 from samikama:custom_optimizer_ordering

Browse Source 
PiperOrigin-RevId: 214132703
This commit is contained in:
TensorFlower Gardener 2018-09-22 15:55:12 -07:00
commit 3aa23d30c2
1 changed files with 10 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
tensorflow/core/grappler/optimizers/meta_optimizer.cc
View File

@ -220,6 +220,16 @@ Status MetaOptimizer::InitializeCustomGraphOptimizers(
TF_RETURN_IF_ERROR(custom_optimizer->Init(&optimizer_config));
optimizers->push_back(std::move(custom_optimizer));
} else {
// If there are no custom optimizers with given name, try to initalize a
// default optimizer. This way, custom configurable optimizers can be
// mixed with default optimizers in any order.
auto optimizer = MakeNewOptimizer(optimizer_config.name());
if (optimizer) {
VLOG(2) << "Registered default graph optimizer: "
<< optimizer_config.name();
optimizers->push_back(std::move(optimizer));
continue;
}
VLOG(2) << "Can't register an optimizer by name: "
<< optimizer_config.name();
}