Skip to content

Commit

Permalink
Optimization of pool2d grad (#35389)
Browse files Browse the repository at this point in the history
* Optimization of pool2d grad, first commit.

* remove useless print codes

* refine codes

* refine codes

* seal more operation into template specialization

* fix template struct error in MaxPool2dGrad.

* Fix header including error

* refine code with comment

* Seal the param-preparation codes into function for common use.

* Seal the param-preparation codes into function for common use.

* Seal the param-preparation into funciton and make it common for other kernels

* polish code and erase useless template speicalization

* Rerun triger

* rerun trigger
  • Loading branch information
JamesLim-sy authored Sep 19, 2021
1 parent 9f88d32 commit 8668519
Show file tree
Hide file tree
Showing 2 changed files with 339 additions and 238 deletions.
Loading

0 comments on commit 8668519

Please sign in to comment.