Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/test/onnx/
H A Dtest_pytorch_onnx_onnxruntime.py8634 class LinalgMultiDimModel(torch.nn.Module): class
8643 self.run_test(LinalgMultiDimModel("fro"), x)
8644 self.run_test(LinalgMultiDimModel(float("inf")), x)
8645 self.run_test(LinalgMultiDimModel(-float("inf")), x)
8646 self.run_test(LinalgMultiDimModel(1), x)
8647 self.run_test(LinalgMultiDimModel(-1), x)