Paddle
Paddle copied to clipboard
[Sparse] Add a batch_norm kernel
PR types
Others
PR changes
OPs
Describe
因为:当前batch_norm是通过前端调用dense的batch_norm,需要多调用一次values和sparse_coo_tensor,也同时引入了对应的反向操作。
所以:添加batch_norm_kernel和batch_norm_grad_kernel,后端调用dense的kernel避免引入values和sparse_coo_tensor的调用。
你的PR提交成功,感谢你对开源项目的贡献! 请关注后续CI自动化测试结果,详情请参考Paddle-CI手册。 Your PR has been submitted. Thanks for your contribution! Please wait for the result of CI firstly. See Paddle CI Manual for details.