Tensorflow Predict Step