code: def testVariadicSortSeveralStable(self, dimension, key_type):
      shape = (2, 3, 4)
      if key_type not in self._supported_key_types():
          return
      for value_type_1 in self._supported_key_types():
          for value_type_2 in self._supported_key_types():
              # The first input is all 0s, there should be no changes for
              # stable sort.
              inputs = [
                  np.zeros(shape, key_type),
                  self._shuffled_arange(shape, value_type_1),
                  self._shuffled_arange(shape, value_type_2),
              ]
  
              @function.Defun(
                  key_type,
                  key_type,
                  value_type_1,
                  value_type_1,
                  value_type_2,
                  value_type_2,
              )
              def compare_lt(x1, x2, y1, y2, z1, z2):
                  del y1, y2, z1, z2
                  return x1 < x2
  
              def wrap_sort(*args):
                  return xla.variadic_sort(
                      args,  # Pass the arguments as a tuple
                      comparator=compare_lt,
                      dimension=dimension,
                      is_stable=True,
                  )
  
              self._assertOpOutputMatchesExpected(
                  wrap_sort, inputs, expected=inputs
              )
  
imports: import unittest
  from absl.testing import parameterized
  import numpy
  from tensorflow.compiler.tests import xla_test
  from tensorflow.compiler.tf2xla.python import xla
  from tensorflow.python.framework import dtypes
  from tensorflow.python.framework import function
  from tensorflow.python.framework import tensor
  from tensorflow.python.framework import test_util
  from tensorflow.python.ops import array_ops
  from tensorflow.python.ops import math_ops
  from tensorflow.python.ops import nn_ops
  from tensorflow.python.ops import sort_ops
  from tensorflow.python.platform import test
label_desc: Use image node_id 10 for calling the XlaSortOpTest obj's underlying member method code with example usage: obj.testVariadicSortSeveralStable(dimension, key_type) and returns: unknown, xla