Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/contrib/factorization/kernels/
Dwals_solver_ops.cc132 ConstEigenMatrixFloatMap factors_mat(factors.matrix<float>().data(), in Compute() local
232 std::forward_as_tuple(factors_mat.rows(), kMaxBatchSize)); in Compute()
264 factors_mat.col(factor_index) * std::sqrt(weight); in Compute()
272 input_value * (w_0 + weight) * factors_mat.col(factor_index); in Compute()
276 factor_batch.block(0, 0, factors_mat.rows(), num_batched); in Compute()