The tf2.1 QuantizedConv2D op only supports uint8 input.
@@ -52,10 +52,9 @@ class FuseNodeStartWithConv2d(QuantizeNodeBase):
matched_node = self.node_name_mapping[match_node_name[0]]
_, normal_inputs = self._get_node_input(matched_node.node.name)
weight_name = normal_inputs[1]
- # TODO this is workaround as the tf 2.1 doesn't support depthwise s8
+ # TODO this is workaround as the tf 2.1 doesn't support depthwise/conv s8
# feature.
- if self.enable_s8 and matched_node.node.op == "DepthwiseConv2dNative" and not self._find_relu_node(
- matched_node.node):
+ if self.enable_s8 and not self._find_relu_node(matched_node.node):
self.output_graph = self.input_graph
return
