Searched refs:triangular_solve_batch_helper (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/pytorch/test/ |
| H A D | test_linalg.py | 4243 def triangular_solve_batch_helper(A_dims, b_dims, upper, unitriangular, transpose): function
4272 triangular_solve_batch_helper((batchsize, 5, 5), (batchsize, 5, 10),
4276 triangular_solve_batch_helper((batchsize, 0, 0), (batchsize, 0, 10),
4278 triangular_solve_batch_helper((batchsize, 0, 0), (batchsize, 0, 0),
|