Home
last modified time | relevance | path

Searched defs:optimizer_name (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/core/grappler/optimizers/data/
Dmeta_optimizer.cc80 const string& optimizer_name = split[0]; in ToConfigMap() local
193 for (const auto& optimizer_name : optimizers) { in Init() local
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dgraph_optimizer_stage.h118 explicit GraphOptimizerStage(const string& optimizer_name, in GraphOptimizerStage()
125 const string& optimizer_name() const { return optimizer_name_; } in optimizer_name() function
Dmeta_optimizer.h97 string optimizer_name; member
Dgraph_optimizer_stage_test.cc39 explicit FakeOptimizerStage(const string& optimizer_name, in FakeOptimizerStage()
Dmeta_optimizer.cc321 for (const string& optimizer_name : cfg_.optimizers()) { in InitializeOptimizersByName() local
/external/tensorflow/tensorflow/python/tpu/
Dtpu_embedding_v2_correctness_test.py134 def _create_strategy_and_mid_level(self, optimizer_name): argument
155 def test_embedding(self, optimizer_name, training, sparse): argument
Dtpu_embedding_v2_test.py1017 def _create_strategy_and_mid_level(self, optimizer_name): argument