Skip to content

Commit 2f9d874

Browse files
committed
Small fixes after merging with master.
Change-Id: I51ce035855ff8e82c339f1cd260b7f5891050ab7
1 parent 69376e4 commit 2f9d874

File tree

3 files changed

+1
-4
lines changed

3 files changed

+1
-4
lines changed

tensorflow_model_optimization/python/core/clustering/keras/cluster_test.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -128,7 +128,6 @@ def setUp(self):
128128
self.keras_custom_layer = KerasCustomLayer()
129129
self.clusterable_layer = MyClusterableLayer(10)
130130

131-
132131
clustering_registry.ClusteringLookupRegistry.register_new_implementation(
133132
{
134133
CustomClusterableLayer: {

tensorflow_model_optimization/python/core/clustering/keras/clusterable_layer.py

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -55,10 +55,8 @@ def get_clusterable_algorithm(self, weight_name): # pylint: disable=unused-argu
5555
5656
The returned class should be derived from AbstractClusteringAlgorithm and
5757
implements the function get_pulling_indices.
58-
5958
This function is used to provide a special lookup function for the custom
6059
weights.
61-
6260
It reshapes and tile centroids the same way as the weights. This allows us
6361
to find pulling indices efficiently.
6462

tensorflow_model_optimization/python/core/clustering/keras/clustering_registry.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -23,7 +23,6 @@
2323
AbstractClusteringAlgorithm = clustering_algorithm.AbstractClusteringAlgorithm
2424

2525

26-
2726
class ConvolutionalWeightsCA(AbstractClusteringAlgorithm):
2827
"""Look-ups for convolutional kernels, e.g. tensors with shape [B,W,H,C]."""
2928

@@ -80,6 +79,7 @@ def get_pulling_indices(self, weight):
8079

8180
return pulling_indices
8281

82+
8383
class ClusteringLookupRegistry(object):
8484
"""Map of layers to strategy.
8585

0 commit comments

Comments
 (0)