Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/python/training/experimental/
Dloss_scale_optimizer_test.py210 run_fn = lambda: opt.minimize(loss, var_list=[var]) function
221 run_fn = lambda: opt.minimize(loss, var_list=[var]) function
244 run_fn = lambda: opt.minimize(loss, var_list=[var]) function
282 run_fn = lambda: opt.minimize(lambda: var + 1., var_list=[var]) function
/external/tensorflow/tensorflow/python/distribute/coordinator/
Dfault_tolerance_test.py336 def run_fn(): function
370 def run_fn(): function
409 def run_fn(): function
/external/tensorflow/tensorflow/python/keras/mixed_precision/
Dlayer_test.py214 def run_fn(): function
/external/tensorflow/tensorflow/python/distribute/
Ddistribute_lib_test.py157 def run_fn(): function
Dstrategy_test_lib.py272 def run_fn(): function
Dtpu_strategy.py1056 def run_fn(inputs): function
Dmirrored_strategy_test.py96 def run_fn(): function