torch.onnx

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

API名称

是否支持

限制与说明

torch.onnx.dynamo_export

支持fp32

torch.onnx.ExportOptions

-

torch.onnx.enable_fake_mode

-

torch.onnx.ExportOutput

支持fp32

torch.onnx.ExportOutput.adapt_torch_inputs_to_onnx

-

torch.onnx.ExportOutput.adapt_torch_outputs_to_onnx

-

torch.onnx.ExportOutput.diagnostic_context

-

torch.onnx.ExportOutput.save

-

torch.onnx.ExportOutputSerializer

-

torch.onnx.ExportOutputSerializer.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

-