TensorRT
TensorRT copied to clipboard
✨[Feature] Add reduction targets in Autocast
Need to add reduction targets in Autocast DepthOfReductionRule
@zewenli98 do you have specific ones we need to add?
Will go over the current reduction ops and see what should be added. btw the definition of reduction op: https://en.wikipedia.org/wiki/Reduction_operator