Amd gpu support

[getnamo]

Likely depends on https://github.com/ROCmSoftwarePlatform/tensorflow-upstream which is available on pip as https://pypi.org/project/tensorflow-rocm/. Atm there are no windows pip wheels for this, but that may change in the future. At that point changing upymodule.json to tensorflow-rocm should suffice.