tfl.premade.get_custom_objects

View source on GitHub

Creates and returns a dictionary mapping names to custom objects.

custom_objects Optional dictionary mapping names (strings) to custom classes or functions to be considered during deserialization. If provided, the returned mapping will be extended to contain this one.

A dictionary mapping names (strings) to tensorflow lattice custom objects.