Make replacement of noisy relus general