Skip to content

Model conversion infinite loop in tf_utils.compute_const_folding_using_tf #1965

Open
@hgong-snap

Description

@hgong-snap

Describe the bug
I tried to convert my model(graphdef) with the tf2onnx tool, but it takes forever to do the conversion. After debugging, it looks like the code stuck at tf2onnx.tf_utils.compute_const_folding_using_tf, particularly this while-loop

Urgency

If there are particular important use cases blocked by this or strict project-related timelines, please share more information and dates. If there are no hard deadlines, please specify none.

We are trying to experiment/benchmark onnxruntime in our service and might want to switch to onnxruntime if it provides better performance v.s. TF. So prefer to get it solved sooner.

System information

  • MacOS
  • Tensorflow Version: 2.7.0
  • Python version: 3.8.13

To Reproduce
a zip file. run with python convert.py (have to use google drive because 37MB > github max file size is 25MB)

Screenshots

Cannot infer shape for LookupTableExportV2_11: LookupTableExportV2_11:0,LookupTableExportV2_11:1
Cannot infer shape for LookupTableExportV2_12: LookupTableExportV2_12:0,LookupTableExportV2_12:1
Cannot infer shape for LookupTableExportV2_13: LookupTableExportV2_13:0,LookupTableExportV2_13:1
Cannot infer shape for LookupTableExportV2_14: LookupTableExportV2_14:0,LookupTableExportV2_14:1
Cannot infer shape for LookupTableExportV2_15: LookupTableExportV2_15:0,LookupTableExportV2_15:1
Cannot infer shape for LookupTableExportV2_16: LookupTableExportV2_16:0,LookupTableExportV2_16:1
Cannot infer shape for LookupTableExportV2_17: LookupTableExportV2_17:0,LookupTableExportV2_17:1
Cannot infer shape for LookupTableExportV2_18: LookupTableExportV2_18:0,LookupTableExportV2_18:1

after these warnings it takes forever.

Metadata

Metadata

Assignees

No one assigned

    Labels

    potential bugError in codebase may cause a bug, but no concrete examples observed

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions