Tensorflow Cross Entropy Loss