若API未标明支持情况则代表该API暂无验证结果,待验证后更新表格。
API名称 |
是否支持 |
限制与说明 |
---|---|---|
torch.export.export |
||
torch.export.dynamic_dim |
||
torch.export.save |
||
torch.export.load |
||
torch.export.register_dataclass |
||
torch.export.Dim |
||
torch.export.dims |
||
torch.export.Constraint |
||
torch.export.ExportedProgram |
||
torch.export.ExportedProgram.module |
||
torch.export.ExportedProgram.buffers |
||
torch.export.ExportedProgram.named_buffers |
||
torch.export.ExportedProgram.parameters |
||
torch.export.ExportedProgram.named_parameters |
||
torch.export.ExportBackwardSignature |
||
torch.export.ExportGraphSignature |
||
torch.export.ModuleCallSignature |
||
torch.export.ModuleCallEntry |
||
torch.export.graph_signature.InputKind |
||
torch.export.graph_signature.InputSpec |
||
torch.export.graph_signature.OutputKind |
||
torch.export.graph_signature.OutputSpec |
||
torch.export.graph_signature.ExportGraphSignature |
||
torch.export.graph_signature.ExportGraphSignature.replace_all_uses |
||
torch._higher_order_ops.cond.cond |