Searched defs:joint_graph (Results 1 – 2 of 2) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/_higher_order_ops/ | ||
| H A D | map.py | 159 def forward(ctx, fw_graph, joint_graph, num_mapped_args, *flat_args): argument |
| H A D | flex_attention.py | 567 joint_graph, argument |