Searched refs:output_context (Results 1 – 1 of 1) sorted by relevance
437 def model_fn(output_context, x): argument449 output_context.set_last_step_output(453 output_context.set_non_tensor_output(key1, value1)456 def step_fn(output_context, inputs): argument458 model_fn, args=(output_context, inputs))459 output_context.set_last_step_output(463 output_context.set_last_step_output(