Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Range analysis improvements and better input shape override #97

Merged
merged 12 commits into from
Feb 5, 2024

Conversation

maltanar
Copy link
Collaborator

@maltanar maltanar commented Feb 5, 2024

  • Support extracting bias from ConvTranspose nodes using ExtractBiasFromConv
  • Allow specifying general input shape as well as bias for input shape override in cleanup_model (please note that override_batchsize is now override_inpsize)
  • Range analysis improvements: support float16 tensors, use calc_monotonic_range for more op types, support Split, support multi-output nodes, support per-channel input range specification

@maltanar maltanar merged commit 3fd9386 into main Feb 5, 2024
5 checks passed
@maltanar maltanar deleted the feature/tensor_stats branch February 5, 2024 14:59
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

1 participant