@tf.function(input_signature=[{'input_ids': tf.TensorSpec((None, None, None), tf.int32, name='input_ids'), 'attention_mask': tf.TensorSpec((None, None, None), tf.int32, name='attention_mask'), 'token_type_ids': tf.TensorSpec((None, None, None), tf.int32, name='token_type_ids')}])
def serving(self, inputs):
    output = self.call(inputs)
    return self.serving_output(output)
@tf.function(input_signature=[{'input_ids': tf.TensorSpec((None, None, None), tf.int32, name='input_ids'), 'attention_mask': tf.TensorSpec((None, None, None), tf.int32, name='attention_mask'), 'token_type_ids': tf.TensorSpec((None, None, None), tf.int32, name='token_type_ids')}])
def serving(self, inputs):
    output = self.call(inputs)
    return self.serving_output(output)