| /aosp_15_r20/external/tensorflow/tensorflow/core/grappler/optimizers/data/ |
| H A D | meta_optimizer.cc | 80 const string& optimizer_name = split[0]; in ToConfigMap() local
197 for (const auto& optimizer_name : optimizers) { in Init() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/tpu/tests/ |
| H A D | tpu_embedding_v1_correctness_test.py | 78 def _create_strategy_and_mid_level(self, optimizer_name): argument
115 def test_embedding(self, optimizer_name, training, sparse, argument
|
| H A D | tpu_embedding_v2_correctness_hd_sparse_forward_test.py | 28 def test_embedding(self, optimizer_name): argument
|
| H A D | tpu_embedding_v2_correctness_ragged_forward_test.py | 28 def test_embedding(self, optimizer_name): argument
|
| H A D | tpu_embedding_v2_correctness_hd_ragged_training_test.py | 28 def test_embedding(self, optimizer_name): argument
|
| H A D | tpu_embedding_v2_correctness_hd_ragged_forward_test.py | 28 def test_embedding(self, optimizer_name): argument
|
| H A D | tpu_embedding_v2_correctness_ragged_training_test.py | 28 def test_embedding(self, optimizer_name): argument
|
| H A D | tpu_embedding_v2_correctness_sparse_training_test.py | 28 def test_embedding(self, optimizer_name): argument
|
| H A D | tpu_embedding_v2_correctness_hd_sparse_training_test.py | 28 def test_embedding(self, optimizer_name): argument
|
| H A D | tpu_embedding_v2_correctness_sparse_forward_test.py | 28 def test_embedding(self, optimizer_name): argument
|
| H A D | tpu_embedding_v2_correctness_base_test.py | 28 def _test_embedding(self, optimizer_name, training, sparse, argument
|
| H A D | tpu_embedding_base_test.py | 168 def _create_strategy_and_mid_level(self, optimizer_name): argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/grappler/optimizers/ |
| H A D | graph_optimizer_stage.h | 118 explicit GraphOptimizerStage(const string& optimizer_name, in GraphOptimizerStage()
125 const string& optimizer_name() const { return optimizer_name_; } in optimizer_name() function
|
| H A D | meta_optimizer.h | 106 string optimizer_name; member
|
| H A D | meta_optimizer.cc | 507 for (const string& optimizer_name : cfg_.optimizers()) { in InitializeOptimizersByName() local
661 for (const string& optimizer_name : cfg_.optimizers()) { in PrintUserAndPluginConfigs() local
|
| H A D | graph_optimizer_stage_test.cc | 39 explicit FakeOptimizerStage(const string& optimizer_name, in FakeOptimizerStage()
|
| /aosp_15_r20/external/pytorch/test/cpp/api/ |
| H A D | optim_baseline.py | 71 def run(optimizer_name, iterations, sample_every): argument
|