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

Add possibility to include default optimizers in custom optimizer list

Browse Source
This commit is contained in:
Sami Kama 2018-09-21 11:17:52 -07:00
parent 23552a8b2f
commit 9bce52aa9f
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

@ -215,6 +215,16 @@ Status MetaOptimizer::InitializeCustomGraphOptimizers(
TF_RETURN_IF_ERROR(custom_optimizer->Init(&optimizer_config)); TF_RETURN_IF_ERROR(custom_optimizer->Init(&optimizer_config));
optimizers->push_back(std::move(custom_optimizer)); optimizers->push_back(std::move(custom_optimizer));
} else { } 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: " VLOG(2) << "Can't register an optimizer by name: "
<< optimizer_config.name(); << optimizer_config.name();
} }