Searched refs:traverse_broadcast_node (Results 1 – 1 of 1) sorted by relevance
27 def traverse_broadcast_node(self, graph_module: torch.fx.GraphModule): member in ExpandBroadcastTensorShape55 self.traverse_broadcast_node(graph_module)