code: def test_reduce_mean(self):
      x = random_ops.random_uniform([8, 3])
  
      @def_function.function(jit_compile=True)
      def f():
          def loop_fn(i, pfor_config):
              x_i = array_ops.gather(x, i)
              return x_i - pfor_config.reduce_mean(x_i)
  
          return pfor_control_flow_ops.pfor(loop_fn, 8)
  
      output = f()
      ans = x - math_ops.reduce_mean(x, axis=0)
      output_val, ans_val = self.evaluate([output, ans])
      self.assertAllClose(ans_val, output_val)
  
imports: from tensorflow.compiler.tf2xla.python import xla
  from tensorflow.python.compiler.xla import jit
  from tensorflow.python.compiler.xla import xla
  from tensorflow.python.eager import context
  from tensorflow.python.eager import def_function
  from tensorflow.python.framework import constant_op
  from tensorflow.python.framework import test_util
  from tensorflow.python.ops import array_ops
  from tensorflow.python.ops import control_flow_v2_toggles
  from tensorflow.python.ops import math_ops
  from tensorflow.python.ops import random_ops
  from tensorflow.python.ops import resource_variable_ops
  from tensorflow.python.ops import while_loop
  from tensorflow.python.ops.parallel_for import control_flow_ops
  from tensorflow.python.ops.parallel_for.test_util import PForTestCase
  from tensorflow.python.platform import test
label_desc: Use image node_id 6 for calling the PForTest obj's underlying member method code with example usage: obj.test_reduce_mean() and returns: pfor_control_flow_ops, unknown