-
Notifications
You must be signed in to change notification settings - Fork 6.8k
[v1.x] ONNX export support for SwapAxis #19789
Conversation
Hey @Zha0q1 , Thanks for submitting the PR
CI supported jobs: [clang, sanity, windows-cpu, windows-gpu, website, miscellaneous, unix-cpu, unix-gpu, centos-gpu, edge, centos-cpu] Note: |
|
||
|
||
@mx_op.register('SwapAxis') | ||
def convert_reshape_like(node, **kwargs): |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
need to rename this to convert_swapaxis
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
LGTM, thanks!
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
LGTM, thanks
Currently we do not support negative axis.
Also the implementation is sub-optimal due to onnx limitations