Custom Loss Function Tensorflow