C++ front-end Bert-loss function: Reshape inputs for SoftmaxCrossEntropyLoss instead of transposing them to improve performance. #4551
Reshape inputs for SoftmaxCrossEntropyLoss instead of transposing them.
f0cb889d
codemzs
requested a review
5 years ago
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
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
codemzs
merged
9d802356
into master 5 years ago
codemzs
deleted the bert_loss_function_perf branch 5 years ago
Assignees
No one assigned
Login to write a write a comment.
Login via GitHub