Searched refs:extract_input (Results 1 – 3 of 3) sorted by relevance
142 def extract_input(node: torch.fx.Node) -> Optional[FakeTensor]: function157 extract_input(node)
286 def extract_input(node: torch.fx.Node) -> Optional[FakeTensor]: function312 extract_input(node)
423 def extract_input(node: torch.fx.Node) -> Optional[FakeTensor]: function449 extract_input(node)