Searched refs:collect_tensor_attrs (Results 1 – 1 of 1) sorted by relevance
764 def collect_tensor_attrs(m: torch.nn.Module, prefix_atoms: List[str]): function769 collect_tensor_attrs(v, prefix_atoms + [k])771 collect_tensor_attrs(self.root, [])