Searched refs:matmul_without_tf32 (Results 1 – 11 of 11) sorted by relevance
/external/tensorflow/tensorflow/python/kernel_tests/linalg/sparse/ |
D | csr_sparse_matrix_test.py | 165 c_dense = test_util.matmul_without_tf32( 205 c_dense = test_util.matmul_without_tf32( 243 c_dense = test_util.matmul_without_tf32(
|
D | csr_sparse_matrix_ops_test.py | 537 math_ops.conj(test_util.matmul_without_tf32(a_dense, b))) 579 c_dense_t = test_util.matmul_without_tf32( 643 c_dense_t = test_util.matmul_without_tf32( 673 c_dense_t = math_ops.conj(test_util.matmul_without_tf32(a_mats, b_mats)) 775 c_dense_t = test_util.matmul_without_tf32( 1146 verification = test_util.matmul_without_tf32( 1200 verification = test_util.matmul_without_tf32( 1241 verification = test_util.matmul_without_tf32(
|
/external/tensorflow/tensorflow/python/kernel_tests/ |
D | self_adjoint_eig_op_test.py | 164 a_ev = test_util.matmul_without_tf32( 165 test_util.matmul_without_tf32(tf_v, array_ops.matrix_diag(tf_e)),
|
D | qr_op_test.py | 114 a_recon = test_util.matmul_without_tf32(q, r) 119 xx = test_util.matmul_without_tf32(x, x, adjoint_a=True)
|
D | matrix_square_root_op_test.py | 40 square = test_util.matmul_without_tf32(sqrt, sqrt)
|
D | cholesky_op_test.py | 109 verification = test_util.matmul_without_tf32(chol, chol, adjoint_b=True) 274 a = test_util.matmul_without_tf32(
|
D | matrix_inverse_op_test.py | 43 tf_ans = test_util.matmul_without_tf32(inv, y, adjoint_b=adjoint)
|
D | lu_op_test.py | 94 verification = test_util.matmul_without_tf32(lower, upper)
|
/external/tensorflow/tensorflow/compiler/tests/ |
D | cholesky_op_test.py | 63 verification = test_util.matmul_without_tf32(chol, chol, adjoint_b=True)
|
D | matrix_triangular_solve_op_test.py | 67 verification = test_util.matmul_without_tf32(
|
/external/tensorflow/tensorflow/python/framework/ |
D | test_util.py | 1970 def matmul_without_tf32(a, b, *args, **kwargs): function
|