Log a warning when device_mesh is specified but variable splits are not for a multi-dimension variable.
Log a warning when device_mesh is specified but variable splits are not for a multi-dimension variable.

PiperOrigin-RevId: 350482818

tensorflow/lingvo,tensorflow/lingvo,tensorflow/lingvo,tensorflow/lingvo