| /external/tensorflow/tensorflow/python/training/experimental/ |
| D | loss_scaling_gradient_tape_test.py | 53 def _run_with_strategy(self, run_fn, strategy, use_tf_function=False): argument
93 def run_fn(): function
113 def run_fn(): function
137 def run_fn(): function
166 def run_fn(): function
193 def run_fn(): function
231 def run_fn(): function
283 def run_fn(): function
318 def run_fn(): function
348 def run_fn(): function
[all …]
|
| D | loss_scale_optimizer_test.py | 215 run_fn = lambda: opt.minimize(loss, var_list=[var]) function
226 run_fn = lambda: opt.minimize(loss, var_list=[var]) function
249 run_fn = lambda: opt.minimize(loss, var_list=[var]) function
287 run_fn = lambda: opt.minimize(lambda: var + 1., var_list=[var]) function
|
| /external/tensorflow/tensorflow/python/keras/mixed_precision/ |
| D | loss_scale_optimizer_test.py | 231 run_fn = lambda: opt.minimize(loss, var_list=[var]) function
253 run_fn = lambda: opt.minimize(loss, var_list=[var]) function
270 run_fn = lambda: opt.minimize(loss, var_list=[var]) function
281 run_fn = lambda: opt.minimize(loss, var_list=[var]) function
301 run_fn = lambda: opt.minimize(loss, var_list=[var]) function
329 run_fn = lambda: opt.minimize(loss, var_list=[var]) function
376 run_fn = lambda: opt.minimize(loss, var_list=[var]) function
414 run_fn = lambda: opt.minimize(loss, [var]) function
424 run_fn = lambda: opt.minimize(loss, [var]) function
538 run_fn = lambda: opt.minimize(loss, [var]) function
[all …]
|
| D | loss_scale_benchmark.py | 121 def run_fn(): function
|
| D | layer_test.py | 214 def run_fn(): function
|
| /external/tensorflow/tensorflow/python/distribute/coordinator/ |
| D | fault_tolerance_test.py | 337 def run_fn(): function
371 def run_fn(): function
410 def run_fn(): function
|
| /external/tensorflow/tensorflow/python/keras/distribute/ |
| D | minimize_loss_test.py | 543 def run_fn(): function
|
| D | distribute_strategy_test.py | 580 def run_fn(): function
|
| /external/tensorflow/tensorflow/python/distribute/ |
| D | distribute_lib_test.py | 160 def run_fn(): function
|
| D | strategy_test_lib.py | 255 def run_fn(): function
|
| D | tpu_strategy.py | 999 def run_fn(inputs): function
|
| D | mirrored_strategy_test.py | 99 def run_fn(): function
|