code: def assertSameValuePerReplica(self, replicator, per_replica):
      per_replica = replicator.experimental_local_results(per_replica)
      first_replica = per_replica[0]
      for nth_replica in per_replica[1:]:
          self.assertAllEqual(first_replica, nth_replica)
  
file_path: sonnet/sonnet/src/conformance/distribute_test.py
imports: from typing import Callable, Tuple
  from absl.testing import parameterized
  import sonnet
  from sonnet.src import test_utils
  from sonnet.src.conformance import descriptors
  from sonnet.src.conformance import goldens
  from sonnet.src.distribute import replicator
  from sonnet.src.distribute import replicator_test_utils
  import tensorflow
label_desc: Use image node_id 2 for calling the TpuReplicatorTest obj's underlying member method code with example usage: obj.assertSameValuePerReplica(replicator, per_replica) without return types