已合并
docs: fix description for api torch.jit.ScriptModule hook API on NPU #35840
考试请神请到孔子发现考的是英语创建于 5月16日
docs: fix description for api torch.jit.ScriptModule hook API on NPU #35840
已合并
共 3 个文件变更+6-0
| @@ -47,10 +47,12 @@ | |||
| 47 | |torch.jit.ScriptModule.register_forward_pre_hook|是|支持fp32| | 47 | |torch.jit.ScriptModule.register_forward_pre_hook|是|支持fp32| |
| 48 | |torch.jit.ScriptModule.register_full_backward_hook|是|支持fp32| | 48 | |torch.jit.ScriptModule.register_full_backward_hook|是|支持fp32| |
| 49 | |torch.jit.ScriptModule.register_full_backward_pre_hook|是|支持fp32| | 49 | |torch.jit.ScriptModule.register_full_backward_pre_hook|是|支持fp32| |
| 50 | +|torch.jit.ScriptModule.register_load_state_dict_pre_hook|是|支持fp32| | ||
| 50 | |torch.jit.ScriptModule.register_load_state_dict_post_hook|是|支持fp32| | 51 | |torch.jit.ScriptModule.register_load_state_dict_post_hook|是|支持fp32| |
| 51 | |torch.jit.ScriptModule.register_module|否|-| | 52 | |torch.jit.ScriptModule.register_module|否|-| |
| 52 | |torch.jit.ScriptModule.register_parameter|否|-| | 53 | |torch.jit.ScriptModule.register_parameter|否|-| |
| 53 | |torch.jit.ScriptModule.register_state_dict_pre_hook|是|支持fp32| | 54 | |torch.jit.ScriptModule.register_state_dict_pre_hook|是|支持fp32| |
| 55 | +|torch.jit.ScriptModule.register_state_dict_post_hook|是|支持fp32| | ||
| 54 | |torch.jit.ScriptModule.requires_grad_|是|支持fp32| | 56 | |torch.jit.ScriptModule.requires_grad_|是|支持fp32| |
| 55 | |torch.jit.ScriptModule.save|是|支持fp32| | 57 | |torch.jit.ScriptModule.save|是|支持fp32| |
| 56 | |torch.jit.ScriptModule.set_extra_state|否|-| | 58 | |torch.jit.ScriptModule.set_extra_state|否|-| |
| @@ -50,10 +50,12 @@ | |||
| 50 | |torch.jit.ScriptModule.register_forward_pre_hook|是|支持fp32| | 50 | |torch.jit.ScriptModule.register_forward_pre_hook|是|支持fp32| |
| 51 | |torch.jit.ScriptModule.register_full_backward_hook|是|支持fp32| | 51 | |torch.jit.ScriptModule.register_full_backward_hook|是|支持fp32| |
| 52 | |torch.jit.ScriptModule.register_full_backward_pre_hook|是|支持fp32| | 52 | |torch.jit.ScriptModule.register_full_backward_pre_hook|是|支持fp32| |
| 53 | +|torch.jit.ScriptModule.register_load_state_dict_pre_hook|是|支持fp32| | ||
| 53 | |torch.jit.ScriptModule.register_load_state_dict_post_hook|是|支持fp32| | 54 | |torch.jit.ScriptModule.register_load_state_dict_post_hook|是|支持fp32| |
| 54 | |torch.jit.ScriptModule.register_module|否|-| | 55 | |torch.jit.ScriptModule.register_module|否|-| |
| 55 | |torch.jit.ScriptModule.register_parameter|否|-| | 56 | |torch.jit.ScriptModule.register_parameter|否|-| |
| 56 | |torch.jit.ScriptModule.register_state_dict_pre_hook|是|支持fp32| | 57 | |torch.jit.ScriptModule.register_state_dict_pre_hook|是|支持fp32| |
| 58 | +|torch.jit.ScriptModule.register_state_dict_post_hook|是|支持fp32| | ||
| 57 | |torch.jit.ScriptModule.requires_grad_|是|支持fp32| | 59 | |torch.jit.ScriptModule.requires_grad_|是|支持fp32| |
| 58 | |torch.jit.ScriptModule.save|是|支持fp32| | 60 | |torch.jit.ScriptModule.save|是|支持fp32| |
| 59 | |torch.jit.ScriptModule.set_extra_state|否|-| | 61 | |torch.jit.ScriptModule.set_extra_state|否|-| |
| @@ -47,10 +47,12 @@ | |||
| 47 | |torch.jit.ScriptModule.register_forward_pre_hook|是|支持fp32| | 47 | |torch.jit.ScriptModule.register_forward_pre_hook|是|支持fp32| |
| 48 | |torch.jit.ScriptModule.register_full_backward_hook|是|支持fp32| | 48 | |torch.jit.ScriptModule.register_full_backward_hook|是|支持fp32| |
| 49 | |torch.jit.ScriptModule.register_full_backward_pre_hook|是|支持fp32| | 49 | |torch.jit.ScriptModule.register_full_backward_pre_hook|是|支持fp32| |
| 50 | +|torch.jit.ScriptModule.register_load_state_dict_pre_hook|是|支持fp32| | ||
| 50 | |torch.jit.ScriptModule.register_load_state_dict_post_hook|是|支持fp32| | 51 | |torch.jit.ScriptModule.register_load_state_dict_post_hook|是|支持fp32| |
| 51 | |torch.jit.ScriptModule.register_module|否|-| | 52 | |torch.jit.ScriptModule.register_module|否|-| |
| 52 | |torch.jit.ScriptModule.register_parameter|否|-| | 53 | |torch.jit.ScriptModule.register_parameter|否|-| |
| 53 | |torch.jit.ScriptModule.register_state_dict_pre_hook|是|支持fp32| | 54 | |torch.jit.ScriptModule.register_state_dict_pre_hook|是|支持fp32| |
| 55 | +|torch.jit.ScriptModule.register_state_dict_post_hook|是|支持fp32| | ||
| 54 | |torch.jit.ScriptModule.requires_grad_|是|支持fp32| | 56 | |torch.jit.ScriptModule.requires_grad_|是|支持fp32| |
| 55 | |torch.jit.ScriptModule.save|是|支持fp32| | 57 | |torch.jit.ScriptModule.save|是|支持fp32| |
| 56 | |torch.jit.ScriptModule.set_extra_state|否|-| | 58 | |torch.jit.ScriptModule.set_extra_state|否|-| |