Searched refs:_get_output_tensor (Results 1 – 1 of 1) sorted by relevance
278 instrumented_outputs.append(self._get_output_tensor(299 def _get_output_tensor(self, member in CheckNumericsCallback