Skip to content

Extension of channels last converter to models with branches #114

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.

Sign up for GitHub

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

Open
wants to merge 12 commits into
base: main
Choose a base branch
from
Open

Extension of channels last converter to models with branches #114

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
12 commits
Select commit Hold shift + click to select a range
c70923b
Some initial modifications to support channels last with branches. Th…
nghielme Apr 17, 2024
fd49030
Cleaned. Should be in good shape for PR
nghielme Apr 19, 2024
b7b6117
Some minor cleaning
nghielme Apr 19, 2024
ec78537
Some other minor aesthetic fixes
nghielme Apr 19, 2024
0125648
added a missing part that caused some tests to fail
nghielme Apr 23, 2024
4b05964
Removed an unused import
nghielme Apr 23, 2024
9d0b4bf
Some fixes, now resize initializer is transposed as well
nghielme May 14, 2024
acdc59b
`Resize` and `Concat` are now handled as special cases. It probably b…
nghielme May 23, 2024
2de8240
The nodes are now handled as special nodes. I also considered the cas…
nghielme May 29, 2024
5463de2
One minor fix on `Resize`
nghielme May 30, 2024
8ab1f28
Added an optional parameter perform
nghielme Jun 27, 2024
177f528
Merge branch 'main' into channels_last_branches
nghielme Dec 20, 2024
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
5 changes: 5 additions & 0 deletions src/qonnx/custom_op/channels_last/base_wrapped_op.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -58,6 +58,11 @@ def to_channels_first_args(ndim):
return tuple(arg_list)


def swap_channels_from_list(l):
l[1], l[-1] = l[-1], l[1]
return l


class ChannelsLastWrappedOp(CustomOp):
# ToDo: _channelsLast_node_types should be loaded / inferred from this file or the registry.
# Standard ONNX nodes which require a ChannelsLast data format to function properly
Expand Down
Loading
Loading