onnxruntime
2c240881 - Add function body for SoftmaxCrossEntropyLossGrad (#10779)

Commit
3 years ago
Add function body for SoftmaxCrossEntropyLossGrad (#10779) * Add function definition for SoftmaxCrossEntropyLossGrad Signed-off-by: Ganesan Ramalingam <grama@microsoft.com> * Cleanup Signed-off-by: Ganesan Ramalingam <grama@microsoft.com> * Eliminate unused variable Signed-off-by: Ganesan Ramalingam <grama@microsoft.com> * Fix index of weight tensor Signed-off-by: Ganesan Ramalingam <grama@microsoft.com> * A few fixes to handle typing and weight Signed-off-by: Ganesan Ramalingam <grama@microsoft.com> * Fix for zero D dimensions Signed-off-by: Ganesan Ramalingam <grama@microsoft.com> * Add function body to internal op also Signed-off-by: Ganesan Ramalingam <grama@microsoft.com> * A few fixes Signed-off-by: Ganesan Ramalingam <grama@microsoft.com> * Fix type variable name Signed-off-by: Ganesan Ramalingam <grama@microsoft.com> * Fix type constraint var Signed-off-by: Ganesan Ramalingam <grama@microsoft.com> * Fix ignore_index handling in testcase Signed-off-by: Ganesan Ramalingam <grama@microsoft.com@orttrainingdev7.d32nl1ml4oruzj4qz3bqlggovf.px.internal.cloudapp.net> * Add fun def for SoftmaxCrossEntropyLossInternal Signed-off-by: Ganesan Ramalingam <grama@microsoft.com@orttrainingdev7.d32nl1ml4oruzj4qz3bqlggovf.px.internal.cloudapp.net> * Specify opset Signed-off-by: Ganesan Ramalingam <grama@microsoft.com@orttrainingdev7.d32nl1ml4oruzj4qz3bqlggovf.px.internal.cloudapp.net> * Handle opset in NLL function Signed-off-by: Ganesan Ramalingam <grama@microsoft.com> * Address PR feedback Signed-off-by: Ganesan Ramalingam <grama@microsoft.com> * Modify onehot Signed-off-by: Ganesan Ramalingam <grama@microsoft.com> * Eliminate duplicate statement Co-authored-by: Ganesan Ramalingam <grama@microsoft.com@orttrainingdev7.d32nl1ml4oruzj4qz3bqlggovf.px.internal.cloudapp.net>
Author
Parents
Loading