Home
last modified time | relevance | path

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

/external/pytorch/torch/_dynamo/
Doutput_graph.py537 def input_name_to_proxy(self): member in OutputGraph
538 return self.current_tracer.input_name_to_proxy
1692 self.input_name_to_proxy.clear()
1815 self.input_name_to_proxy: Dict[str, fx.Proxy] = {}
2067 self.input_name_to_proxy.pop(node.name, None)
2101 if name in self.input_name_to_proxy:
2104 if candidate_name not in self.input_name_to_proxy:
2108 if self.input_name_to_proxy:
2109 prev_name = next(reversed(self.input_name_to_proxy))
2110 node = self.input_name_to_proxy[prev_name].node
[all …]