class_docstr: Tests for non-member functions in the module xla_sharding.py.
code: def test_split_raises_error_with_incommensurate_dimensions(self):
      @def_function.function
      def split_helper(tensor):
          split_tensor = xla_sharding.split(tensor, 0, 8)
          return split_tensor
  
      with self.assertRaises(ValueError):
          _ = split_helper(
              array_ops.ones([4, 5, 6], dtype=dtypes.float32)
          )
  
imports: from absl.testing import absltest
  import numpy
  from google.protobuf.message import DecodeError
  from local_xla.xla import xla_data_pb2
  from tensorflow.python.compiler.xla.experimental import xla_sharding
  from tensorflow.python.eager import def_function
  from tensorflow.python.framework import dtypes
  from tensorflow.python.framework import tensor
  from tensorflow.python.framework import test_util
  from tensorflow.python.ops import array_ops
label_desc: Use image node_id 4 for calling the XlaShardingTest obj's underlying member method code with example usage: obj.test_split_raises_error_with_incommensurate_dimensions() and returns: split_tensor