Lines Matching defs:b_dims
2595 def cholesky_solve_test_helper(self, A_dims, b_dims, upper, device, dtype): argument
2620 def cholesky_solve_batch_helper(A_dims, b_dims, upper): argument
2657 def run_test(A_dims, b_dims, upper): argument
3057 def solve_test_helper(self, A_dims, b_dims, device, dtype): argument
4206 def triangular_solve_test_helper(self, A_dims, b_dims, upper, unitriangular, argument
4243 def triangular_solve_batch_helper(A_dims, b_dims, upper, unitriangular, transpose): argument
4262 def triangular_solve_zero_batch_helper(A_dims, b_dims, upper, unitriangular, transpose): argument
4339 def run_test(A_dims, b_dims, device, upper, transpose, unitriangular): argument
7991 def lu_solve_test_helper(self, A_dims, b_dims, pivot, device, dtype): argument
8024 def lu_solve_batch_test_helper(A_dims, b_dims, pivot): argument
8053 def run_test(A_dims, b_dims): argument
8069 def run_test(A_dims, b_dims, pivot=True): argument
8090 def run_test(A_dims, b_dims): argument