Searched refs:batch_and_row_begin (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()
426 const int64 num_rows, const int64 batch_and_row_begin, in HandleBatchAndRowRange() argument
430 const int64 batch_begin = batch_and_row_begin / num_rows; in HandleBatchAndRowRange()
439 std::max(batch_and_row_begin, batch_idx * num_rows); in HandleBatchAndRowRange()
|