Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/test/onnx/
H A Dtest_pytorch_onnx_onnxruntime.py9725 class TupleModule(torch.nn.Module): class
9750 self.run_test(TupleModule(), (input1, input2, input3))
9753 class TupleModule(torch.nn.Module): class
9763 self.run_test(TupleModule(), (input1, input2))
9766 class TupleModule(torch.nn.Module): class
9787 self.run_test(TupleModule(), (input1, input2))