Searched refs:set_non_tensor_output (Results 1 – 3 of 3) sorted by relevance
103 ctx.set_non_tensor_output(name="value", output=value)
398 output_context.set_non_tensor_output(key1, value1)
643 def set_non_tensor_output(self, name, output): member in MultiStepContext