Enabled gpu support for several ops: including ReLu for bfloat16#2801

i-chaochen merged 2 commits intor2.18-rocm-enhancedROCm/tensorflow-upstream:r2.18-rocm-enhancedfrom

tf2.18-missing-gpu-supportROCm/tensorflow-upstream:tf2.18-missing-gpu-supportCopy head branch name to clipboard