support cpu only in PRLoss layer