Commit 43d232e5 authored by Jaehong Kim's avatar Jaehong Kim Committed by A. Unique TensorFlower
Browse files

Remove duplicated bias and activation from the quantized separable conv.

PiperOrigin-RevId: 446001473
parent 2aff7786
......@@ -64,6 +64,10 @@ class SeparableConv2DQuantized(tf.keras.layers.Layer):
# Depthwise conv input filters is always equal to output filters.
# This filters argument only needed for the point-wise conv2d op.
del dwconv_kwargs['filters']
dwconv_kwargs.update({
'activation': None,
'use_bias': False,
})
self.dw_conv = depthwise_conv2d_quantized(name='dw', **dwconv_kwargs)
conv_kwargs = self._conv_kwargs.copy()
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment