onnxruntime
b14d3f1d - Zhijxu/fix softmax fp16 (#20059)

Commit
1 year ago
Zhijxu/fix softmax fp16 (#20059) in fp16 input, the softmax will return nan in some case, the reason is because in float16 dtype, std::numeric_limits<float16>::infinity() will return 0 instead of inf
Author
Parents
Loading