Skip to content

[webgpu] Add support for SparseToDense#5629

Merged
qjia7 merged 7 commits intotensorflow:masterfrom
haoyunfeix:webgpu_feature_s2d
Sep 24, 2021
Merged

[webgpu] Add support for SparseToDense#5629
qjia7 merged 7 commits intotensorflow:masterfrom
haoyunfeix:webgpu_feature_s2d

Conversation

@haoyunfeix
Copy link
Contributor

@haoyunfeix haoyunfeix commented Sep 16, 2021

FEATURE

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


This change is Reviewable

@google-cla google-cla bot added the cla: yes label Sep 16, 2021
@haoyunfeix haoyunfeix force-pushed the webgpu_feature_s2d branch 2 times, most recently from 64ebc11 to a2ed810 Compare September 20, 2021 11:59
@rthadur rthadur requested a review from pyu10055 September 20, 2021 16:46
@haoyunfeix
Copy link
Contributor Author

@qjia7 @axinging @xhcao @gyagp, PTAL

@rthadur rthadur requested a review from qjia7 September 20, 2021 18:12
size: number;
useWgsl: boolean;
sliceDim: number;
shape: number[];
Copy link
Contributor

Choose a reason for hiding this comment

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

nit: sliceDim/shape are never used members?

Copy link
Contributor Author

Choose a reason for hiding this comment

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

Done.

Copy link
Contributor

@qjia7 qjia7 left a comment

Choose a reason for hiding this comment

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

LGTM. Please resolve the conflicts.

uniforms: string;
uniformsWgsl: string;
outputShape: number[];
userCode: string;
Copy link
Contributor

Choose a reason for hiding this comment

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

nit: userCode is not needed.

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 3 of 4 files at r1, 1 of 1 files at r2, all commit messages.
Reviewable status: :shipit: complete! 2 of 1 approvals obtained (waiting on @haoyunfeix and @qjia7)

@qjia7 qjia7 merged commit 2f59862 into tensorflow:master Sep 24, 2021
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

Projects

None yet

Development

Successfully merging this pull request may close these issues.

3 participants