Thanks for tuning in to Google I/O. View all sessions on demandWatch on demand


The export module for the tagging task.

Inherits From: ExportModule

params A dataclass for parameters to the module.
model A model instance which contains weights and forward computation.
inference_step An optional callable to forward-pass the model. If not specified, it creates a parital function with model as an required kwarg.
preprocessor An optional callable to preprocess the inputs.
postprocessor An optional callable to postprocess the model outputs.

Child Classes

class Params



View source

Get defined function signatures.


View source


View source