code: def testSplitPartialPadding(self, graph_fn):
      for dtype in self.numeric_types:
          with self.session() as sess, self.device_scope():
              split = graph_fn(
                  sess,
                  input_value=[
                      [0, 1, 2],
                      [3, 4, 5],
                      [6, 7, 8],
                  ],
                  input_dtype=dtype,
                  num_outputs=4,
                  num_splits=[2, 2],
                  paddings=[1, 1],
              )
              results = sess.run(split)
          self.assertLen(results, 4)
          self.assertAllClose(results[0], [[0, 1], [3, 4]])
          self.assertAllClose(results[1], [[2, 0], [5, 0]])
          self.assertAllClose(results[2], [[6, 7], [0, 0]])
          self.assertAllClose(results[3], [[8, 0], [0, 0]])
  
imports: from typing import Any, List, Optional
  from absl.testing import parameterized
  import numpy
  from tensorflow.compiler.tests import xla_test
  from tensorflow.python.client.session import Session
  from tensorflow.python.framework import constant_op
  from tensorflow.python.framework.ops import control_dependencies
  from tensorflow.python.framework.ops import Tensor
  from tensorflow.python.ops import gen_tpu_ops
  from tensorflow.python.ops import math_ops
  from tensorflow.python.ops import resource_variable_ops
  from tensorflow.python.ops import variables
  from tensorflow.python.platform import test
label_desc: Use image node_id 12 for calling the XlaSplitNDOpTest obj's underlying member method code with example usage: obj.testSplitPartialPadding(graph_fn) without return types