Home
last modified time | relevance | path

Searched refs:DynamicSliceExportMod (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/pytorch/test/onnx/
H A Dtest_pytorch_onnx_no_runtime.py346 class DynamicSliceExportMod(torch.jit.ScriptModule): class
354 mod = DynamicSliceExportMod()
359 DynamicSliceExportMod(), (input,), opset_version=10
H A Dtest_fx_to_onnx_with_onnxruntime.py450 class DynamicSliceExportMod(torch.nn.Module): class
460 DynamicSliceExportMod(),
H A Dtest_pytorch_onnx_onnxruntime.py2068 class DynamicSliceExportMod(torch.nn.Module): class
2078 DynamicSliceExportMod(),
2111 class DynamicSliceExportMod(torch.nn.Module): class
2120 DynamicSliceExportMod(),
/aosp_15_r20/external/pytorch/test/dynamo/
H A Dtest_export.py3443 class DynamicSliceExportMod(torch.nn.Module): class
3450 gm, _ = torch._dynamo.export(DynamicSliceExportMod(), aten_graph=True)(