class_docstr: Python test for quantization ops.
class_docstr_tok: ["Python","test","for","quantization","ops","."]
code: def test_rounding_is_deterministic_at_integers(self, step_size):
      values = tf.random.uniform((100,), -100, 100, dtype=tf.int32)
      rounded = gen_ops.stochastic_round(
          step_size * tf.cast(values, tf.float32), step_size, ()
      )
      self.assertEqual(rounded.dtype, tf.int32)
      self.assertAllEqual(values, rounded)
  
file_path: compression/tensorflow_compression/python/ops/quantization_ops_test.py
imports: import time
  from absl.testing import parameterized
  import tensorflow
  from tensorflow_compression.python.ops import gen_ops
label_desc: Use image node_id 4 for calling the QuantizationOpsTest obj's underlying member method code with example usage: obj.test_rounding_is_deterministic_at_integers(step_size) without return types