Project import generated by Copybara.

PiperOrigin-RevId: 245842538
Change-Id: Ifa27560e5ef55e14aefefee53cf4b5fb143695be

Author: Googler

BUILD

@@ -5,14 +5,14 @@ package(default_visibility = ["//visibility:public"])
 cc_binary(
     name = "_range_coding_ops.so",
     srcs = [
-        "cc/kernels/range_coder.cc",
-        "cc/kernels/range_coder.h",
-        "cc/kernels/range_coding_helper_kernels.cc",
-        "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",
+        "tensorflow_compression/cc/kernels/range_coder.cc",
+        "tensorflow_compression/cc/kernels/range_coder.h",
+        "tensorflow_compression/cc/kernels/range_coding_helper_kernels.cc",
+        "tensorflow_compression/cc/kernels/range_coding_kernels_util.cc",
+        "tensorflow_compression/cc/kernels/range_coding_kernels_util.h",
+        "tensorflow_compression/cc/kernels/range_coding_kernels.cc",
+        "tensorflow_compression/cc/kernels/unbounded_index_range_coding_kernels.cc",
+        "tensorflow_compression/cc/ops/range_coding_ops.cc",
     ],
     linkshared = 1,
     deps = [
@@ -29,19 +29,20 @@ py_library(
     name = "tensorflow_compression",
     srcs = [
         "__init__.py",
-        "python/__init__.py",
-        "python/layers/__init__.py",
-        "python/layers/entropy_models.py",
-        "python/layers/gdn.py",
-        "python/layers/initializers.py",
-        "python/layers/parameterizers.py",
-        "python/layers/signal_conv.py",
-        "python/ops/__init__.py",
-        "python/ops/_namespace_helper.py",
-        "python/ops/math_ops.py",
-        "python/ops/padding_ops.py",
-        "python/ops/range_coding_ops.py",
-        "python/ops/spectral_ops.py",
+        "tensorflow_compression/__init__.py",
+        "tensorflow_compression/python/__init__.py",
+        "tensorflow_compression/python/layers/__init__.py",
+        "tensorflow_compression/python/layers/entropy_models.py",
+        "tensorflow_compression/python/layers/gdn.py",
+        "tensorflow_compression/python/layers/initializers.py",
+        "tensorflow_compression/python/layers/parameterizers.py",
+        "tensorflow_compression/python/layers/signal_conv.py",
+        "tensorflow_compression/python/ops/__init__.py",
+        "tensorflow_compression/python/ops/namespace_helper.py",
+        "tensorflow_compression/python/ops/math_ops.py",
+        "tensorflow_compression/python/ops/padding_ops.py",
+        "tensorflow_compression/python/ops/range_coding_ops.py",
+        "tensorflow_compression/python/ops/spectral_ops.py",
     ],
     data = [
         ":_range_coding_ops.so"
@@ -51,49 +52,49 @@ py_library(
 py_test(
     name = "entropy_models_test",
     timeout = "long",
-    srcs = ["python/layers/entropy_models_test.py"],
+    srcs = ["tensorflow_compression/python/layers/entropy_models_test.py"],
     deps = [":tensorflow_compression"],
 )
 
 py_test(
     name = "gdn_test",
-    srcs = ["python/layers/gdn_test.py"],
+    srcs = ["tensorflow_compression/python/layers/gdn_test.py"],
     deps = [":tensorflow_compression"],
 )
 
 py_test(
     name = "parameterizers_test",
-    srcs = ["python/layers/parameterizers_test.py"],
+    srcs = ["tensorflow_compression/python/layers/parameterizers_test.py"],
     deps = [":tensorflow_compression"],
 )
 
 py_test(
     name = "signal_conv_test",
     timeout = "long",
-    srcs = ["python/layers/signal_conv_test.py"],
+    srcs = ["tensorflow_compression/python/layers/signal_conv_test.py"],
     deps = [":tensorflow_compression"],
 )
 
 py_test(
     name = "math_ops_test",
-    srcs = ["python/ops/math_ops_test.py"],
+    srcs = ["tensorflow_compression/python/ops/math_ops_test.py"],
     deps = [":tensorflow_compression"],
 )
 
 py_test(
     name = "padding_ops_test",
-    srcs = ["python/ops/padding_ops_test.py"],
+    srcs = ["tensorflow_compression/python/ops/padding_ops_test.py"],
     deps = [":tensorflow_compression"],
 )
 
 py_test(
     name = "spectral_ops_test",
-    srcs = ["python/ops/spectral_ops_test.py"],
+    srcs = ["tensorflow_compression/python/ops/spectral_ops_test.py"],
     deps = [":tensorflow_compression"],
 )
 
 py_test(
     name = "range_coding_ops_test",
-    srcs = ["python/ops/range_coding_ops_test.py"],
+    srcs = ["tensorflow_compression/python/ops/range_coding_ops_test.py"],
     deps = [":tensorflow_compression"],
 )

WORKSPACE

@@ -0,0 +1,8 @@
+local_repository(
+    name = "tf_custom_op",
+    # Change if tensorflow/custom-op repo is cloned into a different directory.
+    path = "/custom-op",
+)
+
+load("//:workspace.bzl", "tf_compression_workspace")
+tf_compression_workspace()

examples/bls2017.py

@@ -30,8 +30,6 @@
 import argparse
 import glob
 
-# Dependency imports
-
 import numpy as np
 import tensorflow as tf
 import tensorflow_compression as tfc

examples/tfci.py

@@ -22,13 +22,12 @@
 import argparse
 import os
 
-# Dependency imports
-
 from absl import app
 from absl.flags import argparse_flags
 import numpy as np
 from six.moves import urllib
 import tensorflow as tf
+
 import tensorflow_compression as tfc  # pylint:disable=unused-import
 
 

python/__init__.py renamed to tensorflow_compression/__init__.py

@@ -28,17 +28,16 @@ limitations under the License.
 #include "tensorflow/core/lib/core/errors.h"
 #include "tensorflow/core/lib/core/threadpool.h"
 #include "tensorflow/core/lib/gtl/array_slice.h"
-#include "tensorflow/core/platform/fingerprint.h"
 #include "tensorflow/core/platform/logging.h"
 #include "tensorflow/core/platform/macros.h"
 #include "tensorflow/core/platform/types.h"
+#include "external/farmhash_archive/src/farmhash.h"
 
 namespace tensorflow_compression {
 namespace {
 namespace gtl = tensorflow::gtl;
 namespace thread = tensorflow::thread;
 using tensorflow::DEVICE_CPU;
-using tensorflow::Fingerprint64;
 using tensorflow::int32;
 using tensorflow::int64;
 using tensorflow::OpKernel;
@@ -229,7 +228,7 @@ class ArrayFingerprintOp : public tensorflow::OpKernel {
                    context->allocate_output(0, TensorShape{}, &output));
 
     output->scalar<int64>()() =
-        Fingerprint64({data.data(), static_cast<size_t>(data.size())});
+        ::util::Fingerprint64(data.data(), static_cast<size_t>(data.size()));
   }
 };
 
@@ -253,11 +252,11 @@ class CheckArrayFingerprintOp : public tensorflow::OpKernel {
         input.shape().num_elements() * tensorflow::DataTypeSize(input.dtype());
     auto data = input.bit_casted_shaped<char, 1>({size});
 
-    OP_REQUIRES(
-        context,
-        fingerprint.scalar<int64>()() ==
-            Fingerprint64({data.data(), static_cast<size_t>(data.size())}),
-        tensorflow::errors::DataLoss("Fingerprint mismatch"));
+    OP_REQUIRES(context,
+                static_cast<uint64>(fingerprint.scalar<int64>()()) ==
+                    ::util::Fingerprint64(data.data(),
+                                            static_cast<size_t>(data.size())),
+                tensorflow::errors::DataLoss("Fingerprint mismatch"));
 
     context->set_output(0, input);
   }