Skip to content

Commit cd7dc5b

Browse files
committed
Make DepthwiseConv2D non-clusterable
1 parent a396089 commit cd7dc5b

File tree

1 file changed

+1
-2
lines changed

1 file changed

+1
-2
lines changed

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

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -155,7 +155,6 @@ class ClusteringLookupRegistry(object):
155155
layers.Conv2DTranspose: {'kernel': ConvolutionalWeightsCA},
156156
layers.Conv3D: {'kernel': ConvolutionalWeightsCA},
157157
layers.Conv3DTranspose: {'kernel': ConvolutionalWeightsCA},
158-
layers.DepthwiseConv2D: {'depthwise_kernel': ConvolutionalWeightsCA},
159158
layers.SeparableConv1D: {'pointwise_kernel': ConvolutionalWeightsCA},
160159
layers.SeparableConv2D: {'pointwise_kernel': ConvolutionalWeightsCA},
161160
layers.Dense: {'kernel': DenseWeightsCA},
@@ -244,7 +243,7 @@ class ClusteringRegistry(object):
244243
layers.Cropping1D: [],
245244
layers.Cropping2D: [],
246245
layers.Cropping3D: [],
247-
layers.DepthwiseConv2D: ['depthwise_kernel'],
246+
layers.DepthwiseConv2D: [],
248247
layers.SeparableConv1D: ['pointwise_kernel'],
249248
layers.SeparableConv2D: ['pointwise_kernel'],
250249
layers.UpSampling1D: [],

0 commit comments

Comments
 (0)