torch.onnx

若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.fake_context

     

torch.onnx.ExportOutput.model_proto

     

torch.onnx.ExportOutput.save

  

torch.onnx.ExportOutput.save_diagnostics

     

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