Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/python/kernel_tests/linalg/sparse/
Dcsr_sparse_matrix_test.py165 c_dense = test_util.matmul_without_tf32(
205 c_dense = test_util.matmul_without_tf32(
243 c_dense = test_util.matmul_without_tf32(
Dcsr_sparse_matrix_ops_test.py537 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/
Dself_adjoint_eig_op_test.py164 a_ev = test_util.matmul_without_tf32(
165 test_util.matmul_without_tf32(tf_v, array_ops.matrix_diag(tf_e)),
Dqr_op_test.py114 a_recon = test_util.matmul_without_tf32(q, r)
119 xx = test_util.matmul_without_tf32(x, x, adjoint_a=True)
Dmatrix_square_root_op_test.py40 square = test_util.matmul_without_tf32(sqrt, sqrt)
Dcholesky_op_test.py109 verification = test_util.matmul_without_tf32(chol, chol, adjoint_b=True)
274 a = test_util.matmul_without_tf32(
Dmatrix_inverse_op_test.py43 tf_ans = test_util.matmul_without_tf32(inv, y, adjoint_b=adjoint)
Dlu_op_test.py94 verification = test_util.matmul_without_tf32(lower, upper)
/external/tensorflow/tensorflow/compiler/tests/
Dcholesky_op_test.py63 verification = test_util.matmul_without_tf32(chol, chol, adjoint_b=True)
Dmatrix_triangular_solve_op_test.py67 verification = test_util.matmul_without_tf32(
/external/tensorflow/tensorflow/python/framework/
Dtest_util.py1970 def matmul_without_tf32(a, b, *args, **kwargs): function