code: def test_reduce_mean_distribute_value(self):
      strategy = mwms.MultiWorkerMirroredStrategy()
  
      @def_function.function
      def value_fn(value_context):
          i = value_context.replica_id_in_sync_group
          return (
              constant_op.constant([[0.0, 1.0], [2.0, 3.0]]) + i * 4.0
          )
  
      distribute_value = (
          strategy.experimental_distribute_values_from_function(
              value_fn
          )
      )
      # replica 0 has [[0.0, 1.0],[2.0, 3.0]] and
      # replica 1 has [[4.0, 5.0],[6.0, 7.0]]. Each worker has 4 replicas.
      # For worker 2, it has replica 4 ~ 7.
  
      result = strategy.reduce(
          reduce_util.ReduceOp.MEAN, distribute_value, axis=None
      )
      # This should be a global reduce and each worker should have same value.
      # [[14.0, 15.0],[16.0, 17.0]]
      final = (self.num_local_devices * self.num_client - 1) * 2.0
      self.assertAllClose(
          result, constant_op.constant([[0.0, 1.0], [2.0, 3.0]]) + final
      )
  
      result = strategy.reduce(
          reduce_util.ReduceOp.MEAN, distribute_value, axis=0
      )
      # [15.0, 16.0]
      self.assertAllClose(
          result, constant_op.constant([0.0, 1.0]) + final + 1
      )
  
      result = strategy.reduce(
          reduce_util.ReduceOp.MEAN, distribute_value, axis=1
      )
  
      self.assertAllClose(
          result, constant_op.constant([0.5, 2.5]) + final
      )
  
imports: import json
  import os
  from absl import flags
  from absl.testing import parameterized
  import numpy
  from tensorflow.dtensor.python import api
  from tensorflow.dtensor.python import d_variable
  from tensorflow.dtensor.python import layout
  from tensorflow.dtensor.python.tests import multi_client_test_util
  from tensorflow.dtensor.python.tests import test_backend_util
  from tensorflow.dtensor.python.tests import test_util
  from tensorflow.python.data.ops import dataset_ops
  from tensorflow.python.distribute import distribute_lib
  from tensorflow.python.distribute import reduce_util
  from tensorflow.python.distribute.cluster_resolver import tfconfig_cluster_resolver
  from tensorflow.python.distribute.experimental import dtensor_util
  from tensorflow.python.distribute.experimental import multi_worker_mirrored_strategy
  from tensorflow.python.eager import def_function
  from tensorflow.python.framework import config
  from tensorflow.python.framework import constant_op
  from tensorflow.python.framework import dtypes
  from tensorflow.python.framework import tensor_spec
  from tensorflow.python.ops import array_ops
  from tensorflow.python.ops import math_ops
  from tensorflow.python.ops import stateless_random_ops
  from tensorflow.python.ops import variables
  from tensorflow.python.platform import test
label_desc: Use image node_id 22 for calling the MultiWorkerMirroredStrategyTest obj's underlying member method code with example usage: obj.test_reduce_mean_distribute_value() and returns: unknown