def serving_output(self, output):
    hs = (tf.convert_to_tensor(output.hidden_states) if self.config.output_hidden_states else None)
    attns = (tf.convert_to_tensor(output.attentions) if self.config.output_attentions else None)
    return TFMaskedLMOutput(logits=output.logits, hidden_states=hs, attentions=attns)
def serving_output(self, output):
    hs = (tf.convert_to_tensor(output.hidden_states) if self.config.output_hidden_states else None)
    attns = (tf.convert_to_tensor(output.attentions) if self.config.output_attentions else None)
    return TFMaskedLMOutput(logits=output.logits, hidden_states=hs, attentions=attns)