Searched defs:graph_module_cls (Results 1 – 2 of 2) sorted by relevance
57 def _make_graph_module(*args, graph_module_cls=None, **kwargs): argument
160 def _deserialize_graph_module(forward, body: Dict[Any, Any], graph_module_cls=None) -> torch.nn.Mod… argument