tf.contrib.eager.inf_nan_callback

View source on GitHub

An execution callback that checks for infs and nans in output tensors.

tf.contrib.eager.inf_nan_callback(
    op_type, inputs, attrs, outputs, op_name, check_inf=True, check_nan=True,
    action=tf.contrib.eager.ExecutionCallback.RAISE
)

This callback can be used with tfe.add_execute_callback to check for invalid numeric values. E.g.,

tfe.add_execute_callback(tfe.inf_nan_callback)

op_type Name of the TFE operation type (e.g., MatMul).
inputs The list of input tensors to the operation, currently unused by this callback.
attrs Attributes of the TFE operation, as a tuple of alternating attribute names and attribute values.
outputs The list of output tensors from the operation, checked by this callback for inf and nan values.
op_name Name of the TFE operation. This name is set by client and can be None if it unset.
check_inf (bool) Whether this callback should check for inf values in the output tensor values.
check_nan (bool) Whether this callback should check for nan values in the output tensor values.
action (ExecutionCallback) Action to be taken by the callback when inf or nan values are detected.

InfOrNanError iff inf or nan values are seen in any of outputs and action is "raise".
ValueError iff the value of action is invalid.