We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 0bb2540 commit 93d2437Copy full SHA for 93d2437
tf2onnx/rewriter/conv2d_with_add_rewriter.py
@@ -32,6 +32,8 @@ def rewrite_biasadd_with_conv2d(g, ops):
32
conv_output = biasadd.output
33
conv_inputs = [conv_input[0], conv_input[1], biasadd.input[1]]
34
35
+ if len(g.find_output_consumers(conv.output[0])) > 1:
36
+ continue
37
# Remove the Conv and BiasAdd node
38
g.remove_node(conv.name)
39
g.remove_node(biasadd.name)
0 commit comments