Allows wrapping for entropy model.

PiperOrigin-RevId: 331651188
Change-Id: Id7d165d11567eec86dda8ee2eac9d3f2fd289043

Author: Xhark

tensorflow_compression/python/entropy_models/continuous_base.py

@@ -103,17 +103,17 @@ def _check_compression(self):
   @property
   def cdf(self):
     self._check_compression()
-    return self._cdf.value()
+    return tf.identity(self._cdf)
 
   @property
   def cdf_offset(self):
     self._check_compression()
-    return self._cdf_offset.value()
+    return tf.identity(self._cdf_offset)
 
   @property
   def cdf_length(self):
     self._check_compression()
-    return self._cdf_length.value()
+    return tf.identity(self._cdf_length)
 
   @property
   def dtype(self):