Skip to content

Commit d0ef35e

Browse files
Merge pull request #437 from arovir01:toupstream/remove_depthwise_conv2d
PiperOrigin-RevId: 319077886
2 parents 99ab1f3 + cd7dc5b commit d0ef35e

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)