PR #9014 Add BatchNorm kernel for ROCm

Add BatchNorm kernel for ROCm #9014

mindest merged 2 commits into master from linmin/bn_rocm

Add BatchNorm kernel for ROCm, update BN test

50182151

mindest marked this pull request as ready for review 4 years ago

mindest requested a review from

weixingzhang 4 years ago

mindest added training

correct epsilon_ setting; limit min epsilon

7d16a9eb

weixingzhang approved these changes on 2021-09-13

mindest merged a1021a1c into master 4 years ago

mindest deleted the linmin/bn_rocm branch 4 years ago

Reviewers

weixingzhang

Assignees

No one assigned

Labels

training

Milestone

No milestone