Searched refs:set_last_step_output (Results 1 – 3 of 3) sorted by relevance
394 output_context.set_last_step_output(404 output_context.set_last_step_output(408 output_context.set_last_step_output(
249 ctx.set_last_step_output(label, output)257 ctx.set_last_step_output(label, output, reduce_op)
599 def set_last_step_output(self, name, output, reduce_op=None): member in MultiStepContext