-
Notifications
You must be signed in to change notification settings - Fork 2k
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
[converter] add TensorScatterUpdate op to converter and executor #7246
Conversation
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!
I think after this PR, TFJS could run models with TensorScatterUpdate op. I am curious, does this PR also enables model converter to convert models with TensorScatterUpdate op?
Reviewable status:
complete! 1 of 1 approvals obtained (waiting on @mattsoulanille)
tfjs-converter/yarn.lock
Outdated
version "0.1.21" | ||
resolved "https://registry.yarnpkg.com/@webgpu/types/-/types-0.1.21.tgz#b181202daec30d66ccd67264de23814cfd176d3a" | ||
integrity sha512-pUrWq3V5PiSGFLeLxoGqReTZmiiXwY3jRkIG5sLLKjyqNxrwm/04b4nw7LSmGWJcKk59XOM/YRTUwOzo4MMlow== | ||
"@webgpu/[email protected]": |
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.
Why it's 0.1.16
not 0.1.21
? The places we used are all 0.1.21
. See https://github.com/tensorflow/tfjs/blob/master/package.json#L29 and https://github.com/tensorflow/tfjs/blob/master/tfjs-core/package.json#L44
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.
updated, thanks
yes, the op_list json file ensure converter the op is supported. |
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
…sorflow#7246) FEATURE * add TensorScatterUpdate op to converter and executor * update the yarn.lock file Co-authored-by: Linchenn <[email protected]>
To see the logs from the Cloud Build CI, please join either our discussion or announcement mailing list.
This change is
fixed #6709
This the converter side for the op.