| /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
69 strategy_fn = lambda: strategy.run(run_fn)
93 def run_fn(): function
97 dy_dx_list = self._run_with_strategy(run_fn, strategy, use_tf_function)
113 def run_fn(): function
117 dy_dx_list = self._run_with_strategy(run_fn, strategy, use_tf_function)
137 def run_fn(): function
142 x1g, x2g, x3g, x4g = self._run_with_strategy(run_fn, strategy,
166 def run_fn(): function
170 dy_dx_list = self._run_with_strategy(run_fn, strategy, use_tf_function)
[all …]
|
| D | loss_scale_optimizer_test.py | 143 run_fn = self._run_fn_with_grad_check(
145 run_op = strategy.experimental_run(run_fn)
183 run_fn = self._run_fn_with_grad_check(strategy, var, opt,
185 run_op = strategy.experimental_run(run_fn)
196 run_op = strategy.experimental_run(run_fn)
215 run_fn = lambda: opt.minimize(loss, var_list=[var]) function
216 run_op = strategy.experimental_run(run_fn)
226 run_fn = lambda: opt.minimize(loss, var_list=[var]) function
227 run_op = strategy.experimental_run(run_fn)
249 run_fn = lambda: opt.minimize(loss, var_list=[var]) function
[all …]
|
| /external/tensorflow/tensorflow/python/keras/mixed_precision/ |
| D | loss_scale_optimizer_test.py | 104 run_fn = self._run_fn_with_grad_check(
106 run_op = strategy.experimental_run(run_fn)
191 run_fn = self._run_fn_with_grad_check(strategy, var, opt,
193 run_op = strategy.experimental_run(run_fn)
203 run_op = strategy.experimental_run(run_fn)
232 run_fn = lambda: opt.minimize(loss, var_list=[var]) function
235 run_op = strategy.experimental_run(run_fn)
245 run_op = strategy.experimental_run(run_fn)
254 run_fn = lambda: opt.minimize(loss, var_list=[var]) function
255 run_op = strategy.experimental_run(run_fn)
[all …]
|
| D | loss_scale_benchmark.py | 124 def run_fn(): function
127 run_fn = def_function.function(run_fn)
130 run_fn()
134 run_fn()
|
| D | keras_test.py | 261 def run_fn(): function
271 op = strategy.experimental_run(run_fn)
|
| /external/tensorflow/tensorflow/python/distribute/coordinator/ |
| D | fault_tolerance_test.py | 300 def run_fn(): function
308 run_thread = threading.Thread(target=run_fn)
334 def run_fn(): function
342 run_thread = threading.Thread(target=run_fn)
373 def run_fn(): function
381 run_thread = threading.Thread(target=run_fn)
|
| /external/tensorflow/tensorflow/python/distribute/ |
| D | distribute_lib_test.py | 160 def run_fn(): function
174 dist.extended.call_for_each_replica(run_fn)
176 dist.extended.call_for_each_replica(run_fn)
|
| D | strategy_test_lib.py | 255 def run_fn(): function
272 output = d.extended.call_for_each_replica(run_fn)
|
| D | tpu_strategy.py | 998 def run_fn(inputs): function
1030 run_fn,
|
| D | mirrored_strategy_test.py | 96 def run_fn(): function
103 distribution.extended.call_for_each_replica(run_fn)
|
| /external/tensorflow/tensorflow/python/keras/distribute/ |
| D | minimize_loss_test.py | 545 def run_fn(): function
549 distribution.run(run_fn)
|
| D | distribute_strategy_test.py | 583 def run_fn(): function
590 run_fn = def_function.function(run_fn)
592 grad_v1, grad_v2 = distribution.run(run_fn)
|
| /external/tensorflow/tensorflow/python/ops/parallel_for/ |
| D | control_flow_ops_test.py | 1853 run_fn = sess.make_callable(targets)
1854 run_fn() # Warm up
1857 run_fn()
|