code: def rt_with_partition_type(self, rt, partition_type):
      if isinstance(rt, tensor_lib.Tensor):
          return rt
      if partition_type == "row_splits":
          return rt
      if partition_type == "value_rowids":
          return ragged_tensor.RaggedTensor.from_value_rowids(
              self.rt_with_partition_type(rt.values, partition_type),
              rt.value_rowids(),
              rt.nrows(),
          )
      raise AssertionError(
          "Unexpected partition_type %r" % partition_type
      )
  
imports: import random
  from absl.testing import parameterized
  import numpy
  from tensorflow.python.client import session
  from tensorflow.python.eager import backprop
  from tensorflow.python.eager import context
  from tensorflow.python.framework import constant_op
  from tensorflow.python.framework import dtypes
  from tensorflow.python.framework import errors
  from tensorflow.python.framework import tensor
  from tensorflow.python.framework import tensor_shape
  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.ragged import ragged_factory_ops
  from tensorflow.python.ops.ragged import ragged_tensor
  from tensorflow.python.ops.ragged.ragged_tensor import RaggedTensor
  from tensorflow.python.platform import benchmark
  from tensorflow.python.platform import googletest
  from tensorflow.python.util import nest
label_desc: Use image node_id 17 for calling the RaggedTensorToTensorOpTest obj's underlying member method code with example usage: obj.rt_with_partition_type(rt, partition_type) and returns: rt, rt, ragged_tensor