Searched defs:optimizer_fn (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/python/keras/distribute/ |
| D | minimize_loss_test.py | 92 def testTrainNetwork(self, distribution, optimizer_fn, use_callable_loss): argument
134 def testTrainNetworkByCallForEachReplica(self, distribution, optimizer_fn, argument
171 def testOptimizerInsideModelFn(self, distribution, optimizer_fn): argument
246 def testTrainNetworkWithBatchNorm(self, distribution, optimizer_fn, momentum, argument
334 def testMeanVsSum(self, distribution, optimizer_fn, loss_reduction, argument
423 def testRunStepsWithOutputContext(self, distribution, optimizer_fn, is_tpu): argument
541 def test_empty_var_list(self, distribution, optimizer_fn): argument
|
| D | ctl_correctness_test.py | 100 def iteration_inside_func(initial_weights, dataset, optimizer_fn, argument
164 def iteration_outside_func(initial_weights, dataset, optimizer_fn, argument
247 def test_dnn_correctness_minus_tpus(self, distribution, optimizer_fn, argument
256 def dnn_correctness(self, distribution, optimizer_fn, iteration_type, argument
|
| D | test_example.py | 59 def batchnorm_example(optimizer_fn, argument
|
| /external/tensorflow/tensorflow/python/distribute/ |
| D | single_loss_example.py | 32 def single_loss_example(optimizer_fn, distribution, use_bias=False, argument
82 def batchnorm_example(optimizer_fn, argument
|