Searched refs:custom_optimizer (Results 1 – 5 of 5) sorted by relevance
90 auto custom_optimizer = rewriter_config.add_custom_optimizers(); in CreateGrapplerRewriteConfig() local91 custom_optimizer->set_name(kOptimizerName); in CreateGrapplerRewriteConfig()94 (*custom_optimizer->mutable_parameter_map())["num_workers"] = in CreateGrapplerRewriteConfig()99 (*custom_optimizer->mutable_parameter_map())["index"] = index_attr; in CreateGrapplerRewriteConfig()
79 auto custom_optimizer = rewriter_config.add_custom_optimizers(); in CreateGrapplerRewriteConfig() local80 custom_optimizer->set_name(kOptimizerName); in CreateGrapplerRewriteConfig()83 (*custom_optimizer->mutable_parameter_map())["num_workers"] = in CreateGrapplerRewriteConfig()
76 auto custom_optimizer = rewriter_config.add_custom_optimizers(); in CreateGrapplerRewriteConfig() local77 custom_optimizer->set_name(kOptimizerName); in CreateGrapplerRewriteConfig()79 (*custom_optimizer->mutable_parameter_map())["optimizers"] in CreateGrapplerRewriteConfig()
235 auto custom_optimizer = in InitializeOptimizersByName() local238 if (custom_optimizer) { in InitializeOptimizersByName()240 TF_RETURN_IF_ERROR(custom_optimizer->Init( in InitializeOptimizersByName()242 optimizers->push_back(std::move(custom_optimizer)); in InitializeOptimizersByName()261 auto custom_optimizer = CustomGraphOptimizerRegistry::CreateByNameOrNull( in InitializeCustomGraphOptimizers() local264 if (custom_optimizer) { in InitializeCustomGraphOptimizers()267 TF_RETURN_IF_ERROR(custom_optimizer->Init(&optimizer_config)); in InitializeCustomGraphOptimizers()268 optimizers->push_back(std::move(custom_optimizer)); in InitializeCustomGraphOptimizers()
174 def custom_optimizer(): function181 optimizer=custom_optimizer,