Skip to content

GitLab

"git@developer.sourcefind.cn:OpenDAS/ollama.git" did not exist on "f335722275d8184836d1b777f356fa7b0a012ede"
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