Lines Matching refs:unitriangular

1768         def run_test(a, b, upper, transpose, unitriangular, op_out):  argument
1769             if unitriangular and self.device_type == 'cpu':
1777 …actual = torch.triangular_solve(b, a, upper=upper, unitriangular=unitriangular, transpose=transpos…
1799                 unitriangular=unitriangular)
1813 …           upper=upper, unitriangular=unitriangular, transpose=transpose, out=(out, actual_A_clone)
1832 …       for (upper, unitriangular, transpose, op_out) in itertools.product([True, False], repeat=4):
1833                 run_test(a, b, upper, unitriangular, transpose, op_out)
2360         def run_test(n, k, upper, unitriangular, transpose, zero):  argument
2361             if not unitriangular:
2383 …expected = torch.triangular_solve(B, A, upper=upper, unitriangular=unitriangular, transpose=transp…
2386 …actual = torch.triangular_solve(B, A_sparse, upper=upper, unitriangular=unitriangular, transpose=t…
2399 …           upper=upper, unitriangular=unitriangular, transpose=transpose, out=(out, actual_A_clone)
2407 …           upper=upper, unitriangular=unitriangular, transpose=transpose, out=(out, actual_A_clone)
2420 …           upper=upper, unitriangular=unitriangular, transpose=transpose, out=(out, actual_A_clone)
2427 …for (k, n), (upper, unitriangular, transpose, zero) in itertools.product(itertools.product(ks, ns),
2429             run_test(n, k, upper, unitriangular, transpose, zero)