tf.compat.v1.losses.sparse_softmax_cross_entropy