tf.contrib.eager.inf_callback

A specialization of inf_nan_callback that checks for infs only.

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