code: def testBig(self):
      # This test was added after discovering a memory allocation bug in the GPU
      # kernel that the existing tests did not catch due to being too small.
      for n in [250, 2500, 25000]:
          indices = np.zeros([n, 2], dtype=np.int64)
          indices[:, 0] = np.arange(n)
          values = np.zeros([n], dtype=np.float32)
          dense_shape = np.array([n, 3], dtype=np.int64)
          sp_input = sparse_tensor.SparseTensor(
              indices, values, dense_shape
          )
          sp_tensors = self.evaluate(
              sparse_ops.sparse_split(
                  sp_input=sp_input, num_split=2, axis=0
              )
          )
          self.assertAllEqual(sp_tensors[0].indices, indices[: n // 2])
          self.assertAllEqual(
              sp_tensors[1].indices, indices[n // 2 :] - [n // 2, 0]
          )
          self.assertAllEqual(sp_tensors[0].values, values[: n // 2])
          self.assertAllEqual(sp_tensors[1].values, values[n // 2 :])
          self.assertAllEqual(sp_tensors[0].dense_shape, [n // 2, 3])
          self.assertAllEqual(sp_tensors[1].dense_shape, [n // 2, 3])
  
imports: import numpy
  from tensorflow.python.framework import errors
  from tensorflow.python.framework import sparse_tensor
  from tensorflow.python.framework import test_util
  from tensorflow.python.ops import gen_sparse_ops
  from tensorflow.python.ops import sparse_ops
  from tensorflow.python.platform import test
label_desc: Use image node_id 17 for calling the SparseSplitOpTest obj's underlying member method code with example usage: obj.testBig() without return types