Skip to content

Commit 61b3545

Browse files
tensorflower-gardenertensorflower-gardener
committed
Merge pull request #1043 from SaoirseARM:toupstream/update_filepaths
PiperOrigin-RevId: 513242106
2 parents 9f9ef85 + 48e1715 commit 61b3545

24 files changed

+28
-28
lines changed

‎tensorflow_model_optimization/python/core/api/BUILD

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -14,7 +14,7 @@ py_strict_library(
1414
"experimental/combine/__init__.py",
1515
"quantization/__init__.py",
1616
"quantization/keras/__init__.py",
17-
"quantization/keras/collaborative_optimizations/__init__.py",
17+
"quantization/keras/collab_opts/__init__.py",
1818
"quantization/keras/default_8bit/__init__.py",
1919
"quantization/keras/default_8bit/default_8bit_transforms/__init__.py",
2020
"quantization/keras/experimental/__init__.py",
@@ -42,9 +42,9 @@ py_strict_library(
4242
"//tensorflow_model_optimization/python/core/quantization/keras:quantize_scheme",
4343
"//tensorflow_model_optimization/python/core/quantization/keras:quantize_wrapper",
4444
"//tensorflow_model_optimization/python/core/quantization/keras:quantizers",
45-
"//tensorflow_model_optimization/python/core/quantization/keras/collaborative_optimizations/cluster_preserve:cluster_utils",
46-
"//tensorflow_model_optimization/python/core/quantization/keras/collaborative_optimizations/cluster_preserve:default_8bit_cluster_preserve_quantize_scheme",
47-
"//tensorflow_model_optimization/python/core/quantization/keras/collaborative_optimizations/prune_preserve:default_8bit_prune_preserve_quantize_scheme",
45+
"//tensorflow_model_optimization/python/core/quantization/keras/collab_opts/cluster_preserve:cluster_utils",
46+
"//tensorflow_model_optimization/python/core/quantization/keras/collab_opts/cluster_preserve:default_8bit_cluster_preserve_quantize_scheme",
47+
"//tensorflow_model_optimization/python/core/quantization/keras/collab_opts/prune_preserve:default_8bit_prune_preserve_quantize_scheme",
4848
"//tensorflow_model_optimization/python/core/quantization/keras/default_8bit:default_8bit_quantize_layout_transform",
4949
"//tensorflow_model_optimization/python/core/quantization/keras/default_8bit:default_8bit_quantize_registry",
5050
"//tensorflow_model_optimization/python/core/quantization/keras/default_8bit:default_8bit_quantize_scheme",

‎tensorflow_model_optimization/python/core/api/experimental/combine/__init__.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -14,13 +14,13 @@
1414
# ==============================================================================
1515
"""Module containing collaborative optimization code."""
1616

17-
from tensorflow_model_optimization.python.core.quantization.keras.collaborative_optimizations.cluster_preserve.cluster_utils import (
17+
from tensorflow_model_optimization.python.core.quantization.keras.collab_opts.cluster_preserve.cluster_utils import (
1818
strip_clustering_cqat,)
1919

20-
from tensorflow_model_optimization.python.core.quantization.keras.collaborative_optimizations.cluster_preserve.default_8bit_cluster_preserve_quantize_scheme import (
20+
from tensorflow_model_optimization.python.core.quantization.keras.collab_opts.cluster_preserve.default_8bit_cluster_preserve_quantize_scheme import (
2121
Default8BitClusterPreserveQuantizeScheme,)
2222

2323
# Deprecated import.
24-
# Please import from tfmot.quantization.keras.collaborative_optimizations
25-
from tensorflow_model_optimization.python.core.quantization.keras.collaborative_optimizations.prune_preserve.default_8bit_prune_preserve_quantize_scheme import (
24+
# Please import from tfmot.quantization.keras.collab_opts
25+
from tensorflow_model_optimization.python.core.quantization.keras.collab_opts.prune_preserve.default_8bit_prune_preserve_quantize_scheme import (
2626
Default8BitPrunePreserveQuantizeScheme,)

‎tensorflow_model_optimization/python/core/api/quantization/keras/__init__.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -19,7 +19,7 @@
1919
from tensorflow_model_optimization.python.core.api.quantization.keras import quantizers
2020
from tensorflow_model_optimization.python.core.api.quantization.keras import default_8bit
2121
from tensorflow_model_optimization.python.core.api.quantization.keras import graph_transformations
22-
from tensorflow_model_optimization.python.core.api.quantization.keras import collaborative_optimizations
22+
from tensorflow_model_optimization.python.core.api.quantization.keras import collab_opts
2323
from tensorflow_model_optimization.python.core.api.quantization.keras import experimental
2424

2525
# quantize all layers with default quantization implementation.

‎tensorflow_model_optimization/python/core/api/quantization/keras/collaborative_optimizations/__init__.py renamed to ‎tensorflow_model_optimization/python/core/api/quantization/keras/collab_opts/__init__.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -14,6 +14,6 @@
1414
# ==============================================================================
1515
"""Module containing collaborative optimization code."""
1616

17-
from tensorflow_model_optimization.python.core.quantization.keras.collaborative_optimizations.prune_preserve.default_8bit_prune_preserve_quantize_scheme import (
17+
from tensorflow_model_optimization.python.core.quantization.keras.collab_opts.prune_preserve.default_8bit_prune_preserve_quantize_scheme import (
1818
Default8BitPrunePreserveQuantizeScheme,)
1919

‎tensorflow_model_optimization/python/core/quantization/keras/BUILD

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -20,7 +20,7 @@ py_strict_library(
2020
"//tensorflow_model_optimization/python/core/quantization/keras/layers", # buildcleaner: keep
2121
"//tensorflow_model_optimization/python/core/quantization/keras/default_8bit", # buildcleaner: keep
2222
"//tensorflow_model_optimization/python/core/quantization/keras/experimental/default_n_bit", # buildcleaner: keep
23-
"//tensorflow_model_optimization/python/core/quantization/keras/collaborative_optimizations", # buildcleaner: keep
23+
"//tensorflow_model_optimization/python/core/quantization/keras/collab_opts", # buildcleaner: keep
2424
],
2525
)
2626

‎tensorflow_model_optimization/python/core/quantization/keras/collaborative_optimizations/BUILD renamed to ‎tensorflow_model_optimization/python/core/quantization/keras/collab_opts/BUILD

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -5,11 +5,11 @@ package(default_visibility = ["//tensorflow_model_optimization:__subpackages__"]
55
licenses(["notice"])
66

77
py_strict_library(
8-
name = "collaborative_optimizations",
8+
name = "collab_opts",
99
srcs = ["__init__.py"],
1010
srcs_version = "PY3",
1111
deps = [
12-
"//tensorflow_model_optimization/python/core/quantization/keras/collaborative_optimizations/cluster_preserve", # buildcleaner: keep
13-
"//tensorflow_model_optimization/python/core/quantization/keras/collaborative_optimizations/prune_preserve", # buildcleaner: keep
12+
"//tensorflow_model_optimization/python/core/quantization/keras/collab_opts/cluster_preserve", # buildcleaner: keep
13+
"//tensorflow_model_optimization/python/core/quantization/keras/collab_opts/prune_preserve", # buildcleaner: keep
1414
],
1515
)

‎tensorflow_model_optimization/python/core/quantization/keras/collaborative_optimizations/__init__.py renamed to ‎tensorflow_model_optimization/python/core/quantization/keras/collab_opts/__init__.py

File renamed without changes.

‎tensorflow_model_optimization/python/core/quantization/keras/collaborative_optimizations/cluster_preserve/BUILD renamed to ‎tensorflow_model_optimization/python/core/quantization/keras/collab_opts/cluster_preserve/BUILD

File renamed without changes.

‎tensorflow_model_optimization/python/core/quantization/keras/collaborative_optimizations/cluster_preserve/__init__.py renamed to ‎tensorflow_model_optimization/python/core/quantization/keras/collab_opts/cluster_preserve/__init__.py

File renamed without changes.

‎tensorflow_model_optimization/python/core/quantization/keras/collaborative_optimizations/cluster_preserve/cluster_preserve_integration_test.py renamed to ‎tensorflow_model_optimization/python/core/quantization/keras/collab_opts/cluster_preserve/cluster_preserve_integration_test.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -21,9 +21,9 @@
2121
from tensorflow_model_optimization.python.core.clustering.keras import cluster_config
2222
from tensorflow_model_optimization.python.core.clustering.keras.experimental import cluster as experimental_cluster
2323
from tensorflow_model_optimization.python.core.quantization.keras import quantize
24-
from tensorflow_model_optimization.python.core.quantization.keras.collaborative_optimizations.cluster_preserve import (
24+
from tensorflow_model_optimization.python.core.quantization.keras.collab_opts.cluster_preserve import (
2525
default_8bit_cluster_preserve_quantize_scheme,)
26-
from tensorflow_model_optimization.python.core.quantization.keras.collaborative_optimizations.cluster_preserve.cluster_utils import (
26+
from tensorflow_model_optimization.python.core.quantization.keras.collab_opts.cluster_preserve.cluster_utils import (
2727
strip_clustering_cqat,)
2828

2929
layers = tf.keras.layers

0 commit comments

Comments
 (0)