Home
last modified time | relevance | path

Searched refs:TorchScriptOnnxExportDiagnostic (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/pytorch/torch/onnx/_internal/diagnostics/
H A D_diagnostic.py46 class TorchScriptOnnxExportDiagnostic(infra.Diagnostic): class
196 ) -> TorchScriptOnnxExportDiagnostic:
202 diagnostic = TorchScriptOnnxExportDiagnostic(
H A D__init__.py7 TorchScriptOnnxExportDiagnostic,
/aosp_15_r20/external/pytorch/test/onnx/internal/
H A Dtest_diagnostics.py252 ) -> diagnostics.TorchScriptOnnxExportDiagnostic:
277 diagnostics.TorchScriptOnnxExportDiagnostic, diagnostic
334 …diagnostic = diagnostics.TorchScriptOnnxExportDiagnostic(self._sample_rule, diagnostics.levels.NOT…