tf.compat.v1.errors.raise_exception_on_not_ok_status