nir: Add colorspace support to YUV lowering pass