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

[wasm] Add ArgMin #7306

Merged
merged 7 commits into from
Mar 14, 2023
Merged

[wasm] Add ArgMin #7306

merged 7 commits into from
Mar 14, 2023

Conversation

chunnienc
Copy link
Collaborator

@chunnienc chunnienc commented Jan 26, 2023

To see the logs from the Cloud Build CI, please join either our discussion or announcement mailing list.


This change is Reviewable

Copy link
Member

@mattsoulanille mattsoulanille left a comment

Choose a reason for hiding this comment

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

LGTM. I have some ideas about how we might simplify some of these kernels that could be applied to future PRs, but this PR looks good to me.

tfjs-backend-wasm/src/kernels/ArgMin.ts Outdated Show resolved Hide resolved
tfjs-backend-wasm/src/cc/kernels/ArgMin.cc Outdated Show resolved Hide resolved
Copy link
Collaborator

@pyu10055 pyu10055 left a comment

Choose a reason for hiding this comment

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

Reviewed 7 of 7 files at r1, all commit messages.
Reviewable status: :shipit: complete! 2 of 1 approvals obtained (waiting on @chunnienc)

@mattsoulanille
Copy link
Member

@chunnienc Feel free to merge this whenever you like.

@chunnienc
Copy link
Collaborator Author

@chunnienc Feel free to merge this whenever you like.

I'm working on the common utils you mentioned. Will append to this PR and request your review again once ready.

Copy link
Member

@mattsoulanille mattsoulanille left a comment

Choose a reason for hiding this comment

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

LGTM

@chunnienc chunnienc merged commit 47b9e0e into tensorflow:master Mar 14, 2023
@chunnienc chunnienc deleted the wasm-argmin branch March 14, 2023 21:50
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.

3 participants