Optimize GatherGrad for AMD GPU (#6381)

Commit

4 years ago

Optimize GatherGrad for AMD GPU (#6381) * optimize gathergrad * address comments Co-authored-by: Weixing Zhang <wezhan@microsoft.com>