code: def testPruneRagged(self):
      x_in = []
      x_out = []
  
      def f(x, y):
          x_in.append(x)
          xx = x * x
          x_out.append(xx)
          return xx, y * y
  
      x_spec = ragged_tensor.RaggedTensorSpec(
          [None, None], dtypes.float32
      )
      y_spec = tensor_spec.TensorSpec((), dtypes.float32)
  
      f_wrapped = wrap_function.wrap_function(f, [x_spec, y_spec])
  
      f_pruned = f_wrapped.prune(x_in[0], x_out[0])
      rt = ragged_factory_ops.constant([[1.0, 2.0], [3.0]])
      expected = ragged_factory_ops.constant_value([[1.0, 4.0], [9.0]])
  
      # Note: when we call f_pruned, we must pass the RaggedTensor in using
      # its components, since that's the current convention for how concrete
      # functions handle structured inputs.
      self.assertAllEqual(f_pruned(rt.values, rt.row_splits), expected)
  
imports: import os
  from tensorflow.core.protobuf import meta_graph_pb2
  from tensorflow.python.data.ops import dataset_ops
  from tensorflow.python.eager import backprop
  from tensorflow.python.eager import def_function
  from tensorflow.python.eager import wrap_function
  from tensorflow.python.framework import constant_op
  from tensorflow.python.framework import dtypes
  from tensorflow.python.framework import importer
  from tensorflow.python.framework import ops
  from tensorflow.python.framework import tensor_spec
  from tensorflow.python.framework import test_util
  from tensorflow.python.ops import array_ops
  from tensorflow.python.ops import init_ops
  from tensorflow.python.ops import state_ops
  from tensorflow.python.ops import variable_scope
  from tensorflow.python.ops import variable_v1
  from tensorflow.python.ops import variables
  from tensorflow.python.ops.ragged import ragged_factory_ops
  from tensorflow.python.ops.ragged import ragged_tensor
  from tensorflow.python.platform import test
  from tensorflow.python.training import saver
label_desc: Use image node_id 3 for calling the WrapFunctionTest obj's underlying member method code with example usage: obj.testPruneRagged() and returns: xx,  unknown