Make RectifiedLinearLayer device agnostic.