Automatic Mixed Precision Pytorch Vs Tensorflow