Searched refs:_maybe_run_in_function (Results 1 – 1 of 1) sorted by relevance
65 def _maybe_run_in_function(fn, run_in_function=False): function456 _maybe_run_in_function(lambda z: z, run_in_function), i)613 out1 = strategy.run(_maybe_run_in_function(618 out2 = strategy.run(_maybe_run_in_function(624 out3 = strategy.run(_maybe_run_in_function(675 _maybe_run_in_function(comm_fn, run_in_function), inputs))))699 _maybe_run_in_function(step, run_in_function), inputs))))721 _maybe_run_in_function(step, run_in_function),