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