Updates deprecated API: tf.random_uniform() -> tf.random.uniform().

PiperOrigin-RevId: 242740993

Author: minnend

python/layers/entropy_models.py

@@ -660,7 +660,7 @@ def _quantize(self, inputs, mode):
     _, _, _, input_slices = self._get_input_dims()
 
     if mode == "noise":
-      noise = tf.random_uniform(tf.shape(inputs), -half, half)
+      noise = tf.random.uniform(tf.shape(inputs), -half, half)
       return tf.math.add_n([inputs, noise])
 
     medians = self._medians[input_slices]
@@ -987,7 +987,7 @@ def _quantize(self, inputs, mode):
     half = tf.constant(.5, dtype=self.dtype)
 
     if mode == "noise":
-      noise = tf.random_uniform(tf.shape(inputs), -half, half)
+      noise = tf.random.uniform(tf.shape(inputs), -half, half)
       return tf.math.add_n([inputs, noise])
 
     outputs = inputs

python/ops/range_coding_ops_test.py

@@ -32,7 +32,7 @@ class RangeCodingOpsTest(tf.test.TestCase):
   """
 
   def test_readme_example(self):
-    data = tf.random_uniform((128, 128), 0, 10, dtype=tf.int32)
+    data = tf.random.uniform((128, 128), 0, 10, dtype=tf.int32)
     histogram = tf.bincount(data, minlength=10, maxlength=10)
     cdf = tf.cumsum(histogram, exclusive=False)
     cdf = tf.pad(cdf, [[1, 0]])