PR #11275 Add ATen export and gradient for torch.max/min

Add ATen export and gradient for torch.max/min #11275

mindest merged 14 commits into master from linmin/atenmax

add aten export for max, max.dim

05b74417

rewrite grad of max (no dim); add cases for min

5c7d1705

update UT cases

d6f9b491

mod sym shape infer

430fc68a

mindest requested a review from

Lafi7e 3 years ago

mindest requested a review from

pengwa 3 years ago

Lafi7e commented on 2022-04-22

resolve comments: shape infer, add comments, etc.

5bea9b19

add test for torch.max of two tensors

fe210fea

mindest marked this pull request as ready for review 3 years ago

minor: identation

259fc325

pengwa commented on 2022-04-26

resolve peng's comments: keepdim; test case

edbd106d

mindest force pushed from 87c535b3 to edbd106d 3 years ago

Merge branch 'master' into linmin/atenmax

da2ec214

correct python format

6f9af641

pengwa dismissed these changes on 2022-04-27

Lafi7e dismissed these changes on 2022-04-27

tmp rm UT for torch.max of two tensors

1272eb7f

mindest dismissed their stale review via 1272eb7f 3 years ago

pengwa commented on 2022-04-28

restore but disable test for max of two tensors

f77803ab

pengwa dismissed these changes on 2022-04-28

Merge branch 'master' into linmin/atenmax

a64accd3

fix recently introduced lint error

8b91ee0b

mindest dismissed their stale review via 8b91ee0b 3 years ago

pengwa approved these changes on 2022-04-28

mindest merged c8270c29 into master 3 years ago

mindest deleted the linmin/atenmax branch 3 years ago

mindest added component:ortmodule

Reviewers

pengwa

Lafi7e

Assignees

No one assigned

Labels

None yet

Milestone

No milestone

onnxruntime Add ATen export and gradient for torch.max/min #11275 Merged

Add ATen export and gradient for torch.max/min #11275

onnxruntime
Add ATen export and gradient for torch.max/min
#11275

Merged