onnxruntime
9d802356 - Reshape inputs for SoftmaxCrossEntropyLoss instead of transposing them. (#4551)

Commit

5 years ago

Reshape inputs for SoftmaxCrossEntropyLoss instead of transposing them. (#4551)

References

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

Author

codemzs

codemzs

Parents

Loading