Searched refs:_detect_nan_inf (Results 1 – 1 of 1) sorted by relevance
763 def _detect_nan_inf(tensor): function828 return {self._parameters.trace_mode: _detect_nan_inf(tensor)}