Searched refs:optimizer_name (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/contrib/learn/python/learn/estimators/ |
| D | composable_model.py | 144 def _get_default_optimizer(self, optimizer_name=None): argument
254 def _get_default_optimizer(self, optimizer_name=None): argument
255 if optimizer_name is None:
256 optimizer_name = "Ftrl"
259 return layers.OPTIMIZER_CLS_NAMES[optimizer_name](
406 def _get_default_optimizer(self, optimizer_name=None): argument
407 if optimizer_name is None:
408 optimizer_name = "Adagrad"
409 return layers.OPTIMIZER_CLS_NAMES[optimizer_name](learning_rate=0.05)
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/data/ |
| D | meta_optimizer.cc | 89 for (const auto& optimizer_name : optimizers) { in Init() local
91 CustomGraphOptimizerRegistry::CreateByNameOrNull(optimizer_name); in Init()
96 enabled_optimizers_[optimizer_name] = std::move(optimizer); in Init()
101 optimizer_name); in Init()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | graph_optimizer_stage.h | 118 explicit GraphOptimizerStage(const string& optimizer_name, in GraphOptimizerStage() argument
121 : optimizer_name_(optimizer_name), stage_name_(stage_name), ctx_(ctx) {} in GraphOptimizerStage()
125 const string& optimizer_name() const { return optimizer_name_; } in optimizer_name() function
264 VLOG(2) << "Failed to run optimizer " << stage->optimizer_name() in PassThroughAllStages()
|
| D | meta_optimizer.cc | 227 for (const string& optimizer_name : cfg_.optimizers()) { in InitializeOptimizersByName() local
228 auto optimizer = MakeNewOptimizer(optimizer_name); in InitializeOptimizersByName()
230 VLOG(2) << "Registered default graph optimizer: " << optimizer_name; in InitializeOptimizersByName()
236 CustomGraphOptimizerRegistry::CreateByNameOrNull(optimizer_name); in InitializeOptimizersByName()
239 VLOG(2) << "Registered custom graph optimizer: " << optimizer_name; in InitializeOptimizersByName()
241 GetCustomGraphOptimizerConfig(optimizer_name))); in InitializeOptimizersByName()
243 initialized_custom_optimizers.insert(optimizer_name); in InitializeOptimizersByName()
245 VLOG(2) << "Can't register an optimizer by name: " << optimizer_name; in InitializeOptimizersByName()
668 LOG(INFO) << " " << result.optimizer_name << ": " << result.result; in PrintResult()
|
| D | graph_optimizer_stage_test.cc | 39 explicit FakeOptimizerStage(const string& optimizer_name, in FakeOptimizerStage() argument
42 : GraphOptimizerStage(optimizer_name, stage_name, ctx) {} in FakeOptimizerStage()
|
| D | meta_optimizer.h | 84 string optimizer_name; member
|
| /external/tensorflow/tensorflow/contrib/opt/python/training/ |
| D | weight_decay_optimizers_test.py | 58 def doTest(self, optimizer, update_fn, optimizer_name, slot_name, argument
117 self.assertEqual("var0_%d/%s:0" % (i, optimizer_name),
|