torch.jit

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

API名称

是否支持

限制与说明

torch.jit.export

  

torch.jit.is_scripting

     

torch.jit.is_tracing

     

torch.jit.script

  

torch.jit.trace

     

torch.jit.script_if_tracing

  

torch.jit.trace_module

  

torch.jit.fork

  

torch.jit.wait

  

torch.jit.ScriptModule

  

torch.jit.ScriptModule.add_module

  

torch.jit.ScriptModule.apply

  

torch.jit.ScriptModule.bfloat16

  

torch.jit.ScriptModule.buffers

  

torch.jit.ScriptModule.children

  

torch.jit.ScriptModule.cpu

  

torch.jit.ScriptModule.cuda

  

torch.jit.ScriptModule.double

  

torch.jit.ScriptModule.eval

  

torch.jit.ScriptModule.extra_repr

  

torch.jit.ScriptModule.float

  

torch.jit.ScriptModule.get_buffer

  

torch.jit.ScriptModule.get_extra_state

  

torch.jit.ScriptModule.get_parameter

  

torch.jit.ScriptModule.get_submodule

  

torch.jit.ScriptModule.half

  

torch.jit.ScriptModule.ipu

  

torch.jit.ScriptModule.load_state_dict

  

torch.jit.ScriptModule.modules

  

torch.jit.ScriptModule.named_buffers

     

torch.jit.ScriptModule.named_children

  

torch.jit.ScriptModule.named_modules

  

torch.jit.ScriptModule.named_parameters

     

torch.jit.ScriptModule.parameters

  

torch.jit.ScriptModule.register_backward_hook

  

torch.jit.ScriptModule.register_buffer

  

torch.jit.ScriptModule.register_forward_hook

     

torch.jit.ScriptModule.register_forward_pre_hook

     

torch.jit.ScriptModule.register_full_backward_hook

     

torch.jit.ScriptModule.register_full_backward_pre_hook

  

torch.jit.ScriptModule.register_load_state_dict_post_hook

  

torch.jit.ScriptModule.register_module

  

torch.jit.ScriptModule.register_parameter

  

torch.jit.ScriptModule.register_state_dict_pre_hook

  

torch.jit.ScriptModule.requires_grad_

  

torch.jit.ScriptModule.save

  

torch.jit.ScriptModule.set_extra_state

  

torch.jit.ScriptModule.share_memory

  

torch.jit.ScriptModule.state_dict

  

torch.jit.ScriptModule.to

  

torch.jit.ScriptModule.to

  

torch.jit.ScriptModule.to

  

torch.jit.ScriptModule.to

  

torch.jit.ScriptModule.to

  

torch.jit.ScriptModule.to_empty

  

torch.jit.ScriptModule.train

  

torch.jit.ScriptModule.type

  

torch.jit.ScriptModule.xpu

  

torch.jit.ScriptModule.zero_grad

  

torch.jit.ScriptFunction

  

torch.jit.ScriptFunction.get_debug_state

  

torch.jit.ScriptFunction.save

  

torch.jit.ScriptFunction.save_to_buffer

  

torch.jit.freeze

  

torch.jit.optimize_for_inference

  

torch.jit.enable_onednn_fusion

  

torch.jit.onednn_fusion_enabled

  

torch.jit.set_fusion_strategy

  

torch.jit.strict_fusion

  

torch.jit.save

  

torch.jit.load

     

torch.jit.ignore

  

torch.jit.unused

  

torch.jit.isinstance

  

torch.jit.Attribute

  

torch.jit.Attribute.count

  

torch.jit.Attribute.index

  

torch.jit.annotate