class_docstr: Model for predicting autofills given context.
class_docstr_tok: ["Model","for","predicting","autofills","given","context","."]
code: def compute_predictions(self, logits):
      if self.hparams.use_softmax_loss:
          return tf.nn.softmax(logits, dim=2)
      return tf.nn.sigmoid(logits)
  
imports: from __future__ import absolute_import
  from __future__ import division
  from __future__ import print_function
  import collections
  import os
  from magenta.models.coconet import lib_hparams
  from magenta.models.coconet import lib_tfutil
  import tensorflow.compat.v1
label_desc: Use image node_id 13 for calling the CoconetGraph obj's underlying member method code with example usage: obj.compute_predictions(logits) and returns: tf, tf