Searched defs:get_node_target (Results 1 – 2 of 2) sorted by relevance
106 def get_node_target(module: torch.nn.Module | GraphModule, target_str: str): function
34 def get_node_target(submodules: Mapping[str, torch.nn.Module], node: torch.fx.Node) -> str: function