Skip to content

Inconsistency in conv+bn fusion and addition of useless reshapes in tf2onnx==1.16.1 #2300

Open
@riccardoalberghi

Description

@riccardoalberghi

Describe the bug
When I convert a u-net architecture a Conv operator and a BN operator are not fused together and 2 Reshape operators are added which reshape to the same shape [-1, 512, 1, 1].

Urgency
none

System information

  • OS Platform and Distribution: macos 14.2.1 arm64
  • TensorFlow Version: 2.13.1
  • Python version: 3.10.13
  • ONNX version: 1.15
  • ONNXRuntime version: 1.16
  • tf2onnx version: 1.16.1 (from github)

To Reproduce
https://drive.google.com/file/d/1xzaAXWxUqXUTBVgDSiyrbjs_TSHXuhwW

by running python -m tf2onnx.convert --saved-model bg_model --output bg_model.onnx

Screenshots
from the top the first conv+bn is fused correctly, then the following is not.
Screenshot 2024-01-31 at 13 10 32
Screenshot 2024-01-31 at 13 12 21

Additional context
Every other conv+bn is fused correctly except one.

Metadata

Metadata

Assignees

Labels

bugAn unexpected problem or unintended behaviorpending on user responseWaiting for more information or validation from user

Type

No type

Projects

No projects

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests

Issue actions