Searched defs:name_to_fused_node (Results 1 – 3 of 3) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/_inductor/ |
| H A D | utils.py | 1674 def buf_name_to_fused_snode(buf_name, name_to_buf, name_to_fused_node): argument
1679 snode, collected_node_set, name_to_buf, name_to_fused_node, criteria_cb=None argument
1700 snode, collected_node_set, name_to_buf, name_to_fused_node, criteria_cb=None argument
|
| H A D | comms.py | 224 nodes: List[BaseSchedulerNode], name_to_buf, name_to_fused_node argument
|
| /aosp_15_r20/external/pytorch/test/distributed/_composable/fsdp/ |
| H A D | test_fully_shard_compile.py | 231 snodes, name_to_buf, name_to_fused_node, orig_fn argument
|