Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/contrib/factorization/kernels/
Dwals_solver_ops.cc248 int num_batched = 0; in Compute() local
263 factor_batch.col(num_batched) = in Compute()
265 ++num_batched; in Compute()
266 if (num_batched == kMaxBatchSize) { in Compute()
268 num_batched = 0; in Compute()
274 if (num_batched != 0) { in Compute()
276 factor_batch.block(0, 0, factors_mat.rows(), num_batched); in Compute()