Searched defs:graph_signature (Results 1 – 8 of 8) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/_export/passes/ |
| H A D | replace_with_hop_pass_util.py | 104 graph_signature, argument
153 graph_signature, argument
|
| H A D | replace_set_grad_with_hop_pass.py | 74 gm: torch.fx.GraphModule, graph_signature argument
101 def replace_set_grad_with_hop_pass(gm: torch.fx.GraphModule, graph_signature): argument
|
| H A D | replace_autocast_with_hop_pass.py | 139 gm: torch.fx.GraphModule, graph_signature argument
170 def replace_autocast_with_hop_pass(gm: torch.fx.GraphModule, graph_signature): argument
|
| /aosp_15_r20/external/pytorch/torch/export/ |
| H A D | exported_program.py | 582 gm: torch.fx.GraphModule, graph_signature, module_call_graph argument
714 def graph_signature(self): member in ExportedProgram
1130 self, graph_module, graph_signature, *, state_dict=None, verifiers=None argument
|
| H A D | unflatten.py | 555 def _inplace_buffer_mutations(graph: torch.fx.Graph, graph_signature) -> None: argument
|
| /aosp_15_r20/external/pytorch/torch/_export/ |
| H A D | utils.py | 49 graph_signature, constants, mod argument
|
| /aosp_15_r20/external/pytorch/torch/_inductor/ |
| H A D | compile_fx.py | 156 def _unlift_graph(mod, gm, graph_signature): argument
|
| /aosp_15_r20/external/executorch/exir/program/ |
| H A D | _program.py | 301 graph_signature, argument
|