Home
last modified time | relevance | path

Searched refs:optimizer_fn (Results 1 – 9 of 9) sorted by relevance

/external/tensorflow/tensorflow/python/keras/distribute/
Dminimize_loss_test.py85 optimizer_fn=optimizer_combinations.optimizers_v2,
89 optimizer_fn=optimizer_combinations.optimizers_v1,
92 def testTrainNetwork(self, distribution, optimizer_fn, use_callable_loss): argument
94 optimizer = optimizer_fn()
134 def testTrainNetworkByCallForEachReplica(self, distribution, optimizer_fn, argument
137 optimizer = optimizer_fn()
169 optimizer_fn=optimizer_combinations.optimizers_v1_and_v2,
171 def testOptimizerInsideModelFn(self, distribution, optimizer_fn): argument
189 optimizer = optimizer_fn()
243 optimizer_fn=optimizer_combinations.optimizers_v1_and_v2,
[all …]
Dctl_correctness_test.py100 def iteration_inside_func(initial_weights, dataset, optimizer_fn, argument
109 optimizer = optimizer_fn()
164 def iteration_outside_func(initial_weights, dataset, optimizer_fn, argument
170 optimizer = optimizer_fn()
231 optimizer_fn=optimizer_combinations.optimizers_v2,
238 optimizer_fn=[
247 def test_dnn_correctness_minus_tpus(self, distribution, optimizer_fn, argument
251 if 'FtrlV1' in optimizer_fn._name and 'TPU' in type(distribution).__name__:
253 self.dnn_correctness(distribution, optimizer_fn, iteration_type,
256 def dnn_correctness(self, distribution, optimizer_fn, iteration_type, argument
[all …]
Doptimizer_combinations.py88 optimizer_fn=optimizers_v1)
99 optimizer_fn=optimizers_v2)
110 optimizer_fn=optimizers_v1_and_v2)
Ddistribute_strategy_test.py475 optimizer_fn = gradient_descent_keras.SGD
476 optimizer = optimizer_fn(0.001)
520 optimizer_fn = gradient_descent_keras.SGD
521 optimizer = optimizer_fn(0.001)
618 optimizer_fn = gradient_descent_keras.SGD
619 optimizer = optimizer_fn(learning_rate=0.001)
692 optimizer_fn = gradient_descent_keras.SGD
693 optimizer = optimizer_fn(learning_rate=0.001)
889 optimizer_fn = gradient_descent_keras.SGD
890 optimizer = optimizer_fn(0.001)
[all …]
Dkeras_stateful_lstm_model_correctness_test.py75 optimizer_fn = gradient_descent_keras.SGD
78 optimizer=optimizer_fn(learning_rate=0.1),
Dcollective_all_reduce_strategy_test.py67 optimizer_fn = gradient_descent_keras.SGD
68 optimizer = optimizer_fn(0.001)
Dtest_example.py59 def batchnorm_example(optimizer_fn, argument
74 optimizer = optimizer_fn()
Dkeras_rnn_model_correctness_test.py64 optimizer_fn = gradient_descent_keras.SGD
67 optimizer=optimizer_fn(learning_rate=0.1),
/external/tensorflow/tensorflow/python/distribute/
Dsingle_loss_example.py32 def single_loss_example(optimizer_fn, distribution, use_bias=False, argument
39 optimizer = optimizer_fn()
82 def batchnorm_example(optimizer_fn, argument
97 optimizer = optimizer_fn()