Export a module and return the graph with ATen ops.
(module, example_inputs)
| 16 | |
| 17 | |
| 18 | def get_exported_graph(module, example_inputs): |
| 19 | """Export a module and return the graph with ATen ops.""" |
| 20 | ep = export(module, example_inputs) |
| 21 | return ep.graph_module.graph |
| 22 | |
| 23 | |
| 24 | def find_node_by_target(graph, target_name): |
no test coverage detected