Searched refs:output_context (Results 1 – 2 of 2) sorted by relevance
386 def model_fn(output_context, x): argument394 output_context.set_last_step_output(398 output_context.set_non_tensor_output(key1, value1)401 def step_fn(output_context, inputs): argument403 model_fn, args=(output_context, inputs))404 output_context.set_last_step_output(408 output_context.set_last_step_output(
2763 self.output_context(f, InspectionWebFormatter.CONTEXT_SHORT)2788 self.output_context(f, InspectionWebFormatter.CONTEXT_FULL)2846 def output_context(self, f, details): member in InspectionWebFormatter