torch.onnx
若API“是否支持”为“是”，“限制与说明”为“-”，说明此API和原生API支持度保持一致。
API名称 |
是否支持 |
限制与说明 |
|---|---|---|
torch.onnx.dynamo_export |
是 |
支持fp32 |
torch.onnx.ExportOptions |
是 |
- |
torch.onnx.enable_fake_mode |
否 |
- |
torch.onnx.ONNXProgram |
是 |
- |
torch.onnx.ONNXProgram.adapt_torch_inputs_to_onnx |
是 |
- |
torch.onnx.ONNXProgram.adapt_torch_outputs_to_onnx |
是 |
- |
torch.onnx.ONNXProgram.diagnostic_context |
否 |
- |
torch.onnx.ONNXProgram.fake_context |
否 |
- |
torch.onnx.ONNXProgram.model_proto |
否 |
- |
torch.onnx.ONNXProgram.save |
是 |
- |
torch.onnx.ONNXProgram.save_diagnostics |
否 |
- |
torch.onnx.ONNXProgramSerializer |
是 |
- |
torch.onnx.ONNXProgramSerializer.serialize |
是 |
- |
torch.onnx.OnnxExporterError |
否 |
- |
torch.onnx.OnnxRegistry |
否 |
- |
torch.onnx.OnnxRegistry.get_op_functions |
否 |
- |
torch.onnx.OnnxRegistry.is_registered_op |
否 |
- |
torch.onnx.OnnxRegistry.opset_version |
否 |
- |
torch.onnx.OnnxRegistry.register_op |
否 |
- |
torch.onnx.DiagnosticOptions |
否 |
- |
torch.onnx.is_onnxrt_backend_supported |
是 |
- |
torch.onnx.export |
是 |
- |
torch.onnx.export_to_pretty_string |
是 |
- |
torch.onnx.register_custom_op_symbolic |
是 |
- |
torch.onnx.unregister_custom_op_symbolic |
是 |
- |
torch.onnx.select_model_mode_for_export |
是 |
- |
torch.onnx.is_in_onnx_export |
是 |
- |
torch.onnx.enable_log |
是 |
- |
torch.onnx.disable_log |
是 |
- |
torch.onnx.verification.find_mismatch |
是 |
- |
torch.onnx.JitScalarType |
是 |
- |
torch.onnx.JitScalarType.dtype |
是 |
- |
torch.onnx.JitScalarType.from_dtype |
是 |
支持bool |
torch.onnx.JitScalarType.from_value |
是 |
支持bool |
torch.onnx.JitScalarType.onnx_compatible |
是 |
- |
torch.onnx.JitScalarType.onnx_type |
是 |
- |
torch.onnx.JitScalarType.scalar_name |
是 |
- |
torch.onnx.JitScalarType.torch_name |
是 |
- |
torch.onnx.verification.GraphInfo |
是 |
- |
torch.onnx.verification.GraphInfo.all_mismatch_leaf_graph_info |
是 |
- |
torch.onnx.verification.GraphInfo.clear |
是 |
- |
torch.onnx.verification.GraphInfo.essential_node_count |
是 |
- |
torch.onnx.verification.GraphInfo.essential_node_kinds |
是 |
- |
torch.onnx.verification.GraphInfo.export_repro |
是 |
- |
torch.onnx.verification.GraphInfo.find_mismatch |
是 |
- |
torch.onnx.verification.GraphInfo.find_partition |
是 |
- |
torch.onnx.verification.GraphInfo.has_mismatch |
是 |
- |
torch.onnx.verification.GraphInfo.pretty_print_mismatch |
是 |
- |
torch.onnx.verification.GraphInfo.pretty_print_tree |
是 |
- |
torch.onnx.verification.GraphInfo.verify_export |
是 |
- |
torch.onnx.verification.VerificationOptions |
是 |
- |