Keep more operations on GPU in vision