Skip to content

Add support for operations needed for well-known transformers e.g. Segment Anything, Stable Diffusion, etc. #478

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.

Sign up for GitHub

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

Merged
wchao1115 merged 19 commits into from
Dec 11, 2023
Merged

Add support for operations needed for well-known transformers e.g. Segment Anything, Stable Diffusion, etc. #478

Changes from all commits
Commits
Show all changes
19 commits
Select commit Hold shift + click to select a range
e284f2a
almost done.
wchao1115 Nov 10, 2023
00c355c
Done.
wchao1115 Nov 11, 2023
cfcbca0
tri -> triangular
wchao1115 Nov 11, 2023
9ab86c2
Fix Bikeshed link errors.
wchao1115 Nov 11, 2023
a0718ad
Minor formatting change.
wchao1115 Nov 11, 2023
8d026d4
Minor code fix. No unsqueeze needed here since the affine params must…
wchao1115 Nov 11, 2023
873eec1
Addressing much of the PR feedback. Still more to go.
wchao1115 Nov 28, 2023
2aa5420
Addressing more PR feedback.
wchao1115 Nov 30, 2023
3bed8a4
More fixes.
wchao1115 Nov 30, 2023
52a7a1e
Fix dropping the mixin.
wchao1115 Nov 30, 2023
c91fda3
Separate out argMin/Max. More 'type' -> 'dataType`.
wchao1115 Dec 5, 2023
06bdb0a
add 'be'
wchao1115 Dec 5, 2023
b491f79
Add dfn for MLOperand.shape() and .dataType(). Fix argMin/Max return …
wchao1115 Dec 6, 2023
2d976cb
Fix `where` emulation code.
wchao1115 Dec 6, 2023
a8a33fd
Fix some wording to clarify the return shape of `gather`.
wchao1115 Dec 6, 2023
9d1ac61
Fix layernorm input shape typo.
wchao1115 Dec 6, 2023
8ce8ed1
Acknowledge Dwayne's contributions.
wchao1115 Dec 6, 2023
38f451d
Wording adjustment.
wchao1115 Dec 6, 2023
e8c9864
}
wchao1115 Dec 7, 2023
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Loading