Searched refs:optimizer_config (Results 1 – 6 of 6) sorted by relevance
255 for (const auto& optimizer_config : cfg_.custom_optimizers()) { in InitializeCustomGraphOptimizers() local256 if (pre_initialized_optimizers.find(optimizer_config.name()) != in InitializeCustomGraphOptimizers()262 optimizer_config.name()); in InitializeCustomGraphOptimizers()266 << optimizer_config.name(); in InitializeCustomGraphOptimizers()267 TF_RETURN_IF_ERROR(custom_optimizer->Init(&optimizer_config)); in InitializeCustomGraphOptimizers()273 auto optimizer = MakeNewOptimizer(optimizer_config.name()); in InitializeCustomGraphOptimizers()276 << optimizer_config.name(); in InitializeCustomGraphOptimizers()281 << optimizer_config.name(); in InitializeCustomGraphOptimizers()
519 optimizer_config = orig_optimizer.get_config()520 optimizer = orig_optimizer.__class__.from_config(optimizer_config)
129 optimizer_config = model.optimizer.get_config()130 optimizer = type(model.optimizer).from_config(optimizer_config)
228 optimizer_config = training_config['optimizer_config']230 optimizer_config, custom_objects=custom_objects)
691 optimizer_config = model.optimizer.get_config()692 optimizer = model.optimizer.__class__.from_config(optimizer_config)
2234 optimizer_config = model.optimizer.get_config()2236 optimizer_config = None2247 _clone_optimizer(model.optimizer, optimizer_config),