tf.errors.error_code_from_exception_type

View source on GitHub