Skip to content

Commit 43c7e28

Browse files
nutsiepullytensorflower-gardener
authored andcommitted
Update comments to suggest current qat support
PiperOrigin-RevId: 367074181
1 parent a6d5c7b commit 43c7e28

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

tensorflow_model_optimization/python/core/quantization/keras/default_8bit/default_8bit_quantize_registry.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -116,9 +116,9 @@ class Default8BitQuantizeRegistry(
116116
_no_quantize(layers.ZeroPadding1D),
117117
_no_quantize(layers.ZeroPadding2D),
118118
# _no_quantize(layers.ZeroPadding3D),
119-
# Enable once verified.
120-
# layers.SeparableConv1D,
121-
# layers.SeparableConv2D,
119+
120+
# Supported via modifications in Transforms.
121+
# layers.SeparableConv1D, layers.SeparableConv2D,
122122

123123
# Core Layers
124124
_no_quantize(layers.ActivityRegularization),
@@ -156,7 +156,7 @@ class Default8BitQuantizeRegistry(
156156
# layers.Embedding: ['embeddings'],
157157

158158
# BatchNormalization is handled elsewhere, in the cases
159-
# where it's preceded by convolutional layers or a Dense layer.
159+
# where it's preceded by convolutional layers.
160160
# layers.BatchNormalization: [],
161161

162162
# Merge layers to be added.

0 commit comments

Comments
 (0)