前往新版Gitcode,体验更适合开发者的 AI 搜索 >>
Created by: suytingwan
Fix fp16 check_grad for ops: softmax, pool2d, pool3d