microsoft / nnfusion

A flexible and efficient deep neural network (DNN) compiler that generates high-performance executable from a DNN model description.
MIT License
958 stars 162 forks source link

Add ArgMax Operator and HLSL kernel. #432

Closed wenxcs closed 2 years ago