Searched refs:last_step_outputs (Results 1 – 6 of 6) sorted by relevance
493 loss_output=ctx.last_step_outputs["replica_loss_reduced"],497 loss_output=ctx.last_step_outputs["cross_replica_loss_reduced"],501 loss_output=ctx.last_step_outputs["cross_replica_loss_not_reduced"],503 return (ctx.run_op, ctx.last_step_outputs["replica_loss_reduced"])
359 flat_last_step_outputs = nest.flatten(ctx.last_step_outputs)386 ctx.last_step_outputs, last_step_tensor_outputs)
611 for (name, output) in ctx.last_step_outputs.items():613 ctx.last_step_outputs[name] = self._local_results(output)614 flat_last_step_outputs = nest.flatten(ctx.last_step_outputs)640 ctx.last_step_outputs, last_step_tensor_outputs)
1001 flat_last_step_outputs = nest.flatten(ctx.last_step_outputs)1617 ctx.last_step_outputs, last_step_tensor_outputs)
2024 def last_step_outputs(self): member in MultiStepContext
197 output_tensors = ctx.last_step_outputs