onnxruntime
1ea462fc - put sce and scr grad into fp32

Commit
4 years ago
put sce and scr grad into fp32 put sce as fp32 node instead of loss node sce gradient doesn't have gpu-fp16 implementation, so also insert cast op for it
Author
Parents
Loading