onnxruntime
Optimize SCE loss compute
#15401
Merged
Overview
Commits
14
Changes
View On GitHub

Optimize SCE loss compute #15401

pengwa merged 14 commits into main from pengwa/sce_opt
pengwa
pengwa introduce shrunken gather operator
b5ae1325
pengwa minor fixes
8cd9fd38
pengwa fixes
6aecb3d1
pengwa sce loss compute optimization
7dc53565
pengwa pengwa requested a review from askhade askhade 2 years ago
pengwa pengwa requested a review from baijumeswani baijumeswani 2 years ago
pengwa pengwa requested a review from guyang3532 guyang3532 2 years ago
pengwa pengwa added training
pengwa increase round of optimization to make optimization fully applied
75ad8ab6
pengwa minors
ef03fed5
Base automatically changed from pengwa/shrunken_gather to main 2 years ago
pengwa Merge branch 'main' of https://github.com/microsoft/onnxruntime into …
5ac1bcd9
pengwa pengwa changed the title sce loss compute optimization Optimize SCE loss compute 2 years ago
pengwa add more tests & upstream shrunken gather & add flags
b2ba9366
pengwa pengwa marked this pull request as ready for review 2 years ago
pengwa pylint
005df100
pengwa refinement
86f5c5e6
baijumeswani
baijumeswani commented on 2023-04-10
pengwa refine based on comments
2d5e7bde
pengwa add support for SoftmaxCrossEntropyLoss
4f63794c
pengwa Merge branch 'main' of https://github.com/microsoft/onnxruntime into …
a9aca4e5
pengwa fix error
b7537111
baijumeswani
baijumeswani approved these changes on 2023-04-13
baijumeswani
baijumeswani commented on 2023-04-13
pengwa
pengwa pengwa merged 516c8e95 into main 2 years ago
pengwa pengwa deleted the pengwa/sce_opt 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