View source on GitHub
|
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.