Searched refs:set_torch_function_state (Results 1 – 2 of 2) sorted by relevance
615 self.set_torch_function_state,1699 def set_torch_function_state(self, enabled: bool) -> None: member in OutputGraph
640 tx.output.set_torch_function_state(values[0])