https://docs.nvidia.com/deeplearning/frameworks/tensorflow-user-guide/index.html#tfamp

See section 6 and 7.