torch.onnx

Note

若API“是否支持”为“是”,“限制与说明”为“-”,说明此API和原生API支持度保持一致。

API名称 是否支持 限制与说明
torch.onnx.dynamo_export 支持fp32
torch.onnx.ExportOptions -
torch.onnx.enable_fake_mode -
torch.onnx.ONNXProgram -
torch.onnx.ONNXProgram.model_proto -
torch.onnx.ONNXProgram.save -
torch.onnx.OnnxExporterError -
torch.onnx.is_onnxrt_backend_supported -
torch.onnx.export -
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.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.VerificationOptions -