Project import generated by Copybara.

PiperOrigin-RevId: 246919162
Change-Id: I35397f2dca2d9a1ef15a436fa0d86a706db57011

Author: Googler

BUILD

@@ -1,34 +1,8 @@
 licenses(["notice"])  # Apache 2.0
 
-package(default_visibility = ["//visibility:public"])
-
-cc_binary(
-    name = "_range_coding_ops.so",
-    srcs = [
-        "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 = [
-        "@local_config_tf//:libtensorflow_framework",
-        "@local_config_tf//:tf_header_lib",
-    ],
-    copts = [
-        "-pthread", "-std=c++11", "-D_GLIBCXX_USE_CXX11_ABI=0",
-        "-Wno-sign-compare", "-Wno-maybe-uninitialized",
-    ],
-)
-
 py_library(
     name = "tensorflow_compression",
     srcs = [
-        "__init__.py",
         "tensorflow_compression/__init__.py",
         "tensorflow_compression/python/__init__.py",
         "tensorflow_compression/python/layers/__init__.py",
@@ -38,15 +12,16 @@ py_library(
         "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/namespace_helper.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"
+        "//tensorflow_compression/cc:libtensorflow_compression.so",
     ],
+    visibility = ["//visibility:public"],
 )
 
 sh_binary(

MANIFEST.in

@@ -1 +1,4 @@
-recursive-include tensorflow_compression/ *.so
+global-include LICENSE
+global-include NOTICE
+global-include *.md
+recursive-include tensorflow_compression/cc/ *.so

__init__.py

@@ -13,9 +13,10 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 # ==============================================================================
-
 # This is based on
 # https://github.com/tensorflow/custom-op/blob/master/build_pip_pkg.sh
+# and modified for this project.
+# ==============================================================================
 
 set -e
 set -x
@@ -36,7 +37,12 @@ function main() {
   echo $(date) : "=== Using tmpdir: ${TMPDIR}"
 
   echo "=== Copying files"
+  PKGDIR="${TMPDIR}/tensorflow_compression"
   rsync -avm -L --exclude='*_test.py' --exclude='build_pip_pkg*' ${PIP_FILE_PREFIX} "${TMPDIR}"
+  for FILENAME in "LICENSE" "README.md"
+  do
+    mv "${TMPDIR}/${FILENAME}" "${PKGDIR}"
+  done
 
   echo $(date) : "=== Building wheel"
   pushd ${TMPDIR}

build_pip_pkg.sh

@@ -1,4 +1,3 @@
-# -*- coding: utf-8 -*-
 # Copyright 2018 Google LLC. All Rights Reserved.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
@@ -13,3 +12,20 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 # ==============================================================================
+"""Data compression tools."""
+
+from __future__ import absolute_import as _absolute_import
+from __future__ import division as _division
+from __future__ import print_function as _print_function
+
+# pylint: disable=wildcard-import
+from tensorflow_compression.python.layers.entropy_models import *
+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_conv import *
+from tensorflow_compression.python.ops.math_ops import *
+from tensorflow_compression.python.ops.padding_ops import *
+from tensorflow_compression.python.ops.range_coding_ops import *
+from tensorflow_compression.python.ops.spectral_ops import *
+# pylint: enable=wildcard-import

tensorflow_compression/__init__.py

@@ -0,0 +1,31 @@
+licenses(["notice"])  # Apache 2.0
+
+cc_binary(
+    name = "libtensorflow_compression.so",
+    srcs = glob(
+        include = [
+            "**/*.h",
+            "**/*.cc",
+        ],
+        exclude = ["**/*test.cc"],
+    ),
+    copts = [
+        "-pthread",
+        "-std=c++11",
+        "-D_GLIBCXX_USE_CXX11_ABI=0",
+        "-Wno-sign-compare",
+        "-Wno-maybe-uninitialized",
+    ],
+    linkshared = 1,
+    linkstatic = 0,
+    visibility = ["//visibility:public"],
+    deps = [
+        # When building this binary together with TensorFlow, i.e., when using
+        # manually built TensorFlow, add the TensorFlow repo in WORKSPACE file
+        # as, say "org_tensorflow" and replace the deps with
+        #   "@org_tensorflow//tensorflow/core:framework",
+        #   "@org_tensorflow//tensorflow/core:lib",
+        "@local_config_tf//:libtensorflow_framework",
+        "@local_config_tf//:tf_header_lib",
+    ],
+)

tensorflow_compression/cc/BUILD

@@ -23,7 +23,8 @@
 
 
 ops = namespace_helper.get_ops(load_library.load_op_library(
-    resource_loader.get_path_to_datafile("../../../_range_coding_ops.so")))
+    resource_loader.get_path_to_datafile(
+        "../../cc/libtensorflow_compression.so")))
 
 globals().update(ops)
 __all__ = list(ops)