Searched defs:run_in_function (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/python/distribute/ |
| H A D | strategy_test_lib.py | 64 def _maybe_run_in_function(fn, run_in_function=False): argument
449 def _test_numpy_dataset(self, strategy, session=None, run_in_function=False): argument
629 def _test_run(self, strategy, run_in_function=False): argument
645 def _test_all_reduce_sum(self, strategy, run_in_function=False): argument
653 def _test_all_reduce_sum_gradients(self, strategy, run_in_function=False): argument
658 def _test_all_reduce_sum_gradient_tape(self, strategy, run_in_function=False): argument
663 def _test_all_reduce_mean(self, strategy, run_in_function=False): argument
671 def _test_all_reduce_mean_gradients(self, strategy, run_in_function=False): argument
677 run_in_function=False): argument
683 run_in_function=False): argument
[all …]
|