Searched refs:last_step_outputs (Results 1 – 6 of 6) sorted by relevance
109 flat_last_step_outputs = nest.flatten(ctx.last_step_outputs)136 ctx.last_step_outputs, last_step_tensor_outputs)
603 for (name, output) in ctx.last_step_outputs.items():605 ctx.last_step_outputs[name] = self._local_results(output)606 flat_last_step_outputs = nest.flatten(ctx.last_step_outputs)632 ctx.last_step_outputs, last_step_tensor_outputs)
303 flat_last_step_outputs = nest.flatten(ctx.last_step_outputs)660 ctx.last_step_outputs, last_step_tensor_outputs)
579 def last_step_outputs(self): member in MultiStepContext
438 loss_output=ctx.last_step_outputs["replica_loss_reduced"],442 loss_output=ctx.last_step_outputs["cross_replica_loss_reduced"],446 loss_output=ctx.last_step_outputs["cross_replica_loss_not_reduced"],448 return (ctx.run_op, ctx.last_step_outputs["replica_loss_reduced"])
350 output_tensors = ctx.last_step_outputs504 output_tensors = ctx.last_step_outputs647 output_tensors = ctx.last_step_outputs