Arg<TF_FloatTensor, [{batch_size x num_classes matrix}]>:$features,
    Arg<TF_I32OrI64Tensor, [{batch_size vector with values in [0, num_classes).
This is the label for the given minibatch entry.}]>:$labels
  );

  let results = (outs
    Res<TF_FloatTensor, [{Per example loss (batch_size vector).}]>:$loss,
    Res<TF_FloatTensor, [{backpropagated gradients (batch_size x num_classes matrix).}]>:$backprop
  );