Searched refs:run_fn (Results 1 – 7 of 7) sorted by relevance
77 def run_fn(): function91 dist.extended.call_for_each_replica(run_fn)93 dist.extended.call_for_each_replica(run_fn)
300 def run_fn(inputs): function330 replicate_outputs = tpu.replicate(run_fn, replicate_inputs)
83 run_fn = self._run_fn_with_grad_check(85 run_op = strategy.experimental_run(run_fn)
225 def run_fn(): function239 op = strategy.experimental_run(run_fn)
220 def run_fn(): function236 output = d.extended.call_for_each_replica(run_fn)
93 def run_fn(): function100 distribution.extended.call_for_each_replica(run_fn)
958 run_fn = sess.make_callable(targets)959 run_fn() # Warm up962 run_fn()