Paddle icon indicating copy to clipboard operation
Paddle copied to clipboard

[Sparse] Add a batch_norm kernel

Open zkh2016 opened this issue 3 years ago • 1 comments

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的调用。

zkh2016 avatar Sep 21 '22 09:09 zkh2016

你的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.

paddle-bot[bot] avatar Sep 21 '22 09:09 paddle-bot[bot]