tf.contrib.eager.nan_callback

A specialization of inf_nan_callback that checks for nans only.

tf.contrib.eager.nan_callback(
    op_type, inputs, attrs, outputs, op_name,
    action=tf.contrib.eager.ExecutionCallback.RAISE
)