Home
last modified time | relevance | path

Searched refs:TRACE_MODE_NAN_INF (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/python/tpu/
Dtensor_tracer_flags.py34 TRACE_MODE_NAN_INF = 'nan-inf' variable
128 self.use_compact_trace = self.trace_mode in (TRACE_MODE_NAN_INF,
197 TRACE_MODE_NAN_INF, TRACE_MODE_PART_TENSOR, TRACE_MODE_FULL_TENSOR,
316 TRACE_MODE_NAN_INF: math_ops.reduce_max,
Dtensor_tracer.py644 tensor_tracer_flags.TRACE_MODE_NAN_INF,
827 if self._parameters.trace_mode == tensor_tracer_flags.TRACE_MODE_NAN_INF:
948 tensor_tracer_flags.TRACE_MODE_NAN_INF,
1105 tensor_tracer_flags.TRACE_MODE_NAN_INF,
1316 tensor_tracer_flags.TRACE_MODE_NAN_INF):
1325 if self._parameters.trace_mode == tensor_tracer_flags.TRACE_MODE_NAN_INF: