Searched refs:non_tensor_outputs (Results 1 – 3 of 3) sorted by relevance
109 value = ctx.non_tensor_outputs["value"]
435 self.assertEqual({key1: (value1,)}, ctx.non_tensor_outputs)
639 def non_tensor_outputs(self): member in MultiStepContext