-
Notifications
You must be signed in to change notification settings - Fork 6.8k
[v1.x] ONNX Supoort for MXNet repeat op #19732
Conversation
Hey @Zha0q1 , Thanks for submitting the PR
CI supported jobs: [windows-cpu, sanity, miscellaneous, centos-gpu, website, centos-cpu, unix-cpu, unix-gpu, clang, edge, windows-gpu] Note: |
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!
|
||
|
||
@pytest.mark.parametrize('dtype', ['float16', 'float32', 'float64', 'int32', 'int64']) | ||
@pytest.mark.parametrize('axis', [None, 0, 1, 2]) |
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.
Shall we also test negative axis like -1 or -2?
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.
ohh good point!
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
No description provided.