Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/
Deigen_mkldnn_contraction_kernel_test.cc122 MatrixMap lhs_mat(lhs.data(), m, k); in TEST() local
127 matmul_result = lhs_mat * rhs_mat; in TEST()
139 sum += std::abs(lhs_mat(i % m, k1) * rhs_mat(k1, i / m)); in TEST()
/external/tensorflow/tensorflow/contrib/factorization/kernels/
Dwals_solver_ops.cc249 EigenMatrixFloatMap lhs_mat(output_lhs_tensor->flat<float>().data() + in Compute() local
252 auto lhs_symm = lhs_mat.selfadjointView<Eigen::Lower>(); in Compute()
281 lhs_mat = lhs_symm; in Compute()