Searched defs:is_graph_output (Results 1 – 2 of 2) sorted by relevance
78 def is_graph_output(tensor: torch.fx.Node) -> bool: function
114 def is_graph_output(self, tensor: torch.fx.Node) -> bool: member in NodeVisitor