Disable variable shape check for better interoperability with SavedModel().

PiperOrigin-RevId: 461452867
Change-Id: Ib9c6b2d7719fd8af28fdbebadf88060d8c059d75

Author: lucastheis

tensorflow_compression/python/entropy_models/continuous_base.py

@@ -209,9 +209,11 @@ def _init_compression(self, cdf, cdf_offset, cdf_shapes):
           cdf_offset, dtype=tf.int32, name="cdf_offset")
     else:
       self._cdf = tf.Variable(
-          cdf, dtype=tf.int32, trainable=False, name="cdf")
+          cdf, dtype=tf.int32, trainable=False, name="cdf",
+          validate_shape=False, shape=[None])
       self._cdf_offset = tf.Variable(
-          cdf_offset, dtype=tf.int32, trainable=False, name="cdf_offset")
+          cdf_offset, dtype=tf.int32, trainable=False, name="cdf_offset",
+          validate_shape=False, shape=[None])
 
   def _build_tables(self, prior, precision, offset=None):
     """Computes integer-valued probability tables used by the range coder.