Home
last modified time | relevance | path

Searched refs:collect_tensor_attrs (Results 1 – 1 of 1) sorted by relevance

/external/pytorch/torch/fx/
D_symbolic_trace.py764 def collect_tensor_attrs(m: torch.nn.Module, prefix_atoms: List[str]): function
769 collect_tensor_attrs(v, prefix_atoms + [k])
771 collect_tensor_attrs(self.root, [])