code: def testSplitVBigTensors(self):
      input_shape = [1, 64, 32768]
      x = np.linspace(
          start=1,
          stop=np.prod(input_shape),
          num=np.prod(input_shape),
          dtype=np.float32,
      ).reshape(input_shape)
      split_axis = 1
      size_splits = [1] * input_shape[split_axis]
  
      y = array_ops.split(
          x, num_or_size_splits=size_splits, axis=split_axis
      )
  
      for i in range(input_shape[split_axis]):
          result = y[i]
          expected = x[:, i : i + 1, :]
          self.assertAllEqual(result, expected)
  
imports: import numpy
  from tensorflow.python.framework import constant_op
  from tensorflow.python.framework import dtypes
  from tensorflow.python.framework import errors_impl
  from tensorflow.python.framework import ops
  from tensorflow.python.framework import test_util
  from tensorflow.python.ops import array_ops
  from tensorflow.python.ops import gradients_impl
  from tensorflow.python.ops import math_ops
  from tensorflow.python.platform import test
label_desc: Use image node_id 29 for calling the SplitOpTest obj's underlying member method code with example usage: obj.testSplitVBigTensors() without return types