original_docstring: Single callable to compute the loss.
docstring: Single callable to compute the loss.
docstring_tokens: Single callable to compute the loss .
code: def _loss_op(model, features, labels, params, mode):
    """Single callable to compute the loss."""
    training = mode == tf.estimator.ModeKeys.TRAIN
    logits, _ = model(features, labels, params, mode)
    return model.compute_loss(logits, labels, training=training, params=params)
code_tokens: def _loss_op ( model , features , labels , params , mode ) : training = mode == tf . estimator . ModeKeys . TRAIN logits , _ = model ( features , labels , params , mode ) return model . compute_loss ( logits , labels , training = training , params = params )
short_docstring: Single callable to compute the loss.
short_docstring_tokens: Single callable to compute the loss .
comment: """Single callable to compute the loss."""