code: def test_scan_partial_eval(self):
      def f_jax(xs, ys):
          body_const = np.ones(
              (2,), dtype=np.float32
          )  # Test constant capture
  
          def body(res0, inputs):
              x, y = inputs
              return res0 + x * y, body_const
  
          c_out, _ = lax.scan(body, 0.0, (xs, ys))
          return c_out
  
      arg = np.arange(10, dtype=np.float32)
      self.ConvertAndCompare(jax.grad(f_jax), arg, arg)
  
imports: from absl.testing import absltest
  import jax
  import jax.lax
  import jax.numpy
  from jax._src import test_util
  import numpy
  from jax.experimental.jax2tf.tests import tf_test_util
  from jax import config
label_desc: Use image node_id 12 for calling the ControlFlowOpsTest obj's underlying member method code with example usage: obj.test_scan_partial_eval() and returns: c_out, unknown,  body_const