I haven't personally tried to do anything on an AMD GPU, but AMD actually has a very active GitHub repository
https://github.com/ROCmSoftwarePlatform/tensorflow-upstream .
If you go through the issues you can get a sense of what it's like from people who do have practical experience.