Fixes for Python 3 compatibility.

Author: Johannes Ballé

tensorflow_compression/__init__.py

@@ -28,7 +28,7 @@
 from tensorflow_compression.python.layers.gdn import *
 from tensorflow_compression.python.layers.initializers import *
 from tensorflow_compression.python.layers.parameterizers import *
-from tensorflow_compression.python.layers.signal import *
+from tensorflow_compression.python.layers.signal_conv import *
 from tensorflow_compression.python.ops.coder_ops import *
 from tensorflow_compression.python.ops.math_ops import *
 from tensorflow_compression.python.ops.padding_ops import *

tensorflow_compression/python/layers/gdn.py

@@ -170,7 +170,7 @@ def call(self, inputs):
       norm_pool += self.beta
       if self.data_format == "channels_first":
         # Return to channels_first format if necessary.
-        axes = range(ndim - 1)
+        axes = list(range(ndim - 1))
         axes.insert(1, ndim - 1)
         norm_pool = array_ops.transpose(norm_pool, axes)