Reenabling unit tests for ROCm
Hi @jurahul / @frgossen Can you please review this PR ? Thank you! Hi @ekuznetsov139 Can you please check @frgossen's [comments](https://github.com/tensorflow/tensorflow/pull/60559#pullrequestreview-1607003961)  and keep us posted? Thank you! This PR is stale because it has been open for 14 days with no activity. It will be closed if no further activity occurs. Thank you.
An upstream of https://github.com/ROCmSoftwarePlatform/tensorflow-upstream/pull/2061

This reenables a number of unit tests or subtests which were disabled for ROCm but were passing, or which could be fixed with minimal effort.
