code: def test_run_with_distribute_value_input(self):
      strategy = mwms.MultiWorkerMirroredStrategy()
  
      def value_fn(value_context):
          return value_context.replica_id_in_sync_group
  
      distributed_values = (
          strategy.experimental_distribute_values_from_function(
              value_fn
          )
      )
  
      @def_function.function
      def replica_fn(inputs):
          return inputs * 2
  
      result = strategy.run(replica_fn, args=(distributed_values,))
      self.assertIsInstance(
          result, dtensor_util.DTensorDistributedValue
      )
      self.assertLen(result.values, self.num_local_devices)
      # Note that the scalar value from
      # experimental_distribute_values_from_function will be up rank to 1D since
      # batched shared dtensor need at least be 1D.
  
      for i in range(self.num_local_devices):
          self.assertAllClose(
              result.values[i],
              constant_op.constant(
                  [(self.client_id * self.num_local_devices + i) * 2]
              ),
          )
  
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 14 for calling the MultiWorkerMirroredStrategyTest obj's underlying member method code with example usage: obj.test_run_with_distribute_value_input() and returns: value_context, unknown