Fixes GCC compilation of unbounded index range coder and adds BUILD rules.

Author: Johannes Ballé

BUILD

@@ -11,6 +11,7 @@ cc_binary(
         "cc/kernels/range_coding_kernels_util.cc",
         "cc/kernels/range_coding_kernels_util.h",
         "cc/kernels/range_coding_kernels.cc",
+        "cc/kernels/unbounded_index_range_coding_kernels.cc",
         "cc/ops/range_coding_ops.cc",
     ],
     linkshared = 1,
@@ -20,7 +21,7 @@ cc_binary(
     ],
     copts = [
         "-pthread", "-std=c++11", "-D_GLIBCXX_USE_CXX11_ABI=0",
-        "-Wno-sign-compare",
+        "-Wno-sign-compare", "-Wno-maybe-uninitialized",
     ],
 )
 
@@ -44,63 +45,54 @@ py_library(
     data = [
         ":_range_coding_ops.so"
     ],
-    srcs_version = "PY2AND3",
 )
 
 py_test(
     name = "entropy_models_test",
     timeout = "long",
     srcs = ["python/layers/entropy_models_test.py"],
     deps = [":tensorflow_compression"],
-    srcs_version = "PY2AND3",
 )
 
 py_test(
     name = "gdn_test",
     srcs = ["python/layers/gdn_test.py"],
     deps = [":tensorflow_compression"],
-    srcs_version = "PY2AND3",
 )
 
 py_test(
     name = "parameterizers_test",
     srcs = ["python/layers/parameterizers_test.py"],
     deps = [":tensorflow_compression"],
-    srcs_version = "PY2AND3",
 )
 
 py_test(
     name = "signal_conv_test",
     timeout = "long",
     srcs = ["python/layers/signal_conv_test.py"],
     deps = [":tensorflow_compression"],
-    srcs_version = "PY2AND3",
 )
 
 py_test(
     name = "math_ops_test",
     srcs = ["python/ops/math_ops_test.py"],
     deps = [":tensorflow_compression"],
-    srcs_version = "PY2AND3",
 )
 
 py_test(
     name = "padding_ops_test",
     srcs = ["python/ops/padding_ops_test.py"],
     deps = [":tensorflow_compression"],
-    srcs_version = "PY2AND3",
 )
 
 py_test(
     name = "spectral_ops_test",
     srcs = ["python/ops/spectral_ops_test.py"],
     deps = [":tensorflow_compression"],
-    srcs_version = "PY2AND3",
 )
 
 py_test(
     name = "range_coding_ops_test",
     srcs = ["python/ops/range_coding_ops_test.py"],
     deps = [":tensorflow_compression"],
-    srcs_version = "PY2AND3",
 )

cc/kernels/unbounded_index_range_coding_kernels.cc

@@ -47,6 +47,13 @@ using tensorflow::Tensor;
 using tensorflow::TensorShape;
 using tensorflow::TensorShapeUtils;
 using tensorflow::TTypes;
+using tensorflow::int16;
+using tensorflow::int32;
+using tensorflow::int64;
+using tensorflow::string;
+using tensorflow::uint8;
+using tensorflow::uint32;
+using tensorflow::uint64;
 
 // Non-incremental encoder op -------------------------------------------------
 class UnboundedIndexRangeEncodeOp : public OpKernel {

cc/kernels/unbounded_index_range_coding_kernels_test.cc

@@ -62,6 +62,13 @@ using tensorflow::Status;
 using tensorflow::Tensor;
 using tensorflow::TensorShape;
 using tensorflow::TTypes;
+using tensorflow::int16;
+using tensorflow::int32;
+using tensorflow::int64;
+using tensorflow::string;
+using tensorflow::uint8;
+using tensorflow::uint32;
+using tensorflow::uint64;
 
 int LogUniform(random::SimplePhilox* gen, uint32 n) {
   CHECK_GT(n, 0);