onnxruntime
C++ front-end Bert-loss function: Reshape inputs for SoftmaxCrossEntropyLoss instead of transposing them to improve performance.
#4551
Merged

C++ front-end Bert-loss function: Reshape inputs for SoftmaxCrossEntropyLoss instead of transposing them to improve performance. #4551

codemzs merged 1 commit into master from bert_loss_function_perf
codemzs
codemzs Reshape inputs for SoftmaxCrossEntropyLoss instead of transposing them.
f0cb889d
codemzs codemzs requested a review from ytaous ytaous 5 years ago
codemzs codemzs requested a review from SherlockNoMad SherlockNoMad 5 years ago
codemzs codemzs requested a review 5 years ago
codemzs codemzs changed the title Reshape inputs for SoftmaxCrossEntropyLoss instead of transposing them to improve performance. C++ Bert-loss function: Reshape inputs for SoftmaxCrossEntropyLoss instead of transposing them to improve performance. 5 years ago
codemzs codemzs changed the title C++ Bert-loss function: Reshape inputs for SoftmaxCrossEntropyLoss instead of transposing them to improve performance. C++ front-end Bert-loss function: Reshape inputs for SoftmaxCrossEntropyLoss instead of transposing them to improve performance. 5 years ago
SherlockNoMad
SherlockNoMad
SherlockNoMad approved these changes on 2020-07-20
codemzs codemzs merged 9d802356 into master 5 years ago
codemzs codemzs deleted the bert_loss_function_perf branch 5 years ago

Login to write a write a comment.

Login via GitHub

Reviewers
Assignees
No one assigned
Labels
Milestone