Skip to content

GitLab

"vscode:/vscode.git/clone" did not exist on "d851bc3173550197c588f5887d89677632faafdb"
Switch branch/tag
Find file BlameHistoryPermalink
  • Qianfeng's avatar
    BatchNorm backward instance/external API/profiler/tests (#519) · 63af525c
    Qianfeng authored 
    * Refine the device batchnorm-backward base API templates and data type assignments

* Remove duplicated kernel file

* Add batchnorm backward instances and external API

* Add batchnorm-backward profiler and tests

* Add client example which uses batchnorm backward external API

* Merge test/batchnorm_fwd and test/batchnorm_bwd into one directory

* Loose the threshold for batchnorm-backward check_err()
    63af525c
profile_batchnorm_bwd.cpp 6.89 KB