mask_weights is still used through get_mask by some Layer subclasses in censor_updates.