Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/test/
H A Dtest_linalg.py4206 def triangular_solve_test_helper(self, A_dims, b_dims, upper, unitriangular, member in TestLinalg
4229 b, A = self.triangular_solve_test_helper((n, n), (n, k), upper,
4244 b, A = self.triangular_solve_test_helper(A_dims, b_dims, upper,
4263 b, A = self.triangular_solve_test_helper(A_dims, b_dims, upper,
4296 b, A = self.triangular_solve_test_helper((256, 256, 5, 5), (5, 1),
4309 b, A = self.triangular_solve_test_helper((3, 3), (512, 512, 3, 1),
4340 b, A = self.triangular_solve_test_helper(A_dims, b_dims, upper,