Searched refs:TrainOutput (Results 1 – 4 of 4) sorted by relevance
236 export_output.TrainOutput(loss=output_1),278 self.assertIsInstance(export_out, export_output.TrainOutput)
287 return {signature_key: export_output_lib.TrainOutput(
387 class TrainOutput(_SupervisedOutput): class
341 outputter = export_output_lib.TrainOutput(loss, predictions, metrics)