code: def test_experimental_local_results_with_inputs(self):
      strategy = mwms.MultiWorkerMirroredStrategy()
  
      def value_fn(ctx):
          value = float(ctx.num_replicas_in_sync)
          return {
              "a": value,
              "b": constant_op.constant([value + 1.0, value + 2.0]),
          }
  
      distributed_values = (
          strategy.experimental_distribute_values_from_function(
              value_fn
          )
      )
  
      @def_function.function
      def replica_fn(inputs):
          result = {}
          for key in inputs:
              result[key] = inputs[key] * 2.0
          return result
  
      result = strategy.run(replica_fn, args=(distributed_values,))
      local_result = strategy.experimental_local_results(result)
      self.assertIsInstance(local_result, tuple)
      self.assertLen(local_result, self.num_local_devices)
      for i in range(self.num_local_devices):
          self.assertDictEqual(
              local_result[i],
              {
                  "a": constant_op.constant([16.0]),
                  "b": constant_op.constant([18.0, 20.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 28 for calling the MultiWorkerMirroredStrategyTest obj's underlying member method code with example usage: obj.test_experimental_local_results_with_inputs() and returns: dict, result