onnxruntime
SCELoss(SCELossGrad) support half(float) input float(half) output
#13972
Merged
Overview
Commits
17
Changes
View On GitHub

SCELoss(SCELossGrad) support half(float) input float(half) output #13972

pengwa merged 17 commits into main from pengwa/sce_fix
pengwa
pengwa softmax cross entropy loss internal support half input float output
973b1d8d
pengwa Merge branch 'main' of https://github.com/microsoft/onnxruntime into …
fe50ab56
pengwa pengwa requested a review from askhade askhade 3 years ago
pengwa pengwa requested a review from baijumeswani baijumeswani 3 years ago
pengwa pengwa requested a review from Lafi7e Lafi7e 3 years ago
pengwa pengwa added training
pengwa fix
a1d63e6b
pengwa Merge branch 'main' of https://github.com/microsoft/onnxruntime into …
e6502722
pengwa Add test for variant input and output for sceloss and scelossinternal…
348cdf2c
askhade
askhade commented on 2023-02-06
pengwa refine the function body type check
4aa20cf2
pengwa Merge branch 'main' of https://github.com/microsoft/onnxruntime into …
3702807b
pengwa remove prefix workaround
3167d248
pengwa Merge branch 'main' of https://github.com/microsoft/onnxruntime into …
e3574b66
pengwa Merge branch 'main' of https://github.com/microsoft/onnxruntime into …
783e7ee9
pengwa fix test runs
fd9c8fdd
pengwa Merge branch 'main' of https://github.com/microsoft/onnxruntime into …
13d4c83b
pengwa fix win build
ce8e99ad
pengwa minor about rocm
04a6c1a0
pengwa Merge branch 'main' of https://github.com/microsoft/onnxruntime into …
900971f2
pengwa kernel registration for rocm
995033b5
askhade
askhade approved these changes on 2023-02-28
askhade
pengwa
pengwa Merge branch 'main' of https://github.com/microsoft/onnxruntime into …
2748dde6
pengwa
pengwa pengwa merged 79aa0acd into main 3 years ago
pengwa pengwa deleted the pengwa/sce_fix branch 3 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