onnxruntime
88c37042
- Add shape inference for additional ops
Go
Login via GitHub
Home
Pricing
FAQ
Install
Login
via GitHub
Commit
View On
GitHub
Commit
5 years ago
Add shape inference for additional ops This commit adds shape inference support for the following ops: SoftmaxCrossEntropy SoftmaxCrossEntropyLossGrad SoftmaxCrossEntropyGrad LayerNormalizationGrad Motivation and Context
References
#5660 - Add shape inference support for a few training ops
Author
alberto-magni
Parents
4e29f480
Loading