onnxruntime
e6a3308d - Optimize cuComputeGradInput performance. (#7479)

Commit

5 years ago

Optimize cuComputeGradInput performance. (#7479) Move the checking of gamma to host and specialize both case through template.

References

Author

sabreshao

Parents