Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/test/
H A Dtest_linalg.py8024 def lu_solve_batch_test_helper(A_dims, b_dims, pivot): function
8036 lu_solve_batch_test_helper((batchsize, 5, 5), (batchsize, 5, 10), pivot)