Searched refs:TrainOutput (Results 1 – 4 of 4) sorted by relevance
216 "train": export_output.TrainOutput(loss=output_1),257 self.assertIsInstance(export_out, export_output.TrainOutput)
286 return {signature_key: export_output_lib.TrainOutput(
386 class TrainOutput(_SupervisedOutput): class
347 outputter = export_output_lib.TrainOutput(loss, predictions, metrics)