"tools/imglab/CMakeLists.txt" did not exist on "754da0ef3ceb598b2e708f1498f6d22c92b290d7"
Commit 9ced825b 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 efe509a0
......@@ -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