Searched refs:test_with_matrix_shapes (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/python/kernel_tests/linalg/ |
| D | tridiagonal_solve_op_test.py | 604 def test_with_matrix_shapes(matrix_shape, rhs_shape=None): function
617 test_with_matrix_shapes(matrix_shape=[4, 4], rhs_shape=[None, None])
618 test_with_matrix_shapes(matrix_shape=[None, 4], rhs_shape=[None, None])
619 test_with_matrix_shapes(matrix_shape=[4, None], rhs_shape=[None, None])
620 test_with_matrix_shapes(matrix_shape=[None, None], rhs_shape=[None, None])
621 test_with_matrix_shapes(matrix_shape=[4, 4])
622 test_with_matrix_shapes(matrix_shape=[None, 4])
623 test_with_matrix_shapes(matrix_shape=[4, None])
624 test_with_matrix_shapes(matrix_shape=[None, None])
625 test_with_matrix_shapes(matrix_shape=None, rhs_shape=[None, None])
[all …]
|