torch.jit
Note
若API“是否支持”为“是”，“限制与说明”为“-”，说明此API和原生API支持度保持一致。
| API名称 | 是否支持 | 限制与说明 |
|---|---|---|
| torch.jit.export | 是 | - |
| torch.jit.is_scripting | 是 | - |
| torch.jit.is_tracing | 是 | 支持fp32 |
| torch.jit.script | 是 | - |
| torch.jit.trace | 是 | 支持fp32 |
| torch.jit.script_if_tracing | 是 | 支持fp32 |
| torch.jit.trace_module | 是 | 支持fp32 |
| torch.jit.fork | 是 | - |
| torch.jit.wait | 是 | - |
| torch.jit.ScriptModule | 是 | 支持fp32 |
| torch.jit.ScriptModule.add_module | 否 | - |
| torch.jit.ScriptModule.apply | 是 | 支持fp32 |
| torch.jit.ScriptModule.bfloat16 | 否 | - |
| torch.jit.ScriptModule.buffers | 是 | 支持fp32 |
| torch.jit.ScriptModule.children | 是 | - |
| torch.jit.ScriptModule.code | 否 | - |
| torch.jit.ScriptModule.code_with_constants | 否 | - |
| torch.jit.ScriptModule.compile | 否 | - |
| torch.jit.ScriptModule.cpu | 是 | 支持fp32 |
| torch.jit.ScriptModule.cuda | 是 | 支持fp32 |
| torch.jit.ScriptModule.double | 是 | 支持fp32 |
| torch.jit.ScriptModule.eval | 是 | 支持fp32 |
| torch.jit.ScriptModule.extra_repr | 是 | 支持fp32 |
| torch.jit.ScriptModule.float | 是 | 支持fp32 |
| torch.jit.ScriptModule.get_buffer | 是 | 支持fp32 |
| torch.jit.ScriptModule.get_extra_state | 否 | - |
| torch.jit.ScriptModule.get_parameter | 是 | 支持fp32 |
| torch.jit.ScriptModule.get_submodule | 是 | 支持fp32 |
| torch.jit.ScriptModule.graph | 否 | - |
| torch.jit.ScriptModule.half | 是 | 支持fp32 |
| torch.jit.ScriptModule.inlined_graph | 否 | - |
| torch.jit.ScriptModule.ipu | 是 | 支持fp32 |
| torch.jit.ScriptModule.load_state_dict | 是 | 支持fp32 |
| torch.jit.ScriptModule.modules | 是 | 支持fp32 |
| torch.jit.ScriptModule.named_buffers | 是 | 支持fp32 |
| torch.jit.ScriptModule.named_children | 是 | 支持fp32 |
| torch.jit.ScriptModule.named_modules | 是 | 支持fp32 |
| torch.jit.ScriptModule.named_parameters | 是 | 支持fp32 |
| torch.jit.ScriptModule.parameters | 是 | 支持fp32 |
| torch.jit.ScriptModule.register_backward_hook | 是 | 支持fp32 |
| torch.jit.ScriptModule.register_buffer | 否 | - |
| torch.jit.ScriptModule.register_forward_hook | 是 | 支持fp32 |
| torch.jit.ScriptModule.register_forward_pre_hook | 是 | 支持fp32 |
| torch.jit.ScriptModule.register_full_backward_hook | 是 | 支持fp32 |
| torch.jit.ScriptModule.register_full_backward_pre_hook | 是 | 支持fp32 |
| torch.jit.ScriptModule.register_load_state_dict_post_hook | 是 | 支持fp32 |
| torch.jit.ScriptModule.register_module | 否 | - |
| torch.jit.ScriptModule.register_parameter | 否 | - |
| torch.jit.ScriptModule.register_state_dict_pre_hook | 是 | 支持fp32 |
| torch.jit.ScriptModule.requires_grad_ | 是 | 支持fp32 |
| torch.jit.ScriptModule.save | 是 | 支持fp32 |
| torch.jit.ScriptModule.set_extra_state | 否 | - |
| torch.jit.ScriptModule.share_memory | 否 | - |
| torch.jit.ScriptModule.state_dict | 是 | 支持fp32 |
| torch.jit.ScriptModule.to | 是 | 支持fp32 |
| torch.jit.ScriptModule.to_empty | 是 | 支持fp32 |
| torch.jit.ScriptModule.train | 是 | 支持fp32 |
| torch.jit.ScriptModule.type | 是 | 支持fp16，fp32，uint8，int8，int16，int32，int64，bool |
| torch.jit.ScriptModule.xpu | 是 | 支持fp32 |
| torch.jit.ScriptModule.zero_grad | 是 | 支持fp32 |
| torch.jit.ScriptFunction | 是 | 支持fp32 |
| torch.jit.ScriptFunction.get_debug_state | 是 | 支持fp32 |
| torch.jit.ScriptFunction.save | 是 | 支持fp32 |
| 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 | 是 | 支持fp32 |
| torch.jit.Attribute.count | 是 | 支持fp32 |
| torch.jit.Attribute.index | 是 | 支持fp32 |
| torch.jit.Attribute.type | 否 | - |
| torch.jit.Attribute.value | 否 | - |
| torch.jit.annotate | 是 | 支持fp32 |