Searched refs:_check_is_graph_module (Results 1 – 2 of 2) sorted by relevance
35 def _check_is_graph_module(model: torch.nn.Module) -> None: function84 _check_is_graph_module(model)532 _check_is_graph_module(graph_module)
12 _check_is_graph_module,