torch.fx

若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,int32

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

-