Module: tfm.nlp.serving_modules

Serving export modules for TF Model Garden NLP models.

Classes

class MaskedLM: The export module for the Bert Pretrain (MaskedLM) task.

class QuestionAnswering: The export module for the question answering task.

class SentencePrediction: The export module for the sentence prediction task.

class Tagging: The export module for the tagging task.

class Translation: The export module for the translation task.

Functions

features_to_int32(...): Converts tf.int64 features to tf.int32, keep other features the same.