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