torch.fx
Note
若API“是否支持”为“是”，“限制与说明”为“-”，说明此API和原生API支持度保持一致。
| API名称 | 是否支持 | 限制与说明 |
|---|---|---|
| torch.fx.symbolic_trace | 是 | 支持fp32 |
| torch.fx.wrap | 是 | 支持fp32 |
| torch.fx.GraphModule | 是 | 支持fp32 |
| torch.fx.GraphModule.init | 是 | 支持fp32 |
| torch.fx.GraphModule.add_submodule | 是 | 支持fp32 |
| torch.fx.GraphModule.code | 否 | - |
| torch.fx.GraphModule.delete_all_unused_submodules | 是 | 支持fp32 |
| torch.fx.GraphModule.delete_submodule | 是 | 支持fp32 |
| torch.fx.GraphModule.graph | 否 | - |
| torch.fx.GraphModule.print_readable | 是 | - |
| torch.fx.GraphModule.recompile | 是 | - |
| torch.fx.GraphModule.to_folder | 是 | 支持fp32 |
| torch.fx.Graph | 是 | - |
| torch.fx.Graph.init | 是 | - |
| torch.fx.Graph.call_function | 是 | - |
| torch.fx.Graph.call_method | 是 | - |
| torch.fx.Graph.call_module | 是 | - |
| torch.fx.Graph.create_node | 是 | - |
| torch.fx.Graph.eliminate_dead_code | 是 | - |
| torch.fx.Graph.erase_node | 是 | - |
| torch.fx.Graph.get_attr | 是 | - |
| torch.fx.Graph.graph_copy | 是 | - |
| torch.fx.Graph.inserting_after | 是 | - |
| torch.fx.Graph.inserting_before | 是 | - |
| torch.fx.Graph.lint | 是 | 支持fp32 |
| torch.fx.Graph.node_copy | 是 | 支持fp32 |
| torch.fx.Graph.nodes | 否 | - |
| torch.fx.Graph.on_generate_code | 是 | 支持fp32 |
| torch.fx.Graph.output | 是 | - |
| torch.fx.Graph.placeholder | 是 | - |
| torch.fx.Graph.print_tabular | 是 | 支持fp32 |
| torch.fx.Graph.process_inputs | 是 | - |
| torch.fx.Graph.process_outputs | 是 | - |
| torch.fx.Graph.python_code | 是 | 支持fp32 |
| torch.fx.Graph.set_codegen | 是 | - |
| torch.fx.Node | 是 | 支持fp32 |
| torch.fx.Node.all_input_nodes | 否 | - |
| torch.fx.Node.append | 是 | 支持fp32 |
| torch.fx.Node.args | 否 | - |
| torch.fx.Node.format_node | 是 | 支持fp32 |
| torch.fx.Node.is_impure | 是 | 支持fp32 |
| torch.fx.Node.kwargs | 否 | - |
| torch.fx.Node.next | 否 | - |
| torch.fx.Node.normalized_arguments | 是 | - |
| torch.fx.Node.prepend | 是 | - |
| torch.fx.Node.prev | 否 | - |
| torch.fx.Node.replace_all_uses_with | 是 | - |
| torch.fx.Node.replace_input_with | 是 | 支持fp32 |
| torch.fx.Node.stack_trace | 否 | - |
| torch.fx.Node.update_arg | 是 | 支持fp32 |
| torch.fx.Node.update_kwarg | 是 | 支持fp32 |
| torch.fx.Tracer | 是 | - |
| torch.fx.Tracer.call_module | 是 | 支持fp32 |
| torch.fx.Tracer.create_arg | 是 | - |
| torch.fx.Tracer.create_args_for_root | 是 | 支持fp32 |
| torch.fx.Tracer.create_node | 是 | - |
| torch.fx.Tracer.create_proxy | 是 | 支持fp32 |
| torch.fx.Tracer.getattr | 是 | - |
| torch.fx.Tracer.is_leaf_module | 是 | - |
| torch.fx.Tracer.iter | 是 | 支持fp32 |
| torch.fx.Tracer.keys | 是 | 支持fp32 |
| torch.fx.Tracer.path_of_module | 是 | - |
| torch.fx.Tracer.proxy | 是 | 支持fp32 |
| torch.fx.Tracer.to_bool | 是 | 支持fp32 |
| torch.fx.Tracer.trace | 是 | - |
| torch.fx.Proxy | 是 | - |
| torch.fx.Interpreter | 是 | - |
| torch.fx.Interpreter.call_function | 是 | 支持fp32 |
| torch.fx.Interpreter.call_method | 是 | 支持fp32 |
| torch.fx.Interpreter.call_module | 是 | 支持fp32 |
| torch.fx.Interpreter.fetch_args_kwargs_from_env | 是 | 支持fp32 |
| torch.fx.Interpreter.fetch_attr | 是 | 支持fp32 |
| torch.fx.Interpreter.get_attr | 是 | 支持fp32 |
| torch.fx.Interpreter.map_nodes_to_values | 是 | 支持fp32 |
| torch.fx.Interpreter.output | 是 | 支持fp32 |
| torch.fx.Interpreter.placeholder | 是 | 支持fp32 |
| torch.fx.Interpreter.run | 是 | - |
| torch.fx.Interpreter.run_node | 是 | 支持fp32 |
| torch.fx.Transformer | 是 | - |
| torch.fx.Transformer.call_function | 是 | 支持fp32 |
| torch.fx.Transformer.call_module | 是 | 支持fp32 |
| torch.fx.Transformer.get_attr | 是 | 支持fp32 |
| torch.fx.Transformer.placeholder | 是 | 支持fp32 |
| torch.fx.Transformer.transform | 是 | - |
| torch.fx.replace_pattern | 是 | - |