Skip to content

feat(atenlib): add ops (ArgMax, ArgMin, Detach) #324

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

Merged
merged 14 commits into from
Jan 18, 2023
Merged

feat(atenlib): add ops (ArgMax, ArgMin, Detach) #324

merged 14 commits into from
Jan 18, 2023

Conversation

xiaowuhu
Copy link
Contributor

@xiaowuhu xiaowuhu commented Jan 17, 2023

add ops: argmax, argmin, but use trace_only = True mode, due to OptionalHasElement(dim) cannot work now.

@xiaowuhu xiaowuhu marked this pull request as draft January 17, 2023 08:21
@xiaowuhu xiaowuhu changed the title Xiaowu/add ops(Argmax) Xiaowu/add ops(ArgMax, ArgMin) Jan 17, 2023
@xiaowuhu xiaowuhu marked this pull request as ready for review January 17, 2023 08:32
@xiaowuhu xiaowuhu mentioned this pull request Jan 17, 2023
@xiaowuhu xiaowuhu changed the title Xiaowu/add ops(ArgMax, ArgMin) Xiaowu/add ops(ArgMax, ArgMin, Detach) Jan 17, 2023
@codecov
Copy link

codecov bot commented Jan 17, 2023

Codecov Report

Merging #324 (99af280) into main (be4ea50) will increase coverage by 0.07%.
The diff coverage is 95.83%.

@@            Coverage Diff             @@
##             main     #324      +/-   ##
==========================================
+ Coverage   73.17%   73.24%   +0.07%     
==========================================
  Files          96       96              
  Lines        9476     9495      +19     
==========================================
+ Hits         6934     6955      +21     
+ Misses       2542     2540       -2     
Impacted Files Coverage Δ
...t/function_libs/torch_aten/ops_correctness_test.py 95.76% <ø> (ø)
onnxscript/function_libs/torch_aten/ops/core.py 63.76% <95.83%> (+0.60%) ⬆️

Help us with your feedback. Take ten seconds to tell us how you rate us. Have a feature suggestion? Share it here.

@justinchuby justinchuby changed the title Xiaowu/add ops(ArgMax, ArgMin, Detach) feat(atenlib): add ops (ArgMax, ArgMin, Detach) Jan 17, 2023
@xiaowuhu xiaowuhu merged commit 000d657 into microsoft:main Jan 18, 2023
@xiaowuhu xiaowuhu deleted the xiaowu/addOps(6) branch January 18, 2023 01:44
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.

2 participants