Using Swift on Tensorflow