onnxruntime
Use PadAndUnflatten to replace GatherGrad for restore
#16429
Merged
Overview
Commits
16
Changes
View On GitHub

Use PadAndUnflatten to replace GatherGrad for restore #16429

pengwa merged 16 commits into main from pengwa/unflatten_with_index
pengwa
pengwa use PadByAxis to replace GatherGrad restore
ad17c25a
pengwa pengwa added training
pengwa pengwa requested a review from guyang3532 guyang3532 2 years ago
pengwa fix tests
67d972ea
pengwa add unit test
8bda8b41
pengwa add cuda/rcom marco check
4674f449
pengwa Merge branch 'main' of https://github.com/microsoft/onnxruntime into …
8d113e09
guyang3532
guyang3532 commented on 2023-06-25
guyang3532
guyang3532 commented on 2023-06-25
pengwa rename and refine
f48e8e82
pengwa remove axis attribute
4c9208b9
pengwa refine a bit
053eda09
github-advanced-security
github-advanced-security commented on 2023-06-26
pengwa Merge branch 'main' of https://github.com/microsoft/onnxruntime into …
625c4748
pengwa fix lintrunner
39dd13ff
pengwa refine op schema doc
6e7e9671
pengwa pengwa changed the title Use PadByAxis to replace GatherGrad for restore Use PadAndUnflatten to replace GatherGrad for restore 2 years ago
pengwa minor
c01fd038
pengwa format
93722be5
pengwa fix amd failure
8970639c
guyang3532
guyang3532 commented on 2023-06-26
pengwa disable test for ROCM temporary, renaming
c9def2be
pengwa Merge branch 'main' of https://github.com/microsoft/onnxruntime into …
1aeb34e6
guyang3532
guyang3532 approved these changes on 2023-06-27
pengwa pengwa merged 403bebfb into main 2 years ago
pengwa pengwa deleted the pengwa/unflatten_with_index branch 2 years ago

Login to write a write a comment.

Login via GitHub

Reviewers
Assignees
No one assigned
Labels
training
Milestone
No milestone
Terms Of Use Privacy Policy Refund Policy Impressum