Skip to content

GitLab

Switch branch/tag
Find file BlameHistoryPermalink
  • Qianfeng's avatar
    Add pooling example (#257) · 97c4d486
    Qianfeng authored 
    * Add example for computing LayerNorm mean and meansquare

* Refactor the pool2d_fwd example and add example for float type testing

* Revert "Add example for computing LayerNorm mean and meansquare"

This reverts commit df52e6f9d897b00c981baa48f291450bcd60925d.

* Tiny fix in pool2d_fwd_common.hpp
    97c4d486
pool2d_fwd_fp16.cpp 3.64 KB