Remove tfc.Round

Callers can instead use:
	tf.keras.layers.Lambda(tf.round)
or
	tf.keras.layers.Lambda(tfc.round_st).

PiperOrigin-RevId: 424283975
Change-Id: I71f555fe84884c23839e86b297266e2ae159ca67

Author: relational

tensorflow_compression/python/layers/soft_round.py

@@ -19,22 +19,11 @@
 
 
 __all__ = [
-    "Round",
     "SoftRound",
     "SoftRoundConditionalMean",
 ]
 
 
-class Round(tf.keras.layers.Layer):
-  """Applies rounding."""
-
-  def call(self, inputs):
-    return tf.round(inputs)
-
-  def compute_output_shape(self, input_shape):
-    return input_shape
-
-
 class SoftRound(tf.keras.layers.Layer):
   """Applies a differentiable approximation of rounding."""
 

tensorflow_compression/python/layers/soft_round_test.py

@@ -21,12 +21,6 @@
 
 class SoftRoundTest(tf.test.TestCase):
 
-  def test_round_layer_rounds(self):
-    layer = soft_round.Round()
-    x = tf.linspace(-5.0, 5.0, num=50)
-    y = layer(x)
-    self.assertAllClose(y, tf.math.round(x))
-
   def test_soft_round_layer_soft_rounds(self):
     alpha = 5.0
     layer = soft_round.SoftRound(alpha=alpha)