Tensorflow Mnist Training and Predictions