(ctx, inputs)
| 102 | if isinstance(distribution, (tpu_strategy.TPUStrategy, |
| 103 | tpu_strategy.TPUStrategyV1)): |
| 104 | def step_fn(ctx, inputs): |
| 105 | value, update = distribution.extended.call_for_each_replica( |
| 106 | metric_fn, args=(inputs,)) |
| 107 | ctx.set_non_tensor_output(name="value", output=value) |
| 108 | return distribution.group(update) |
| 109 | |
| 110 | ctx = distribution.extended.experimental_run_steps_on_iterator( |
| 111 | step_fn, iterator, iterations=distribution.extended.steps_per_run) |
nothing calls this directly
no test coverage detected