/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 | 232 run_fn = lambda: opt.minimize(loss, var_list=[var]) function 254 run_fn = lambda: opt.minimize(loss, var_list=[var]) function 271 run_fn = lambda: opt.minimize(loss, var_list=[var]) function 282 run_fn = lambda: opt.minimize(loss, var_list=[var]) function 302 run_fn = lambda: opt.minimize(loss, var_list=[var]) function 325 run_fn = lambda: opt.minimize(loss, var_list=[var]) function 363 run_fn = lambda: opt.minimize(loss, [var]) function 373 run_fn = lambda: opt.minimize(loss, [var]) function 487 run_fn = lambda: opt.minimize(loss, [var]) function 525 run_fn = lambda: opt.minimize(loss, var_list=[var]) function [all …]
|
D | loss_scale_benchmark.py | 124 def run_fn(): function
|
D | keras_test.py | 261 def run_fn(): function
|
/external/tensorflow/tensorflow/python/distribute/coordinator/ |
D | fault_tolerance_test.py | 300 def run_fn(): function 334 def run_fn(): function 373 def run_fn(): function
|
/external/tensorflow/tensorflow/python/keras/distribute/ |
D | minimize_loss_test.py | 545 def run_fn(): function
|
D | distribute_strategy_test.py | 583 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 | 998 def run_fn(inputs): function
|
D | mirrored_strategy_test.py | 96 def run_fn(): function
|