Searched refs:_wrap_and_check_outputs (Results 1 – 1 of 1) sorted by relevance
65 def _wrap_and_check_outputs( member in ExportOutput223 self._outputs = self._wrap_and_check_outputs(270 loss_dict = self._wrap_and_check_outputs(loss, self.LOSS_NAME)273 pred_dict = self._wrap_and_check_outputs(