torch.fx

若API未标明支持情况,则代表该API的支持情况待验证。

API名称

是否支持

限制与说明

torch.fx.symbolic_trace

是

  

torch.fx.wrap

是

  

torch.fx.GraphModule

是

  

torch.fx.GraphModule.__init__

是

  

torch.fx.GraphModule.add_submodule

是

  

torch.fx.GraphModule.code

     

torch.fx.GraphModule.delete_all_unused_submodules

是

  

torch.fx.GraphModule.delete_submodule

是

  

torch.fx.GraphModule.graph

     

torch.fx.GraphModule.print_readable

是

  

torch.fx.GraphModule.recompile

是

  

torch.fx.GraphModule.to_folder

是

  

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

是

  

torch.fx.Graph.node_copy

是

  

torch.fx.Graph.nodes

     

torch.fx.Graph.on_generate_code

是

  

torch.fx.Graph.output

是

  

torch.fx.Graph.placeholder

是

  

torch.fx.Graph.print_tabular

是

  

torch.fx.Graph.process_inputs

是

  

torch.fx.Graph.process_outputs

是

  

torch.fx.Graph.python_code

是

  

torch.fx.Graph.set_codegen

是

  

torch.fx.Node

是

  

torch.fx.Node.all_input_nodes

     

torch.fx.Node.append

是

  

torch.fx.Node.args

     

torch.fx.Node.format_node

是

  

torch.fx.Node.is_impure

是

  

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

是

  

torch.fx.Node.stack_trace

     

torch.fx.Node.update_arg

是

  

torch.fx.Node.update_kwarg

是

  

torch.fx.Tracer

是

  

torch.fx.Tracer.call_module

是

  

torch.fx.Tracer.create_arg

是

  

torch.fx.Tracer.create_args_for_root

是

  

torch.fx.Tracer.create_node

是

  

torch.fx.Tracer.create_proxy

是

  

torch.fx.Tracer.getattr

是

  

torch.fx.Tracer.is_leaf_module

是

  

torch.fx.Tracer.iter

是

  

torch.fx.Tracer.keys

是

  

torch.fx.Tracer.path_of_module

是

  

torch.fx.Tracer.proxy

是

  

torch.fx.Tracer.to_bool

是

  

torch.fx.Tracer.trace

是

  

torch.fx.Proxy

是

  

torch.fx.Interpreter

是

  

torch.fx.Interpreter.boxed_run

     

torch.fx.Interpreter.call_function

是

  

torch.fx.Interpreter.call_method

是

  

torch.fx.Interpreter.call_module

是

  

torch.fx.Interpreter.fetch_args_kwargs_from_env

是

  

torch.fx.Interpreter.fetch_attr

是

  

torch.fx.Interpreter.get_attr

是

  

torch.fx.Interpreter.map_nodes_to_values

是

  

torch.fx.Interpreter.output

是

  

torch.fx.Interpreter.placeholder

是

  

torch.fx.Interpreter.run

是

  

torch.fx.Interpreter.run_node

是

  

torch.fx.Transformer

是

  

torch.fx.Transformer.call_function

是

  

torch.fx.Transformer.call_module

是

  

torch.fx.Transformer.get_attr

是

  

torch.fx.Transformer.placeholder

是

  

torch.fx.Transformer.transform

是

  

torch.fx.replace_pattern

是