Searched refs:batch_and_row_end (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/sparse/ |
| D | mat_mul_op.cc | 304 [&](int64 batch_and_row_begin, int64 batch_and_row_end) { in SparseDenseMatMulWithoutTransposedLHS() argument
306 num_lhs_rows, batch_and_row_begin, batch_and_row_end, in SparseDenseMatMulWithoutTransposedLHS()
375 [&](int64 batch_and_row_begin, int64 batch_and_row_end, int tid) { in SparseDenseMatMulWithTransposedLHS() argument
377 num_lhs_cols, batch_and_row_begin, batch_and_row_end, in SparseDenseMatMulWithTransposedLHS()
427 const int64 batch_and_row_end, in HandleBatchAndRowRange() argument
431 const int64 batch_end_inclusive = batch_and_row_end / num_rows; in HandleBatchAndRowRange()
441 std::min(batch_and_row_end, (batch_idx + 1) * num_rows); in HandleBatchAndRowRange()
|