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

change argmin/argmax to take scalar axis #724

Merged
merged 1 commit into from
Jul 18, 2024

Conversation

philloooo
Copy link
Contributor

@philloooo philloooo commented Jul 15, 2024

fixes #629

  • Changes argmin/argmax to take scalar axis instead of axes list.
  • Remove default value for axis, so now it becomes positional argument

@fdwr @huningxin please take a look :)


Preview | Diff

Copy link
Collaborator

@fdwr fdwr left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Thanks Phillis for the update. FYI to @BruceDai and @Honry since this impacts WPT's and ORT WebNN EP.

Copy link
Member

@inexorabletash inexorabletash left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

LGTM!

@fdwr fdwr requested a review from huningxin July 16, 2024 22:38
@fdwr
Copy link
Collaborator

fdwr commented Jul 16, 2024

Note forced pushes make it harder to compare iterations in the Files tab 🔍😐, as the "Show changes since your last review" button doesn't work correctly.

Copy link
Contributor

@huningxin huningxin left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

LGTM!

@huningxin huningxin merged commit 92ff390 into webmachinelearning:main Jul 18, 2024
2 checks passed
github-actions bot added a commit that referenced this pull request Jul 18, 2024
SHA: 92ff390
Reason: push, by huningxin

Co-authored-by: github-actions[bot] <41898282+github-actions[bot]@users.noreply.github.com>
chromium-wpt-export-bot pushed a commit to web-platform-tests/wpt that referenced this pull request Jul 22, 2024
This CL changes argMin/argMax to take scalar axis to align with the spec
change [1].

The changes include the definition of IDL and mojo, the implementation
of DirectML, TfLite and CoreML backend and the related unit tests and
WPT tests.

[1] webmachinelearning/webnn#724

Bug: 352359898
Change-Id: Iee68b568331570a6ea51fd17b2195c602d4cff59
Cq-Include-Trybots: luci.chromium.try:win11-blink-rel,mac14.arm64-blink-rel,mac14-blink-rel
chromium-wpt-export-bot pushed a commit to web-platform-tests/wpt that referenced this pull request Jul 23, 2024
This CL changes argMin/argMax to take scalar axis to align with the spec
change [1].

The changes include the definition of IDL and mojo, the implementation
of DirectML, TfLite and CoreML backend and the related unit tests and
WPT tests.

[1] webmachinelearning/webnn#724

Bug: 352359898
Change-Id: Iee68b568331570a6ea51fd17b2195c602d4cff59
Cq-Include-Trybots: luci.chromium.try:win11-blink-rel,mac14.arm64-blink-rel,mac14-blink-rel
aarongable pushed a commit to chromium/chromium that referenced this pull request Jul 23, 2024
This CL changes argMin/argMax to take scalar axis to align with the spec
change [1].

The changes include the definition of IDL and mojo, the implementation
of DirectML, TfLite and CoreML backend and the related unit tests and
WPT tests.

[1] webmachinelearning/webnn#724

Bug: 352359898
Change-Id: Iee68b568331570a6ea51fd17b2195c602d4cff59
Cq-Include-Trybots: luci.chromium.try:win11-blink-rel,mac14.arm64-blink-rel,mac14-blink-rel
Reviewed-on: https://chromium-review.googlesource.com/c/chromium/src/+/5721028
Reviewed-by: Alex Gough <ajgo@chromium.org>
Reviewed-by: Phillis Tang <phillis@chromium.org>
Commit-Queue: Bin Miao <bin.miao@intel.com>
Reviewed-by: ningxin hu <ningxin.hu@intel.com>
Cr-Commit-Position: refs/heads/main@{#1331567}
chromium-wpt-export-bot pushed a commit to web-platform-tests/wpt that referenced this pull request Jul 23, 2024
This CL changes argMin/argMax to take scalar axis to align with the spec
change [1].

The changes include the definition of IDL and mojo, the implementation
of DirectML, TfLite and CoreML backend and the related unit tests and
WPT tests.

[1] webmachinelearning/webnn#724

Bug: 352359898
Change-Id: Iee68b568331570a6ea51fd17b2195c602d4cff59
Cq-Include-Trybots: luci.chromium.try:win11-blink-rel,mac14.arm64-blink-rel,mac14-blink-rel
Reviewed-on: https://chromium-review.googlesource.com/c/chromium/src/+/5721028
Reviewed-by: Alex Gough <ajgo@chromium.org>
Reviewed-by: Phillis Tang <phillis@chromium.org>
Commit-Queue: Bin Miao <bin.miao@intel.com>
Reviewed-by: ningxin hu <ningxin.hu@intel.com>
Cr-Commit-Position: refs/heads/main@{#1331567}
chromium-wpt-export-bot pushed a commit to web-platform-tests/wpt that referenced this pull request Jul 23, 2024
This CL changes argMin/argMax to take scalar axis to align with the spec
change [1].

The changes include the definition of IDL and mojo, the implementation
of DirectML, TfLite and CoreML backend and the related unit tests and
WPT tests.

[1] webmachinelearning/webnn#724

Bug: 352359898
Change-Id: Iee68b568331570a6ea51fd17b2195c602d4cff59
Cq-Include-Trybots: luci.chromium.try:win11-blink-rel,mac14.arm64-blink-rel,mac14-blink-rel
Reviewed-on: https://chromium-review.googlesource.com/c/chromium/src/+/5721028
Reviewed-by: Alex Gough <ajgo@chromium.org>
Reviewed-by: Phillis Tang <phillis@chromium.org>
Commit-Queue: Bin Miao <bin.miao@intel.com>
Reviewed-by: ningxin hu <ningxin.hu@intel.com>
Cr-Commit-Position: refs/heads/main@{#1331567}
moz-v2v-gh pushed a commit to mozilla/gecko-dev that referenced this pull request Jul 25, 2024
…ar axis, a=testonly

Automatic update from web-platform-tests
WebNN: Change argMin/argMax to take scalar axis

This CL changes argMin/argMax to take scalar axis to align with the spec
change [1].

The changes include the definition of IDL and mojo, the implementation
of DirectML, TfLite and CoreML backend and the related unit tests and
WPT tests.

[1] webmachinelearning/webnn#724

Bug: 352359898
Change-Id: Iee68b568331570a6ea51fd17b2195c602d4cff59
Cq-Include-Trybots: luci.chromium.try​:win11-blink-rel,mac14.arm64-blink-rel,mac14-blink-rel
Reviewed-on: https://chromium-review.googlesource.com/c/chromium/src/+/5721028
Reviewed-by: Alex Gough <ajgo@chromium.org>
Reviewed-by: Phillis Tang <phillis@chromium.org>
Commit-Queue: Bin Miao <bin.miao@intel.com>
Reviewed-by: ningxin hu <ningxin.hu@intel.com>
Cr-Commit-Position: refs/heads/main@{#1331567}

--

wpt-commits: acd785862cc0cd8a52e83ac6db95403e0e6e5450
wpt-pr: 47235
jamienicol pushed a commit to jamienicol/gecko that referenced this pull request Jul 26, 2024
…ar axis, a=testonly

Automatic update from web-platform-tests
WebNN: Change argMin/argMax to take scalar axis

This CL changes argMin/argMax to take scalar axis to align with the spec
change [1].

The changes include the definition of IDL and mojo, the implementation
of DirectML, TfLite and CoreML backend and the related unit tests and
WPT tests.

[1] webmachinelearning/webnn#724

Bug: 352359898
Change-Id: Iee68b568331570a6ea51fd17b2195c602d4cff59
Cq-Include-Trybots: luci.chromium.try​:win11-blink-rel,mac14.arm64-blink-rel,mac14-blink-rel
Reviewed-on: https://chromium-review.googlesource.com/c/chromium/src/+/5721028
Reviewed-by: Alex Gough <ajgo@chromium.org>
Reviewed-by: Phillis Tang <phillis@chromium.org>
Commit-Queue: Bin Miao <bin.miao@intel.com>
Reviewed-by: ningxin hu <ningxin.hu@intel.com>
Cr-Commit-Position: refs/heads/main@{#1331567}

--

wpt-commits: acd785862cc0cd8a52e83ac6db95403e0e6e5450
wpt-pr: 47235
ChunMinChang pushed a commit to ChunMinChang/gecko-dev that referenced this pull request Jul 26, 2024
…ar axis, a=testonly

Automatic update from web-platform-tests
WebNN: Change argMin/argMax to take scalar axis

This CL changes argMin/argMax to take scalar axis to align with the spec
change [1].

The changes include the definition of IDL and mojo, the implementation
of DirectML, TfLite and CoreML backend and the related unit tests and
WPT tests.

[1] webmachinelearning/webnn#724

Bug: 352359898
Change-Id: Iee68b568331570a6ea51fd17b2195c602d4cff59
Cq-Include-Trybots: luci.chromium.try​:win11-blink-rel,mac14.arm64-blink-rel,mac14-blink-rel
Reviewed-on: https://chromium-review.googlesource.com/c/chromium/src/+/5721028
Reviewed-by: Alex Gough <ajgo@chromium.org>
Reviewed-by: Phillis Tang <phillis@chromium.org>
Commit-Queue: Bin Miao <bin.miao@intel.com>
Reviewed-by: ningxin hu <ningxin.hu@intel.com>
Cr-Commit-Position: refs/heads/main@{#1331567}

--

wpt-commits: acd785862cc0cd8a52e83ac6db95403e0e6e5450
wpt-pr: 47235
i3roly pushed a commit to i3roly/firefox-dynasty that referenced this pull request Jul 26, 2024
…ar axis, a=testonly

Automatic update from web-platform-tests
WebNN: Change argMin/argMax to take scalar axis

This CL changes argMin/argMax to take scalar axis to align with the spec
change [1].

The changes include the definition of IDL and mojo, the implementation
of DirectML, TfLite and CoreML backend and the related unit tests and
WPT tests.

[1] webmachinelearning/webnn#724

Bug: 352359898
Change-Id: Iee68b568331570a6ea51fd17b2195c602d4cff59
Cq-Include-Trybots: luci.chromium.try​:win11-blink-rel,mac14.arm64-blink-rel,mac14-blink-rel
Reviewed-on: https://chromium-review.googlesource.com/c/chromium/src/+/5721028
Reviewed-by: Alex Gough <ajgo@chromium.org>
Reviewed-by: Phillis Tang <phillis@chromium.org>
Commit-Queue: Bin Miao <bin.miao@intel.com>
Reviewed-by: ningxin hu <ningxin.hu@intel.com>
Cr-Commit-Position: refs/heads/main@{#1331567}

--

wpt-commits: acd785862cc0cd8a52e83ac6db95403e0e6e5450
wpt-pr: 47235
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.

argMax/Min only support scalar axis in TFLite runtime
4 participants