Searched refs:fn_eager (Results 1 – 2 of 2) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/python/distribute/ |
| H A D | strategy_common_test.py | 228 def fn_eager(): function
233 fn_graph = def_function.function(fn_eager)
237 self.assertEqual(fn_eager().numpy(), 1.0 * strategy.num_replicas_in_sync)
242 self.assertEqual(fn_eager().numpy(), 1.0 * strategy.num_replicas_in_sync)
|
| /aosp_15_r20/external/pytorch/test/inductor/ |
| H A D | test_cudagraph_trees.py | 1960 def _assert_equal_multi_loop(self, param, fn_eager, fn_compiled): argument
1961 exp_output, exp_grad = self._run_iter(param, fn_eager)
1967 def run_static_input_param_test(self, fn_eager, num_graphs): argument
1969 fn_compiled = torch.compile(fn_eager, mode="reduce-overhead")
1972 self._assert_equal_multi_loop(p1, fn_eager, fn_compiled)
1975 self._assert_equal_multi_loop(p2, fn_eager, fn_compiled)
1978 self._assert_equal_multi_loop(p1, fn_eager, fn_compiled)
|