Searched refs:collected_node_set (Results 1 – 1 of 1) sorted by relevance
| /external/pytorch/torch/_inductor/ |
| D | utils.py | 1679 snode, collected_node_set, name_to_buf, name_to_fused_node, criteria_cb=None argument
1683 collected_node_set.add(snode)
1688 if defining_op_for_dep in collected_node_set:
1692 collected_node_set,
1700 snode, collected_node_set, name_to_buf, name_to_fused_node, criteria_cb=None argument
1704 collected_node_set.add(snode)
1713 if user_op in collected_node_set:
1717 collected_node_set,
|