cost = tf.reduce_mean(tf.nn.softmax_cross_entropy_with_logits(logits=pred, labels=y))

auto cost = Mean(root.WithOpName("cost"), SoftmaxCrossEntropyWithLogits(root, output_layer, y));