class_docstr: Layer returning a tensor with random values distributed uniformly.
class_docstr_tok: ["Layer","returning","a","tensor","with","random","values","distributed","uniformly","."]
code: def _get_conditionally_synced_rng(self):
      if self._sync and fastmath.global_device_count() > 1:
          return fastmath.psum(self.rng, "batch")
      else:
          return self.rng
  
imports: from absl import logging
  import numpy
  import tensorflow
  from trax import fastmath
  from trax.fastmath import numpy
  from trax.layers import base
  from trax.layers import initializers
  from trax.layers.assert_shape import assert_shape
  from trax.layers.base import Fn
label_desc: Use image node_id 3 for calling the RandomUniform obj's underlying member method code with example usage: obj._get_conditionally_synced_rng() and returns: fastmath, self