code: def test_reduce_sum_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.
      # The shape of the global tensor is [16, 2], and each worker gets [8, 2].
  
      result = strategy.reduce(
          reduce_util.ReduceOp.SUM, distribute_value, axis=None
      )
      self.assertAllClose(result, [[112.0, 120.0], [128.0, 136.0]])
  
      result = strategy.reduce(
          reduce_util.ReduceOp.SUM, distribute_value, axis=0
      )
      self.assertAllClose(result, constant_op.constant([240.0, 256.0]))
  
      result = strategy.reduce(
          reduce_util.ReduceOp.SUM, distribute_value, axis=1
      )
      self.assertAllClose(result, constant_op.constant([232.0, 264.0]))
  
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 23 for calling the MultiWorkerMirroredStrategyTest obj's underlying member method code with example usage: obj.test_reduce_sum_distribute_value() and returns: unknown