Searched refs:rhs_mat (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | eigen_mkldnn_contraction_kernel_test.cc | 131 MatrixMap rhs_mat(rhs.data(), k, n); in TEST() local
135 matmul_result = lhs_mat * rhs_mat; in TEST()
147 sum += std::abs(lhs_mat(i % m, k1) * rhs_mat(k1, i / m)); in TEST()
189 Matrix rhs_mat(k, n); in TEST() local
199 rhs_mat(i, j) = static_cast<int32_t>(rhs(i, j)); in TEST()
205 matmul_result = lhs_mat * rhs_mat; in TEST()
|
| /external/tensorflow/tensorflow/python/ops/linalg/ |
| D | linear_operator_block_diag.py | 599 rhs_mat = [array_ops.expand_dims(block, axis=-1) for block in rhs]
600 solution_mat = self.solve(rhs_mat, adjoint=adjoint)
608 rhs_mat = array_ops.expand_dims(rhs, axis=-1)
609 solution_mat = self.solve(rhs_mat, adjoint=adjoint)
|
| D | linear_operator_block_lower_triangular.py | 812 rhs_mat = [array_ops.expand_dims(block, axis=-1) for block in rhs]
813 solution_mat = self.solve(rhs_mat, adjoint=adjoint)
820 rhs_mat = array_ops.expand_dims(rhs, axis=-1)
821 solution_mat = self.solve(rhs_mat, adjoint=adjoint)
|
| D | linear_operator.py | 881 rhs_mat = array_ops.expand_dims(rhs, axis=-1)
882 solution_mat = self.solve(rhs_mat, adjoint=adjoint)
|