Searched defs:get_submodule (Results 1 – 4 of 4) sorted by relevance
322 def get_submodule(node: torch.fx.Node) -> torch.fx.GraphModule: function
120 def get_submodule(self, name: str) -> "torch.nn.Module": member in NamedMemberAccessor
721 def get_submodule(self, keys): member in OutputGraph
663 def get_submodule(self, target: str) -> "Module": member in Module