Searched refs:ModuleWrapper (Results 1 – 2 of 2) sorted by relevance
183 layer = functional.ModuleWrapper(layer)
1404 class ModuleWrapper(base_layer.Layer): class1420 super(ModuleWrapper, self).__init__(**kwargs)