Searched refs:run_in_function (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/python/distribute/ |
| D | strategy_test_lib.py | 65 def _maybe_run_in_function(fn, run_in_function=False): argument
66 if not run_in_function or not context.executing_eagerly():
432 def _test_numpy_dataset(self, strategy, session=None, run_in_function=False): argument
456 _maybe_run_in_function(lambda z: z, run_in_function), i)
612 def _test_run(self, strategy, run_in_function=False): argument
615 run_in_function))
619 lambda x: {"a": x * 2, "b": x * x}, run_in_function), args=(out1,))
625 lambda b, a: a + 2 * b + 2, run_in_function), kwargs=out2)
628 def _test_all_reduce_sum(self, strategy, run_in_function=False): argument
634 run_in_function=run_in_function)
[all …]
|
| D | tpu_strategy_test.py | 1027 self._test_numpy_dataset(strategy, run_in_function=True)
1035 self._test_run(strategy, run_in_function=True)
1043 self._test_all_reduce_sum(strategy, run_in_function=True)
1047 self._test_all_reduce_sum_gradients(strategy, run_in_function=True)
1051 self._test_all_reduce_sum_gradient_tape(strategy, run_in_function=True)
1055 self._test_all_reduce_mean(strategy, run_in_function=True)
1059 self._test_all_reduce_mean_gradients(strategy, run_in_function=True)
1063 self._test_all_reduce_mean_gradient_tape(strategy, run_in_function=True)
|