Home
last modified time | relevance | path

Searched defs:run_fn (Results 1 – 12 of 12) sorted by relevance

/external/tensorflow/tensorflow/python/training/experimental/
Dloss_scaling_gradient_tape_test.py53 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 …]
Dloss_scale_optimizer_test.py215 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/
Dloss_scale_optimizer_test.py232 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 …]
Dloss_scale_benchmark.py124 def run_fn(): function
Dkeras_test.py261 def run_fn(): function
/external/tensorflow/tensorflow/python/distribute/coordinator/
Dfault_tolerance_test.py300 def run_fn(): function
334 def run_fn(): function
373 def run_fn(): function
/external/tensorflow/tensorflow/python/keras/distribute/
Dminimize_loss_test.py545 def run_fn(): function
Ddistribute_strategy_test.py583 def run_fn(): function
/external/tensorflow/tensorflow/python/distribute/
Ddistribute_lib_test.py160 def run_fn(): function
Dstrategy_test_lib.py255 def run_fn(): function
Dtpu_strategy.py998 def run_fn(inputs): function
Dmirrored_strategy_test.py96 def run_fn(): function